[plugin.video.opentakserver] 1.0.0

plugin.video.opentakserver/Readme.md

@@ -0,0 +1,3 @@
+# OpenTAKServer Kodi Plugin
+
+A simple plugin that allows you to watch live streams and recordings from your [OpenTAKServer](https://github.com/brian7704/OpenTAKServer) in [Kodi](https://kodi.tv).

plugin.video.opentakserver/addon.xml

@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<addon id="plugin.video.opentakserver"
+version="1.0.0"
+name="OpenTAKServer Video Plugin"
+provider-name="OpenTAKServer">
+<requires>
+  <import addon="xbmc.python" version="3.0.1"/>
+  <import addon="script.module.requests" version="2.31.0"/>
+</requires>
+<extension point="xbmc.python.pluginsource" library="main.py">
+  <provides>video</provides>
+</extension>
+<extension point="xbmc.addon.metadata">
+  <summary lang="en_GB">OpenTAKServer Video Plugin</summary>
+  <description lang="en_GB">View your OpenTAKServer video streams and recordings in Kodi</description>
+  <license>GPL-3.0-only</license>
+  <assets>
+    <icon>resources/images/ots-logo.png</icon>
+    <fanart>resources/images/ots-logo.png</fanart>
+  </assets>
+</extension>
+</addon>

plugin.video.opentakserver/main.py

@@ -0,0 +1,185 @@
+import datetime
+import os
+import sys
+import uuid
+from urllib.parse import urlencode, parse_qsl
+import requests
+import pickle
+
+import xbmc
+import xbmcgui
+import xbmcplugin
+import xbmcvfs
+import xbmcaddon
+from xbmcaddon import Addon
+from xbmcvfs import translatePath
+
+# Get the plugin url in plugin:// notation.
+URL = sys.argv[0]
+# Get a plugin handle as an integer number.
+HANDLE = int(sys.argv[1])
+# Get addon base path
+ADDON_PATH = translatePath(Addon().getAddonInfo('path'))
+ICONS_DIR = os.path.join(ADDON_PATH, 'resources', 'images', 'icons')
+FANART_DIR = os.path.join(ADDON_PATH, 'resources', 'images', 'fanart')
+__addon__ = xbmcaddon.Addon(id='plugin.video.opentakserver')
+__addondir__ = xbmcvfs.translatePath( __addon__.getAddonInfo('profile'))
+
+s = requests.session()
+csrf_token = None
+
+def get(url):
+    f = xbmcvfs.File(os.path.join(__addondir__, "cookies.bin"), 'rb')
+    s.cookies = pickle.loads(f.readBytes())
+    f.close()
+
+    # TODO: Certificates
+    return s.get(url, verify=False)
+
+def login():
+    r = s.post(xbmcplugin.getSetting(HANDLE, "server_url") + "/api/login",
+               json={'username': xbmcplugin.getSetting(HANDLE, "username"),
+                     'password': xbmcplugin.getSetting(HANDLE, "password")})
+    if r.status_code == 200:
+        global csrf_token
+        csrf_token = r.json()['response']['csrf_token']
+        xbmc.log(csrf_token, xbmc.LOGINFO)
+        xbmc.log(str(s.cookies), xbmc.LOGINFO)
+        f = xbmcvfs.File(os.path.join(__addondir__, 'cookies.bin'), 'wb')
+        f.write(pickle.dumps(s.cookies))
+        f.close()
+    else:
+        dialog = xbmcgui.Dialog()
+        dialog.notification('Login Failed', 'Check the settings and try again', xbmcgui.NOTIFICATION_INFO, 5000)
+
+
+def format_url(**kwargs):
+    """
+    Create a URL for calling the plugin recursively from the given set of keyword arguments.
+
+    :param kwargs: "argument=value" pairs
+    :return: plugin call URL
+    :rtype: str
+    """
+    return '{}?{}'.format(URL, urlencode(kwargs))
+
+
+def list_videos():
+    """
+    Create the list of playable videos in the Kodi interface.
+    """
+    #genre_info = get_videos()
+    # Set plugin category. It is displayed in some skins as the name
+    # of the current section.
+    xbmcplugin.setPluginCategory(HANDLE, "OpenTAKServer")
+    # Set plugin content. It allows Kodi to select appropriate views
+    # for this type of content.
+    xbmcplugin.setContent(HANDLE, 'videos')
+    # Finish creating a virtual folder.
+    xbmcplugin.endOfDirectory(HANDLE)
+
+
+
+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))
+    xbmc.log("paramstring " + str(paramstring), xbmc.LOGINFO)
+    xbmcplugin.setPluginCategory(HANDLE, "OpenTAKServer")
+    xbmcplugin.setContent(HANDLE, 'videos')
+
+    window = xbmcgui.Window(10000)
+
+    # Check the parameters passed to the plugin
+    if not params:
+        login()
+        xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="streams", page="1"), xbmcgui.ListItem(label="Streams"), True)
+        xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="recordings", page="1"), xbmcgui.ListItem(label="Recordings"), True)
+        xbmcplugin.endOfDirectory(HANDLE)
+
+    elif params['choice'] == "streams":
+        page = params["page"]
+        if not page:
+            page = 1
+        else:
+            page = int(page)
+
+        if page > 1:
+            list_item = xbmcgui.ListItem(label="Previous Page")
+            xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="streams", page=str(page - 1)), list_item, isFolder=True)
+
+        streams = get(xbmcplugin.getSetting(HANDLE, "server_url") + "/api/video_streams").json()
+        for stream in streams['results']:
+            list_item = xbmcgui.ListItem(label=stream['path'])
+
+            # The random UUID prevents Kodi from pulling the thumbnail from cache. The server ignores it
+            thumbnail = xbmcplugin.getSetting(HANDLE,"server_url") + '/api/videos/thumbnail?path={}&random={}|Cookie=Cookie: '.format(stream['path'], str(uuid.uuid4()))
+            for cookie in s.cookies:
+                thumbnail += cookie.name + "=" + cookie.value + "; "
+            list_item.setArt({'thumb': thumbnail, 'fanart': thumbnail})
+
+            link = stream['rtsp_link'].split("//", 1)
+            xbmcplugin.addDirectoryItem(HANDLE, "rtsp://{}:{}@{}".format(xbmcplugin.getSetting(HANDLE, "username"), xbmcplugin.getSetting(HANDLE, "password"), link[-1]), list_item, False)
+
+        if page < streams['total_pages']:
+            list_item = xbmcgui.ListItem(label="Next Page")
+            xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="streams", page=str(page + 1)), list_item, isFolder=True)
+        xbmcplugin.endOfDirectory(HANDLE)
+
+    elif params['choice'] == 'recordings':
+        page = params["page"]
+        if not page:
+            page = 1
+        else:
+            page = int(page)
+
+        if page > 1:
+            list_item = xbmcgui.ListItem(label="Previous Page")
+            xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="recordings", page=str(page - 1)), list_item, isFolder=True)
+
+        server_url = xbmcplugin.getSetting(HANDLE, "server_url")
+        recordings = get("{}/api/videos/recordings?page={}".format(server_url, page))
+        for recording in recordings.json()['results']:
+            url = "{}/api/videos/recording?id={}|Cookie=Cookie: ".format(server_url, recording['id'])
+            thumbnail = '{}/api/videos/thumbnail?path={}&recording={}&random={}|Cookie=Cookie: '.format(server_url, recording['path'], recording['filename'], str(uuid.uuid4()))
+            for cookie in s.cookies:
+                url += cookie.name + "=" + cookie.value + "; "
+                thumbnail += cookie.name + "=" + cookie.value + "; "
+            list_item = xbmcgui.ListItem(label=recording['path'] + " - " + recording['start_time'])
+            list_item.setArt({'thumb': thumbnail})
+            try:
+                start_time = datetime.datetime.strptime(recording['start_time'], "%Y-%m-%dT%H:%M:%SZ")
+            except:
+                start_time = datetime.datetime.now()
+
+            tag = list_item.getVideoInfoTag()
+            tag.setDateAdded(start_time.strftime("%Y-%m-%d %H:%M:%S"))
+            tag.setPremiered(start_time.strftime("%Y-%m-%d %H:%M:%S"))
+            tag.setDuration(recording['duration'])
+            xbmcplugin.addDirectoryItem(HANDLE, url, list_item)
+
+        if page < recordings.json()['total_pages']:
+            list_item = xbmcgui.ListItem(label="Next Page")
+            xbmcplugin.addDirectoryItem(HANDLE, format_url(choice="recordings", page=str(page + 1)), list_item, isFolder=True)
+
+        xbmcplugin.endOfDirectory(HANDLE)
+    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}!')
+
+
+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
+    if not xbmcplugin.getSetting(HANDLE, "server_url") or not xbmcplugin.getSetting(HANDLE, "username") or not xbmcplugin.getSetting(HANDLE, "password"):
+        xbmcaddon.Addon().openSettings()
+    router(sys.argv[2][1:])

plugin.video.opentakserver/resources/settings.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="utf-8" standalone="yes"?>
+<settings>
+    <category label="32001">
+        <setting label="Username" type="text"   id="username" default="" />
+        <setting label="Password" type="text"   id="password" option="hidden"  enable="!eq(-1,)" default="" />
+        <setting label="Server URL" type="text" id="server_url" default="https://public.opentakserver.io" />
+    </category>
+</settings>