import click
from flask.cli import with_appcontext

from .extensions import db
from .models import User


@click.command("init-db")
@click.option(
    "--admin-email",
    required=False,
    default="beheer@alphen.cloud",
    show_default=True,
    help="Email for the initial admin",
)
@click.option("--admin-pass", required=True, help="Password for the initial admin")
@with_appcontext
def init_db_command(admin_email: str, admin_pass: str):
    """Create tables and ensure an admin account exists."""
    db.create_all()

    # Lightweight migration for older SQLite DBs: ensure User.email column exists.
    # This avoids requiring Alembic for this small project.
    try:
        cols = [r[1] for r in db.session.execute(db.text("PRAGMA table_info(user)")).fetchall()]
        if "email" not in cols:
            db.session.execute(db.text("ALTER TABLE user ADD COLUMN email VARCHAR(255)"))
            db.session.commit()

        display_cols = [r[1] for r in db.session.execute(db.text("PRAGMA table_info(display)")).fetchall()]
        if "description" not in display_cols:
            db.session.execute(db.text("ALTER TABLE display ADD COLUMN description VARCHAR(200)"))
            db.session.commit()

        company_cols = [r[1] for r in db.session.execute(db.text("PRAGMA table_info(company)")).fetchall()]
        if "storage_max_bytes" not in company_cols:
            db.session.execute(db.text("ALTER TABLE company ADD COLUMN storage_max_bytes BIGINT"))
            db.session.commit()
    except Exception:
        # Best-effort; if it fails we continue so fresh DBs still work.
        db.session.rollback()

    admin_email = (admin_email or "").strip().lower()
    if not admin_email:
        raise click.UsageError("--admin-email is required")

    existing = User.query.filter_by(email=admin_email).first()
    if existing:
        if not existing.is_admin:
            existing.is_admin = True
        existing.email = admin_email
        existing.username = admin_email
        existing.set_password(admin_pass)
        db.session.commit()
        click.echo(f"Updated admin user '{admin_email}'.")
        return

    u = User(is_admin=True)
    u.email = admin_email
    u.username = admin_email
    u.set_password(admin_pass)
    db.session.add(u)
    db.session.commit()
    click.echo(f"Created admin user '{admin_email}'.")