#! /usr/bin/python

# Copyright (C) 2005 Ross Burton <ross@burtonini.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

import os, sys, datetime, ConfigParser

import musicbrainz
from musicbrainz.queries import *

import scrobbler

config = ConfigParser.SafeConfigParser()
config.read(os.path.expanduser('~/.cdscrobbler'))

if not config.has_option("CDScrobbler", "username") or not config.has_option("CDScrobbler", "password"):
    print "No complete configuration file found, see the README."
    sys.exit(1)

current = datetime.datetime.utcnow()

scrob = scrobbler.Scrobbler(config.get("CDScrobbler", "username"),
                            config.get("CDScrobbler", "password"),
                            client="cds", version="0.1")
scrob.handshake()

mb = musicbrainz.mb()

def upload_album():
    global current
    count = mb.GetResultInt(MBE_GetNumAlbums)
    if count == 0:
        print "Found no albums"
        return
    if count > 1:
        print "TODO: found %d albums (skipping)" % count
        return

    tracks = []

    mb.Select1(MBS_SelectAlbum, 1)
    album = mb.GetResultData(MBE_AlbumGetAlbumName)
    for ii in range(1, mb.GetResultInt(MBE_AlbumGetNumTracks) + 1):
        mbid = mb.GetIDFromURL(mb.GetResultData1(MBE_AlbumGetTrackId, ii))
        artist = mb.GetResultData1(MBE_AlbumGetArtistName, ii)
        name = mb.GetResultData1(MBE_AlbumGetTrackName, ii)
        dura = mb.GetResultInt1(MBE_AlbumGetTrackDuration, ii) / 1000
        track = mb.GetResultInt1(MBE_AlbumGetTrackNum, ii)
        tracks.append(scrobbler.Track(artist, name, album, dura, mbid, None))

    # Fix up times on tracks
    tracks.reverse()
    for t in tracks:
        current = current - datetime.timedelta(seconds=t.length)
        t.tracktime = current
    tracks.reverse()
    
    # Submit the songs to Audioscrobbler
    scrob.submit(tracks)


if len(sys.argv) >= 2:
    mb.SetDepth(4)
    for i in reversed(sys.argv[1:]):
        mb.QueryWithArgs(MBQ_GetAlbumById, [i])
        upload_album()
else:
    mb.SetDepth(2)
    mb.Query(MBQ_GetCDTOC)
    cdid = mb.GetResultData(MBE_TOCGetCDIndexId)
    mb.QueryWithArgs(MBQ_GetCDInfoFromCDIndexId, [cdid])
    upload_album()