File Coverage

blib/lib/MusicBrainz/Collection.pm

Criterion	Covered	Total	%
statement	33	61	54.1
branch	3	18	16.6
condition	5	12	41.6
subroutine	8	10	80.0
pod	2	2	100.0
total	51	103	49.5

package MusicBrainz::Collection;

use strict;

use Audio::Scan;
use File::Next;
use HTTP::Request::Common qw(POST);
use LWP::UserAgent;

our $VERSION = '0.01';

sub new {
    my ( $class, %opts ) = @_;

    my $self = {
        user    => $opts{user} || '',
        pass    => $opts{pass} || '',
        albums  => {},
        verbose => $opts{verbose} || 0,
    };

    bless $self, $class;

    if ( !$self->{user} ) {
        $self->_load_auth;

        unless ( $self->{user} && $self->{pass} ) {
            die "No login information found\n";
        }
    }

    return $self;
}

sub process {
    my ( $self, $dir ) = @_;

    if ( my $albums = $self->_find_albums($dir) ) {
        print "Uploading " . scalar( @{$albums} ) . " albums to collection...\n";

        my $ua = LWP::UserAgent->new;
        $ua->credentials(
            'musicbrainz.org:80',
            'musicbrainz.org',
            $self->{user} => $self->{pass}
        );

        my $req = POST "http://musicbrainz.org/ws/1/collection/", [
            addAlbums => join( ',', @{$albums} ),
        ];

        $self->{verbose} && print $req->as_string;

        my $response = $ua->request($req);

        if ( $response->is_success ) {
            print "Done!\n";
        }
        else {
            print "Error: " . $response->status_line . "\n";
        }
    }
    else {
        print "No album ID tags found\n";
    }
}

sub _load_auth {
    my $self = shift;

    require File::Spec;
    my $file = File::Spec->catfile( $ENV{HOME}, '.musicbrainz' );

    if ( -e $file ) {
        open my $fh, '<', $file or die "Unable to read .musicbrainz file: $!\n";
        my $prefs = do { local $/; <$fh> };
        close $fh;

        ($self->{user}) = $prefs =~ m/user\s+(.+)/;
        ($self->{pass}) = $prefs =~ m/pass\s+(.+)/;
    }
}

sub _find_albums {
    my ( $self, $dir ) = @_;

    my $iter = File::Next::files( {
        file_filter => sub { Audio::Scan->is_supported( $File::Next::name ) },
    }, $dir );

    # Speed up scanning a bit by ignoring artwork
    local $ENV{AUDIO_SCAN_NO_ARTWORK} = 1;

    while ( defined ( my $file = $iter->() ) ) {
        $self->{verbose} && print "$file\n";

        my $s = Audio::Scan->scan_tags($file);
        my $tags = $s->{tags};

        my $albumid
            =  $tags->{'MusicBrainz Album Id'}
            || $tags->{'MUSICBRAINZ ALBUM ID'}
            || $tags->{'MUSICBRAINZ_ALBUMID'}
            || $tags->{'MusicBrainz/Album Id'};

        if ( $albumid ) {
            $self->{albums}->{$albumid} = 1;
        }
    }

    return [ keys %{ $self->{albums} } ];
}

1;
__END__