# This isn't even really started.

import gpod
import types
import eyeD3
import gtkpod
import os

class DatabaseException(RuntimeError):
    pass

class TrackException(RuntimeError):
    pass

class Database:
    def __init__(self, mountpoint="/mnt/ipod"):
        self._itdb = itdb = gpod.itdb_parse(mountpoint,
                                            None)
        if not self._itdb:
            raise DatabaseException("Unable to parse iTunes database at mount point %s" % mountpoint)
        else:
            self._itdb.mountpoint = mountpoint

        self._load_gtkpod_extended_info()

    def __str__(self):
        return self.__repr__()
    
    def __repr__(self):
        return "<Database Filename:%s Playlists:%s Tracks:%s>" % (
            repr(self._itdb.filename),
            gpod.sw_get_list_len(self._itdb.playlists),
            len(self))

    def _load_gtkpod_extended_info(self):
        self._itdb_file    = os.path.join(self._itdb.mountpoint,
                                          "iPod_Control","iTunes","iTunesDB")
        self._itdbext_file = os.path.join(self._itdb.mountpoint,
                                          "iPod_Control","iTunes","iTunesDB.ext")
        if os.path.exists(self._itdb_file) and os.path.exists(self._itdbext_file):
            gtkpod.parse(self._itdbext_file, self, self._itdb_file)

    def close(self):
        if not gpod.itdb_write(self._itdb, None):
            raise DatabaseException("Unable to save iTunes database at %s" % mountpoint)
        gtkpod.write(self._itdbext_file, self, self._itdb_file)

    def __getitem__(self, index):
        if type(index) == types.SliceType:
            return [self[i] for i in xrange(*index.indices(len(self)))]
        else:
            if index < 0:
                index += len(self)
            return Track(proxied_track=gpod.sw_get_track(self._itdb.tracks, index))

    def __len__(self):
        return gpod.sw_get_list_len(self._itdb.tracks)

    def import_file(self, filename):
        track = Track(filename)
        track.copy_to_ipod()
        gpod.itdb_playlist_add_track(gpod.itdb_playlist_mpl(self._itdb),
                                     track._track, -1)

    def __del__(self):
        gpod.itdb_free(self._itdb)

    def add(self, track, pos=-1):
        gpod.itdb_track_add(self._itdb, track._track, pos)

    def get_master(self):
        return Playlist(self,proxied_playlist=gpod.itdb_playlist_mpl(self._itdb))

    def get_podcasts(self):
        return Playlist(self,proxied_playlist=gpod.itdb_playlist_podcasts(self._itdb))

    def get_playlists(self):
        l = []
        for playlist in gpod.sw_get_playlists(self._itdb):
            l.append(Playlist(self,proxied_playlist=playlist))
        return tuple(l)

    Master   = property(get_master)
    Podcasts = property(get_podcasts)
    Playlists= property(get_playlists) 

    def smart_update(self):
        gpod.itdb_spl_update_all(self._itdb)

class Track:
    # Note we don't free the underlying structure, as it's still used
    # by the itdb.
    
    
    _proxied_attributes = ("title","ipod_path","album","artist","genre","filetype",
                           "comment","category","composer","grouping","description",
                           "podcasturl","podcastrss","chapterdata","subtitle","id",
                           "size","tracklen","cd_nr","cds","track_nr","tracks",
                           "bitrate","samplerate","samplerate_low","year","volume",
                           "soundcheck","time_added","time_played","time_modified",
                           "bookmark_time","rating","playcount","playcount2",
                           "recent_playcount","transferred","BPM","app_rating",
                           "type1","type2","compilation","starttime","stoptime",
                           "checked","dbid","drm_userid","visible","filetype_marker",
                           "artwork_count","artwork_size","samplerate2",
                           "time_released","has_artwork","flag1","flag2","flag3","flag4",
                           "lyrics_flag","movie_flag","mark_unplayed","samplecount",
                           "chapterdata_raw","chapterdata_raw_length","artwork",
                           "usertype")

    def __init__(self, from_file=None, proxied_track=None, podcast=False):
        if from_file:
            self._track = gpod.itdb_track_new()
            self['userdata'] = {'filename_locale': from_file,
                                'transferred': 0}
            audiofile = eyeD3.Mp3AudioFile(self['userdata']['filename_locale'])
            tag = audiofile.getTag()
            for func, attrib in (('getArtist','artist'),
                                 ('getTitle','title'),
                                 ('getBPM','BPM'),
                                 ('getPlayCount','playcount'),
                                 ('getAlbum','album')):
                value = getattr(tag,func)()
                if value:
                    self[attrib] = value
            self['genre'] = tag.getGenre().name
            disc, of = tag.getDiscNum()
            if disc is not None:
                self['cd_nr'] = disc
            if of is not None:
                self['cds'] = of
            n, of = tag.getTrackNum()
            if n is not None:
                self['track_nr'] = n
            if of is not None:
                self['tracks'] = of
            self['tracklen'] = audiofile.getPlayTime() * 1000
            if podcast:
                self['flag2'] = 0x01 # skip when shuffling
                self['flag3'] = 0x01 # remember playback position
                self['flag4'] = 0x01 # Show Title/Album on the 'Now Playing' page
            else:
                self['flag2'] = 0x00 # do not skip when shuffling
                self['flag3'] = 0x00 # do not remember playback position
                self['flag4'] = 0x00 # Show Title/Album/Artist on the 'New Playing' page
            
        elif proxied_track:
            self._track = proxied_track
        else:
            self._track = gpod.itdb_track_new()
        self.set_podcast(podcast)

    def copy_to_ipod(self):
        self['userdata']['md5_hash'] = gtkpod.sha1_hash(
            self['userdata']['filename_locale'])
        self['userdata']['transferred'] = 1
        if gpod.itdb_cp_track_to_ipod(self._track,
                                      self['userdata']['filename_locale'], None) != 1:
            raise TrackException('Unable to copy %s to iPod as %s' % (
                self['userdata']['filename_locale'],
                self))

    def ipod_filename(self):
        return gpod.itdb_filename_on_ipod(self._track)

    def set_podcast(self, value):
        if value:
            self['flag1'] = 0x02 # unknown
            self['flag2'] = 0x01 # skip when shuffling
            self['flag3'] = 0x01 # remember playback position
            self['flag4'] = 0x01 # Show Title/Album on the 'Now Playing' page            
        else:
            self['flag1'] = 0x02 # unknown
            self['flag2'] = 0x00 # do not skip when shuffling
            self['flag3'] = 0x00 # do not remember playback position
            self['flag4'] = 0x00 # Show Title/Album/Artist on the 'New Playing' page

    def __str__(self):
        return self.__repr__()
    
    def __repr__(self):
        return "<Track Artist:%s Title:%s Album:%s>" % (
            repr(self['artist']),
            repr(self['title']),
            repr(self['album']))

    def keys(self):
        return list(self._proxied_attributes)

    def items(self):
        return [self[k] for k in self._proxied_attributes]

    def pairs(self):
        return [(k, self[k]) for k in self._proxied_attributes]        

    def __getitem__(self, item):
        if item == "userdata":
            return gpod.sw_get_track_userdata(self._track)
        elif item in self._proxied_attributes:
            return getattr(self._track, item)
        else:
            raise KeyError('No such key: %s' % item)

    def __setitem__(self, item, value):
        #print item, value
        if item == "userdata":
            gpod.sw_set_track_userdata(self._track, value)
            return
        if type(value) == types.UnicodeType:
            value = value.encode()
        if item in self._proxied_attributes:
            return setattr(self._track, item, value)
        else:
            raise KeyError('No such key: %s' % item)

_playlist_sorting = {
    1:'playlist',
    2:'unknown2',
    3:'songtitle',
    4:'album',
    5:'artist',
    6:'bitrate',
    7:'genre',
    8:'kind',
    9:'modified',
    10:'track',
    11:'size',
    12:'time',
    13:'year',
    14:'rate',
    15:'comment',
    16:'added',
    17:'equalizer',
    18:'composer',
    19:'unknown19',
    20:'count',
    21:'last',
    22:'disc',
    23:'rating',
    24:'release',
    25:'BPM',
    26:'grouping',
    27:'category',
    28:'description'}
        
class Playlist:
    def __init__(self, parent_db, proxied_playlist=None,
                 title="New Playlist", smart=False, pos=-1):
        self._db = parent_db
        if proxied_playlist:
            self._pl = proxied_playlist
        else:
            if smart:
                smart = 1
            else:
                smart = 0
            self._pl = gpod.itdb_playlist_new(title, smart)
            gpod.itdb_playlist_add(self._db._itdb, self._pl, pos)

    def smart_update(self):
        gpod.itdb_spl_update(self._pl)

    def randomize(self):
        gpod.itdb_playlist_randomize(self._pl)

    def get_name(self):
        return self._pl.name
    def set_name(self, name):
        self._pl.name = name
    def get_id(self):
        return self._pl.id
    def get_smart(self):
        if self._pl.is_spl == 1:
            return True
        return False
    def get_master(self):
        if gpod.itdb_playlist_is_mpl(self._pl) == 1:
            return True
        return False
    def get_podcast(self):
        if gpod.itdb_playlist_is_podcasts(self._pl) == 1:
            return True
        return False
    def get_sort(self):
        return _playlist_sorting[self._pl.sortorder]
    def set_sort(self, order):
        order = order.lower()
        for k, v in _playlist_sorting.items():
            if v == order:
                self._pl.sortorder = v
                return
        return ValueError("Unknown playlist sorting '%s'" % order)

    name   = property(get_name, set_name)
    id     = property(get_id)
    smart  = property(get_smart)
    master = property(get_master)
    podcast= property(get_podcast)    
    order  = property(get_sort, set_sort)

    def __str__(self):
        return self.__repr__()
    
    def __repr__(self):
        return "<Playlist Title:%s Sort:%s Smart:%s Master:%s Podcast:%s Tracks:%d>" % (
            repr(self.name),
            repr(self.order),
            repr(self.smart),
            repr(self.master),
            repr(self.podcast),
            len(self))
        
    def __getitem__(self, index):
        if type(index) == types.SliceType:
            return [self[i] for i in xrange(*index.indices(len(self)))]
        else:
            if index < 0:
                index += len(self)
            return Track(proxied_track=gpod.sw_get_track(self._pl.members, index))

    def __len__(self):
        #return self._pl.num # Always 0 ?
        return gpod.sw_get_list_len(self._pl.members)

    def add(self, track, pos=-1):
        gpod.itdb_playlist_add_track(self._pl, track._track, pos)