computer-playbook/plugins/lookup/status.py

from __future__ import annotations

from typing import Any

from ansible.plugins.lookup import LookupBase

from utils.roles.validation.invokable import list_invokable_app_ids

_STATUS_CACHE: dict[tuple, dict[str, list[str]]] = {}


def _reset_cache_for_tests() -> None:
    _STATUS_CACHE.clear()


def _coerce_to_list(raw: Any) -> list[str]:
    # CSV input (`-e APPLICATIONS_WHITELIST=a,b`) → tokenise; `list(str)` would yield chars.
    if raw is None:
        return []
    if isinstance(raw, str):
        return [item.strip() for item in raw.split(",") if item.strip()]
    return [str(item) for item in raw]


class LookupModule(LookupBase):
    def run(
        self,
        terms: list[Any],
        variables: dict[str, Any] | None = None,
        **kwargs: Any,
    ) -> list[dict[str, list[str]]]:
        vars_ = variables or getattr(self._templar, "available_variables", {}) or {}

        whitelist = _coerce_to_list(vars_.get("APPLICATIONS_WHITELIST"))
        groups = _coerce_to_list(vars_.get("group_names"))
        key = (tuple(whitelist), tuple(groups))

        cached = _STATUS_CACHE.get(key)
        if cached is not None:
            return [cached]

        running = whitelist or groups
        result = {
            "whitelist": whitelist,
            "running": list(running),
            "groups": groups,
            "all": list(list_invokable_app_ids()),
        }
        _STATUS_CACHE[key] = result
        return [result]