"""Central registry for strategy identity and display metadata."""

from __future__ import annotations

from dataclasses import dataclass


MAIN_COMPOSITE_STRATEGY = "main_composite_v1"
BOX_RETEST_1H_STRATEGY = "box_retest_1h_v1"
BOX_RETEST_4H_STRATEGY = "box_retest_4h_v1"


@dataclass(frozen=True)
class StrategyDefinition:
    strategy_code: str
    strategy_name: str
    description: str = ""
    mode: str = "paper_only"
    status: str = "active"


STRATEGY_DEFINITIONS: dict[str, StrategyDefinition] = {
    MAIN_COMPOSITE_STRATEGY: StrategyDefinition(
        strategy_code=MAIN_COMPOSITE_STRATEGY,
        strategy_name="综合确认主链路",
        description="迁移期兼容主链路，承载现有综合筛选与确认逻辑。",
        mode="paper_enabled",
    ),
    BOX_RETEST_1H_STRATEGY: StrategyDefinition(
        strategy_code=BOX_RETEST_1H_STRATEGY,
        strategy_name="1H箱体突破回踩",
        description="小时级底部箱体突破后回踩箱体上沿或EMA承接的早期结构策略。",
        mode="paper_only",
    ),
    BOX_RETEST_4H_STRATEGY: StrategyDefinition(
        strategy_code=BOX_RETEST_4H_STRATEGY,
        strategy_name="4H箱体突破回踩",
        description="底部箱体突破后回踩箱体上沿或EMA承接的结构策略。",
        mode="paper_only",
    ),
}


def normalize_strategy_code(strategy_code: str | None) -> str:
    code = str(strategy_code or "").strip()
    return code or MAIN_COMPOSITE_STRATEGY


def strategy_definition(strategy_code: str | None) -> StrategyDefinition:
    code = normalize_strategy_code(strategy_code)
    return STRATEGY_DEFINITIONS.get(
        code,
        StrategyDefinition(
            strategy_code=code,
            strategy_name=code,
            description="未注册策略，请补充 strategy_registry。",
            status="unknown",
        ),
    )


def strategy_label(strategy_code: str | None) -> str:
    return strategy_definition(strategy_code).strategy_name


def registered_strategy_codes() -> list[str]:
    return list(STRATEGY_DEFINITIONS.keys())


def strategy_catalog_seed_rows(strategy_version: str = "") -> list[dict]:
    rows = []
    for item in STRATEGY_DEFINITIONS.values():
        rows.append(
            {
                "strategy_code": item.strategy_code,
                "strategy_name": item.strategy_name,
                "strategy_version": strategy_version or "",
                "status": item.status,
                "mode": item.mode,
                "description": item.description,
            }
        )
    return rows