kodi-jacktook / plugin.video.jacktook / lib / gui / source_manager_dialog.py

import json
import os
from datetime import timedelta

import xbmcgui

from lib.db.cached import cache
from lib.utils.kodi.settings import get_setting
from lib.utils.kodi.utils import ADDON_PATH, kodilog, translation

BUILTIN_SOURCE_SETTINGS = [
    ("jackett_enabled", "Jackett"),
    ("prowlarr_enabled", "Prowlarr"),
    ("jacktookburst_enabled", "Burst"),
    ("jackgram_enabled", "Jackgram"),
    ("easynews_enabled", "Easynews"),
    ("stremio_enabled", "Stremio"),
    ("external_scraper_enabled", "External Scraper"),
]

CACHE_KEY = "source_manager_selection"
KNOWN_CACHE_KEY = "source_manager_known_keys"


def _parse_selection(raw):
    if raw is None:
        return []
    if isinstance(raw, str):
        try:
            return json.loads(raw)
        except (ValueError, TypeError):
            return []
    try:
        return list(raw)
    except TypeError:
        return []


def _get_icon_path(name):
    icon_name = f"{name.lower()}.png"
    media_path = os.path.join(ADDON_PATH, "resources", "media", icon_name)
    if os.path.exists(media_path):
        return media_path
    img_path = os.path.join(ADDON_PATH, "resources", "img", icon_name)
    if os.path.exists(img_path):
        return img_path
    return os.path.join(ADDON_PATH, "icon.png")


def _load_stremio_addons():
    """Import and return the Stremio addon selector, or None on failure."""
    try:
        from lib.clients.stremio.helpers import get_selected_stream_addons

        return get_selected_stream_addons
    except Exception as e:
        kodilog(f"Failed to import get_selected_stream_addons: {e}")
        return None


def _add_stremio_sources(items, cache_keys, get_selected_stream_addons):
    """Append enabled Stremio addon entries to items/cache_keys."""
    if get_selected_stream_addons is None:
        return
    try:
        addons = get_selected_stream_addons()
        for addon in addons:
            key = f"Stremio:{addon.key()}"
            addon_name = addon.manifest.name
            addon_icon_path = _get_icon_path(addon_name)
            li_addon = xbmcgui.ListItem(label=addon_name)
            li_addon.setArt({"icon": addon_icon_path})
            items.append(li_addon)
            cache_keys.append(key)
    except Exception as e:
        kodilog(f"Error loading Stremio addons: {e}")


def _build_source_items():
    """Return (ListItem list, cache key list) for enabled sources."""
    get_selected_stream_addons = _load_stremio_addons()
    items = []
    cache_keys = []

    for setting_key, display_name in BUILTIN_SOURCE_SETTINGS:
        if not get_setting(setting_key):
            continue

        if setting_key == "external_scraper_enabled":
            module_name = get_setting("external_scraper_module_name")
            if module_name:
                display_name = str(module_name)

        if setting_key == "stremio_enabled":
            _add_stremio_sources(items, cache_keys, get_selected_stream_addons)
            continue

        icon_path = _get_icon_path(display_name)
        li = xbmcgui.ListItem(label=display_name)
        li.setArt({"icon": icon_path})
        items.append(li)
        cache_keys.append(display_name)

    return items, cache_keys


def _resolve_selection(cache_keys):
    """Load saved selection, auto-select newly enabled sources, and persist changes.

    Returns the current selection list.
    """
    current_selection = _parse_selection(cache.get(CACHE_KEY))
    known_keys = _parse_selection(cache.get(KNOWN_CACHE_KEY))
    modified = False

    if not current_selection:
        current_selection = list(cache_keys)
        known_keys = list(cache_keys)
        modified = True
    else:
        # Only auto-select sources that were not known the last time the
        # dialog was saved. This avoids re-selecting sources the user
        # deliberately deselected.
        newly_enabled = [key for key in cache_keys if key not in known_keys]
        if newly_enabled:
            current_selection.extend(newly_enabled)
            known_keys = list(cache_keys)
            modified = True

    if modified:
        cache.set(CACHE_KEY, json.dumps(current_selection), expires=timedelta(days=365))
        cache.set(KNOWN_CACHE_KEY, json.dumps(known_keys), expires=timedelta(days=365))

    return current_selection


def _persist_selection(selected, cache_keys):
    """Save the user's selection and the set of known source keys."""
    cache.set(CACHE_KEY, json.dumps(selected), expires=timedelta(days=365))
    cache.set(KNOWN_CACHE_KEY, json.dumps(cache_keys), expires=timedelta(days=365))


def open_source_manager_dialog():
    items, cache_keys = _build_source_items()
    if not items:
        xbmcgui.Dialog().notification("Jacktook", translation(90756))
        return

    current_selection = _resolve_selection(cache_keys)
    preselect = [i for i, key in enumerate(cache_keys) if key in current_selection]

    result = xbmcgui.Dialog().multiselect(
        translation(90755),
        items,
        preselect=preselect,
        useDetails=True,
    )

    if result is None or not result:
        return

    selected = [cache_keys[i] for i in result]
    _persist_selection(selected, cache_keys)