abcde-musicbrainz-tool

   1 #!/usr/bin/perl
   2 # Copyright (c) 2012 Steve McIntyre <93sam@debian.org>
   3 # This code is hereby licensed for public consumption under either the
   4 # GNU GPL v2 or greater, or Larry Wall's Artistic license - your choice.
   5 #
   6 # You should have received a copy of the GNU General Public License along
   7 # with this program; if not, write to the Free Software Foundation, Inc.,
   8 # 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
   9 #
  10 # abcde-musicbrainz-tool
  11 #
  12 # Helper script for abcde to work with the MusicBrainz WS API (v2)
  13
  14 use strict;
  15 use encoding "utf8";
  16 use POSIX qw(ceil);
  17 use Digest::SHA;
  18 use MusicBrainz::DiscID;
  19 use WebService::MusicBrainz::Release;
  20 use WebService::MusicBrainz::Artist;
  21 use WebService::MusicBrainz::Response::Track;
  22 use WebService::MusicBrainz::Response::TrackList;
  23 use Getopt::Long;
  24
  25 my $FRAMES_PER_SEC = 75;
  26
  27 my ($device, $command, $discid, @discinfo, $workdir);
  28 Getopt::Long::Configure ('no_ignore_case');
  29 Getopt::Long::Configure ('no_auto_abbrev');
  30 GetOptions ("device=s"       => \$device,
  31             "command=s"      => \$command,
  32             "discid=s"       => \$discid,
  33             "discinfo=i{5,}" => \@discinfo,
  34             "workdir=s"      => \$workdir);
  35
  36 if (!defined($device)) {
  37     $device = "/dev/cdrom";
  38 }
  39 if (!defined($command)) {
  40     $command = "id";
  41 }
  42 if (!defined($workdir)) {
  43     $workdir = "/tmp";
  44 }
  45
  46 sub calc_sha1($) {
  47     my $filename = shift;
  48     my $s = Digest::SHA->new(1);
  49     $s->addfile($filename);
  50     return $s->hexdigest;
  51 }
  52
  53 if ($command =~ m/^id/) {
  54     my $disc = new MusicBrainz::DiscID($device);
  55
  56     # read the disc in the default disc drive */
  57     if ( $disc->read() == 0 ) {
  58         printf STDERR "Error: %s\n", $disc->error_msg();
  59         exit(1);
  60     }
  61
  62     printf("%s ", $disc->id());
  63     printf("%d ", $disc->last_track_num() + 1 - $disc->first_track_num());
  64
  65     for ( my $i = $disc->first_track_num;
  66           $i <= $disc->last_track_num; $i++ ) {
  67         printf("%d ", $disc->track_offset($i));
  68     }
  69     printf("%d\n", $disc->sectors() / $FRAMES_PER_SEC);
  70     undef $disc;
  71
  72 } elsif ($command =~ m/data/) {
  73     my $ws = WebService::MusicBrainz::Release->new();
  74     my $response = $ws->search({ DISCID => $discid });
  75     my @releases = $response->release_list();
  76     my $releasenum = 0;
  77     my @sums;
  78
  79     foreach my $release (@releases) {
  80         my $a_artist = $release->artist()->name();
  81         my $va = 0;
  82         if ($a_artist =~ /Various Artists/) {
  83             $va = 1;
  84         }
  85         $releasenum++;
  86         open (OUT, "> $workdir/cddbread.$releasenum");
  87         binmode OUT, ":utf8";
  88         print OUT "# xmcd style database file\n";
  89         print OUT "#\n";
  90         print OUT "# Track frame offsets:\n";
  91         # Assume standard pregap
  92         my $total_len = 2000;
  93         my @tracks = @{$release->track_list()->tracks()};
  94         for (my $i = 0; $i < scalar(@tracks); $i++) {
  95             printf OUT "#       %d\n", ceil($total_len * $FRAMES_PER_SEC / 1000.0);
  96             $total_len += $tracks[$i]->duration();
  97         }
  98         print OUT "#\n";
  99         printf OUT "# Disc length: %d seconds\n", $total_len / 1000.0;
 100         print OUT "#\n";
 101         print OUT "# Submitted via: XXXXXX\n";
 102         print OUT "#\n";
 103         print OUT "#blues,classical,country,data,folk,jazz,newage,reggae,rock,soundtrack,misc\n";
 104         print OUT "#CATEGORY=none\n";
 105         print OUT "DISCID=" . $discid . "\n";
 106         print OUT "DTITLE=" . $a_artist. " / " . $release->title() . "\n";
 107         print OUT "DYEAR=\n";
 108         print OUT "DGENRE=\n";
 109
 110         my @tracks = @{$release->track_list()->tracks()};
 111         for (my $i = 0; $i < scalar(@tracks); $i++) {
 112             my $track = $tracks[$i];
 113             my $t_name = $track->title;
 114             if ($va) {
 115                 my $t_artist = $track->artist->name;
 116                 printf OUT "TTITLE%d=%s / %s\n", $i, $t_artist, $t_name;
 117             } else {
 118                 printf OUT "TTITLE%d=%s\n", $i, $t_name;
 119             }
 120         }
 121
 122         print OUT "EXTD=\n";
 123         for (my $i = 0; $i < scalar(@tracks); $i++) {
 124             printf OUT "EXTT%d=\n", $i;
 125         }
 126         print OUT "PLAYORDER=\n";
 127         print OUT ".\n";
 128         close OUT;
 129
 130         # Check to see that this entry is unique; generate a checksum
 131         # and compare to any previous checksums
 132         my $checksum = calc_sha1("$workdir/cddbread.$releasenum");
 133         foreach my $sum (@sums) {
 134             if ($checksum eq $sum) {
 135                 unlink("$workdir/cddbread.$releasenum");
 136                 $releasenum--;
 137                 last;
 138             }
 139         }
 140         push (@sums, $checksum);
 141     }
 142 } elsif ($command =~ m/calcid/) {
 143 # Calculate MusicBrainz ID from disc offsets; see
 144 # http://musicbrainz.org/doc/DiscIDCalculation
 145
 146     my ($first, $last, $leadin, $leadout, @offsets) = @discinfo;
 147
 148     my $s = Digest::SHA->new(1);
 149     $s->add(sprintf "%02X", int($first));
 150     $s->add(sprintf "%02X", int($last));
 151
 152     my @a;
 153     for (my $i = 0; $i < 100; $i++) {
 154         $a[$i] = 0;
 155     }
 156     my $i = 0;
 157     foreach my $o ($leadout, @offsets) {
 158        $a[$i++] = int($o) + int($leadin);
 159     }
 160     for (my $i = 0; $i < 100; $i++) {
 161        $s->add(sprintf "%08X", $a[$i]);
 162     }
 163
 164     my $id = $s->b64digest;
 165     # CPAN Digest modules do not pad their Base64 output, so we have to do it.
 166     while (length($id) % 4) {
 167         $id .= '=';
 168     }
 169
 170     $id =~ tr#+#.#;
 171     $id =~ tr#/#_#;
 172     $id =~ tr#=#-#;
 173
 174     print $id;
 175 }
 176