import os
import re
import unittest

from plugins.filter.get.all.application_ids import get_all_application_ids
from utils.cache.files import iter_project_files_with_content


def collect_domain_keys():
    """
    Scan all YAML and Python files under the project for usages of
    domains.get('...') and domains['...'] and return a dict mapping each
    domain key to a list of file:line locations where it's used.
    Ignores the integration test file itself.
    """
    pattern = re.compile(
        r"domains(?:\.get\(\s*['\"](?P<id>[^'\"]+)['\"]\s*\)|\[['\"](?P<id2>[^'\"]+)['\"]\])"
    )
    locations: dict[str, list[str]] = {}
    ignore_path = os.path.normpath(__file__)

    for path, text in iter_project_files_with_content(
        extensions=(".yml", ".yaml", ".py")
    ):
        if os.path.normpath(path) == ignore_path:
            continue
        for lineno, line in enumerate(text.splitlines(), start=1):
            for m in pattern.finditer(line):
                key = m.group("id") or m.group("id2")
                locations.setdefault(key, []).append(f"{path}:{lineno}")
    return locations


class TestDomainApplicationIds(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.valid_ids = set(get_all_application_ids(roles_dir="roles"))
        if not cls.valid_ids:
            raise RuntimeError(
                "No application_ids found in roles/*/vars/main.yml"  # nocheck: role-file-spot
            )
        cls.domain_locations = collect_domain_keys()
        if not cls.domain_locations:
            raise RuntimeError(
                "No domains.get(...) or domains[...] usages found to validate"
            )

        cls.ignore_keys = {"canonical", "aliases"}

    def test_all_keys_are_valid(self):
        """Ensure every domains.get/[] key matches a valid application_id (excluding ignored keys)."""

        def is_placeholder(key):
            return bool(re.match(r"^\{.+\}$", key))

        invalid = {}
        for key, locs in self.domain_locations.items():
            if key in self.ignore_keys or is_placeholder(key):
                continue
            if key not in self.valid_ids:
                invalid[key] = locs

        if invalid:
            details = []
            for key, locs in invalid.items():
                locations_str = ", ".join(locs)
                details.append(f"'{key}' at {locations_str}")
            detail_msg = "; ".join(details)
            self.fail(
                f"Found usages of domains with invalid application_ids: {detail_msg}. "
                f"Valid application_ids are: {sorted(self.valid_ids)}"
            )


if __name__ == "__main__":
    unittest.main()