from django.contrib.auth import get_user_model
from django.core.management import call_command

VALID_SKILL = {
    "code": "issue-structure",
    "name": "问题结构化",
    "description": "提取问题关键信息",
    "prompt_template": "请分析：{{ input.problem }}",
    "input_schema": {
        "type": "object",
        "properties": {"problem": {"type": "string"}},
        "required": ["problem"],
    },
    "output_schema": {
        "type": "object",
        "properties": {"summary": {"type": "string"}},
        "required": ["summary"],
        "additionalProperties": False,
    },
    "example_input": {"problem": "采集任务没有进度"},
}


def login_admin(client):
    call_command("bootstrap_governance")
    user = get_user_model().objects.get(username="admin")
    client.force_login(user)
    return user


def test_skill_draft_publish_clone_and_rollback(client, db):
    login_admin(client)
    created = client.post("/api/skills", data=VALID_SKILL, content_type="application/json")
    assert created.status_code == 200
    skill_id = created.json()["id"]

    published_v1 = client.post(f"/api/skills/{skill_id}/publish")
    assert published_v1.status_code == 200
    assert published_v1.json()["version"] == 1
    assert published_v1.json()["status"] == "published"

    draft_v2 = client.post(f"/api/skills/{skill_id}/draft")
    assert draft_v2.status_code == 200
    assert draft_v2.json()["version"] == 2
    updated = client.patch(
        f"/api/skills/{skill_id}/draft",
        data={"prompt_template": "新版：{{ input.problem }}"},
        content_type="application/json",
    )
    assert updated.status_code == 200
    assert updated.json()["prompt_template"].startswith("新版")

    published_v2 = client.post(f"/api/skills/{skill_id}/publish")
    assert published_v2.status_code == 200
    rolled_back = client.post(f"/api/skills/{skill_id}/rollback/1")
    assert rolled_back.status_code == 200
    assert rolled_back.json()["active_version"] == 1


def test_invalid_schema_or_template_cannot_be_saved(client, db):
    login_admin(client)
    created = client.post("/api/skills", data=VALID_SKILL, content_type="application/json")
    skill_id = created.json()["id"]

    invalid = client.patch(
        f"/api/skills/{skill_id}/draft",
        data={"prompt_template": "{{ input.problem.upper() }}"},
        content_type="application/json",
    )

    assert invalid.status_code == 400


def test_skill_permission_is_required(client, db):
    user = get_user_model().objects.create_user(username="requester", password="password")
    client.force_login(user)

    response = client.get("/api/skills")

    assert response.status_code == 403