#!/usr/bin/env python3
"""parse_outline 窗覆盖门离线测试：不连库、不调用模型。"""

import pathlib
import sys

import click

HERE = pathlib.Path(__file__).resolve().parent
sys.path.insert(0, str(HERE))

import parse_outline as po  # noqa: E402


def check(name, condition):
    """输出与现有 parse-book 离线测试一致的简明结果。"""

    if not condition:
        raise AssertionError(name)
    print(f"[PASS] {name}")


def main():
    """覆盖连续、头部、中部、尾部和重叠窗五类边界。"""

    check("coverage-连续无缝", po.outline_coverage_gaps([(1, 33), (34, 65)], 1, 65) == [])
    check(
        "coverage-识别中部缝隙",
        po.outline_coverage_gaps([(951, 983), (1015, 1047)], 951, 1047) == [(984, 1014)],
    )
    check("coverage-识别头部缝隙", po.outline_coverage_gaps([(5, 10)], 1, 10) == [(1, 4)])
    check("coverage-识别尾部缝隙", po.outline_coverage_gaps([(1, 6)], 1, 10) == [(7, 10)])
    check(
        "coverage-重叠窗不误报",
        po.outline_coverage_gaps([(1, 20), (15, 30), (31, 40)], 1, 40) == [],
    )
    po.ensure_outline_coverage([(1, 33), (34, 65)], 1, 65, context="测试")
    try:
        po.ensure_outline_coverage([(951, 983), (1015, 1047)], 951, 1047, context="测试")
    except click.ClickException as error:
        check("coverage-缝隙非零阻断", "984-1014" in str(error))
    else:
        raise AssertionError("coverage-缝隙非零阻断")

    original_chat = po.chat_degrade
    try:
        replies = iter([("{}", {}), ('{"outline":"补窗成功"}', {"prompt_tokens": 1})])
        po.chat_degrade = lambda _prompt, _model: next(replies)
        outline, usage = po.generate_outline_with_retry("prompt", "model", max_attempts=2)
        check("window-空输出重试后成功", outline == "补窗成功" and usage["prompt_tokens"] == 1)

        calls = {"count": 0}

        def always_empty(_prompt, _model):
            calls["count"] += 1
            return "{}", {}

        po.chat_degrade = always_empty
        try:
            po.generate_outline_with_retry("prompt", "model", max_attempts=2)
        except click.ClickException as error:
            check("window-有限重试耗尽即停书", calls["count"] == 2 and "2 次" in str(error))
        else:
            raise AssertionError("window-有限重试耗尽即停书")

        def exhausted_chain(_prompt, _model):
            raise po.SensitiveError("全部渠道耗尽")

        po.chat_degrade = exhausted_chain
        try:
            po.generate_outline_with_retry("prompt", "model", max_attempts=2)
        except click.ClickException as error:
            check("window-治理链耗尽即停书", "治理链耗尽" in str(error))
        else:
            raise AssertionError("window-治理链耗尽即停书")
    finally:
        po.chat_degrade = original_chat

    unordered = [
        (3, 83, 984, 1014, "补插窗", "revised"),
        (1, 1, 1, 33, "首窗", "revised"),
        (4, 52, 1015, 1047, "后续窗", "revised"),
        (2, 82, 2226, 2250, "末窗", "revised"),
    ]
    ordered = po.order_outline_windows(unordered)
    check("check-按绝对章号排序", [row[2] for row in ordered] == [1, 984, 1015, 2226])
    rendered = po.render_coherence_outlines(ordered)
    check(
        "check-阶段号按章序重编",
        "【阶段2（第984–1014章）】" in rendered and "【阶段83" not in rendered,
    )

    print("\n全部离线自测通过：11 项（未连库、未发任何 LLM 调用）")


if __name__ == "__main__":
    main()