git.sesse.net Git - ccbs/blob - parse/parse-ddrfreak-songlist.pl

   1 #! /usr/bin/perl
   2 use strict;
   3 use warnings;
   4
   5 my $machine = shift;
   6 die "Missing machine (first argument on the command line)" if (!defined($machine));
   7
   8 # Parses songlist from ddrfreak.com
   9
  10 print "begin;\n";
  11
  12 while (<>) {
  13         while (not m/<tr bgcolor=\#CFDCF8>/) {
  14                 $_ = <>;
  15                 last if not $_;
  16         }
  17         $_ = <>;
  18         last if not $_;
  19         next if /<td bgcolor=\#DFECF8>Beginner<\/td>/;
  20         m/ <td \s* align=center> .*? <\/td> /x or die "Parse error, line 1";
  21         $_ = <>;
  22         m/ <td> (?: <font \s* color=.*?> )? (.*?) (?: <\/font> )? <\/td> /x or die "Parse error, line 2";
  23
  24         my $songname = $1;
  25
  26         $_ = <>;
  27         m/ <td \s* align=center> (.*?) <\/td> \s*            # artist
  28            <td \s* align=center \s* bgcolor=\#DFECF8>
  29              (?: <a \s* href=" .*? "> ([0-9-] | 10) <\/a>)?  # beginner single
  30            <\/td>
  31          /x or die "Parse error, line 3";
  32
  33         my ($artist, $sb) = ($1, $2);
  34
  35         # single standard
  36         $_ = <>;
  37         m/ <td \s* align=center \s* bgcolor=\#CFDCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  38                 or die "Parse error, line 4";
  39         my $ss = $1;
  40
  41         # single difficult
  42         $_ = <>;
  43         m/ <td \s* align=center \s* bgcolor=\#BFCCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  44                 or die "Parse error, line 5";
  45         my $sd = $1;
  46
  47         # single expert
  48         $_ = <>;
  49         m/ <td \s* align=center \s* bgcolor=\#AFBCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  50                 or die "Parse error, line 6";
  51         my $se = $1;
  52
  53         # single challenge
  54         $_ = <>;
  55         m/ <td \s* align=center \s* bgcolor=\#9FACF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  56                 or die "Parse error, line 7";
  57         my $sc = $1;
  58
  59         # double standard
  60         $_ = <>;
  61         m/ <td \s* align=center \s* bgcolor=\#CFDCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  62                 or die "Parse error, line 8";
  63         my $ds = $1;
  64
  65         # double difficult
  66         $_ = <>;
  67         m/ <td \s* align=center \s* bgcolor=\#BFCCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  68                 or die "Parse error, line 9";
  69         my $dd = $1;
  70
  71         # double expert
  72         $_ = <>;
  73         m/ <td \s* align=center \s* bgcolor=\#AFBCF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  74                 or die "Parse error, line 10";
  75         my $de = $1;
  76
  77         # double challenge
  78         $_ = <>;
  79         m/ <td \s* align=center \s* bgcolor=\#9FACF8> (?: <a \s* href=" .*? "> )? (\d*) ( ?: <\/a> )? <\/td> /x
  80                 or die "Parse error, line 11";
  81         my $dc = $1;
  82
  83         # Blarg, hack
  84         if ($songname eq 'FANTASY') {
  85                 $songname = "$songname ($artist)";
  86         }
  87
  88         # minimal SQL escaping
  89         $songname =~ s/'/\\'/g;
  90         $artist =~ s/'/\\'/g;
  91
  92         # Insert the song only if it doesn't already exist
  93         printf "INSERT INTO songs SELECT nextval('songs_song_seq') AS song, '%s' AS title, '%s' AS artist, 0 AS minbpm, 0 AS maxbpm WHERE '%s' NOT IN ( SELECT title FROM songs );\n",
  94                 $songname, $artist, $songname;
  95
  96         for my $t (['single', 'beginner', $sb],
  97                    ['single', 'standard', $ss],
  98                    ['single', 'difficult', $sd],
  99                    ['single', 'expert', $se],
 100                    ['single', 'challenge', $sc],
 101                    ['double', 'standard', $ds],
 102                    ['double', 'difficult', $dd],
 103                    ['double', 'expert', $de],
 104                    ['double', 'challenge', $dc]) {
 105                 next if (!defined($t->[2]) || $t->[2] eq '');
 106                 printf "INSERT INTO songratings (song,machine,playmode,difficulty,feetrating) VALUES ((SELECT song FROM songs WHERE title='%s'),(SELECT machine FROM machines WHERE machinename='%s'),'%s','%s',%u);\n",
 107                         $songname, $machine, $t->[0], $t->[1], $t->[2];
 108         }
 109 }
 110
 111 printf "commit;\n";