from __future__ import annotations

import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import Session, sessionmaker
from sqlalchemy.pool import StaticPool

from app.core.agent_enums import AgentAssetDomain, AgentAssetStatus
from app.db.base import Base
from app.models.agent_asset import AgentAsset, AgentAssetVersion
from app.schemas.agent_asset import (
    AgentAssetRiskRuleDraftUpdate,
    AgentAssetRiskRuleGenerateRequest,
    AgentAssetRiskRuleRevisionCreate,
)
from app.services.agent_asset_rule_library import AgentAssetRuleLibraryManager
from app.services.risk_rule_generation import RiskRuleGenerationService
from app.services.agent_asset_risk_rule_revision import AgentAssetRiskRuleRevisionService


class NullRuntimeChatService:
    def complete(self, *args, **kwargs) -> None:
        return None


def build_session() -> Session:
    engine = create_engine(
        "sqlite+pysqlite:///:memory:",
        connect_args={"check_same_thread": False},
        poolclass=StaticPool,
    )
    Base.metadata.create_all(bind=engine)
    return sessionmaker(bind=engine, autoflush=False, autocommit=False)()


def test_update_unpublished_risk_rule_draft_updates_business_fields(tmp_path) -> None:
    with build_session() as db:
        asset_id = _create_rule(db, tmp_path)
        updated = AgentAssetRiskRuleRevisionService(db).update_unpublished_draft(
            asset_id,
            AgentAssetRiskRuleDraftUpdate(
                rule_title="差旅绕行说明校验",
                expense_category="travel",
                natural_language="差旅报销存在绕行但未说明原因时，进入风险复核。",
                requires_attachment=True,
            ),
            actor="finance",
        )

        assert updated.name == "差旅绕行说明校验"
        assert updated.description == "差旅报销存在绕行但未说明原因时，进入风险复核。"
        assert updated.scenario_json == ["差旅费"]
        assert updated.config_json["requires_attachment"] is True
        assert updated.config_json["generation_request"]["natural_language"] == updated.description
        assert updated.config_json["last_operation"]["action"] == "update_draft"


def test_update_published_rule_requires_revision(tmp_path) -> None:
    with build_session() as db:
        asset_id = _create_rule(db, tmp_path)
        asset = db.get(AgentAsset, asset_id)
        assert asset is not None
        asset.status = AgentAssetStatus.ACTIVE.value
        asset.published_version = asset.current_version
        db.add(asset)
        db.flush()

        with pytest.raises(PermissionError):
            AgentAssetRiskRuleRevisionService(db).update_unpublished_draft(
                asset_id,
                AgentAssetRiskRuleDraftUpdate(natural_language="已上线规则不能直接覆盖。"),
                actor="finance",
            )


def test_create_revision_draft_for_published_rule_does_not_overwrite_active_version(tmp_path) -> None:
    with build_session() as db:
        asset_id = _create_rule(db, tmp_path)
        asset = db.get(AgentAsset, asset_id)
        assert asset is not None
        asset.status = AgentAssetStatus.ACTIVE.value
        asset.published_version = "v0.1.0"
        asset.current_version = "v0.1.0"
        asset.working_version = "v0.1.0"
        db.add(asset)
        db.flush()

        updated = AgentAssetRiskRuleRevisionService(db).create_revision_draft(
            asset_id,
            AgentAssetRiskRuleRevisionCreate(
                rule_title="差旅票据城市复核",
                natural_language="票据城市与申报目的地不一致时，要求补充说明。",
                requires_attachment=True,
                change_reason="补充城市一致性判断。",
            ),
            actor="manager",
        )

        revision = updated.config_json["revision_draft"]
        assert updated.status == AgentAssetStatus.ACTIVE.value
        assert updated.published_version == "v0.1.0"
        assert updated.working_version == "v0.1.1"
        assert revision["version"] == "v0.1.1"
        assert revision["base_version"] == "v0.1.0"
        assert revision["generation_request"]["natural_language"] == "票据城市与申报目的地不一致时，要求补充说明。"
        assert updated.config_json["last_operation"]["action"] == "create_revision"
        assert db.query(AgentAssetVersion).filter_by(asset_id=asset_id, version="v0.1.1").one()


def _create_rule(db: Session, tmp_path) -> str:
    return RiskRuleGenerationService(
        db,
        rule_library_manager=AgentAssetRuleLibraryManager(rule_root=tmp_path / "rules"),
        runtime_chat_service=NullRuntimeChatService(),
    ).generate_rule_asset(
        AgentAssetRiskRuleGenerateRequest(
            business_domain=AgentAssetDomain.EXPENSE,
            expense_category="travel",
            rule_title="差旅规则草稿",
            natural_language="差旅报销事由缺失时，提示补充说明。",
        ),
        actor="pytest",
    )