main.py

'''Kodi video plugin for Intergalactic FM'''

__author__ = 'Shogutora, Pander, Dreamer'
__copyright__ = 'GPL v.3 https://www.gnu.org/copyleft/gpl.html'

from json import load
from sys import argv
from os.path import isfile
import sys
from urllib.parse import parse_qsl
import requests

import xbmcgui
import xbmcplugin
import xbmcaddon
from xbmc import log, LOGDEBUG, LOGERROR
from xbmcvfs import translatePath

__addonid__ = "plugin.video.intergalacticfm"
base = translatePath(f'special://home/addons/{__addonid__}/resources/')

# Get the plugin url in plugin:// notation.
_url = argv[0]
# Get the plugin handle as an integer number.
_handle = int(argv[1])

_addon = xbmcaddon.Addon()


FM = 'https://www.intergalactic.fm/'
TV = 'https://intergalactic.tv/'
PN = 'ifm-system/playingnow.json'
PL = 'playlist.m3u8'


#TODO If JSON on ifm-site is younger, load that file. Otherwise, load local file.
streams = load(open(f'{base}streams.json'))  # pylint:disable=consider-using-with,unspecified-encoding

def now_videos(streams):
    '''
    Create list of available streams
    '''

    req = requests.get(f'{FM}{PN}')
    # log(f'{__addonid__} JSON: {req.json()}', LOGDEBUG)

    try:
        nowplay = req.json()
        npvids = nowplay['11']
        log(f'{__addonid__} npvids: {npvids}', LOGDEBUG)
    except Exception as exc:
        npvids = []
        log(f'{__addonid__} Error with nowplay[\'11\'] {exc}', LOGERROR)

    listvids = []

    for key in streams.keys(): # order is determined in streams.json
        if key in set(npvids):
            listvids.append(streams[key])
# Uncomment the next two lines to show offline streams for development only
#        else:
#            listvids.append(streams[key])

    return listvids


def list_videos():
    '''
    Create the list of playable streams in the Kodi interface.
    '''

    listing = []

    xbmcplugin.setPluginCategory(_handle, 'Channels')
    xbmcplugin.setContent(_handle, 'videos')

    videos = now_videos(streams)

    for video in videos:
        label = f"Live - {video['label']}"
        list_item = xbmcgui.ListItem(label=label)
        list_item.setInfo(type='video', infoLabels={'genre': video['genre'],
                                                    'plot': video['plot'],
                                                    'tagline': video['tagline']})

        # see https://kodi.wiki/view/Movie_artwork
        # only poster, fanart and clearlogo is supported/needed

        art = {}

        # poster 1000x1500 1:1.5 PNG
        poster = base + video['label'].lower().replace(' ', '_') + '-poster.png'
        if isfile(poster):
            art['poster'] = poster
        else: # note: specific fallback
            art['poster'] = f'{base}intergalactic_tv-poster.png'
        #log(f"{__addonid__} poster: {art['poster']}", LOGDEBUG)

        # fanart 1920x1080 16:9 JPG
        fanart = base + video['label'].lower().replace(' ', '_') + '-fanart.jpg'
        if isfile(fanart):
            art['fanart'] = fanart
        else: # note: specific fallback
            art['fanart'] = f'{base}cbs_tv-fanart.jpg'
        #log(f"{__addonid__} fanart: {art['fanart']}", LOGDEBUG)

        # clearlogo 800x310 1:0.388 transparent PNG (is top-left corner overlay)
        clearlogo = base + video['label'].lower().replace(' ', '_') + '-clearlogo.png'
        if isfile(clearlogo):
            art['clearlogo'] = clearlogo
        else: # note: specific fallback
            art['clearlogo'] = f'{base}intergalactic_tv-clearlogo.png'
        #log(f"{__addonid__} clearlogo: {art['clearlogo']}", LOGDEBUG)

        list_item.setArt(art)
        list_item.setProperty('IsPlayable', 'true')

        url = f"{TV}{video['url']}{PL}"
        #log(f'{__addonid__} url: {url}', LOGDEBUG)
        url = f'{_url}?action=play&video={url}'
        is_folder = False

        listing.append((url, list_item, is_folder))

    xbmcplugin.addDirectoryItems(_handle, listing, len(listing))
    xbmcplugin.addSortMethod(_handle, xbmcplugin.SORT_METHOD_LABEL_IGNORE_THE)
    xbmcplugin.endOfDirectory(_handle)


def play_video(path):
    '''
    Play a video by the provided path.
    :param path: Fully-qualified video URL
    :type path: str
    '''
    # Create a playable item with a path to play.
    play_item = xbmcgui.ListItem(path=path)
    # Pass the item to the Kodi player.
    xbmcplugin.setResolvedUrl(_handle, True, listitem=play_item)


def router(paramstring):
    '''
    Router function that calls other functions
    depending on the provided paramstring
    :param paramstring: URL encoded plugin paramstring
    :type paramstring: str
    '''
    # Parse a URL-encoded paramstring to the dictionary of
    # {<parameter>: <value>} elements
    params = dict(parse_qsl(paramstring[1:]))
    # Check the parameters passed to the plugin
    if params:
        if params['action'] == 'listing':
            # Display the list of videos in a provided category.
            list_videos()
        elif params['action'] == 'play':
            # Play a video from a provided URL.
            play_video(params['video'])
        else:
            # If the provided paramstring does not contain a supported action
            # we raise an exception. This helps to catch coding errors,
            # e.g. typos in action names.
            raise ValueError(f'Invalid paramstring: {paramstring}!')
    else:
        list_videos()


if __name__ == '__main__':
    # Call the router function and pass the plugin call parameters to it.
    # We use string slicing to trim the leading '?' from the plugin call paramstring

    router(sys.argv[2])