"""从 `alembic/baseline_generated.sql` 生成基线初始迁移（方案 A）。

背景：39 张基线表此前没有任何 Alembic 迁移，空库无法重建。本脚本把基线 DDL 转成一条
初始迁移（`down_revision=None`），使 `alembic upgrade head` 能在空库上建出全部业务表。

设计取舍：迁移体内使用 `op.execute` 原样执行基线 SQL，而不是手工翻译成 `op.create_table`。
39 张表包含生成列、字符集与排序规则、联合唯一键等方言细节，逐表翻译容易引入新的结构漂移；
原样执行可保证"迁移产生的结构与 baseline_generated.sql 一致"，而该文件由
`tools/generate_baseline_sql.py` 从基线文档生成，并由 `tools/audit_constraints.py` 审计。

默认只打印摘要，加 `--write` 才落盘，避免在 Alembic 链上意外产生双 head。

用法：
    python tools/generate_baseline_migration.py            # 预览
    python tools/generate_baseline_migration.py --write    # 写入迁移文件
"""

from __future__ import annotations

import re
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
SQL_FILE = ROOT / "alembic" / "baseline_generated.sql"
TARGET = ROOT / "alembic" / "versions" / "20260909_baseline_schema.py"
REVISION = "20260909_baseline_schema"

TABLE_RE = re.compile(r"CREATE TABLE\s+`?([A-Za-z0-9_]+)`?")

HEADER = '''"""基线结构初始迁移（方案 A）：使空库可重建全部基线表。

本文件由 `tools/generate_baseline_migration.py` 从 `alembic/baseline_generated.sql` 生成，
请勿手工编辑；结构变更请改 `docs/00-新数据库基线设计.md` 后重新生成基线 SQL 与迁移。

设计说明：基线表此前不在 Alembic 链上，导致空库无法重建（迁移链只创建平台增量表）。
本迁移作为链首（`down_revision=None`）：既有库的 `alembic_version` 已位于本迁移的后代，
Alembic 不会重复执行；空库则从本迁移开始依次建表。

约束：只创建基线表，不修改任何既有表名或既有字段定义。
"""
from alembic import op

revision = "{revision}"
down_revision = None
branch_labels = None
depends_on = None

STATEMENTS: list[str] = [
'''

TAIL = '''

def upgrade() -> None:
    for statement in STATEMENTS:
        op.execute(statement)


def downgrade() -> None:
    # 仅用于整体回退到空库；真实环境回退请使用前向兼容迁移。
    for table in reversed(TABLES):
        op.execute(f"DROP TABLE IF EXISTS `{table}`")
'''


def split_statements(sql_text: str) -> list[str]:
    statements: list[str] = []
    current: list[str] = []
    for line in sql_text.splitlines():
        stripped = line.strip()
        if not stripped or stripped.startswith("--"):
            continue
        if stripped.startswith("SET FOREIGN_KEY_CHECKS"):
            continue
        current.append(line)
        if stripped.endswith(";"):
            statements.append("\n".join(current).rstrip().rstrip(";"))
            current = []
    if current:
        statements.append("\n".join(current).rstrip())
    return statements


def table_names(statements: list[str]) -> list[str]:
    names: list[str] = []
    for statement in statements:
        match = TABLE_RE.match(statement.strip())
        if match:
            names.append(match.group(1))
    return names


def render(statements: list[str], tables: list[str]) -> str:
    body = "".join(
        '    """' + statement.replace('"""', '\\"\\"\\"') + '""",\n' for statement in statements
    )
    tables_block = "TABLES: list[str] = [\n" + "".join(
        f'    "{name}",\n' for name in tables
    ) + "]\n"
    return HEADER.format(revision=REVISION) + body + "]\n\n" + tables_block + TAIL


def main() -> int:
    if not SQL_FILE.exists():
        print(f"baseline sql missing: {SQL_FILE}")
        return 1
    statements = split_statements(SQL_FILE.read_text(encoding="utf-8"))
    tables = table_names(statements)
    if len(statements) != len(tables):
        print(f"unexpected: statements={len(statements)} tables={len(tables)}")
        return 1
    if "--write" not in sys.argv:
        print(f"preview: statements={len(statements)} tables={len(tables)}")
        print(f"target : {TARGET}")
        print("run with --write to create the migration file")
        return 0
    TARGET.write_text(render(statements, tables), encoding="utf-8")
    print(f"written: {TARGET} (statements={len(statements)}, tables={len(tables)})")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())