from __future__ import annotations

import os

from sqlalchemy import inspect, text

from ..extensions import db
from ..models import QuickWin, User
from .app_settings import ensure_app_settings


def ensure_schema_and_admins() -> None:
    inspector = inspect(db.engine)
    column_names = {column["name"] for column in inspector.get_columns("user")}

    if "is_admin" not in column_names:
        db.session.execute(text("ALTER TABLE user ADD COLUMN is_admin BOOLEAN NOT NULL DEFAULT 0"))
        db.session.commit()

    if "calendar_feed_token" not in column_names:
        db.session.execute(text("ALTER TABLE user ADD COLUMN calendar_feed_token VARCHAR(255)"))
        db.session.commit()

    ensure_app_settings()

    users_without_feed = User.query.filter(User.calendar_feed_token.is_(None)).all()
    if users_without_feed:
        for user in users_without_feed:
            user.ensure_calendar_feed_token()
        db.session.commit()

    admin_exists = User.query.filter_by(is_admin=True).first()
    default_quick_win_user = admin_exists

    preferred_admin_email = os.getenv("DEFAULT_ADMIN_EMAIL", "mail@hnz.io").lower().strip()
    preferred_user = User.query.filter(User.email.ilike(preferred_admin_email)).first()
    if preferred_user and not admin_exists:
        preferred_user.is_admin = True
        db.session.commit()
        default_quick_win_user = preferred_user

    first_user = User.query.order_by(User.id.asc()).first()
    if first_user and not User.query.filter_by(is_admin=True).first():
        first_user.is_admin = True
        db.session.commit()
        default_quick_win_user = first_user

    _ensure_default_quick_wins(default_quick_win_user or User.query.order_by(User.id.asc()).first())


def _ensure_default_quick_wins(default_user: User | None) -> None:
    if not default_user:
        return

    defaults = [
        ("Schnell Aufräumen", "fast"),
        ("Spülmaschine ausräumen", "normal"),
        ("Bett machen", "normal"),
        ("Lüften", "fast"),
        ("Wäsche zusammenlegen", "medium"),
    ]

    existing_titles = {quick_win.title for quick_win in QuickWin.query.all()}
    created = False
    for title, effort in defaults:
        if title not in existing_titles:
            db.session.add(
                QuickWin(
                    title=title,
                    effort=effort,
                    active=True,
                    created_by_user_id=default_user.id,
                )
            )
            created = True
    if created:
        db.session.commit()