addon.py

import re, os, sys
import urllib.parse
import xbmc
import xbmcgui
import xbmcaddon
import xbmcplugin
import xbmcvfs

import tempfile

_locallib_path = os.path.dirname(xbmcvfs.translatePath("special://home/addons/script.anyurl.player/lib"))
sys.path.append(_locallib_path)

def reencodeYT(video_id):
    return "https://www.youtube.com/watch?v="+video_id

''' Extend Kodi player in order to resolve playlist as soon as playback starts
'''
class MyPlayer(xbmc.Player):
    def __init__( self):
        xbmc.Player.__init__(self)
        self._processed = False
        self._playlist = xbmc.PLAYLIST_VIDEO
        self._wait = 120 # Start by waiting 2 minutes

    def onPlayBackStarted(self):
        xbmc.Player.onPlayBackStarted(self)
        # Hack here: dig into the playlists,
        #   and resolve Youtube (and others in the future) videos in advance
        playlist = xbmc.PlayList(self._playlist)
        position = playlist.getposition() + 1
        if (int(position) >= int(playlist.size())):
            position = 0
        resolvePlaylist(self._playlist, position)
        self._processed = True

    def onPlayBackEnded(self):
        xbmc.Player.onPlayBackEnded(self)
        self._wait = 30

    def isProcessed(self):
        return self._processed

    def setPlaylist(self, playlist):
        self._playlist = playlist

    def getWait(self):
        self._wait -= 1
        return self._wait

''' Read from incoming arguments '''
def getArg(args, label, default=None):
    for arg in args:
        match = re.match(label+"=", arg)
        if match:
            return arg[match.end():]
    return default

''' Try to resolve URL '''
def URLResolve(url, label, description=''):
    try:
        import resolveurl
    except:
        xbmc.log("Something crashed %s %s %s"%(url, label, description), xbmc.LOGINFO)
        return (None, '')

    li = None
    file_url = ""
    if True:
        pass
    try:
        pass
        xbmc.log("Resolving %s" % url, xbmc.LOGINFO)

        # Acquire lock
        # Be sure only one script runs at a time
        tmpfile = tempfile.gettempdir()+"/.anyurl.resolver.lock"
        counter = 0
        while (os.path.isfile(tmpfile) and counter < 2000):
            xbmc.log("%s Waiting for lock file %d - %s" % (addon_id, counter, url), xbmc.LOGDEBUG)
            xbmc.sleep(2000)
            counter = counter + 1
        if (os.path.isfile(tmpfile)):
            xbmc.log("%s Failed to lock %s" % (addon_id, url), xbmc.LOGINFO)
            os.remove(tmpfile) # Unlock after fail
            return (None, '') # Lock removed afer two minutes, but fail anyway
        xbmc.log("%s Locking %s" % (addon_id, url), xbmc.LOGDEBUG)
        open(tmpfile, 'a').close()

        file_url = resolveurl.resolve(url)

        if counter > 0: # Wait a bit if we had to wait before to be sure resolver is ready for the next one
            xbmc.sleep(3000)
        xbmc.log("%s Unlocking %s" % (addon_id, tmpfile), xbmc.LOGDEBUG)
        os.remove(tmpfile) # Unlock

        if file_url: # Resolved, but not with metadata
            li = xbmcgui.ListItem(label = label, path = file_url)
            infolabels={"Studio":"","Title":label,
                        "plot":description, 'plotoutline': description}
            li.setInfo(type="video", infoLabels=infolabels)
            li.setProperty('IsPlayable', 'true')
        else: # Unable to resolve
            xbmc.log("%s: Non resolvable URL: %s %s" % (addon_id, url, label), xbmc.LOGINFO)
    except KeyError:
        xbmc.log("%s: Missing URL" % (addon_id), xbmc.LOGINFO)
    except:
        xbmc.log("%s: Unhandled exception @URLResolve %s" % (addon_id, sys.exc_info()[0]), xbmc.LOGINFO)

    return (li, file_url)

''' Play a single video without touching the current playlist '''
def playVideo(url, label, playlist_id, description=''):
    if True:
        pass
    try:
        pass
        li,file_url = URLResolve(url, label)

        if li is None: return # Non playable URL

        if (re.match('plugin:', file_url)):
            player = MyPlayer()
            xbmc.log("%s: Play video: %s %s" % (addon_id, li.getLabel(), file_url), xbmc.LOGINFO)
            player.play(item = file_url, listitem = li)
            player.setPlaylist(playlist_id)
            while(not player.isProcessed() and player.getWait() > 0):
                xbmc.sleep(1000)
        elif file_url:
            xbmc.log("%s: Resolved URL: %s" % (addon_id, li.getLabel()), xbmc.LOGINFO)
            xbmcplugin.setResolvedUrl(addon_handle, succeeded=True, listitem=li)
        else:
            pass
    except:
        xbmc.log("%s: Unhandled exception @playVideo %s" % (addon_id, sys.exc_info()[0]), xbmc.LOGINFO)

''' Queue a new URL into the playlist '''
def queueVideo(url, label, playlist_id, position, old_url=""):
    if True:
        pass
    try:
        pass
        li, file_url = URLResolve(url, label)

        if li is None:
            return False

        if old_url:
            li.setProperty("original_listitem_url",old_url)

        if file_url:
            if (file_url == old_url):
                return False
            xbmc.PlayList(playlist_id).add(file_url, li, position)
            xbmc.log("%s: Queue resolved URI: %s %s" % (addon_id, li.getLabel(), file_url), xbmc.LOGINFO)
    except:
        xbmc.log("%s: Unhandled exception @queueVideo %s" % (addon_id, sys.exc_info()[0]), xbmc.LOGINFO)
        return False
    return True


''' Recursive function to resolve "plugin://" adresses '''
def resolvePlaylist(playlist_id, position):
    xbmc.log("%s A Looking for plugins in playlist %s from %s" % (addon_id, playlist_id, position), xbmc.LOGINFO)
    playlist = xbmc.PlayList(playlist_id)
    matched = False
    while (not matched and (int(position) < int(playlist.size()))):
        # Try next item on the list
        if resolvePlaylistElement(playlist_id, position):
            matched = True
        position += 1
    if (int(position) < int(playlist.size())):
        xbmc.sleep(10000) # Resolve one video every 10 seconds
        xbmc.executebuiltin("RunScript(script.anyurl.player,mode=resolve_plugin,position=%d,playlist=%d)" % (position, playlist_id))
    else:
        xbmc.log("%s Finished scanning playlist" % (addon_id), xbmc.LOGDEBUG)

def resolvePlaylistElement(playlist_id, position, url='', label='', description=''):
    xbmc.log("%s B Looking for plugins in playlist %s from %s" % (addon_id, playlist_id, position), xbmc.LOGINFO)
    # Process item
    xbmc.log("YY%s"%url, xbmc.LOGINFO)
    playlist = xbmc.PlayList(playlist_id)
    item = playlist[position]
    url_parts = item.getPath().split('?',1)
    if len(url_parts) > 1:
        args = urllib.parse.parse_qs(url_parts[1])
    else:
        return False # Nothing to resolve
    if (re.match('plugin://plugin.video.youtube', url_parts[0])):
        url = reencodeYT(args.get('video_id', [''])[0])
        print(url)
        if not label: label = args.get('label',[''])[0]
        if not label: label = item.getLabel()
        if not description: description = args.get('description',[''])[0]
    elif (re.match('plugin://script.anyurl.player', url_parts[0])):
        url = args.get('url',[''])[0]
        if not label: label = args.get('label',[''])[0]
        if not description: description = args.get('description',[''])[0]
    if (url and label):
        xbmc.log("XX%s"%url, xbmc.LOGINFO)
        replaceItem(playlist_id, position, url, label)
        return True
    return False # Nothing changed

def replaceItem(playlist, position, url, label):
    item = xbmc.PlayList(playlist)[position]
    if not item:
        xbmc.log("%s: No item \"%s\"in playlist %s" % (addon_id, label, playlist))
        return False
    orig_url = item.getPath()
    resolved = queueVideo(url, label, playlist, position, orig_url)
    xbmc.PlayList(playlist).remove(orig_url)
    if not resolved: # Just update infolabels
        infolabels={"Studio":"","Title":label,
                    "plot":description, 'plotoutline': description}
        item.setLabel(label)
        item.setInfo(type="video", infoLabels=infolabels)
        item.setProperty('IsPlayable', 'true')
        xbmc.PlayList(playlist).add(orig_url, item, position)
        xbmc.log("%s: ReplaceItem resolved URI: %s %s" % (addon_id, item.getLabel(), orig_url), xbmc.LOGINFO)
    return resolved


addon_id = 'script.anyurl.player'
addon = xbmcaddon.Addon(id=addon_id)

try:
    # Parse arguments as a handled plugin (e.g: Playable URI)
    base_url = sys.argv[0]
    addon_handle = int(sys.argv[1])
    args = urllib.parse.parse_qs(sys.argv[2][1:])
    mode = args.get('mode',[None])[0]
    url = args.get('url',[''])[0]
    label = args.get('title',[''])[0]
    description = args.get('description',[label])[0]
    position = int(args.get('position',['0'])[0])
    playlist = int(args.get('playlistid',['1'])[0])
    full_url = base_url + sys.argv[2]
except:
    # Parse arguments as a non-handled plugin (e.g. script)
    base_url = "plugin://"+sys.argv[0]
    addon_handle = None
    mode = getArg(sys.argv, 'mode', '')
    url = urllib.parse.unquote(getArg(sys.argv, 'url', ''))
    label = urllib.parse.unquote(getArg(sys.argv, 'title', ''))
    description = urllib.parse.unquote(getArg(sys.argv, 'description', label))
    position = int(getArg(sys.argv, 'position', '-1'))
    playlist = int(getArg(sys.argv, 'playlistid', '1'))
    full_url = "plugin://script.anyurl.resolver/?mode=play_video&url="+urllib.parse.quote(url)

if mode == 'play_video':
    if addon_handle: # Can't play video without a handle
        playVideo(url=url, label=label, playlist_id=playlist)
elif mode == 'queue_video':
    queueVideo(url=url, label=label, playlist_id=playlist, position=position, old_url=full_url)
elif mode == 'resolve_plugin':
    resolvePlaylist(playlist_id=playlist, position=position)
elif mode == 'resolve_single_plugin':
    resolvePlaylistElement(playlist_id=playlist, position=position, url=url, label=label, description=description)
elif mode == 'update_title':
    resolvePlaylistElement(playlist_id=playlist, position=position, url="fake://fake", label=label, description=description)
elif mode == 'test':
    print("Do nothing, yet")
    pass
else:
    xbmc.log("%s: Nothing to play" % (addon_id), xbmc.LOGINFO)