"""Chinese appendix facts for issue #91 section A.

Dosha, friendship, and yoga cells come from structured fields. The Chinese
appendix is not an English render with the short-word sanitizer applied.
"""

from __future__ import annotations

from scripts.pl9_full_data_export import (
    _render_pl9_ai_density_raw_data_markdown_zh,
    render_full_data_markdown,
)
from scripts.report_render.appendix_language import (
    chinese_prose_gates,
    dosha_row,
    friendship_table,
)
from tests.test_report_full_data_fix import _packet


def test_dosha_rows_use_structured_fields() -> None:
    missing = dosha_row("mangal_dosha", {}, "zh")
    assert missing is not None
    assert missing[1] == "未判定"
    assert dosha_row("mangal_dosha", {}, "en")[1] == "Not determined"

    clear = dosha_row("mangal_dosha", {"has_dosha": False}, "zh")
    assert clear is not None and clear[1] == "无"
    assert "火星不在煞宫" in clear[2]
    english = dosha_row("mangal_dosha", {"has_dosha": False}, "en")
    assert english is not None and english[1] == "No"
    assert "not in a dosha house" in english[2]

    partial = dosha_row("kaal_sarp_dosha", {
        "has_dosha": True,
        "type": "partial",
        "planets_in_axis": ["Sun", "Moon", "Mars", "Mercury", "Jupiter"],
        "planets_outside": ["Venus", "Saturn"],
        "rahu_house": 1,
        "ketu_house": 7,
    }, "zh")
    assert partial is not None
    assert partial[1] == "部分"
    assert "7 颗中 5 颗" in partial[2]
    assert "北交点/罗睺在第 1 宫" in partial[2]
    partial_en = dosha_row("kaal_sarp_dosha", {
        "has_dosha": True,
        "type": "partial",
        "planets_in_axis": ["Sun", "Moon", "Mars", "Mercury", "Jupiter"],
        "planets_outside": ["Venus", "Saturn"],
        "rahu_house": 1,
        "ketu_house": 7,
    }, "en")
    assert partial_en is not None and "5 of 7" in partial_en[2]


def test_friendship_reads_rows_not_snapshot_keys() -> None:
    snapshot = {
        "status": "parameter_sensitive",
        "rows": [
            {
                "planet": "Sun",
                "natural": {"friends": ["Moon", "Mars"], "enemies": ["Venus"], "neutral": ["Mercury"]},
                "temporary": {"friends": ["Jupiter"], "enemies": ["Saturn"]},
                "compound": {"great_friends": ["Moon"], "friends": [], "neutral": [], "enemies": [], "great_enemies": ["Saturn"]},
            }
        ],
    }
    headers, rows = friendship_table(snapshot, "zh")
    assert headers[0] == "行星"
    assert headers[-1] == "复合关系"
    assert rows[0][0] == "太阳"
    assert "月亮" in rows[0][1]
    assert "极友：月亮" in rows[0][-1]
    assert "GREAT_FRIEND" not in rows[0][-1]
    _, english_rows = friendship_table(snapshot, "en")
    assert english_rows[0][0] == "Sun"
    assert "Great friend: Moon" in english_rows[0][-1]


def test_chinese_appendix_is_not_back_translated() -> None:
    packet = _packet("zh")
    packet["worksheets"]["strengths_and_scores"]["planetary_friendship"] = {
        "rows": [{
            "planet": "Sun",
            "natural": {"friends": ["Moon"], "enemies": [], "neutral": ["Mars"]},
            "temporary": {"friends": ["Jupiter"], "enemies": ["Venus"]},
            "compound": {"great_friends": ["Moon"], "friends": [], "neutral": [], "enemies": [], "great_enemies": []},
        }],
    }
    appendix = _render_pl9_ai_density_raw_data_markdown_zh(packet)
    assert "## 行星友敌关系矩阵" in appendix
    assert "## Planetary Friendship Matrix" not in appendix
    assert "Natural friends" not in appendix
    assert "极友：月亮" in appendix
    assert "Aries" not in appendix
    assert "白羊座" in appendix
    assert "第 1 宫宫主" in appendix
    assert "house " not in appendix
    gates = chinese_prose_gates(appendix)
    assert gates == {"english_signs": 0, "han_function_glue": 0}

    rendered = render_full_data_markdown(packet)
    rendered_gates = chinese_prose_gates(rendered)
    assert rendered_gates["english_signs"] == 0
    assert rendered_gates["han_function_glue"] == 0
    assert "Raja Yoga" not in rendered
    assert "blocked" not in rendered.lower()


def test_chinese_prose_gates_catch_sign_names_and_glue() -> None:
    sample = "```jyotish-chart\nAries\n```\n正文 Aries。火星 in 12，in )，, :"
    gates = chinese_prose_gates(sample)
    assert gates["english_signs"] == 1
    assert gates["han_function_glue"] == 3