#! /usr/bin/env python2.4

'''
=head1 NAME

B<fa-rip> - rip a CD for B<fa-flacd>

=head1 SYNOPSIS

B<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]

=head1 DESCRIPTION

B<fa-rip> creates a temporary directory for storage of its
intermediate files, uses MusicBrainz to create the "cue" file and
candidate tags files, and runs C<cdparanoia(1)> to rip the CD to the
"wav" file.

In order for this CD to be processed by B<fa-flacd>, you must create a
"tags" file.  This is usually done by renaming one of the
candidate-tags files and deleting the others.  Don't forget to fill in
the DATE tag in the selected candidate before renaming it.  If
B<fa-rip> could not find any tag information from MusicBrainz, you'll
have to fill out the candidate-tags-0 template.

=head1 OPTIONS

=over 4

=item B<-d> [B<--device>] I<device>

Use I<device> as the CD-ROM device, instead of the default
"/dev/cdrom" or the environment variable CDDEV.

=item B<-p> [B<--post-processor>] I<post-processor>

Create a "post-processor" file in the temporary directory containing
the line 'I<post-processor> "$@"'.  See B<fa-flacd>'s man page for
information about this hook.

=item B<-t> [B<--tracks>] I<track-count>

Archive only the first I<track-count> tracks.  This is handy for
ignoring data tracks.

=back

=head1 ENVIRONMENT

=over 4

=item CDDEV

B<fa-rip> uses this to rip audio and save the cuesheet for a CD.
MusicBrainz::Client can usually figure this out automatically.

=back

=head1 AUTHORS

Written by Eric Gillespie <epg@pretzelnet.org>.

flac-archive is free software; you may redistribute it and/or modify
it under the same terms as Perl itself.

=cut

''' #' # python-mode is sucks

import os, re, sys, tempfile, traceback
from optparse import OptionParser
import urllib

import musicbrainz2.disc
import musicbrainz2.webservice

from org.diplodocus.util import catch_EnvironmentError as c

def mkcue(disc, trackcount=None):
    fp = c(file, 'cue', 'w')
    c(fp.write, 'FILE "dummy.wav" WAVE\n')
    c(fp.write, '  TRACK 01 AUDIO\n')
    c(fp.write, '    INDEX 01 00:00:00\n')

    if trackcount == None:
        trackcount = disc.lastTrackNum
    else:
        trackcount = min(trackcount, disc.lastTrackNum)

    pregap = disc.tracks[0][0]
    for i in xrange(disc.firstTrackNum, trackcount):
        offset = disc.tracks[i][0]
        offset -= pregap

        minutes = seconds = 0
        sectors = offset % 75
        if offset >= 75:
            seconds = offset / 75
            if seconds >= 60:
                minutes = seconds / 60
                seconds = seconds % 60

        c(fp.write, '  TRACK %02d AUDIO\n' % (i + 1,))
        c(fp.write,
          '    INDEX 01 %02d:%02d:%02d\n' % (minutes, seconds, sectors))

    c(fp.close)

    return trackcount

def tags_file(fn, trackcount, various, artist=None, album=None,
              release_dates={}, tracks=[]):
    fp = c(file, fn, 'w')
    c(fp.write, 'ARTIST=')
    if artist != None:
        c(fp.write, artist)
    c(fp.write, '\nALBUM=')
    if album != None:
        c(fp.write, album)
    c(fp.write, '\n')

    have_date = False
    for (country, date) in release_dates.items():
        have_date = True
        c(fp.write, 'DATE[%s]=%s\n' % (country, date))
    have_date or c(fp.write, 'DATE=\n')

    if len(tracks) > 0:
        trackcount = min(trackcount, len(tracks))
    for i in xrange(1, trackcount + 1):
        try:
            track = tracks.pop(0)
            title = track.title
            artist = track.artist
        except IndexError:
            title = ''
            artist = ''
        various and c(fp.write, 'ARTIST[%d]=%s\n' % (i, artist))
        c(fp.write, 'TITLE[%d]=%s\n' % (i, title))

    c(fp.close)

def cover_art(i, asin):
    url = 'http://images.amazon.com/images/P/%s.01.MZZZZZZZ.jpg' % (asin,)
    fp = file('cover.front-' + i, 'w')
    fp.write(urllib.urlopen(url).read())
    fp.close()

def tags(q, releases, trackcount):
    results = []
    seen_various = False

    tags_file('candidate-tags-0', trackcount, False)

    include = musicbrainz2.webservice.ReleaseIncludes(tracks=True)

    i = 0
    for album in releases:
        i += 1
        various = not album.release.isSingleArtistRelease()

        if various and not seen_various:
            seen_various = True
            tags_file('candidate-tags-0v', trackcount, True)

        tags_file('candidate-tags-' + str(i), trackcount, various,
                  album.release.artist.name, album.release.title,
                  album.release.getReleaseEventsAsDict(),
                  q.getReleaseById(album.release.id, include).tracks)

        cover_art(str(i), album.release.asin)

def rip(device, trackcount, single_file):
    if device == None:
        device = '/dev/cdrom'

    argv = ['cdparanoia', '-d', device, '1-' + str(trackcount)]

    if single_file:
        argv.append('wav')
    else:
        argv.append('-B')

    c(os.execvp, argv[0], argv)

def make_post_processor(command):
    if command == None:
        return

    fd = c(os.open, 'post-processor', os.O_CREAT | os.O_WRONLY, 0555)
    fp = c(os.fdopen, fd, 'w')
    c(fp.write, command +' "$@"\n')
    c(fp.close)

def releases_by_disc(q, disc):
    filter = musicbrainz2.webservice.ReleaseFilter(discId=disc.getId())
    return q.getReleases(filter)

def releases_by(q, title, artist=None):
    r = q.getReleases(musicbrainz2.webservice.ReleaseFilter(title=title))
    if artist == None:
        return r

    artist = re.sub(r'\s+', r'\s+', artist.strip())
    return [x for x in r if re.match(artist, x.release.artist.name,
                                     re.IGNORECASE) != None]

def main(argv):
    # Control the exit code for any uncaught exceptions.
    try:
        parser = OptionParser()
        parser.disable_interspersed_args()
        parser.add_option('--artist')
        parser.add_option('--title')
        parser.add_option('-d', '--device')
        parser.add_option('-m', '--no-musicbrainz',
                          action='store_true', default=False)
        parser.add_option('-p', '--post-processor')
        parser.add_option('-s', '--single-file',
                          action='store_true', default=False)
        parser.add_option('-t', '--tracks', type='int', default=99)
    except:
        traceback.print_exc()
        return 2

    try:
        # Raises SystemExit on invalid options in argv.
        (options, args) = parser.parse_args(argv[1:])
    except Exception, error:
        if isinstance(error, SystemExit):
            return 1
        traceback.print_exc()
        return 2

    try:
        device = options.device
        trackcount = options.tracks

        tempdir = c((lambda x: tempfile.mkdtemp(prefix=x, dir='.')),
                    'flac-archive.')
        c(os.chdir, tempdir)

        make_post_processor(options.post_processor)

        q = musicbrainz2.webservice.Query()
        if options.title != None:
            releases = releases_by(q, options.title, options.artist)
        else:
            disc = musicbrainz2.disc.readDisc(device)
            trackcount = mkcue(disc, trackcount)
            if options.no_musicbrainz:
                releases = []
            else:
                releases = releases_by_disc(q, disc)

        tags(q, releases, trackcount)

        if options.title == None:
            rip(device, trackcount, options.single_file)
    except Exception, error:
        if isinstance(error, SystemExit):
            raise
        # check all print_exc and format_exc in fa-flacd.py; i think
        # for some i don't do this msg print check
        sys.stderr.write(getattr(error, 'msg', ''))
        traceback.print_exc()
        return 2

    return 0

if __name__ == '__main__':
    sys.exit(main(sys.argv))