import json
import pytest
import httpx
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.pool import StaticPool

from app.database import Base
from app.models.upstream import Upstream
from app.models.upstream_key import UpstreamGeneratedKey
from app.models.website import Website
from app.routers.websites import preview_cleanup_invalid_accounts, execute_cleanup_invalid_accounts
from app.services.website_client import Sub2ApiWebsiteClient


@pytest.fixture()
def db_session():
    engine = create_engine(
        "sqlite://",
        connect_args={"check_same_thread": False},
        poolclass=StaticPool,
    )
    from app.models import admin_user, upstream, snapshot, webhook_config, notification_log, custom_page, website, revoked_token, upstream_key
    Base.metadata.create_all(bind=engine)
    TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
    db = TestingSessionLocal()
    try:
        yield db
    finally:
        db.close()
        Base.metadata.drop_all(bind=engine)


class MockResponse:
    def __init__(self, status_code, lines, raise_exc=None):
        self.status_code = status_code
        self._lines = lines
        self._raise_exc = raise_exc

    def __enter__(self):
        if self._raise_exc:
            raise self._raise_exc
        return self

    def __exit__(self, *args):
        pass

    def iter_lines(self):
        return self._lines


def mock_stream(status_code, lines, raise_exc=None):
    def mock_fn(*args, **kwargs):
        return MockResponse(status_code, lines, raise_exc)
    return mock_fn


def test_website_client_test_account_sse_scenarios(monkeypatch):
    """测试 Sub2ApiWebsiteClient.test_account 在各种 SSE 场景下的解析、处理、ID 转义和 API 日志记录。"""
    c = Sub2ApiWebsiteClient(
        base_url="http://w1",
        api_prefix="api/v1",
        auth_type="bearer",
        auth_config={"token": "t1"},
    )

    log_calls = []
    def mock_log(**kwargs):
        log_calls.append(kwargs)
    monkeypatch.setattr("app.services.website_client.log_external_api_call", mock_log)

    # 1. success=true，包含带有斜杠的 ID 校验以测试 URL quote 转义
    monkeypatch.setattr(c._client, "stream", mock_stream(200, ["event: test_complete", "data: {\"success\": true}"]))
    res = c.test_account("acc/1")
    assert res["status"] == "success"
    assert res["cleanup_allowed"] is False
    # 验证日志记录和 URL 转义：acc/1 应被转义为 acc%2F1
    assert len(log_calls) == 1
    assert log_calls[-1]["method"] == "POST"
    assert log_calls[-1]["path"] == "/api/v1/accounts/acc%2F1/test"
    assert log_calls[-1]["success"] is True

    # 2. success=false
    monkeypatch.setattr(c._client, "stream", mock_stream(200, ["event: test_complete", "data: {\"success\": false}"]))
    res = c.test_account("acc/1")
    assert res["status"] == "failed"
    assert res["cleanup_allowed"] is True
    assert len(log_calls) == 2
    assert log_calls[-1]["success"] is True

    # 3. HTTP 404
    monkeypatch.setattr(c._client, "stream", mock_stream(404, []))
    res = c.test_account("acc/1")
    assert res["status"] == "404"
    assert res["cleanup_allowed"] is True
    assert len(log_calls) == 3
    assert log_calls[-1]["status_code"] == 404
    assert log_calls[-1]["success"] is False
    assert log_calls[-1]["error_type"] == "HTTPStatus"
    assert log_calls[-1]["error_message"] == "HTTP 404"

    # 4. error event
    monkeypatch.setattr(c._client, "stream", mock_stream(200, ["event: error", "data: \"internal error\""]))
    res = c.test_account("acc/1")
    assert res["status"] == "error"
    assert res["cleanup_allowed"] is False
    assert len(log_calls) == 4
    assert log_calls[-1]["success"] is False
    assert log_calls[-1]["error_type"] == "WebsiteError"

    # 5. 超时
    monkeypatch.setattr(c._client, "stream", mock_stream(200, [], raise_exc=httpx.TimeoutException("Connection timed out")))
    res = c.test_account("acc/1")
    assert res["status"] == "timeout"
    assert res["cleanup_allowed"] is False
    assert len(log_calls) == 5
    assert log_calls[-1]["success"] is False
    assert log_calls[-1]["error_type"] == "TimeoutException"

    # 6. 非法 SSE (没有 test_complete)
    monkeypatch.setattr(c._client, "stream", mock_stream(200, ["data: hello world"]))
    res = c.test_account("acc/1")
    assert res["status"] == "invalid_sse"
    assert res["cleanup_allowed"] is False
    assert len(log_calls) == 6
    assert log_calls[-1]["success"] is False
    assert log_calls[-1]["error_type"] == "WebsiteError"


def test_cleanup_invalid_preview_and_execute(db_session, monkeypatch):
    """测试预览与执行清理的完整工作流。
    - 包含本地 orphaned 键。
    - 包含远端 SmartUp 备注孤立账号。
    - 包含无 SmartUp 备注的手工账号（不参与候选）。
    - 包含测试通过与失败的不同流程分支。
    """
    w = Website(
        name="W1",
        site_type="sub2api",
        base_url="http://w1",
        enabled=True,
        auth_config_json="{}",
        timeout_seconds=30
    )
    u1 = Upstream(name="U1", base_url="http://u1")
    db_session.add_all([w, u1])
    db_session.commit()
    db_session.refresh(w)
    db_session.refresh(u1)

    # 1. 本地 orphaned 键 k1：对应的远端账号测试将失败 (可清理)
    k1 = UpstreamGeneratedKey(
        upstream_id=u1.id,
        group_id="G1",
        group_name="G1-Name",
        key_name="Key-G1",
        key_value="sk-g1",
        status="orphaned",
        imported_website_id=w.id,
        imported_account_id="acc-k1",
        error="original-error-k1"
    )
    # 2. 本地 orphaned 键 k2：对应的远端账号测试将成功 (不可清理)
    k2 = UpstreamGeneratedKey(
        upstream_id=u1.id,
        group_id="G2",
        group_name="G2-Name",
        key_name="Key-G2",
        key_value="sk-g2",
        status="orphaned",
        imported_website_id=w.id,
        imported_account_id="acc-k2",
        error="original-error-k2"
    )
    # 3. 本地 imported 正常键 k3（不进入候选）
    k3 = UpstreamGeneratedKey(
        upstream_id=u1.id,
        group_id="G3",
        group_name="G3-Name",
        key_name="Key-G3",
        key_value="sk-g3",
        status="imported",
        imported_website_id=w.id,
        imported_account_id="acc-k3"
    )
    db_session.add_all([k1, k2, k3])
    db_session.commit()

    deleted_accounts = []
    test_calls = {}

    class MockClient:
        def __init__(self, **kwargs): pass
        def __enter__(self): return self
        def __exit__(self, *a): pass

        def list_accounts(self):
            # 远端列表：
            # acc-k1 (Condition A, k1)
            # acc-k2 (Condition A, k2)
            # acc-k3 (k3 正常)
            # acc-orphan-remote (Condition B, notes 带备注，本地不存在关联)
            # acc-manual (无备注手工账号，notes 没匹配上，不应进候选)
            return [
                {"id": "acc-k1", "name": "Acc-K1", "notes": f"Imported by SmartUp from upstream key #{k1.id}"},
                {"id": "acc-k2", "name": "Acc-K2", "notes": f"Imported by SmartUp from upstream key #{k2.id}"},
                {"id": "acc-k3", "name": "Acc-K3", "notes": f"Imported by SmartUp from upstream key #{k3.id}"},
                {"id": "acc-orphan-remote", "name": "Orphan-Remote", "notes": "Imported by SmartUp from upstream key #9999"},
                {"id": "acc-manual", "name": "Manual-Acc", "notes": "Manually created account"},
            ]

        def extract_id(self, val):
            return val.get("id") if isinstance(val, dict) else str(val)

        def test_account(self, account_id):
            test_calls[account_id] = test_calls.get(account_id, 0) + 1
            if account_id in ("acc-k1", "acc-orphan-remote"):
                # 测试失败 -> 可删除
                return {"status": "failed", "cleanup_allowed": True, "message": "API check failed"}
            else:
                # 其他测试成功
                return {"status": "success", "cleanup_allowed": False, "message": "API check OK"}

        def delete_account(self, account_id):
            deleted_accounts.append(account_id)
            return {"status": "deleted", "message": f"Account {account_id} has been deleted"}

    monkeypatch.setattr("app.routers.websites._client", lambda row: MockClient())

    # ── 1. 测试预览 ──
    preview_res = preview_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert preview_res.success is True
    # 候选账号应为 3 个：acc-k1, acc-k2, acc-orphan-remote (acc-k3 正常, acc-manual 无备注，均不进入候选)
    items = preview_res.items
    assert len(items) == 3

    item_k1 = next(i for i in items if i.account_id == "acc-k1")
    assert item_k1.cleanup_allowed is True
    assert item_k1.test_status == "failed"

    item_k2 = next(i for i in items if i.account_id == "acc-k2")
    assert item_k2.cleanup_allowed is True
    assert item_k2.test_status == "test_passed"
    assert item_k2.cleanup_action == "delete_remote_and_clear_marker"

    item_orphan = next(i for i in items if i.account_id == "acc-orphan-remote")
    assert item_orphan.cleanup_allowed is True
    assert item_orphan.test_status == "failed"
    assert item_orphan.upstream_key_id == 9999
    assert item_orphan.cleanup_action == "delete_remote_account"

    # 确认没有调用过删除
    assert len(deleted_accounts) == 0

    # ── 2. 测试执行 ──
    execute_res = execute_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert execute_res.success is True
    assert "成功清理 3 个" in execute_res.message

    # 验证被删除的账号（所有候选都按 cleanup_action 执行删除）
    assert len(deleted_accounts) == 3
    assert "acc-k1" in deleted_accounts
    assert "acc-k2" in deleted_accounts
    assert "acc-orphan-remote" in deleted_accounts

    # 验证本地数据库的 DB key 清空逻辑：
    db_session.refresh(k1)
    db_session.refresh(k2)
    db_session.refresh(k3)

    # k1 应该被清空 imported 字段（delete_remote_and_clear_marker），但保留 status='orphaned' 及原始 error
    assert k1.imported_website_id is None
    assert k1.imported_account_id is None
    assert k1.status == "orphaned"
    assert k1.error == "original-error-k1"

    # k2 也是 delete_remote_and_clear_marker，删除成功后清空导入标记
    assert k2.imported_website_id is None
    assert k2.imported_account_id is None
    assert k2.status == "orphaned"


def test_cleanup_list_accounts_failure(db_session, monkeypatch):
    """测试 list_accounts() 失败时，整个清理预览与执行直接阻断报错。"""
    w = Website(
        name="W1",
        site_type="sub2api",
        base_url="http://w1",
        enabled=True,
        auth_config_json="{}",
        timeout_seconds=30
    )
    db_session.add(w)
    db_session.commit()

    class FailClient:
        def __init__(self, **kwargs): pass
        def __enter__(self): return self
        def __exit__(self, *a): pass
        def list_accounts(self):
            return None # 模拟获取失败

    monkeypatch.setattr("app.routers.websites._client", lambda row: FailClient())

    preview_res = preview_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert preview_res.success is False
    assert "拉取远端账号列表失败" in preview_res.message
    assert len(preview_res.items) == 0

    execute_res = execute_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert execute_res.success is False
    assert "拉取远端账号列表失败" in execute_res.message
    assert len(execute_res.items) == 0


def test_cleanup_missing_remote_account_clear_local_marker(db_session, monkeypatch):
    """status=imported 且本地有 imported_account_id，但远端账号列表已不存在该账号
    → 走 clear_local_marker，不调用 delete_account，只清 SmartUp 本地导入标记。
    """
    w = Website(
        name="W-clear",
        site_type="sub2api",
        base_url="http://w-clear",
        enabled=True,
        auth_config_json="{}",
        timeout_seconds=30
    )
    u = Upstream(name="U-clear", base_url="http://u-clear")
    db_session.add_all([w, u])
    db_session.commit()
    db_session.refresh(w)
    db_session.refresh(u)

    # 正常 imported 的 Key，关联到远端账号 acc-gone
    k = UpstreamGeneratedKey(
        upstream_id=u.id,
        group_id="G-clear",
        group_name="G-Clear-Name",
        key_name="Key-Clear",
        key_value="sk-clear",
        status="imported",
        imported_website_id=w.id,
        imported_account_id="acc-gone",
        imported_target_group_id="tg-1",
        imported_target_group_name="Target Group 1",
    )
    db_session.add(k)
    db_session.commit()

    deleted_accounts: list[str] = []

    class MockClient:
        def __init__(self, **kwargs): pass
        def __enter__(self): return self
        def __exit__(self, *a): pass

        def list_accounts(self):
            # 远端列表中不包含 acc-gone
            return [
                {"id": "acc-other", "name": "Other-Acc", "notes": ""},
            ]

        def extract_id(self, val):
            return val.get("id") if isinstance(val, dict) else str(val)

        def delete_account(self, account_id: str):
            deleted_accounts.append(account_id)
            return {"status": "deleted", "message": "deleted"}

        def test_account(self, account_id: str):
            return {"status": "404", "cleanup_allowed": True, "message": "not found"}

    monkeypatch.setattr("app.routers.websites._client", lambda row: MockClient())

    # ── Preview ──
    preview_res = preview_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert preview_res.success is True
    assert len(preview_res.items) == 1

    item = preview_res.items[0]
    assert item.account_id == "acc-gone"
    assert item.cleanup_action == "clear_local_marker"
    assert item.cleanup_allowed is True
    assert item.test_status == "not_needed"
    assert item.reason == "目标账号已被手动删除"

    # ── Execute ──
    execute_res = execute_cleanup_invalid_accounts(wid=w.id, db=db_session)
    assert execute_res.success is True
    assert "清除本地标记 1 个" in execute_res.message

    # 不应调用删除
    assert len(deleted_accounts) == 0

    # 本地导入标记应被清空
    db_session.refresh(k)
    assert k.imported_website_id is None
    assert k.imported_account_id is None
    assert k.imported_at is None
    assert k.imported_target_group_id is None
    assert k.imported_target_group_name is None
    # status 不受影响
    assert k.status == "imported"