Initial commit

lib/main.py

@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+import xbmc
+import xbmcgui
+import xbmcaddon
+import json
+import urllib.request
+import logging
+
+# Initialize the logger
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+ADDON = xbmcaddon.Addon()
+ADDONID = xbmcaddon.Addon().getAddonInfo('id')
+CWD = xbmcaddon.Addon().getAddonInfo('path')
+API_URL = ADDON.getSetting('api_url')
+
+CANCEL_DIALOG = (9, 10, 92, 216, 247, 257, 275, 61467, 61448)
+
+def send_request(endpoint, data=None):
+    url = f"{API_URL}{endpoint}"
+    if data is not None:
+        data = json.dumps(data).encode('utf-8')
+    req = urllib.request.Request(url, data, headers={'Content-Type': 'application/json'})
+    response = urllib.request.urlopen(req)
+    if response.status > 300:
+         res = json.loads(response.read().decode('utf-8'))
+         xbmcgui.Dialog().notification('Error', f"Failed to create dialog: {res['errmsg']}", xbmcgui.NOTIFICATION_ERROR)
+         raise Exception(res["errmsg"])
+    return json.loads(response.read().decode('utf-8'))
+
+
+# Sources ##########################################################
+
+def get_sources():
+    sources = send_request("/sources")
+    res = []
+    for src in sources:
+        source = sources[src]
+        source["id"] = src
+        res.append(source)
+    return res
+    
+def get_current_source():
+    sources = send_request("/sources")
+    for src in sources:
+        source = sources[src]
+        if source["enabled"]:
+            return sources[src]["name"]
+    return None
+
+def enable_source(src):
+    send_request("/sources/" + src + "/enable", {})
+    return True
+
+
+# Mixer ############################################################
+
+def get_mixers():
+    return send_request("/mixer")
+
+def mixer_value(mixer):
+    if mixer["Type"] == "ENUMERATED":
+        return str(mixer["items"][mixer["values"][0]])
+    elif mixer["Type"] == "BOOLEAN":
+        return "Oui" if mixer["items"][mixer["values"][0]] else "Non"
+    elif mixer["Type"] == "INTEGER" and "DBScale" in mixer:
+        return "%0.1f dB" % (mixer["DBScale"]["Min"] + mixer["DBScale"]["Step"] * mixer["values"][0])
+    else:
+        return str(mixer["values"][0])
+
+def set_mixer(mixer, value):
+    val = []
+    for v in mixer["values"]:
+        val.append(value)
+    send_request("/mixer/%s/values" % mixer["NumID"], val)
+    return True
+
+
+class AmplifierControlDialog(xbmcgui.WindowXMLDialog):
+    def __init__(self, *args, **kwargs):
+        super(AmplifierControlDialog, self).__init__(*args, **kwargs)
+        self.data = kwargs['optional1']
+    
+    def onInit(self):
+        self.getControl(2).setLabel("Hathoris")
+
+        self.list = self.getControl(10)
+        self.list.reset()
+        self.list.addItem(xbmcgui.ListItem("Source", get_current_source()))
+
+        self.mixers = get_mixers()
+        for mixer in self.mixers:
+            self.list.addItem(xbmcgui.ListItem(mixer["Name"], mixer_value(mixer)))
+
+        #self.getControl(100).setLabel("Volume")
+        #self.getControl(200).setLabel("Source")
+        return
+
+    def onClick(self, controlId):
+        if controlId == 10:
+            if self.list.getSelectedPosition() == 0:
+                sources = get_sources()
+                selected = xbmcgui.Dialog().select('Sélectionnez la source à utiliser', [s["name"] for s in sources])
+                if selected != -1:
+                    enable_source(sources[selected]["id"])
+                    self.onInit()
+            elif self.mixers[self.list.getSelectedPosition() - 1]["Type"] == "ENUMERATED":
+                items = self.mixers[self.list.getSelectedPosition() - 1]["items"]
+                selected = xbmcgui.Dialog().select(self.mixers[self.list.getSelectedPosition() - 1]["Name"], items, preselect=self.mixers[self.list.getSelectedPosition() - 1]["values"][0])
+                if selected != -1:
+                    set_mixer(self.mixers[self.list.getSelectedPosition() - 1], selected)
+                    self.onInit()
+            elif self.mixers[self.list.getSelectedPosition() - 1]["Type"] == "BOOLEAN":
+                items = ["Non", "Oui"]
+                selected = xbmcgui.Dialog().select(self.mixers[self.list.getSelectedPosition() - 1]["Name"], items, preselect=1 if self.mixers[self.list.getSelectedPosition() - 1]["values"][0] else 0)
+                if selected != -1:
+                    set_mixer(self.mixers[self.list.getSelectedPosition() - 1], True if selected else False)
+                    self.onInit()
+            else:
+                newval = xbmcgui.Dialog().numeric(0, self.mixers[self.list.getSelectedPosition() - 1]["Name"], str(self.mixers[self.list.getSelectedPosition() - 1]["values"][0]))
+                set_mixer(self.mixers[self.list.getSelectedPosition() - 1], int(newval))
+                self.onInit()
+        elif controlId == 100:
+            volume = self.getControl(101).getValue()
+            send_request('set_volume', {'volume': volume})
+        elif controlId == 200:
+            source = self.getControl(201).getSelectedItem().getLabel()
+            send_request('set_source', {'source': source})
+        elif controlId == 18:
+            self.close()
+
+    def onAction(self, action):
+        if action.getButtonCode() in CANCEL_DIALOG:
+            self.close()
+
+if __name__ == '__main__':
+    try:
+      dialog = AmplifierControlDialog('amplifier_control.xml', CWD, 'default', optional1='some data')
+      dialog.doModal()
+      del dialog
+    except Exception as e:
+        logger.error(f"Failed to create dialog: {e}")
+        xbmcgui.Dialog().notification('Error', f"Failed to create dialog: {e}", xbmcgui.NOTIFICATION_ERROR)