import hashlib
import importlib.machinery
import importlib.util
import io
import json
import os
from pathlib import Path
import subprocess
import tarfile
import tempfile
import unittest
from contextlib import redirect_stdout
from unittest import mock


SCRIPT_DIR = Path(__file__).resolve().parent
RUNNER_PATH = SCRIPT_DIR / "nodedc-deploy"
BUILDER_PATH = (
    SCRIPT_DIR / "build-engine-mcp-normalized-identity-search-artifact.mjs"
)
ENGINE_ROOT = SCRIPT_DIR.parent.parent.parent / "NODEDC_ENGINE_INFRA"
PATCH_ID = "engine-mcp-normalized-identity-search-20991231-999"


def load_runner():
    loader = importlib.machinery.SourceFileLoader(
        "nodedc_engine_mcp_normalized_identity_search",
        str(RUNNER_PATH),
    )
    spec = importlib.util.spec_from_loader(loader.name, loader)
    module = importlib.util.module_from_spec(spec)
    loader.exec_module(module)
    return module


RUNNER = load_runner()


class EngineMcpNormalizedIdentitySearchTest(unittest.TestCase):
    def require_current_target_source(self):
        for relative_path, expected in (
            RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_TARGET_SHA256.items()
        ):
            path = ENGINE_ROOT / relative_path
            if (
                not path.is_file()
                or hashlib.sha256(path.read_bytes()).hexdigest() != expected
            ):
                self.skipTest("normalized identity search source has advanced")

    def build(self, artifact_dir, engine_root=ENGINE_ROOT):
        environment = os.environ.copy()
        environment["NODEDC_ENGINE_SOURCE_ROOT"] = str(engine_root)
        environment["NODEDC_DEPLOY_ARTIFACT_DIR"] = str(artifact_dir)
        completed = subprocess.run(
            ["node", str(BUILDER_PATH), PATCH_ID],
            check=True,
            capture_output=True,
            text=True,
            env=environment,
        )
        return json.loads(completed.stdout)

    def test_builder_emits_exact_deterministic_backend_only_successor(self):
        self.require_current_target_source()
        with tempfile.TemporaryDirectory(
            prefix="nodedc-engine-mcp-normalized-identity-search-"
        ) as directory:
            root = Path(directory)
            first = self.build(root / "first")
            second = self.build(root / "second")
            first_artifact = Path(first["artifact"])
            second_artifact = Path(second["artifact"])
            self.assertEqual(first_artifact.read_bytes(), second_artifact.read_bytes())
            self.assertEqual(first["services"], ["nodedc-backend"])
            self.assertEqual(first["mcpVersion"], "0.11.0")
            self.assertEqual(first["tool"], "engine_find_normalized_subjects")
            self.assertEqual(first["commandSurface"], "absent")

            extract = root / "extract"
            with tarfile.open(first_artifact, "r:gz") as archive:
                archive.extractall(extract, filter="data")
            entries = tuple(
                (extract / "files.txt").read_text(encoding="utf-8").splitlines()
            )
            payload = extract / "payload"
            self.assertEqual(
                entries,
                RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_ARTIFACT_ENTRIES,
            )
            self.assertTrue(
                RUNNER.is_engine_mcp_normalized_identity_search_slice(
                    "engine",
                    entries,
                )
            )
            self.assertEqual(
                RUNNER.component_services("engine", entries),
                ("nodedc-backend",),
            )
            RUNNER.validate_engine_mcp_normalized_identity_search_slice(
                payload,
                entries,
            )
            (root / "load").mkdir()
            manifest, loaded_entries, _ = RUNNER.load_artifact(
                first_artifact,
                root / "load",
            )
            self.assertEqual(manifest["component"], "engine")
            self.assertEqual(tuple(loaded_entries), entries)

    def test_builder_rejects_source_drift(self):
        self.require_current_target_source()
        with tempfile.TemporaryDirectory(
            prefix="nodedc-engine-mcp-normalized-identity-drift-"
        ) as directory:
            root = Path(directory)
            engine_copy = root / "engine"
            for relative_path in (
                RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_ARTIFACT_ENTRIES
            ):
                source = ENGINE_ROOT / relative_path
                destination = engine_copy / relative_path
                destination.parent.mkdir(parents=True, exist_ok=True)
                destination.write_bytes(source.read_bytes())
            gateway = engine_copy / RUNNER.ENGINE_NODE_INTELLIGENCE_GATEWAY_REL
            gateway.write_text(
                gateway.read_text(encoding="utf-8") + "\n// drift\n",
                encoding="utf-8",
            )
            with self.assertRaises(subprocess.CalledProcessError):
                self.build(root / "artifact", engine_copy)

    def test_descriptor_and_activation_are_canonically_attested(self):
        self.require_current_target_source()
        descriptor = json.loads(
            (
                ENGINE_ROOT
                / RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_DESCRIPTOR_REL
            ).read_text(encoding="utf-8")
        )
        activation = RUNNER.read_engine_node_intelligence_descriptor(
            ENGINE_ROOT / RUNNER.ENGINE_NODE_INTELLIGENCE_DESCRIPTOR_REL
        )
        self.assertEqual(descriptor["mcpVersion"], "0.11.0")
        self.assertFalse(descriptor["normalizedFactSearch"]["commandSurface"])
        self.assertEqual(
            activation["source"]["gatewaySha256"],
            RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_TARGET_SHA256[
                RUNNER.ENGINE_NODE_INTELLIGENCE_GATEWAY_REL
            ],
        )
        self.assertEqual(
            activation["predecessor"]["gatewaySha256"],
            RUNNER.ENGINE_NODE_INTELLIGENCE_PREDECESSOR_GATEWAY_SHA256,
        )

    def test_preflight_requires_exact_foundation_and_absent_descriptor(self):
        with tempfile.TemporaryDirectory(
            prefix="nodedc-engine-mcp-normalized-identity-preflight-"
        ) as directory:
            root = Path(directory)
            hashes = {}
            for mapping in (
                RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_PREDECESSOR_SHA256,
                RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_FOUNDATION_SHA256,
            ):
                for relative_path, expected in mapping.items():
                    path = root / relative_path
                    path.parent.mkdir(parents=True, exist_ok=True)
                    path.write_text("installed\n", encoding="utf-8")
                    hashes[path] = expected
            with (
                mock.patch.object(RUNNER, "component_root", return_value=root),
                mock.patch.object(
                    RUNNER,
                    "sha256_file",
                    side_effect=lambda path: hashes[Path(path)],
                ),
                mock.patch.object(
                    RUNNER,
                    "preflight_engine_credential_backend_runtime",
                    return_value={"mode": "verified-derived-retry"},
                ),
            ):
                result = (
                    RUNNER
                    .preflight_engine_mcp_normalized_identity_search_predecessor()
                )
            self.assertEqual(
                result["mode"],
                "classified-aspects-v1-to-normalized-identity-search-v1",
            )
            self.assertEqual(
                result["foundation_sha256"],
                RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_FOUNDATION_SHA256,
            )

    def test_plan_renders_closed_read_only_identity_boundary(self):
        self.require_current_target_source()
        with tempfile.TemporaryDirectory(
            prefix="nodedc-engine-mcp-normalized-identity-plan-"
        ) as directory:
            root = Path(directory)
            artifact = Path(self.build(root / "artifacts")["artifact"])
            live_root = root / "live"
            live_root.mkdir()
            preflight = {
                "mode": "classified-aspects-v1-to-normalized-identity-search-v1",
                "predecessor_sha256": dict(
                    RUNNER
                    .ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_PREDECESSOR_SHA256
                ),
                "foundation_sha256": dict(
                    RUNNER
                    .ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_FOUNDATION_SHA256
                ),
                "target_sha256": dict(
                    RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_TARGET_SHA256
                ),
                "new_paths": tuple(
                    RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_NEW_PATHS
                ),
                "backend_mode": "verified-derived-retry",
            }
            output = io.StringIO()
            with (
                mock.patch.object(RUNNER, "validate_artifact_location"),
                mock.patch.object(RUNNER, "ensure_layout"),
                mock.patch.object(RUNNER, "TMP_DIR", root),
                mock.patch.object(RUNNER, "component_root", return_value=live_root),
                mock.patch.object(
                    RUNNER,
                    "component_compose_root",
                    return_value=live_root,
                ),
                mock.patch.object(
                    RUNNER,
                    "preflight_engine_mcp_normalized_identity_search_predecessor",
                    return_value=preflight,
                ),
                mock.patch.object(
                    RUNNER,
                    "preflight_engine_credential_backend_runtime",
                    return_value={"mode": "verified-derived-retry"},
                ),
                mock.patch.object(RUNNER, "state_has_sha", return_value=False),
                mock.patch.object(RUNNER, "state_has_patch_id", return_value=False),
                redirect_stdout(output),
            ):
                RUNNER.plan_artifact(artifact)
            plan = output.getvalue()
            self.assertIn("services=nodedc-backend", plan)
            self.assertIn("engine_mcp_version=0.11.0", plan)
            self.assertIn(
                "engine_mcp_tool=engine_find_normalized_subjects",
                plan,
            )
            self.assertIn("engine_mcp_command_surface_included=no", plan)
            self.assertIn(
                "engine_data_product_added=fleet.units.identity.current.v1",
                plan,
            )
            self.assertIn("l2_graph=untouched", plan)
            self.assertIn("state=new", plan)

    def test_healthchecks_dispatch_identity_search_acceptance(self):
        entries = RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_ARTIFACT_ENTRIES
        with tempfile.TemporaryDirectory(
            prefix="nodedc-engine-mcp-normalized-identity-health-"
        ) as directory:
            root = Path(directory)
            all_hashes = {
                **RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_TARGET_SHA256,
                **RUNNER.ENGINE_MCP_NORMALIZED_IDENTITY_SEARCH_FOUNDATION_SHA256,
            }
            for relative_path in all_hashes:
                path = root / relative_path
                path.parent.mkdir(parents=True, exist_ok=True)
                path.write_text("target\n", encoding="utf-8")

            def target_hash(path):
                relative_path = Path(path).relative_to(root).as_posix()
                return all_hashes[relative_path]

            with (
                mock.patch.object(RUNNER, "component_root", return_value=root),
                mock.patch.object(RUNNER, "sha256_file", side_effect=target_hash),
                mock.patch.object(RUNNER, "healthcheck_compose_service"),
                mock.patch.object(RUNNER, "component_healthchecks", return_value=()),
                mock.patch.object(
                    RUNNER,
                    "preflight_engine_credential_backend_runtime",
                    return_value={"mode": "verified-derived-retry"},
                ),
                mock.patch.object(
                    RUNNER,
                    "accept_engine_mcp_normalized_identity_search_runtime",
                    return_value={"live": "accepted"},
                ) as acceptance,
                mock.patch.object(RUNNER, "healthcheck_container"),
            ):
                RUNNER.run_healthchecks(
                    "engine",
                    entries,
                    ("nodedc-backend",),
                )
            acceptance.assert_called_once_with()

    def test_live_acceptance_probes_read_only_normalized_boundary(self):
        expected_live = (
            "engine-mcp-normalized-identity-search:"
            "0.11.0:canonical-facts:full-admin-identifiers:read-only:v1"
        )
        with (
            mock.patch.object(RUNNER, "component_root", return_value=ENGINE_ROOT),
            mock.patch.object(
                RUNNER,
                "validate_engine_mcp_normalized_identity_search_slice",
            ),
            mock.patch.object(
                RUNNER,
                "engine_backend_container_id",
                return_value="backend",
            ),
            mock.patch.object(
                RUNNER,
                "run_engine_backend_probe",
                return_value=expected_live,
            ) as backend_probe,
        ):
            result = RUNNER.accept_engine_mcp_normalized_identity_search_runtime()
        self.assertEqual(result["live"], expected_live)
        probe_source = backend_probe.call_args.args[0][-1]
        self.assertIn("engine_find_normalized_subjects", probe_source)
        self.assertIn("commandSurfaceIncluded: false", probe_source)
        self.assertIn("classification!=='read'", probe_source)


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