"""学习账号初始化与清理（用户掌握度按全局 Knowledge 节点记录）。"""
from __future__ import annotations

from sqlalchemy.orm import Session
from sqlalchemy import text

from models import (
    AttemptItem,
    AttemptSession,
    DailyCompletion,
    ErrorEntry,
    Role,
    User,
    UserKnowledge,
    UserRole,
)
from seed.rbac import DEFAULT_USERS


def user_has_admin_role(db: Session, user_id: int) -> bool:
    return (
        db.query(Role.id)
        .join(UserRole, UserRole.role_id == Role.id)
        .filter(Role.code == "admin", UserRole.user_id == user_id)
        .first()
        is not None
    )


def ensure_profile(db: Session, user_id: int) -> None:
    """新学习账号无需预置掌握度：从第一次章节测试起由答题结果生成。"""
    db.commit()


def ensure_default_profiles(db: Session) -> None:
    """默认学生账号同样从空学习记录开始，管理员不建学习档案。"""
    for username, _, _, _ in DEFAULT_USERS:
        user = db.query(User).filter(User.username == username).first()
        if user is None or user_has_admin_role(db, user.id):
            continue
        ensure_profile(db, user.id)


def remove_unused_admin_profile(db: Session, user_id: int) -> None:
    """管理员不保留学习数据。"""
    db.query(UserKnowledge).filter(UserKnowledge.user_id == user_id).delete()
    db.query(ErrorEntry).filter(ErrorEntry.user_id == user_id).delete()
    db.query(AttemptItem).filter(AttemptItem.user_id == user_id).delete()
    db.query(AttemptSession).filter(AttemptSession.user_id == user_id).delete()
    db.query(DailyCompletion).filter(DailyCompletion.user_id == user_id).delete()
    db.commit()


def reset_user_learning(db: Session, user_id: int) -> None:
    remove_unused_admin_profile(db, user_id)


def reset_all_learning(db: Session) -> None:
    db.query(UserKnowledge).delete()
    db.query(ErrorEntry).delete()
    db.query(AttemptItem).delete()
    db.query(AttemptSession).delete()
    db.query(DailyCompletion).delete()
    db.execute(text("DROP TABLE IF EXISTS knowledge_points"))
    for table_name, column in (
        ("chapters", "knowledge_names"),
        ("questions", "knowledge_names"),
    ):
        try:
            db.execute(
                text(f"ALTER TABLE {table_name} DROP COLUMN {column}")
            )
        except Exception:
            # 新库本来就没有这些旧文本列，忽略
            pass
    db.commit()