rename snippets to utils

utils/icon_resolver.py

@@ -0,0 +1,103 @@
+import json
+import os
+import re
+
+import gi
+
+gi.require_version("Gtk", "3.0")
+from gi.repository import GLib, Gtk
+from loguru import logger
+
+
+# TODO WIP
+# Idea: nearest string matching algorithm
+#       if already exists: retrieve from json
+#       if found: stor in json: app_id -> icon_name
+#       if not found: store in json -> misisng-icon
+
+
+CACHE_DIR = str(GLib.get_user_cache_dir()) + "/modus"
+ICON_CACHE_FILE = CACHE_DIR + "/icons.json"
+if not os.path.exists(CACHE_DIR):
+    os.makedirs(CACHE_DIR)
+
+
+class IconResolver:
+    def __init__(
+        self, default_applicaiton_icon: str = "application-x-executable-symbolic"
+    ):
+        if os.path.exists(ICON_CACHE_FILE):
+            f = open(ICON_CACHE_FILE)
+            try:
+                self._icon_dict = json.load(f)
+            except json.JSONDecodeError:
+                logger.info("[ICONS] Cache file does not exist or is corrupted")
+            f.close()
+        else:
+            self._icon_dict = {}
+
+        self.default_applicaiton_icon = default_applicaiton_icon
+
+    def get_icon_name(self, app_id: str):
+        if app_id in self._icon_dict:
+            return self._icon_dict[app_id]
+        new_icon = self._compositor_find_icon(app_id)
+        logger.info(
+            f"[ICONS] found new icon: '{new_icon}' for app id: '{app_id}', storing..."
+        )
+        self._store_new_icon(app_id, new_icon)
+        return new_icon
+
+    def get_icon_pixbuf(self, app_id: str, size: int = 16):
+        return Gtk.IconTheme.get_default().load_icon(
+            self.get_icon_name(app_id),
+            size,
+            Gtk.IconLookupFlags.FORCE_SIZE,
+        )
+
+    def _store_new_icon(self, app_id: str, icon: str):
+        self._icon_dict[app_id] = icon
+        with open(ICON_CACHE_FILE, "w") as f:
+            json.dump(self._icon_dict, f)
+            f.close()
+
+    def _get_icon_from_desktop_file(self, desktop_file_path: str):
+        # TODO: get icon in the [Desktop Entry] section only
+        with open(desktop_file_path) as f:
+            for line in f.readlines():
+                if "Icon=" in line:
+                    return "".join(line[5:].split())
+            return self.default_applicaiton_icon
+
+    def _get_desktop_file(self, app_id: str) -> str | None:
+        data_dirs = GLib.get_system_data_dirs()
+        for data_dir in data_dirs:
+            data_dir = data_dir + "/applications/"
+            if os.path.exists(data_dir):
+                # Do name resolving here
+
+                files = os.listdir(data_dir)
+                matching = [
+                    s for s in files if "".join(app_id.lower().split()) in s.lower()
+                ]
+                if matching:
+                    return data_dir + matching[0]
+
+                for word in list(filter(None, re.split(r"-|\.|_|\s", app_id))):
+                    matching = [s for s in files if word.lower() in s.lower()]
+                    if matching:
+                        return data_dir + matching[0]
+
+        return None
+
+    def _compositor_find_icon(self, app_id: str):
+        if Gtk.IconTheme.get_default().has_icon(app_id):
+            return app_id
+        if Gtk.IconTheme.get_default().has_icon(app_id + "-desktop"):
+            return app_id + "-desktop"
+        desktop_file = self._get_desktop_file(app_id)
+        return (
+            self._get_icon_from_desktop_file(desktop_file)
+            if desktop_file
+            else self.default_applicaiton_icon
+        )