Fix for multiuser environment.
[clinton/xbmc-groove.git] / default.py
1 # Copyright 2011 Stephen Denham
2
3 # This file is part of xbmc-groove.
4 #
5 # xbmc-groove is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # xbmc-groove is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with xbmc-groove. If not, see <http://www.gnu.org/licenses/>.
17
18
19 import urllib, sys, os, shutil, re, pickle, time, xbmcaddon, xbmcplugin, xbmcgui, xbmc
20
21 __addon__ = xbmcaddon.Addon('plugin.audio.groove')
22 __addonname__ = __addon__.getAddonInfo('name')
23 __cwd__ = __addon__.getAddonInfo('path')
24 __author__ = __addon__.getAddonInfo('author')
25 __version__ = __addon__.getAddonInfo('version')
26 __language__ = __addon__.getLocalizedString
27 __debugging__ = __addon__.getSetting('debug')
28
29 MODE_SEARCH_SONGS = 1
30 MODE_SEARCH_ALBUMS = 2
31 MODE_SEARCH_ARTISTS = 3
32 MODE_SEARCH_ARTISTS_ALBUMS = 4
33 MODE_SEARCH_PLAYLISTS = 5
34 MODE_ARTIST_POPULAR = 6
35 MODE_POPULAR_SONGS = 7
36 MODE_FAVORITES = 8
37 MODE_PLAYLISTS = 9
38 MODE_ALBUM = 10
39 MODE_ARTIST = 11
40 MODE_PLAYLIST = 12
41 MODE_SONG_PAGE = 13
42 MODE_SIMILAR_ARTISTS = 14
43 MODE_SONG = 15
44 MODE_FAVORITE = 16
45 MODE_UNFAVORITE = 17
46 MODE_MAKE_PLAYLIST = 18
47 MODE_REMOVE_PLAYLIST = 19
48 MODE_RENAME_PLAYLIST = 20
49 MODE_REMOVE_PLAYLIST_SONG = 21
50 MODE_ADD_PLAYLIST_SONG = 22
51
52 ACTION_MOVE_LEFT = 1
53 ACTION_MOVE_UP = 3
54 ACTION_MOVE_DOWN = 4
55 ACTION_PAGE_UP = 5
56 ACTION_PAGE_DOWN = 6
57 ACTION_SELECT_ITEM = 7
58 ACTION_PREVIOUS_MENU = 10
59
60 # Formats for track labels
61 ARTIST_ALBUM_NAME_LABEL = 0
62 NAME_ALBUM_ARTIST_LABEL = 1
63
64 # Stream marking time (seconds)
65 STREAM_MARKING_TIME = 30
66
67 # Timeout
68 STREAM_TIMEOUT = 30
69
70 songMarkTime = 0
71 player = xbmc.Player()
72 playTimer = None
73
74 baseDir = __cwd__
75 resDir = xbmc.translatePath(os.path.join(baseDir, 'resources'))
76 libDir = xbmc.translatePath(os.path.join(resDir, 'lib'))
77 imgDir = xbmc.translatePath(os.path.join(resDir, 'img'))
78 cacheDir = os.path.join(xbmc.translatePath('special://masterprofile/addon_data/'), os.path.basename(baseDir))
79 thumbDirName = 'thumb'
80 thumbDir = os.path.join(xbmc.translatePath('special://masterprofile/addon_data/'), os.path.basename(baseDir), thumbDirName)
81
82 baseModeUrl = 'plugin://plugin.audio.groove/'
83 playlistUrl = baseModeUrl + '?mode=' + str(MODE_PLAYLIST)
84 playlistsUrl = baseModeUrl + '?mode=' + str(MODE_PLAYLISTS)
85 favoritesUrl = baseModeUrl + '?mode=' + str(MODE_FAVORITES)
86
87 searchArtistsAlbumsName = __language__(30006)
88
89 thumbDef = os.path.join(imgDir, 'default.tbn')
90 listBackground = os.path.join(imgDir, 'listbackground.png')
91
92 sys.path.append (libDir)
93 from GroovesharkAPI import GrooveAPI
94 from threading import Event, Thread
95
96 if __debugging__ == 'true':
97 __debugging__ = True
98 else:
99 __debugging__ = False
100
101 try:
102 groovesharkApi = GrooveAPI(__debugging__)
103 if groovesharkApi.pingService() != True:
104 raise StandardError(__language__(30007))
105 except:
106 dialog = xbmcgui.Dialog(__language__(30008),__language__(30009),__language__(30010))
107 dialog.ok(__language__(30008),__language__(30009))
108 sys.exit(-1)
109
110 # Mark song as playing or played
111 def markSong(songid, duration, streamKey, streamServerID):
112 global songMarkTime
113 global playTimer
114 global player
115 if player.isPlayingAudio():
116 tNow = player.getTime()
117 if tNow >= STREAM_MARKING_TIME and songMarkTime == 0:
118 groovesharkApi.markStreamKeyOver30Secs(streamKey, streamServerID)
119 songMarkTime = tNow
120 elif duration > tNow and duration - tNow < 2 and songMarkTime >= STREAM_MARKING_TIME:
121 playTimer.cancel()
122 songMarkTime = 0
123 groovesharkApi.markSongComplete(songid, streamKey, streamServerID)
124 else:
125 playTimer.cancel()
126 songMarkTime = 0
127
128 class _Info:
129 def __init__( self, *args, **kwargs ):
130 self.__dict__.update( kwargs )
131
132 # Window dialog to select a grooveshark playlist
133 class GroovesharkPlaylistSelect(xbmcgui.WindowDialog):
134
135 def __init__(self, items=[]):
136 gap = int(self.getHeight()/100)
137 w = int(self.getWidth()*0.5)
138 h = self.getHeight()-30*gap
139 rw = self.getWidth()
140 rh = self.getHeight()
141 x = rw/2 - w/2
142 y = rh/2 -h/2
143
144 self.imgBg = xbmcgui.ControlImage(x+gap, 5*gap+y, w-2*gap, h-5*gap, listBackground)
145 self.addControl(self.imgBg)
146
147 self.playlistControl = xbmcgui.ControlList(2*gap+x, y+3*gap+30, w-4*gap, h-10*gap, textColor='0xFFFFFFFF', selectedColor='0xFFFF4242')
148 self.playlistControl.setItemHeight(50)
149 self.addControl(self.playlistControl)
150
151 self.lastPos = 0
152 self.isSelecting = False
153 self.selected = -1
154 listitems = []
155 for playlist in items:
156 listitems.append(xbmcgui.ListItem(playlist[0]))
157 listitems.append(xbmcgui.ListItem(__language__(30011)))
158 self.playlistControl.addItems(listitems)
159 self.setFocus(self.playlistControl)
160 self.playlistControl.selectItem(0)
161 item = self.playlistControl.getListItem(self.lastPos)
162 item.select(True)
163
164 # Highlight selected item
165 def setHighlight(self):
166 if self.isSelecting:
167 return
168 else:
169 self.isSelecting = True
170
171 pos = self.playlistControl.getSelectedPosition()
172 if pos >= 0:
173 item = self.playlistControl.getListItem(self.lastPos)
174 item.select(False)
175 item = self.playlistControl.getListItem(pos)
176 item.select(True)
177 self.lastPos = pos
178 self.isSelecting = False
179
180 # Control - select
181 def onControl(self, control):
182 if control == self.playlistControl:
183 self.selected = self.playlistControl.getSelectedPosition()
184 self.close()
185
186 # Action - close or up/down
187 def onAction(self, action):
188 if action == ACTION_PREVIOUS_MENU:
189 self.selected = -1
190 self.close()
191 elif action == ACTION_MOVE_UP or action == ACTION_MOVE_DOWN or action == ACTION_PAGE_UP or action == ACTION_PAGE_DOWN == 6:
192 self.setFocus(self.playlistControl)
193 self.setHighlight()
194
195
196 class PlayTimer(Thread):
197 # interval -- floating point number specifying the number of seconds to wait before executing function
198 # function -- the function (or callable object) to be executed
199
200 # iterations -- integer specifying the number of iterations to perform
201 # args -- list of positional arguments passed to function
202 # kwargs -- dictionary of keyword arguments passed to function
203
204 def __init__(self, interval, function, iterations=0, args=[], kwargs={}):
205 Thread.__init__(self)
206 self.interval = interval
207 self.function = function
208 self.iterations = iterations
209 self.args = args
210 self.kwargs = kwargs
211 self.finished = Event()
212
213 def run(self):
214 count = 0
215 while not self.finished.isSet() and (self.iterations <= 0 or count < self.iterations):
216 self.finished.wait(self.interval)
217 if not self.finished.isSet():
218 self.function(*self.args, **self.kwargs)
219 count += 1
220
221 def cancel(self):
222 self.finished.set()
223
224 def setIterations(self, iterations):
225 self.iterations = iterations
226
227
228 def getTime(self):
229 return self.iterations * self.interval
230
231
232 class Grooveshark:
233
234 albumImg = xbmc.translatePath(os.path.join(imgDir, 'album.png'))
235 artistImg = xbmc.translatePath(os.path.join(imgDir, 'artist.png'))
236 artistsAlbumsImg = xbmc.translatePath(os.path.join(imgDir, 'artistsalbums.png'))
237 favoritesImg = xbmc.translatePath(os.path.join(imgDir, 'favorites.png'))
238 playlistImg = xbmc.translatePath(os.path.join(imgDir, 'playlist.png'))
239 usersplaylistsImg = xbmc.translatePath(os.path.join(imgDir, 'usersplaylists.png'))
240 popularSongsImg = xbmc.translatePath(os.path.join(imgDir, 'popularSongs.png'))
241 popularSongsArtistImg = xbmc.translatePath(os.path.join(imgDir, 'popularSongsArtist.png'))
242 songImg = xbmc.translatePath(os.path.join(imgDir, 'song.png'))
243 defImg = xbmc.translatePath(os.path.join(imgDir, 'default.tbn'))
244 fanImg = xbmc.translatePath(os.path.join(baseDir, 'fanart.jpg'))
245
246 settings = xbmcaddon.Addon(id='plugin.audio.groove')
247 songsearchlimit = int(settings.getSetting('songsearchlimit'))
248 albumsearchlimit = int(settings.getSetting('albumsearchlimit'))
249 artistsearchlimit = int(settings.getSetting('artistsearchlimit'))
250 songspagelimit = int(settings.getSetting('songspagelimit'))
251 username = settings.getSetting('username')
252 password = settings.getSetting('password')
253
254 userid = 0
255
256 def __init__( self ):
257 self._handle = int(sys.argv[1])
258 if os.path.isdir(cacheDir) == False:
259 os.makedirs(cacheDir)
260 if __debugging__ :
261 xbmc.log(__language__(30012) + " " + cacheDir)
262 artDir = xbmc.translatePath(thumbDir)
263 if os.path.isdir(artDir) == False:
264 os.makedirs(artDir)
265 if __debugging__ :
266 xbmc.log(__language__(30012) + " " + artDir)
267
268 # Top-level menu
269 def categories(self):
270
271 self.userid = self._get_login()
272
273 # Setup
274 xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
275
276 self._add_dir(__language__(30013), '', MODE_SEARCH_SONGS, self.songImg, 0)
277 self._add_dir(__language__(30014), '', MODE_SEARCH_ALBUMS, self.albumImg, 0)
278 self._add_dir(__language__(30015), '', MODE_SEARCH_ARTISTS, self.artistImg, 0)
279 self._add_dir(searchArtistsAlbumsName, '', MODE_SEARCH_ARTISTS_ALBUMS, self.artistsAlbumsImg, 0)
280 # Not supported by key
281 #self._add_dir("Search for user's playlists...", '', MODE_SEARCH_PLAYLISTS, self.usersplaylistsImg, 0)
282 self._add_dir(__language__(30016), '', MODE_ARTIST_POPULAR, self.popularSongsArtistImg, 0)
283 self._add_dir(__language__(30017), '', MODE_POPULAR_SONGS, self.popularSongsImg, 0)
284 if (self.userid != 0):
285 self._add_dir(__language__(30018), '', MODE_FAVORITES, self.favoritesImg, 0)
286 self._add_dir(__language__(30019), '', MODE_PLAYLISTS, self.playlistImg, 0)
287
288 # Search for songs
289 def searchSongs(self):
290 query = self._get_keyboard(default="", heading=__language__(30020))
291 if (query != ''):
292 songs = groovesharkApi.getSongSearchResults(query, limit = self.songsearchlimit)
293 if (len(songs) > 0):
294 self._add_songs_directory(songs)
295 else:
296 dialog = xbmcgui.Dialog()
297 dialog.ok(__language__(30008), __language__(30021))
298 self.categories()
299 else:
300 self.categories()
301
302 # Search for albums
303 def searchAlbums(self):
304 query = self._get_keyboard(default="", heading=__language__(30022))
305 if (query != ''):
306 albums = groovesharkApi.getAlbumSearchResults(query, limit = self.albumsearchlimit)
307 if (len(albums) > 0):
308 self._add_albums_directory(albums)
309 else:
310 dialog = xbmcgui.Dialog()
311 dialog.ok(__language__(30008), __language__(30023))
312 self.categories()
313 else:
314 self.categories()
315
316 # Search for artists
317 def searchArtists(self):
318 query = self._get_keyboard(default="", heading=__language__(30024))
319 if (query != ''):
320 artists = groovesharkApi.getArtistSearchResults(query, limit = self.artistsearchlimit)
321 if (len(artists) > 0):
322 self._add_artists_directory(artists)
323 else:
324 dialog = xbmcgui.Dialog()
325 dialog.ok(__language__(30008), __language__(30025))
326 self.categories()
327 else:
328 self.categories()
329
330 # Search for playlists
331 def searchPlaylists(self):
332 query = self._get_keyboard(default="", heading=__language__(30026))
333 if (query != ''):
334 playlists = groovesharkApi.getUserPlaylistsByUsername(query)
335 if (len(playlists) > 0):
336 self._add_playlists_directory(playlists)
337 else:
338 dialog = xbmcgui.Dialog()
339 dialog.ok(__language__(30008), __language__(30027))
340 self.categories()
341 else:
342 self.categories()
343
344 # Search for artists albums
345 def searchArtistsAlbums(self, artistName = None):
346 if artistName == None or artistName == searchArtistsAlbumsName:
347 query = self._get_keyboard(default="", heading=__language__(30028))
348 else:
349 query = artistName
350 if (query != ''):
351 artists = groovesharkApi.getArtistSearchResults(query, limit = self.artistsearchlimit)
352 if (len(artists) > 0):
353 artist = artists[0]
354 artistID = artist[1]
355 if __debugging__ :
356 xbmc.log("Found " + artist[0] + "...")
357 albums = groovesharkApi.getArtistAlbums(artistID, self.albumsearchlimit)
358 if (len(albums) > 0):
359 self._add_albums_directory(albums, artistID)
360 else:
361 dialog = xbmcgui.Dialog()
362 dialog.ok(__language__(30008), __language__(30029))
363 self.categories()
364 else:
365 dialog = xbmcgui.Dialog()
366 dialog.ok(__language__(30008), __language__(30030))
367 self.categories()
368 else:
369 self.categories()
370
371 # Get my favorites
372 def favorites(self):
373 userid = self._get_login()
374 if (userid != 0):
375 favorites = groovesharkApi.getUserFavoriteSongs()
376 if (len(favorites) > 0):
377 self._add_songs_directory(favorites, isFavorites=True)
378 else:
379 dialog = xbmcgui.Dialog()
380 dialog.ok(__language__(30008), __language__(30031))
381 self.categories()
382
383 # Get popular songs
384 def popularSongs(self):
385 popular = groovesharkApi.getPopularSongsToday(limit = self.songsearchlimit)
386 if (len(popular) > 0):
387 self._add_songs_directory(popular)
388 else:
389 dialog = xbmcgui.Dialog()
390 dialog.ok(__language__(30008), __language__(30032))
391 self.categories()
392
393 # Get my playlists
394 def playlists(self):
395 userid = self._get_login()
396 if (userid != 0):
397 playlists = groovesharkApi.getUserPlaylists()
398 if (len(playlists) > 0):
399 self._add_playlists_directory(playlists)
400 else:
401 dialog = xbmcgui.Dialog()
402 dialog.ok(__language__(30008), __language__(30033))
403 self.categories()
404 else:
405 dialog = xbmcgui.Dialog()
406 dialog.ok(__language__(30008), __language__(30034), __language__(30035))
407
408 # Make songs a favorite
409 def favorite(self, songid):
410 userid = self._get_login()
411 if (userid != 0):
412 if __debugging__ :
413 xbmc.log("Favorite song: " + str(songid))
414 groovesharkApi.addUserFavoriteSong(songID = songid)
415 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ', ' + __language__(30036) + ', 1000, ' + thumbDef + ')')
416 else:
417 dialog = xbmcgui.Dialog()
418 dialog.ok(__language__(30008), __language__(30034), __language__(30037))
419
420 # Remove song from favorites
421 def unfavorite(self, songid, prevMode=0):
422 userid = self._get_login()
423 if (userid != 0):
424 if __debugging__ :
425 xbmc.log("Unfavorite song: " + str(songid) + ', previous mode was ' + str(prevMode))
426 groovesharkApi.removeUserFavoriteSongs(songIDs = songid)
427 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ', ' + __language__(30038) + ', 1000, ' + thumbDef + ')')
428 # Refresh to remove item from directory
429 if (int(prevMode) == MODE_FAVORITES):
430 xbmc.executebuiltin("Container.Refresh(" + favoritesUrl + ")")
431 else:
432 dialog = xbmcgui.Dialog()
433 dialog.ok(__language__(30008), __language__(30034), __language__(30039))
434
435
436 # Show selected album
437 def album(self, albumid):
438 album = groovesharkApi.getAlbumSongs(albumid, limit = self.songsearchlimit)
439 self._add_songs_directory(album, trackLabelFormat=NAME_ALBUM_ARTIST_LABEL)
440
441 # Show selected artist
442 def artist(self, artistid):
443 albums = groovesharkApi.getArtistAlbums(artistid, limit = self.albumsearchlimit)
444 self._add_albums_directory(albums, artistid, True)
445
446 # Show selected playlist
447 def playlist(self, playlistid, playlistname):
448 userid = self._get_login()
449 if (userid != 0):
450 songs = groovesharkApi.getPlaylistSongs(playlistid)
451 self._add_songs_directory(songs, trackLabelFormat=NAME_ALBUM_ARTIST_LABEL, playlistid=playlistid, playlistname=playlistname)
452 else:
453 dialog = xbmcgui.Dialog()
454 dialog.ok(__language__(30008), __language__(30034), __language__(30040))
455
456 # Show popular songs of the artist
457 def artistPopularSongs(self):
458 query = self._get_keyboard(default="", heading=__language__(30041))
459 if (query != ''):
460 artists = groovesharkApi.getArtistSearchResults(query, limit = self.artistsearchlimit)
461 if (len(artists) > 0):
462 artist = artists[0]
463 artistID = artist[1]
464 if __debugging__ :
465 xbmc.log("Found " + artist[0] + "...")
466 songs = groovesharkApi.getArtistPopularSongs(artistID, limit = self.songsearchlimit)
467 if (len(songs) > 0):
468 self._add_songs_directory(songs, trackLabelFormat=NAME_ALBUM_ARTIST_LABEL)
469 else:
470 dialog = xbmcgui.Dialog()
471 dialog.ok(__language__(30008), __language__(30042))
472 self.categories()
473 else:
474 dialog = xbmcgui.Dialog()
475 dialog.ok(__language__(30008), __language__(30043))
476 self.categories()
477 else:
478 self.categories()
479
480 # Play a song
481 def playSong(self, item):
482 global playTimer
483 global player
484 if item != None:
485 # Get stream as it could have expired
486 item.select(True)
487 url = ''
488 songid = item.getProperty('songid')
489 stream = groovesharkApi.getSubscriberStreamKey(songid)
490 if stream != False:
491 url = stream['url']
492 key = stream['StreamKey']
493 server = stream['StreamServerID']
494 duration = int(self._setDuration(stream['uSecs']))
495 stream = [songid, duration, url, key, server]
496 self._setSongStream(stream)
497 if url != '':
498 item.setPath(url)
499 xbmcplugin.setResolvedUrl(handle=int(sys.argv[1]), succeeded=True, listitem=item)
500 if __debugging__ :
501 xbmc.log("Grooveshark playing: " + url)
502 # Wait for play then start timer
503 seconds = 0
504 while seconds < STREAM_TIMEOUT:
505 try:
506 if player.isPlayingAudio() == True:
507 if playTimer != None:
508 playTimer.cancel()
509 songMarkTime = 0
510 playTimer = PlayTimer(1, markSong, self._setDuration(duration), [songid, duration, key, server])
511 playTimer.start()
512 break
513 except: pass
514 time.sleep(1)
515 seconds = seconds + 1
516 else:
517 xbmc.log("No song URL")
518 else:
519 xbmc.log("No song stream")
520 else:
521 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ', ' + __language__(30044) + ', 1000, ' + thumbDef + ')')
522
523 # Make a song directory item
524 def songItem(self, songid, name, album, artist, coverart, trackLabelFormat=ARTIST_ALBUM_NAME_LABEL, tracknumber=1):
525
526 stream = self._getSongStream(songid)
527 if stream != False:
528 duration = stream[1]
529 url = stream[2]
530 key = stream[3]
531 server = stream[4]
532 songImg = self._get_icon(coverart, 'song-' + str(songid) + "-image")
533 if int(trackLabelFormat) == NAME_ALBUM_ARTIST_LABEL:
534 trackLabel = name + " - " + album + " - " + artist
535 else:
536 trackLabel = artist + " - " + album + " - " + name
537 item = xbmcgui.ListItem(label = trackLabel, thumbnailImage=songImg, iconImage=songImg)
538 item.setPath(url)
539 item.setInfo( type="music", infoLabels={ "title": name, "album": album, "artist": artist, "duration": duration, "tracknumber" : tracknumber} )
540 item.setProperty('mimetype', 'audio/mpeg')
541 item.setProperty("IsPlayable", "true")
542 item.setProperty('songid', str(songid))
543 item.setProperty('coverart', songImg)
544 item.setProperty('title', name)
545 item.setProperty('album', album)
546 item.setProperty('artist', artist)
547 item.setProperty('duration', str(duration))
548 item.setProperty('key', str(key))
549 item.setProperty('server', str(server))
550 item.setProperty('fanart_image', self.fanImg)
551 return item
552 else:
553 xbmc.log("No song URL")
554 return None
555
556 # Next page of songs
557 def songPage(self, offset, trackLabelFormat, playlistid = 0, playlistname = ''):
558 self._add_songs_directory([], trackLabelFormat, offset, playlistid = playlistid, playlistname = playlistname)
559
560 # Make a playlist from an album
561 def makePlaylist(self, albumid, name):
562 userid = self._get_login()
563 if (userid != 0):
564 re.split(' - ',name,1)
565 nameTokens = re.split(' - ',name,1) # suggested name
566 name = self._get_keyboard(default=nameTokens[0], heading=__language__(30045))
567 if name != '':
568 album = groovesharkApi.getAlbumSongs(albumid, limit = self.songsearchlimit)
569 songids = []
570 for song in album:
571 songids.append(song[1])
572 if groovesharkApi.createPlaylist(name, songids) == 0:
573 dialog = xbmcgui.Dialog()
574 dialog.ok(__language__(30008), __language__(30046), name)
575 else:
576 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ',' + __language__(30047)+ ', 1000, ' + thumbDef + ')')
577 else:
578 dialog = xbmcgui.Dialog()
579 dialog.ok(__language__(30008), __language__(30034), __language__(30048))
580
581 # Rename a playlist
582 def renamePlaylist(self, playlistid, name):
583 userid = self._get_login()
584 if (userid != 0):
585 newname = self._get_keyboard(default=name, heading=__language__(30049))
586 if newname == '':
587 return
588 elif groovesharkApi.playlistRename(playlistid, newname) == 0:
589 dialog = xbmcgui.Dialog()
590 dialog.ok(__language__(30008), __language__(30050), name)
591 else:
592 # Refresh to show new item name
593 xbmc.executebuiltin("Container.Refresh")
594 else:
595 dialog = xbmcgui.Dialog()
596 dialog.ok(__language__(30008), __language__(30034), __language__(30051))
597
598 # Remove a playlist
599 def removePlaylist(self, playlistid, name):
600 dialog = xbmcgui.Dialog()
601 if dialog.yesno(__language__(30008), name, __language__(30052)) == True:
602 userid = self._get_login()
603 if (userid != 0):
604 if groovesharkApi.playlistDelete(playlistid) == 0:
605 dialog = xbmcgui.Dialog()
606 dialog.ok(__language__(30008), __language__(30053), name)
607 else:
608 # Refresh to remove item from directory
609 xbmc.executebuiltin("Container.Refresh(" + playlistsUrl + ")")
610 else:
611 dialog = xbmcgui.Dialog()
612 dialog.ok(__language__(30008), __language__(30034), __language__(30054))
613
614 # Add song to playlist
615 def addPlaylistSong(self, songid):
616 userid = self._get_login()
617 if (userid != 0):
618 playlists = groovesharkApi.getUserPlaylists()
619 if (len(playlists) > 0):
620 ret = 0
621 # Select the playlist
622 playlistSelect = GroovesharkPlaylistSelect(items=playlists)
623 playlistSelect.setFocus(playlistSelect.playlistControl)
624 playlistSelect.doModal()
625 i = playlistSelect.selected
626 del playlistSelect
627 if i > -1:
628 # Add a new playlist
629 if i >= len(playlists):
630 name = self._get_keyboard(default='', heading=__language__(30055))
631 if name != '':
632 songIds = []
633 songIds.append(songid)
634 if groovesharkApi.createPlaylist(name, songIds) == 0:
635 dialog = xbmcgui.Dialog()
636 dialog.ok(__language__(30008), __language__(30056), name)
637 else:
638 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ',' + __language__(30057) + ', 1000, ' + thumbDef + ')')
639 # Existing playlist
640 else:
641 playlist = playlists[i]
642 playlistid = playlist[1]
643 if __debugging__ :
644 xbmc.log("Add song " + str(songid) + " to playlist " + str(playlistid))
645 songIDs=[]
646 songs = groovesharkApi.getPlaylistSongs(playlistid)
647 for song in songs:
648 songIDs.append(song[1])
649 songIDs.append(songid)
650 ret = groovesharkApi.setPlaylistSongs(playlistid, songIDs)
651 if ret == False:
652 dialog = xbmcgui.Dialog()
653 dialog.ok(__language__(30008), __language__(30058))
654 else:
655 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ',' + __language__(30059) + ', 1000, ' + thumbDef + ')')
656 else:
657 dialog = xbmcgui.Dialog()
658 dialog.ok(__language__(30008), __language__(30060))
659 self.categories()
660 else:
661 dialog = xbmcgui.Dialog()
662 dialog.ok(__language__(30008), __language__(30034), __language__(30061))
663
664 # Remove song from playlist
665 def removePlaylistSong(self, playlistid, playlistname, songid):
666 dialog = xbmcgui.Dialog()
667 if dialog.yesno(__language__(30008), __language__(30062), __language__(30063)) == True:
668 userid = self._get_login()
669 if (userid != 0):
670 songs = groovesharkApi.getPlaylistSongs(playlistID)
671 songIDs=[]
672 for song in songs:
673 if (song[1] != songid):
674 songIDs.append(song[1])
675 ret = groovesharkApi.setPlaylistSongs(playlistID, songIDs)
676 if ret == False:
677 dialog = xbmcgui.Dialog()
678 dialog.ok(__language__(30008), __language__(30064), __language__(30065))
679 else:
680 # Refresh to remove item from directory
681 xbmc.executebuiltin('XBMC.Notification(' + __language__(30008) + ',' + __language__(30066)+ ', 1000, ' + thumbDef + ')')
682 xbmc.executebuiltin("Container.Update(" + playlistUrl + "&id="+str(playlistid) + "&name=" + str(playlistname) + ")")
683 else:
684 dialog = xbmcgui.Dialog()
685 dialog.ok(__language__(30008), __language__(30034), __language__(30067))
686
687 # Find similar artists to searched artist
688 def similarArtists(self, artistId):
689 similar = groovesharkApi.getSimilarArtists(artistId, limit = self.artistsearchlimit)
690 if (len(similar) > 0):
691 self._add_artists_directory(similar)
692 else:
693 dialog = xbmcgui.Dialog()
694 dialog.ok(__language__(30008), __language__(30068))
695 self.categories()
696
697 # Get keyboard input
698 def _get_keyboard(self, default="", heading="", hidden=False):
699 kb = xbmc.Keyboard(default, heading, hidden)
700 kb.doModal()
701 if (kb.isConfirmed()):
702 return unicode(kb.getText(), "utf-8")
703 return ''
704
705 # Login to grooveshark
706 def _get_login(self):
707 if (self.username == "" or self.password == ""):
708 dialog = xbmcgui.Dialog()
709 dialog.ok(__language__(30008), __language__(30069), __language__(30070), __language__(30082))
710 return 0
711 else:
712 if self.userid == 0:
713 uid = groovesharkApi.login(self.username, self.password)
714 if (uid != 0):
715 return uid
716 else:
717 dialog = xbmcgui.Dialog()
718 dialog.ok(__language__(30008), __language__(30069), __language__(30070), __language__(30082))
719 return 0
720
721 # File download
722 def _get_icon(self, url, songid):
723 if url != 'None':
724 localThumb = os.path.join(xbmc.translatePath(os.path.join(thumbDir, str(songid)))) + '.tbn'
725 try:
726 if os.path.isfile(localThumb) == False:
727 loc = urllib.URLopener()
728 loc.retrieve(url, localThumb)
729 except:
730 shutil.copy2(thumbDef, localThumb)
731 return os.path.join(os.path.join(thumbDir, str(songid))) + '.tbn'
732 else:
733 return thumbDef
734
735 # Add songs to directory
736 def _add_songs_directory(self, songs, trackLabelFormat=ARTIST_ALBUM_NAME_LABEL, offset=0, playlistid=0, playlistname='', isFavorites=False):
737
738 totalSongs = len(songs)
739 offset = int(offset)
740 start = 0
741 end = totalSongs
742
743 # No pages needed
744 if offset == 0 and totalSongs <= self.songspagelimit:
745 if __debugging__ :
746 xbmc.log("Found " + str(totalSongs) + " songs...")
747 # Pages
748 else:
749 # Cache all next pages songs
750 if offset == 0:
751 self._setSavedSongs(songs)
752 else:
753 songs = self._getSavedSongs()
754 totalSongs = len(songs)
755
756 if totalSongs > 0:
757 start = offset
758 end = min(start + self.songspagelimit,totalSongs)
759
760 n = start
761 items = end - start
762 while n < end:
763 song = songs[n]
764 name = song[0]
765 songid = song[1]
766 album = song[2]
767 artist = song[4]
768 coverart = song[6]
769 item = self.songItem(songid, name, album, artist, coverart, trackLabelFormat, (n+1))
770 if item != None:
771 coverart = item.getProperty('coverart')
772 songname = song[0]
773 songalbum = song[2]
774 songartist = song[4]
775 u=sys.argv[0]+"?mode="+str(MODE_SONG)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid) \
776 +"&album="+urllib.quote_plus(songalbum) \
777 +"&artist="+urllib.quote_plus(songartist) \
778 +"&coverart="+urllib.quote_plus(coverart)
779 fav=sys.argv[0]+"?mode="+str(MODE_FAVORITE)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid)
780 unfav=sys.argv[0]+"?mode="+str(MODE_UNFAVORITE)+"&name="+urllib.quote_plus(songname)+"&id="+str(songid)+"&prevmode="
781 menuItems = []
782 if isFavorites == True:
783 unfav = unfav +str(MODE_FAVORITES)
784 else:
785 menuItems.append((__language__(30071), "XBMC.RunPlugin("+fav+")"))
786 menuItems.append((__language__(30072), "XBMC.RunPlugin("+unfav+")"))
787 if playlistid > 0:
788 rmplaylstsong=sys.argv[0]+"?playlistid="+str(playlistid)+"&id="+str(songid)+"&mode="+str(MODE_REMOVE_PLAYLIST_SONG)+"&name="+str(playlistname)
789 menuItems.append((__language__(30073), "XBMC.RunPlugin("+rmplaylstsong+")"))
790 else:
791 addplaylstsong=sys.argv[0]+"?id="+str(songid)+"&mode="+str(MODE_ADD_PLAYLIST_SONG)
792 menuItems.append((__language__(30074), "XBMC.RunPlugin("+addplaylstsong+")"))
793 item.addContextMenuItems(menuItems, replaceItems=False)
794 xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]),url=u,listitem=item,isFolder=False, totalItems=items)
795 else:
796 end = min(end + 1,totalSongs)
797 if __debugging__ :
798 xbmc.log(song[0] + " does not exist.")
799 n = n + 1
800
801 if totalSongs > end:
802 u=sys.argv[0]+"?mode="+str(MODE_SONG_PAGE)+"&id=playlistid"+"&offset="+str(end)+"&label="+str(trackLabelFormat)+"&name="+str(playlistname)
803 self._add_dir(__language__(30075) + '...', u, MODE_SONG_PAGE, self.songImg, 0, totalSongs - end)
804
805 xbmcplugin.setContent(self._handle, 'songs')
806 xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
807
808 # Add albums to directory
809 def _add_albums_directory(self, albums, artistid=0, isverified=False):
810 n = len(albums)
811 itemsExisting = n
812 if __debugging__ :
813 xbmc.log("Found " + str(n) + " albums...")
814 i = 0
815 while i < n:
816 album = albums[i]
817 albumID = album[3]
818 if isverified or groovesharkApi.getDoesAlbumExist(albumID):
819 albumArtistName = album[0]
820 albumName = album[2]
821 albumImage = self._get_icon(album[4], 'album-' + str(albumID))
822 self._add_dir(albumName + " - " + albumArtistName, '', MODE_ALBUM, albumImage, albumID, itemsExisting)
823 else:
824 itemsExisting = itemsExisting - 1
825 i = i + 1
826 # Not supported by key
827 #if artistid > 0:
828 # self._add_dir('Similar artists...', '', MODE_SIMILAR_ARTISTS, self.artistImg, artistid)
829 xbmcplugin.setContent(self._handle, 'albums')
830 xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_ALBUM_IGNORE_THE)
831 xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
832
833 # Add artists to directory
834 def _add_artists_directory(self, artists):
835 n = len(artists)
836 itemsExisting = n
837 if __debugging__ :
838 xbmc.log("Found " + str(n) + " artists...")
839 i = 0
840 while i < n:
841 artist = artists[i]
842 artistID = artist[1]
843 if groovesharkApi.getDoesArtistExist(artistID):
844 artistName = artist[0]
845 self._add_dir(artistName, '', MODE_ARTIST, self.artistImg, artistID, itemsExisting)
846 else:
847 itemsExisting = itemsExisting - 1
848 i = i + 1
849 xbmcplugin.setContent(self._handle, 'artists')
850 xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_ARTIST_IGNORE_THE)
851 xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
852
853 # Add playlists to directory
854 def _add_playlists_directory(self, playlists):
855 n = len(playlists)
856 if __debugging__ :
857 xbmc.log("Found " + str(n) + " playlists...")
858 i = 0
859 while i < n:
860 playlist = playlists[i]
861 playlistName = playlist[0]
862 playlistID = playlist[1]
863 self._add_dir(playlistName, '', MODE_PLAYLIST, self.playlistImg, playlistID, n)
864 i = i + 1
865 xbmcplugin.setContent(self._handle, 'files')
866 xbmcplugin.addSortMethod(self._handle, xbmcplugin.SORT_METHOD_LABEL)
867 xbmcplugin.setPluginFanart(int(sys.argv[1]), self.fanImg)
868
869 # Add whatever directory
870 def _add_dir(self, name, url, mode, iconimage, itemId, items=1):
871
872 if url == '':
873 u=sys.argv[0]+"?mode="+str(mode)+"&name="+urllib.quote_plus(name)+"&id="+str(itemId)
874 else:
875 u = url
876 directory=xbmcgui.ListItem(name, iconImage=iconimage, thumbnailImage=iconimage)
877 directory.setInfo( type="Music", infoLabels={ "title": name } )
878 directory.setProperty('fanart_image', self.fanImg)
879
880 # Custom menu items
881 menuItems = []
882 if mode == MODE_ALBUM:
883 mkplaylst=sys.argv[0]+"?mode="+str(MODE_MAKE_PLAYLIST)+"&name="+name+"&id="+str(itemId)
884 menuItems.append((__language__(30076), "XBMC.RunPlugin("+mkplaylst+")"))
885 if mode == MODE_PLAYLIST:
886 rmplaylst=sys.argv[0]+"?mode="+str(MODE_REMOVE_PLAYLIST)+"&name="+urllib.quote_plus(name)+"&id="+str(itemId)
887 menuItems.append((__language__(30077), "XBMC.RunPlugin("+rmplaylst+")"))
888 mvplaylst=sys.argv[0]+"?mode="+str(MODE_RENAME_PLAYLIST)+"&name="+urllib.quote_plus(name)+"&id="+str(itemId)
889 menuItems.append((__language__(30078), "XBMC.RunPlugin("+mvplaylst+")"))
890
891 directory.addContextMenuItems(menuItems, replaceItems=False)
892
893 return xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]),url=u,listitem=directory,isFolder=True, totalItems=items)
894
895 def _getSavedSongs(self):
896 path = os.path.join(cacheDir, 'songs.dmp')
897 try:
898 f = open(path, 'rb')
899 songs = pickle.load(f)
900 f.close()
901 except:
902 songs = []
903 pass
904 return songs
905
906 def _setSavedSongs(self, songs):
907 try:
908 # Create the 'data' directory if it doesn't exist.
909 if not os.path.exists(cacheDir):
910 os.makedirs(cacheDir)
911 path = os.path.join(cacheDir, 'songs.dmp')
912 f = open(path, 'wb')
913 pickle.dump(songs, f, protocol=pickle.HIGHEST_PROTOCOL)
914 f.close()
915 except:
916 xbmc.log("An error occurred saving songs")
917 pass
918
919 # Duration to seconds
920 def _setDuration(self, usecs):
921 if usecs < 60000000:
922 usecs = usecs * 10 # Some durations are 10x to small
923 return int(usecs / 1000000)
924
925 def _getSongStream(self, songid):
926 idSong = int(songid)
927 stream = None
928 streams = []
929 path = os.path.join(cacheDir, 'streams.dmp')
930 try:
931 f = open(path, 'rb')
932 streams = pickle.load(f)
933 for song in streams:
934 if song[0] == idSong:
935 duration = song[1]
936 url = song[2]
937 key = song[3]
938 server = song[4]
939 stream = [idSong, duration, url, key, server]
940 if __debugging__ :
941 xbmc.log("Found " + str(idSong) + " in stream cache")
942 break;
943 f.close()
944 except:
945 pass
946
947 # Not in cache
948 if stream == None:
949 stream = groovesharkApi.getSubscriberStreamKey(songid)
950 if stream != False and stream['url'] != '':
951 duration = self._setDuration(stream['uSecs'])
952 url = stream['url']
953 key = stream['StreamKey']
954 server = stream['StreamServerID']
955 stream = [idSong, duration, url, key, server]
956 self._addSongStream(stream)
957
958 return stream
959
960 def _addSongStream(self, stream):
961 streams = self._getStreams()
962 streams.append(stream)
963 path = os.path.join(cacheDir, 'streams.dmp')
964 try:
965 f = open(path, 'wb')
966 pickle.dump(streams, f, protocol=pickle.HIGHEST_PROTOCOL)
967 f.close()
968 if __debugging__ :
969 xbmc.log("Added " + str(stream[0]) + " to stream cache")
970 except:
971 xbmc.log("An error occurred adding to stream")
972
973 def _setSongStream(self, stream):
974 idStream = int(stream[0])
975 stream[1] = self._setDuration(stream[1])
976 streams = self._getStreams()
977 path = os.path.join(cacheDir, 'streams.dmp')
978 i = 0
979
980 for song in streams:
981 if song[0] == idStream:
982 streams[i] = stream
983 try:
984 f = open(path, 'wb')
985 pickle.dump(streams, f, protocol=pickle.HIGHEST_PROTOCOL)
986 f.close()
987 if __debugging__ :
988 xbmc.log("Updated " + str(idStream) + " in stream cache")
989 break;
990 except:
991 xbmc.log("An error occurred setting stream")
992 i = i + 1
993
994 def _getStreams(self):
995 path = os.path.join(cacheDir, 'streams.dmp')
996 try:
997 f = open(path, 'rb')
998 streams = pickle.load(f)
999 f.close()
1000 except:
1001 streams = []
1002 pass
1003 return streams
1004
1005
1006 # Parse URL parameters
1007 def get_params():
1008 param=[]
1009 paramstring=sys.argv[2]
1010 if __debugging__ :
1011 xbmc.log(paramstring)
1012 if len(paramstring)>=2:
1013 params=sys.argv[2]
1014 cleanedparams=params.replace('?','')
1015 if (params[len(params)-1]=='/'):
1016 params=params[0:len(params)-2]
1017 pairsofparams=cleanedparams.split('&')
1018 param={}
1019 for i in range(len(pairsofparams)):
1020 splitparams={}
1021 splitparams=pairsofparams[i].split('=')
1022 if (len(splitparams))==2:
1023 param[splitparams[0]]=splitparams[1]
1024 return param
1025
1026 # Main
1027 grooveshark = Grooveshark();
1028
1029 params=get_params()
1030 mode=None
1031 try: mode=int(params["mode"])
1032 except: pass
1033 itemId=0
1034 try: itemId=int(params["id"])
1035 except: pass
1036 name = None
1037 try: name=urllib.unquote_plus(params["name"])
1038 except: pass
1039
1040 # Call function for URL
1041 if mode==None:
1042 grooveshark.categories()
1043
1044 elif mode==MODE_SEARCH_SONGS:
1045 grooveshark.searchSongs()
1046
1047 elif mode==MODE_SEARCH_ALBUMS:
1048 grooveshark.searchAlbums()
1049
1050 elif mode==MODE_SEARCH_ARTISTS:
1051 grooveshark.searchArtists()
1052
1053 elif mode==MODE_SEARCH_ARTISTS_ALBUMS:
1054 grooveshark.searchArtistsAlbums(name)
1055
1056 elif mode==MODE_SEARCH_PLAYLISTS:
1057 grooveshark.searchPlaylists()
1058
1059 elif mode==MODE_POPULAR_SONGS:
1060 grooveshark.popularSongs()
1061
1062 elif mode==MODE_ARTIST_POPULAR:
1063 grooveshark.artistPopularSongs()
1064
1065 elif mode==MODE_FAVORITES:
1066 grooveshark.favorites()
1067
1068 elif mode==MODE_PLAYLISTS:
1069 grooveshark.playlists()
1070
1071 elif mode==MODE_SONG_PAGE:
1072 try: offset=urllib.unquote_plus(params["offset"])
1073 except: pass
1074 try: label=urllib.unquote_plus(params["label"])
1075 except: pass
1076 grooveshark.songPage(offset, label, itemId, name)
1077
1078 elif mode==MODE_SONG:
1079 try: album=urllib.unquote_plus(params["album"])
1080 except: pass
1081 try: artist=urllib.unquote_plus(params["artist"])
1082 except: pass
1083 try: coverart=urllib.unquote_plus(params["coverart"])
1084 except: pass
1085 song = grooveshark.songItem(itemId, name, album, artist, coverart)
1086 grooveshark.playSong(song)
1087
1088 elif mode==MODE_ARTIST:
1089 grooveshark.artist(itemId)
1090
1091 elif mode==MODE_ALBUM:
1092 grooveshark.album(itemId)
1093
1094 elif mode==MODE_PLAYLIST:
1095 grooveshark.playlist(itemId, name)
1096
1097 elif mode==MODE_FAVORITE:
1098 grooveshark.favorite(itemId)
1099
1100 elif mode==MODE_UNFAVORITE:
1101 try: prevMode=int(urllib.unquote_plus(params["prevmode"]))
1102 except:
1103 prevMode = 0
1104 grooveshark.unfavorite(itemId, prevMode)
1105
1106 elif mode==MODE_SIMILAR_ARTISTS:
1107 grooveshark.similarArtists(itemId)
1108
1109 elif mode==MODE_MAKE_PLAYLIST:
1110 grooveshark.makePlaylist(itemId, name)
1111
1112 elif mode==MODE_REMOVE_PLAYLIST:
1113 grooveshark.removePlaylist(itemId, name)
1114
1115 elif mode==MODE_RENAME_PLAYLIST:
1116 grooveshark.renamePlaylist(itemId, name)
1117
1118 elif mode==MODE_REMOVE_PLAYLIST_SONG:
1119 try: playlistID=urllib.unquote_plus(params["playlistid"])
1120 except: pass
1121 grooveshark.removePlaylistSong(playlistID, name, itemId)
1122
1123 elif mode==MODE_ADD_PLAYLIST_SONG:
1124 grooveshark.addPlaylistSong(itemId)
1125
1126 if mode < MODE_SONG:
1127 xbmcplugin.endOfDirectory(int(sys.argv[1]))