"""初始化数据库：建库 + 建表 + 建初始管理员。

用法：
    python -m app.scripts.init_db            # 建表（不会删数据）
    python -m app.scripts.init_db --drop     # 先删表再建（危险，会清空数据）
"""

from __future__ import annotations

import argparse
import sys

from app.core.config import settings
from app.core.database import SessionLocal, engine, ensure_database_exists
from app.core.security import hash_password
from app.dao.account_dao import AccountDao
from app.model import Account, Base, Role


def init(drop: bool = False) -> None:
    ensure_database_exists()
    if drop:
        confirm = input("这会删除 wolin 库里的全部表和数据，输入 yes 继续：")
        if confirm.strip().lower() != "yes":
            print("已取消")
            return
        Base.metadata.drop_all(bind=engine)
        print("已删除所有表")

    Base.metadata.create_all(bind=engine)
    tables = ", ".join(sorted(Base.metadata.tables.keys()))
    print(f"建表完成：{tables}")

    with SessionLocal() as db:
        if AccountDao.get_by_username(db, settings.ADMIN_USERNAME, with_deleted=True) is None:
            db.add(
                Account(
                    username=settings.ADMIN_USERNAME,
                    hashed_password=hash_password(settings.ADMIN_PASSWORD),
                    real_name="系统管理员",
                    role=Role.ADMIN.value,
                )
            )
            db.commit()
            print(f"初始管理员：{settings.ADMIN_USERNAME} / {settings.ADMIN_PASSWORD}")
        else:
            print(f"管理员 {settings.ADMIN_USERNAME} 已存在，跳过")

        if AccountDao.get_by_username(db, "viewer", with_deleted=True) is None:
            db.add(
                Account(
                    username="viewer",
                    hashed_password=hash_password("viewer123"),
                    real_name="只读访客",
                    role=Role.VIEWER.value,
                )
            )
            db.commit()
            print("只读账号：viewer / viewer123")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="初始化沃林学生管理系统数据库")
    parser.add_argument("--drop", action="store_true", help="先删除所有表再重建（会丢数据）")
    args = parser.parse_args()
    try:
        init(drop=args.drop)
    except Exception as exc:  # noqa: BLE001
        print(f"初始化失败：{exc}", file=sys.stderr)
        raise