from laya_router.skills import decide_skill, discover_skills, suggest_skill


def _skill(root, name, description, *, disabled=False, body="Do the documented steps."):
    folder = root / name
    folder.mkdir(parents=True)
    flag = "true" if disabled else "false"
    (folder / "SKILL.md").write_text(
        f"---\nname: {name}\ndescription: {description}\ndisable-model-invocation: {flag}\n---\n{body}\n",
        encoding="utf-8",
    )


def test_discover_skips_disabled_skills(tmp_path):
    home = tmp_path / "home"
    _skill(home / ".claude" / "skills", "commit", "Create a git commit")
    _skill(home / ".agents" / "skills", "hidden", "Should stay hidden", disabled=True)
    found = discover_skills(None, home)
    assert [skill.name for skill in found] == ["commit"]


def test_gate_and_fits_thresholds():
    assert decide_skill(
        ranked_names=["commit", "review"],
        gate_scores={"acts_on_user_system": 0.1, "would_follow_documented_procedure": 0.1, "prose_suffices": 0.9},
        rerank_choice="commit",
        fits=[0.9, 0.1],
    ) is None
    assert decide_skill(
        ranked_names=["commit", "review"],
        gate_scores={"acts_on_user_system": 0.8, "would_follow_documented_procedure": 0.8, "prose_suffices": 0.1},
        rerank_choice="review",
        fits=[0.2, 0.1],
    ) is None
    assert decide_skill(
        ranked_names=["commit", "review"],
        gate_scores={"acts_on_user_system": 0.8, "would_follow_documented_procedure": 0.8, "prose_suffices": 0.1},
        rerank_choice="review",
        fits=[0.4, 0.7],
    ) == "review"


class _Agent:
    def __init__(self, answers):
        self.answers = list(answers)
        self.calls = 0

    def predict(self, state, questions):
        payload = self.answers[self.calls % len(self.answers)]
        self.calls += 1
        return {"answers": payload}


def test_suggest_skips_a_skill_already_named(tmp_path):
    home = tmp_path / "home"
    _skill(home / ".claude" / "skills", "commit", "Create a git commit")
    skills = discover_skills(None, home)
    agent = _Agent(
        [
            {
                "which": {"choice": "commit", "probabilities": {"commit": 0.9}},
                "acts_on_user_system": {"noul": 0.9},
                "would_follow_documented_procedure": {"noul": 0.9},
                "prose_suffices": {"noul": 0.1},
            },
            {
                "which": {"choice": "commit"},
                "fits_0": {"noul": 0.8},
            },
        ]
    )
    first = suggest_skill(agent, skills, "commit these changes", inject="suggest", already=set())
    assert first["name"] == "commit"
    assert first["mode"] == "suggest"
    again = suggest_skill(agent, skills, "commit these changes", inject="suggest", already={"commit"})
    assert again is None