# -*- coding: iso8859-1 -*-
# PyCDDB v 0.1.1
# Written by Roland Schäuble <rschaeuble@gmx.ch>
# http://pycddb.sourceforge.net

# Modified 6 August 2004 by Marek Hnilica <aiyyo@centrum.cz>
# Added AlbumID support
# procedures in read function so it joins information in two lines with same 
# TTITLE index

""" Python CDDB-Access
Access a freedb service from Python.
"""
import urllib
import os
import getpass
import string
import re
import socket
import version
__version__ = version.mBoxVersion

class PyCDDB:
    def __init__(self, cddb_server = 'http://freedb.freedb.org/~cddb/cddb.cgi',
                 app = "mBox", version = __version__):
        self.cddb_server = cddb_server
        self.user = getpass.getuser()
        self.host = socket.gethostname()
        self.app = app
        self.version = version
        self.protocol = 5
        self.code = 0

    def status(self):
        return self.code

    def message(self):
        if self.code == 0:
            return ""
        elif self.code == 200:
            return "Found exact match"
        elif self.code == 202:
            return "No match found"
        elif self.code == 210:
            return "Ok"
        elif self.code == 211:
            return "Found inexact matches"
        elif self.code == 401:
            return "Specified CDDB entry not found"
        elif self.code == 402:
            return "Server error"
        elif self.code == 403:
            return "Database entry is corrupt"
        elif self.code == 409:
            return "No handshake"

    def query(self, discid):
        if discid != "":
            result = []
            self.track_offset = map(string.atoi, string.split(discid)[2:-1])
            self.disc_length = string.atoi(string.split(discid)[-1:][0]) * 75
            query = urllib.quote_plus(string.rstrip(discid))
            url = "%s?cmd=cddb+query+%s&hello=%s+%s+%s+%s&proto=%d" % \
                  (self.cddb_server, query, self.user, self.host, self.app,
                   self.version, self.protocol)
            response = urllib.urlopen(url)
            header = response.readline()
            if re.match("[0-9]+.*", header):
                self.code = string.atoi(string.split(header, ' ', 1)[0])
                if self.code == 200: # Exact match
                    info = string.split(header, ' ', 3)
                    result.append( { 'category': info[1], 'disc_id': info[2], 'title': info[3] } )
                elif self.code == 210 or self.code == 211: # Multiple exact mattches or inexact match
                    line = string.rstrip(response.readline())
                    while line != ".":
                        info = string.split(line, ' ', 2)
                        result.append( { 'category': info[0], 'disc_id': info[1], 'title': info[2] } )
                        line = string.rstrip(response.readline())
	    return result


    def read(self, query_item):
        result = {}
        url = "%s?cmd=cddb+read+%s+%s&hello=%s+%s+%s+%s&proto=%d" % \
              (self.cddb_server, query_item['category'], query_item['disc_id'],
               self.user, self.host, self.app, self.version, self.protocol)
        response = urllib.urlopen(url)
        header = response.readline()
        try:
		self.code = string.atoi(string.split(header, ' ', 1)[0])
		if self.code == 210:
			re_indexed_key = re.compile("([^=0123456789]+)([0-9]+)=(.*)")
			re_key = re.compile("([^=]+)=(.*)")
			previndex=-1
			ttitleedit=0
			for line in response.readlines():
				line = string.rstrip(line)
				m = re_indexed_key.match(line)
				if m:
					(key, index, data) = m.groups()
					if result.has_key(key):
						if key=='TTITLE':
							if not int(index) == previndex:
								previndex=int(index)
								result[key].append(data)
								ttitleedit=1
								continue
							else:
								result['TTITLE'][len(result['TTITLE'])-1]=result['TTITLE'][len(result['TTITLE'])-1]+data
								previndex=int(index)
						if not ttitleedit:
							result[key].append(data)
								
					else:
						result[key] = []
						result[key].append(data)
				else:
					m = re_key.match(line)
					if m:
						(key, data) = m.groups()
						result[key] = data
        except:
            result = {}
        return result



def RaiseCheckSum(value):
	increment=0
	while value>0:
		increment=increment+(value%10)
		value=value/10
	return increment
def GetAlbumId(tracks):
	TotalTime=0
	CheckSum=2
	StartOffset=150
	Offsets=[150]
	for i in range(len(tracks)):
		CurrentTime=tracks[i][0]
		TotalTime+=CurrentTime
		if not i==len(tracks)-1:
			StartOffset+=(CurrentTime*75)
			CheckSum=CheckSum+RaiseCheckSum(StartOffset)
			Offsets.append(StartOffset)
	AlbumId=(CheckSum%0xff)<<24 | TotalTime<<8 | len(tracks)
	return [AlbumId,Offsets,TotalTime]