default.py

   1 import urllib, sys, os, shutil, re, xbmcaddon, xbmcplugin, xbmcgui, xbmc
   2
   3 MODE_SEARCH_SONGS = 1
   4 MODE_SEARCH_ALBUMS = 2
   5 MODE_SEARCH_ARTISTS = 3
   6 MODE_POPULAR_SONGS = 4
   7 MODE_FAVORITES = 5
   8 MODE_PLAYLISTS = 6
   9 MODE_ALBUM = 7
  10 MODE_ARTIST = 8
  11 MODE_PLAYLIST = 9
  12 MODE_SIMILAR_ARTISTS = 10
  13 MODE_SONG = 11
  14 MODE_FAVORITE = 12
  15 MODE_UNFAVORITE = 13
  16 MODE_MAKE_PLAYLIST = 14
  17 MODE_REMOVE_PLAYLIST = 15
  18 MODE_RENAME_PLAYLIST = 16
  19 MODE_REMOVE_PLAYLIST_SONG = 17
  20 MODE_ADD_PLAYLIST_SONG = 18
  21
  22 ACTION_MOVE_LEFT = 1
  23 ACTION_MOVE_UP = 3
  24 ACTION_MOVE_DOWN = 4
  25 ACTION_PAGE_UP = 5
  26 ACTION_PAGE_DOWN = 6
  27 ACTION_SELECT_ITEM = 7
  28 ACTION_PREVIOUS_MENU = 10
  29
  30 baseDir = os.getcwd()
  31 resDir = xbmc.translatePath(os.path.join(baseDir, 'resources'))
  32 libDir = xbmc.translatePath(os.path.join(resDir,  'lib'))
  33 imgDir = xbmc.translatePath(os.path.join(resDir,  'img'))
  34 thumbDir = os.path.join('special://masterprofile/addon_data/', os.path.join(os.path.basename(os.getcwd()), 'thumb'))
  35
  36 baseModeUrl = 'plugin://plugin.audio.groove/'
  37 playlistUrl = baseModeUrl + '?mode=' + str(MODE_PLAYLIST)
  38 playlistsUrl = baseModeUrl + '?mode=' + str(MODE_PLAYLISTS)
  39 favoritesUrl = baseModeUrl + '?mode=' + str(MODE_FAVORITES)
  40
  41 thumbDef = os.path.join(os.path.basename(os.getcwd()), 'default.tbn')
  42 listBackground = os.path.join(imgDir, 'listbackground.png')
  43
  44 sys.path.append (libDir)
  45 from GrooveAPI import GrooveAPI
  46 groovesharkApi = GrooveAPI()
  47
  48 class _Info:
  49     def __init__( self, *args, **kwargs ):
  50         self.__dict__.update( kwargs )
  51
  52 # Window dialog to select a grooveshark playlist
  53 class GroovesharkPlaylistSelect(xbmcgui.WindowDialog):
  54
  55     def __init__(self, items=[]):
  56         gap = int(self.getHeight()/100)
  57         w = int(self.getWidth()*0.5)
  58         h = self.getHeight()-30*gap
  59         rw = self.getWidth()
  60         rh = self.getHeight()
  61         x = rw/2 - w/2
  62         y = rh/2 -h/2
  63
  64         self.imgBg = xbmcgui.ControlImage(x+gap, 5*gap+y, w-2*gap, h-5*gap, listBackground)
  65         self.addControl(self.imgBg)
  66
  67         self.playlistControl = xbmcgui.ControlList(2*gap+x, y+3*gap+30, w-4*gap, h-10*gap, textColor='0xFFFFFFFF', selectedColor='0xFFFF4242', itemTextYOffset=0, itemHeight=50, alignmentY = 0)
  68         self.addControl(self.playlistControl)
  69
  70         self.lastPos = 0
  71         self.isSelecting = False
  72         self.selected = -1
  73         listitems = []
  74         for playlist in items:
  75             listitems.append(xbmcgui.ListItem(playlist[0]))
  76         listitems.append(xbmcgui.ListItem('New...'))
  77         self.playlistControl.addItems(listitems)
  78         self.setFocus(self.playlistControl)
  79         self.playlistControl.selectItem(0)
  80         item = self.playlistControl.getListItem(self.lastPos)
  81         item.select(True)
  82
  83     # Highlight selected item
  84     def setHighlight(self):
  85         if self.isSelecting:
  86             return
  87         else:
  88             self.isSelecting = True
  89
  90         pos = self.playlistControl.getSelectedPosition()
  91         if pos >= 0:
  92             item = self.playlistControl.getListItem(self.lastPos)
  93             item.select(False)
  94             item = self.playlistControl.getListItem(pos)
  95             item.select(True)
  96             self.lastPos = pos
  97         self.isSelecting = False
  98
  99     # Control - select
 100     def onControl(self, control):
 101         if control == self.playlistControl:
 102             self.selected = self.playlistControl.getSelectedPosition()
 103             self.close()
 104
 105     # Action - close or up/down
 106     def onAction(self, action):
 107         if action == ACTION_PREVIOUS_MENU:
 108             self.selected = -1
 109             self.close()
 110         elif action == ACTION_MOVE_UP or action == ACTION_MOVE_DOWN or action == ACTION_PAGE_UP or action == ACTION_PAGE_DOWN == 6:
 111             self.setFocus(self.playlistControl)
 112             self.setHighlight()
 113
 114 class Groveshark:
 115
 116     albumImg = xbmc.translatePath(os.path.join(imgDir, 'album.png'))
 117     artistImg = xbmc.translatePath(os.path.join(imgDir, 'artist.png'))
 118     favoritesImg = xbmc.translatePath(os.path.join(imgDir, 'favorites.png'))
 119     playlistImg = xbmc.translatePath(os.path.join(imgDir, 'playlist.png'))
 120     popularSongsImg = xbmc.translatePath(os.path.join(imgDir, 'popularSongs.png'))
 121     songImg = xbmc.translatePath(os.path.join(imgDir, 'song.png'))
 122     defImg = xbmc.translatePath(os.path.join(imgDir, 'default.tbn'))
 123     fanImg = xbmc.translatePath(os.path.join(baseDir, 'fanart.png'))
 124
 125     settings = xbmcaddon.Addon(id='plugin.audio.groove')
 126     songsearchlimit = settings.getSetting('songsearchlimit')
 127     albumsearchlimit = settings.getSetting('albumsearchlimit')
 128     artistsearchlimit = settings.getSetting('artistsearchlimit')
 129     username = settings.getSetting('username')
 130     password = settings.getSetting('password')
 131     userid = 0
 132
 133     def __init__( self ):
 134         self._handle = int(sys.argv[1])
 135
 136     # Top-level menu
 137     def categories(self):
 138
 139         self.userid = self._get_login()
 140
 141         # Setup
 142         xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
 143         if os.path.isdir(thumbDir) == False:
 144             os.makedirs(thumbDir)
 145
 146         self._add_dir('Search songs...', '', MODE_SEARCH_SONGS, self.songImg, 0)
 147         self._add_dir('Search albums...', '', MODE_SEARCH_ALBUMS, self.albumImg, 0)
 148         self._add_dir('Search artists...', '', MODE_SEARCH_ARTISTS, self.artistImg, 0)
 149         self._add_dir('Popular songs', '', MODE_POPULAR_SONGS, self.popularSongsImg, 0)
 150         if (self.userid != 0):
 151             self._add_dir('Favorites', '', MODE_FAVORITES, self.favoritesImg, 0)
 152             self._add_dir('Playlists', '', MODE_PLAYLISTS, self.playlistImg, 0)
 153
 154     # Search for songs
 155     def searchSongs(self):
 156         query = self._get_keyboard(default="", heading="Search songs")
 157         if (query != ''):
 158             groovesharkApi.setRemoveDuplicates(True)
 159             songs = groovesharkApi.searchSongs(query, limit = self.songsearchlimit)
 160             if (len(songs) > 0):
 161                 self._add_songs_directory(songs)
 162             else:
 163                 dialog = xbmcgui.Dialog()
 164                 dialog.ok('Grooveshark XBMC', 'No matching songs.')
 165                 self.categories()
 166         else:
 167             self.categories()
 168
 169     # Search for albums
 170     def searchAlbums(self):
 171         query = self._get_keyboard(default="", heading="Search albums")
 172         if (query != ''):
 173             albums = groovesharkApi.searchAlbums(query, limit = self.albumsearchlimit)
 174             if (len(albums) > 0):
 175                 self._add_albums_directory(albums)
 176             else:
 177                 dialog = xbmcgui.Dialog()
 178                 dialog.ok('Grooveshark XBMC', 'No matching albums.')
 179                 self.categories()
 180         else:
 181             self.categories()
 182
 183     # Search for artists
 184     def searchArtists(self):
 185         query = self._get_keyboard(default="", heading="Search artists")
 186         if (query != ''):
 187             artists = groovesharkApi.searchArtists(query, limit = self.artistsearchlimit)
 188             if (len(artists) > 0):
 189                 self._add_artists_directory(artists)
 190             else:
 191                 dialog = xbmcgui.Dialog()
 192                 dialog.ok('Grooveshark XBMC', 'No matching artists.')
 193                 self.categories()
 194         else:
 195             self.categories()
 196
 197     # Get my favorites
 198     def favorites(self):
 199         userid = self._get_login()
 200         if (userid != 0):
 201             favorites = groovesharkApi.userGetFavoriteSongs(userid)
 202             if (len(favorites) > 0):
 203                 self._add_songs_directory(favorites, isFavorites=True)
 204             else:
 205                 dialog = xbmcgui.Dialog()
 206                 dialog.ok('Grooveshark XBMC', 'You have no favorites.')
 207                 self.categories()
 208
 209     # Get popular songs
 210     def popularSongs(self):
 211         popular = groovesharkApi.popularGetSongs(limit = self.songsearchlimit)
 212         if (len(popular) > 0):
 213             self._add_songs_directory(popular)
 214         else:
 215             dialog = xbmcgui.Dialog()
 216             dialog.ok('Grooveshark XBMC', 'No popular songs.')
 217             self.categories()
 218
 219     # Get my playlists
 220     def playlists(self):
 221         userid = self._get_login()
 222         if (userid != 0):
 223             playlists = groovesharkApi.userGetPlaylists()
 224             if (len(playlists) > 0):
 225                 self._add_playlists_directory(playlists)
 226             else:
 227                 dialog = xbmcgui.Dialog()
 228                 dialog.ok('Grooveshark XBMC', 'You have no Grooveshark playlists.')
 229                 self.categories()
 230         else:
 231             dialog = xbmcgui.Dialog()
 232             dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to see your Grooveshark playlists.')
 233
 234     # Make songs a favorite
 235     def favorite(self, songid):
 236         userid = self._get_login()
 237         if (userid != 0):
 238             xbmc.log("Favorite song: " + str(songid))
 239             groovesharkApi.favoriteSong(songID = songid)
 240             xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Added to favorites, 1000, ' + thumbDef + ')')
 241         else:
 242             dialog = xbmcgui.Dialog()
 243             dialog.ok('Grooveshark XBMC', 'You must be logged in', 'to add favorites.')
 244
 245     # Remove song from favorites
 246     def unfavorite(self, songid, prevMode=0):
 247         userid = self._get_login()
 248         if (userid != 0):
 249             xbmc.log("Unfavorite song: " + str(songid) + ', previous mode was ' + str(prevMode))
 250             groovesharkApi.unfavoriteSong(songID = songid)
 251             xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Removed from Grooveshark favorites, 1000, ' + thumbDef + ')')
 252             # Refresh to remove item from directory
 253             if (int(prevMode) == MODE_FAVORITES):
 254                 xbmc.executebuiltin("Container.Refresh(" + favoritesUrl + ")")
 255         else:
 256             dialog = xbmcgui.Dialog()
 257             dialog.ok('Grooveshark XBMC', 'You must be logged in', 'to remove Grooveshark favorites.')
 258
 259     # 'Frown' a song
 260     def frown(self, songid):
 261         userid = self._get_login()
 262         if (userid != 0):
 263             xbmc.log("Frown song: " + str(songid))
 264             if groovesharkApi.radioFrown(songId = songid) != True:
 265                 xbmc.log("Unable to frown song " + str(songid))
 266             else:
 267                 xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Frowned, 1000, ' + thumbDef + ')')
 268         else:
 269             dialog = xbmcgui.Dialog()
 270             dialog.ok('Grooveshark XBMC', 'You must be logged in', 'to frown a Grooveshark song.')
 271
 272     # Find similar artists to searched artist
 273     def similarArtists(self, artistId):
 274         similar = groovesharkApi.artistGetSimilar(artistId, limit = self.artistsearchlimit)
 275         if (len(similar) > 0):
 276             self._add_artists_directory(similar)
 277         else:
 278             dialog = xbmcgui.Dialog()
 279             dialog.ok('Grooveshark XBMC', 'No similar artists.')
 280             self.categories()
 281
 282     # Make a playlist from an album
 283     def makePlaylist(self, albumid, name):
 284         userid = self._get_login()
 285         if (userid != 0):
 286             re.split(' - ',name,1)
 287             nameTokens = re.split(' - ',name,1) # suggested name
 288             name = self._get_keyboard(default=nameTokens[0], heading="Grooveshark playlist name")
 289             if name != '':
 290                 groovesharkApi.setRemoveDuplicates(True)
 291                 album = groovesharkApi.albumGetSongs(albumid, self.songsearchlimit)
 292                 songids = []
 293                 for song in album:
 294                     songids.append(song[1])
 295                 if groovesharkApi.playlistCreateUnique(name, songids) == 0:
 296                     dialog = xbmcgui.Dialog()
 297                     dialog.ok('Grooveshark XBMC', 'Cannot create Grooveshark playlist ', name)
 298                 else:
 299                     xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Grooveshark playlist created, 1000, ' + thumbDef + ')')
 300         else:
 301             dialog = xbmcgui.Dialog()
 302             dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to create a Grooveshark playlist.')
 303
 304     # Rename a playlist
 305     def renamePlaylist(self, playlistid, name):
 306         userid = self._get_login()
 307         if (userid != 0):
 308             newname = self._get_keyboard(default=name, heading="Grooveshark playlist name")
 309             if newname == '':
 310                 return
 311             elif groovesharkApi.playlistRename(playlistid, newname) == 0:
 312                 dialog = xbmcgui.Dialog()
 313                 dialog.ok('Grooveshark XBMC', 'Cannot rename Grooveshark playlist ', name)
 314             else:
 315                 # Refresh to show new item name
 316                 xbmc.executebuiltin("Container.Refresh")
 317         else:
 318             dialog = xbmcgui.Dialog()
 319             dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to rename a Grooveshark playlist.')
 320
 321     # Remove a playlist
 322     def removePlaylist(self, playlistid, name):
 323         dialog = xbmcgui.Dialog()
 324         if dialog.yesno('Grooveshark XBMC', name, 'Delete this Grooveshark playlist?') == True:
 325             userid = self._get_login()
 326             if (userid != 0):
 327                 if groovesharkApi.playlistDelete(playlistid) == 0:
 328                     dialog = xbmcgui.Dialog()
 329                     dialog.ok('Grooveshark XBMC', 'Cannot remove Grooveshark playlist ', name)
 330                 else:
 331                     # Refresh to remove item from directory
 332                     xbmc.executebuiltin("Container.Refresh(" + playlistsUrl + ")")
 333             else:
 334                 dialog = xbmcgui.Dialog()
 335                 dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to delete a Grooveshark playlist.')
 336
 337     # Add song to playlist
 338     def addPlaylistSong(self, songid):
 339         userid = self._get_login()
 340         if (userid != 0):
 341             playlists = groovesharkApi.userGetPlaylists()
 342             if (len(playlists) > 0):
 343                 ret = 0
 344                 # Select the playlist
 345                 playlistSelect = GroovesharkPlaylistSelect(items=playlists)
 346                 playlistSelect.setFocus(playlistSelect.playlistControl)
 347                 playlistSelect.doModal()
 348                 i = playlistSelect.selected
 349                 del playlistSelect
 350                 if i > -1:
 351                     # Add a new playlist
 352                     if i >= len(playlists):
 353                         name = self._get_keyboard(default='', heading="Grooveshark playlist name")
 354                         if name != '':
 355                             songIds = []
 356                             songIds.append(songid)
 357                             if groovesharkApi.playlistCreateUnique(name, songIds) == 0:
 358                                 dialog = xbmcgui.Dialog()
 359                                 dialog.ok('Grooveshark XBMC', 'Cannot create Grooveshark playlist ', name)
 360                             else:
 361                                 xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Grooveshark playlist created, 1000, ' + thumbDef + ')')
 362                     # Existing playlist
 363                     else:
 364                         playlist = playlists[i]
 365                         playlistid = playlist[1]
 366                         xbmc.log("Add song " + str(songid) + " to playlist " + str(playlistid))
 367                         ret = groovesharkApi.playlistAddSong(playlistid, songid, 0)
 368                         if ret == 0:
 369                             dialog = xbmcgui.Dialog()
 370                             dialog.ok('Grooveshark XBMC', 'Cannot add to playlist ')
 371                         else:
 372                             xbmc.executebuiltin('XBMC.Notification(Grooveshark XBMC, Added song to Grooveshark playlist, 1000, ' + thumbDef + ')')
 373             else:
 374                 dialog = xbmcgui.Dialog()
 375                 dialog.ok('Grooveshark XBMC', 'You have no Grooveshark playlists.')
 376                 self.categories()
 377         else:
 378             dialog = xbmcgui.Dialog()
 379             dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to add a song to a Grooveshark playlist.')
 380
 381     # Remove song from playlist
 382     def removePlaylistSong(self, playlistid, playlistname, songpos):
 383         dialog = xbmcgui.Dialog()
 384         if dialog.yesno('Grooveshark XBMC', 'Delete this song from the Grooveshark playlist?') == True:
 385             userid = self._get_login()
 386             if (userid != 0):
 387                 if groovesharkApi.playlistDeleteSong(playlistid, songpos) == 0:
 388                     dialog = xbmcgui.Dialog()
 389                     dialog.ok('Grooveshark XBMC', 'Failed to remove ', ' song from Grooveshark playlist.')
 390                 else:
 391                     # Refresh to remove item from directory
 392                     xbmc.executebuiltin("Container.Refresh(" + playlistUrl + "&id="+str(playlistid) + "&name=" + playlistname + ")")
 393             else:
 394                 dialog = xbmcgui.Dialog()
 395                 dialog.ok('Grooveshark XBMC', 'You must be logged in ', ' to delete a song from a Grooveshark playlist.')
 396
 397     # Show selected album
 398     def album(self, albumid):
 399         groovesharkApi.setRemoveDuplicates(True)
 400         album = groovesharkApi.albumGetSongs(albumId = albumid, limit = self.songsearchlimit)
 401         self._add_songs_directory(album)
 402
 403     # Show selected artist
 404     def artist(self, artistid):
 405         albums = groovesharkApi.artistGetAlbums(artistId = artistid, limit = self.albumsearchlimit)
 406         self._add_albums_directory(albums, artistid)
 407
 408     # Show selected playlist
 409     def playlist(self, playlistid, playlistname):
 410         userid = self._get_login()
 411         if (userid != 0):
 412             songs = groovesharkApi.playlistGetSongs(playlistId = playlistid, limit = self.songsearchlimit)
 413             self._add_songs_directory(songs, playlistid, playlistname)
 414         else:
 415             dialog = xbmcgui.Dialog()
 416             dialog.ok('Grooveshark XBMC', 'You must be logged in', 'to get Grooveshark playlists.')
 417
 418     # Play a song
 419     def playSong(self, item):
 420         url = groovesharkApi.getStreamURL(item.getProperty('id'))
 421         item.setPath(url)
 422         xbmc.log("Playing: " + url)
 423         xbmcplugin.setResolvedUrl(handle=int(sys.argv[1]), succeeded=True, listitem=item)
 424
 425     # Make a song directory item
 426     def songItem(self, id, name, album, artist, duration, thumb, image):
 427         # Only try to get the image
 428         if image != "":
 429             songImg = self._get_icon(image, 'song-' + str(id) + "-image")
 430             songThm = songImg
 431         else:
 432             songThm = self._get_icon(thumb, 'song-' + str(id) + "-thumb")
 433             songImg = songThm
 434         item = xbmcgui.ListItem(label = artist + " - " + album + " - " + name, thumbnailImage=songThm, iconImage=songImg)
 435         item.setInfo( type="music", infoLabels={ "title": name, "duration": duration, "album": album, "artist": artist} )
 436         item.setProperty('mimetype', 'audio/mpeg')
 437         item.setProperty("IsPlayable", "true")
 438         item.setProperty('id', str(id))
 439         item.setProperty('thumb', songThm)
 440         item.setProperty('image', songImg)
 441         return item
 442
 443     # Get keyboard input
 444     def _get_keyboard(self, default="", heading="", hidden=False):
 445         kb = xbmc.Keyboard(default, heading, hidden)
 446         kb.doModal()
 447         if (kb.isConfirmed()):
 448             return unicode(kb.getText(), "utf-8")
 449         return ''
 450
 451     # Login to grooveshark
 452     def _get_login(self):
 453         if (self.username == "" or self.password == ""):
 454             dialog = xbmcgui.Dialog()
 455             dialog.ok('Grooveshark XBMC', 'Unable to login.', 'Check username and password in settings.')
 456             return 0
 457         else:
 458             if self.userid == 0:
 459                 uid = groovesharkApi.loginExt(self.username, self.password)
 460             if (uid != 0):
 461                 return uid
 462             else:
 463                 dialog = xbmcgui.Dialog()
 464                 dialog.ok('Grooveshark XBMC', 'Unable to login.', 'Check username and password in settings.')
 465                 return 0
 466
 467     # Get a song directory item
 468     def _get_song_item(self, song):
 469         name = song[0]
 470         id = song[1]
 471         duration = song[2]
 472         album = song[3]
 473         artist = song[6]
 474         thumb = song[8]
 475         image = song[9]
 476         return self.songItem(id, name, album, artist, duration, thumb, image)
 477
 478     # File download
 479     def _get_icon(self, url, id):
 480         localThumb = os.path.join(xbmc.translatePath(os.path.join(thumbDir, str(id)))) + '.tbn'
 481         try:
 482             if os.path.isfile(localThumb) == False:
 483                 xbmc.log('Downloading ' + url + ' to ' + localThumb)
 484                 loc = urllib.URLopener()
 485                 loc.retrieve(url, localThumb)
 486         except:
 487             shutil.copy2(self.defImg, localThumb)
 488
 489         return os.path.join(os.path.join(thumbDir, str(id))) + '.tbn'
 490
 491     # Add songs to directory
 492     def _add_songs_directory(self, songs, playlistid=0, playlistname='', isFavorites=False):
 493         n = len(songs)
 494         xbmc.log("Found " + str(n) + " songs...")
 495         i = 0
 496         while i < n:
 497             song = songs[i]
 498             item = self._get_song_item(song)
 499             songthumb = item.getProperty('thumb')
 500             songimage = item.getProperty('image')
 501             songname = song[0]
 502             songid = song[1]
 503             songduration = song[2]
 504             songalbum = song[3]
 505             songartist = song[6]
 506             u=sys.argv[0]+"?mode="+str(MODE_SONG)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid) \
 507             +"&album="+urllib.quote_plus(songalbum) \
 508             +"&artist="+urllib.quote_plus(songartist) \
 509             +"&duration="+str(songduration) \
 510             +"&thumb="+urllib.quote_plus(songthumb) \
 511             +"&image="+urllib.quote_plus(songimage)
 512             fav=sys.argv[0]+"?mode="+str(MODE_FAVORITE)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid)
 513             unfav=sys.argv[0]+"?mode="+str(MODE_UNFAVORITE)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid)+"&prevmode="
 514             menuItems = []
 515             if isFavorites == True:
 516                 unfav = unfav +str(MODE_FAVORITES)
 517             else:
 518                 menuItems.append(("Grooveshark favorite", "XBMC.RunPlugin("+fav+")"))
 519             menuItems.append(("Not Grooveshark favorite", "XBMC.RunPlugin("+unfav+")"))
 520             if playlistid > 0:
 521                 rmplaylstsong=sys.argv[0]+"?playlistid="+str(playlistid)+"&id="+str(i+1)+"&mode="+str(MODE_REMOVE_PLAYLIST_SONG)+"&name="+playlistname
 522                 menuItems.append(("Remove from Grooveshark playlist", "XBMC.RunPlugin("+rmplaylstsong+")"))
 523             else:
 524                 addplaylstsong=sys.argv[0]+"?id="+str(songid)+"&mode="+str(MODE_ADD_PLAYLIST_SONG)
 525                 menuItems.append(("Add to Grooveshark playlist", "XBMC.RunPlugin("+addplaylstsong+")"))
 526             item.addContextMenuItems(menuItems, replaceItems=False)
 527             xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]),url=u,listitem=item,isFolder=False, totalItems=n)
 528             i = i + 1
 529
 530         xbmcplugin.setContent(self._handle, 'songs')
 531         xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
 532
 533     # Add albums to directory
 534     def _add_albums_directory(self, albums, artistid=0):
 535         n = len(albums)
 536         xbmc.log("Found " + str(n) + " albums...")
 537         i = 0
 538         while i < n:
 539             album = albums[i]
 540             albumArtistName = album[0]
 541             albumName = album[2]
 542             albumID = album[3]
 543             albumImage = self._get_icon(album[4], 'album-' + str(albumID))
 544             self._add_dir(albumName + " - " + albumArtistName, '', MODE_ALBUM, albumImage, albumID, n)
 545             i = i + 1
 546         if artistid > 0:
 547             self._add_dir('Similar artists...', '', MODE_SIMILAR_ARTISTS, self.artistImg, artistid)
 548         xbmcplugin.setContent(self._handle, 'albums')
 549         xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_ALBUM_IGNORE_THE)
 550         xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
 551
 552     # Add artists to directory
 553     def _add_artists_directory(self, artists):
 554         n = len(artists)
 555         xbmc.log("Found " + str(n) + " artists...")
 556         i = 0
 557         while i < n:
 558             artist = artists[i]
 559             artistName = artist[0]
 560             artistID = artist[1]
 561             self._add_dir(artistName, '', MODE_ARTIST, self.artistImg, artistID, n)
 562             i = i + 1
 563         xbmcplugin.setContent(self._handle, 'artists')
 564         xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_ARTIST_IGNORE_THE)
 565         xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
 566
 567     # Add playlists to directory
 568     def _add_playlists_directory(self, playlists):
 569         n = len(playlists)
 570         xbmc.log("Found " + str(n) + " playlists...")
 571         i = 0
 572         while i < n:
 573             playlist = playlists[i]
 574             playlistName = playlist[0]
 575             playlistID = playlist[1]
 576             self._add_dir(playlistName, '', MODE_PLAYLIST, self.playlistImg, playlistID, n)
 577             i = i + 1
 578         xbmcplugin.setContent(self._handle, 'files')
 579         xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_PLAYLIST_ORDER)
 580         xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
 581
 582     # Add whatever directory
 583     def _add_dir(self, name, url, mode, iconimage, id, items=1):
 584
 585         u=sys.argv[0]+"?mode="+str(mode)+"&name="+urllib.quote_plus(name)+"&id="+str(id)
 586         dir=xbmcgui.ListItem(name, iconImage=iconimage, thumbnailImage=iconimage)
 587         dir.setInfo( type="Music", infoLabels={ "title": name } )
 588
 589         # Custom menu items
 590         menuItems = []
 591         if mode == MODE_ALBUM:
 592             mkplaylst=sys.argv[0]+"?mode="+str(MODE_MAKE_PLAYLIST)+"&name="+name+"&id="+str(id)
 593             menuItems.append(("Make Grooveshark playlist", "XBMC.RunPlugin("+mkplaylst+")"))
 594         if mode == MODE_PLAYLIST:
 595             rmplaylst=sys.argv[0]+"?mode="+str(MODE_REMOVE_PLAYLIST)+"&name="+urllib.quote_plus(name)+"&id="+str(id)
 596             menuItems.append(("Delete Grooveshark playlist", "XBMC.RunPlugin("+rmplaylst+")"))
 597             mvplaylst=sys.argv[0]+"?mode="+str(MODE_RENAME_PLAYLIST)+"&name="+urllib.quote_plus(name)+"&id="+str(id)
 598             menuItems.append(("Rename Grooveshark playlist", "XBMC.RunPlugin("+mvplaylst+")"))
 599
 600         dir.addContextMenuItems(menuItems, replaceItems=False)
 601         return xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]),url=u,listitem=dir,isFolder=True, totalItems=items)
 602
 603
 604 # Parse URL parameters
 605 def get_params():
 606     param=[]
 607     paramstring=sys.argv[2]
 608     xbmc.log(paramstring)
 609     if len(paramstring)>=2:
 610         params=sys.argv[2]
 611         cleanedparams=params.replace('?','')
 612         if (params[len(params)-1]=='/'):
 613             params=params[0:len(params)-2]
 614         pairsofparams=cleanedparams.split('&')
 615         param={}
 616         for i in range(len(pairsofparams)):
 617             splitparams={}
 618             splitparams=pairsofparams[i].split('=')
 619             if (len(splitparams))==2:
 620                 param[splitparams[0]]=splitparams[1]
 621     return param
 622
 623 # Main
 624 grooveshark = Groveshark();
 625
 626 params=get_params()
 627 mode=None
 628 try: mode=int(params["mode"])
 629 except: pass
 630
 631 if mode > MODE_FAVORITES:
 632     name=None
 633     id=None
 634     try: name=urllib.unquote_plus(params["name"])
 635     except: pass
 636     try: id=int(params["id"])
 637     except: pass
 638
 639 # Call function for URL
 640 if mode==None:
 641     grooveshark.categories()
 642
 643 elif mode==MODE_SEARCH_SONGS:
 644     grooveshark.searchSongs()
 645
 646 elif mode==MODE_SEARCH_ALBUMS:
 647     grooveshark.searchAlbums()
 648
 649 elif mode==MODE_SEARCH_ARTISTS:
 650     grooveshark.searchArtists()
 651
 652 elif mode==MODE_POPULAR_SONGS:
 653     grooveshark.popularSongs()
 654
 655 elif mode==MODE_FAVORITES:
 656     grooveshark.favorites()
 657
 658 elif mode==MODE_PLAYLISTS:
 659     grooveshark.playlists()
 660
 661 elif mode==MODE_SONG:
 662     try: album=urllib.unquote_plus(params["album"])
 663     except: pass
 664     try: artist=urllib.unquote_plus(params["artist"])
 665     except: pass
 666     try: duration=int(params["duration"])
 667     except: pass
 668     try: thumb=urllib.unquote_plus(params["thumb"])
 669     except: pass
 670     try: image=urllib.unquote_plus(params["image"])
 671     except: pass
 672     song = grooveshark.songItem(id, name, album, artist, duration, thumb, image)
 673     grooveshark.playSong(song)
 674
 675 elif mode==MODE_ARTIST:
 676     grooveshark.artist(id)
 677
 678 elif mode==MODE_ALBUM:
 679     grooveshark.album(id)
 680
 681 elif mode==MODE_PLAYLIST:
 682     grooveshark.playlist(id, name)
 683
 684 elif mode==MODE_FAVORITE:
 685     grooveshark.favorite(id)
 686
 687 elif mode==MODE_UNFAVORITE:
 688     try: prevMode=int(urllib.unquote_plus(params["prevmode"]))
 689     except:
 690         prevMode = 0
 691     grooveshark.unfavorite(id, prevMode)
 692
 693 elif mode==MODE_SIMILAR_ARTISTS:
 694     grooveshark.similarArtists(id)
 695
 696 elif mode==MODE_MAKE_PLAYLIST:
 697     grooveshark.makePlaylist(id, name)
 698
 699 elif mode==MODE_REMOVE_PLAYLIST:
 700     grooveshark.removePlaylist(id, name)
 701
 702 elif mode==MODE_RENAME_PLAYLIST:
 703     grooveshark.renamePlaylist(id, name)
 704
 705 elif mode==MODE_REMOVE_PLAYLIST_SONG:
 706     try: playlistID=urllib.unquote_plus(params["playlistid"])
 707     except: pass
 708     grooveshark.removePlaylistSong(playlistID, name, id)
 709
 710 elif mode==MODE_ADD_PLAYLIST_SONG:
 711     grooveshark.addPlaylistSong(id)
 712
 713 if mode < MODE_SONG:
 714     xbmcplugin.endOfDirectory(int(sys.argv[1]))