from __future__ import annotations

from flask_login import UserMixin
from werkzeug.security import check_password_hash, generate_password_hash

from .extensions import db
from .time_utils import utc_now


class User(UserMixin, db.Model):
    __tablename__ = "users"

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), unique=True, nullable=False)
    password_hash = db.Column(db.String(255), nullable=False)
    is_admin = db.Column(db.Boolean, default=False, nullable=False)
    created_at = db.Column(db.DateTime, default=utc_now)

    def set_password(self, password: str) -> None:
        self.password_hash = generate_password_hash(password)

    def check_password(self, password: str) -> bool:
        return check_password_hash(self.password_hash, password)


class PasswordResetToken(db.Model):
    __tablename__ = "password_reset_tokens"

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False, index=True)
    created_by_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
    token_hash = db.Column(db.String(64), unique=True, nullable=False, index=True)
    expires_at = db.Column(db.DateTime, nullable=False)
    used_at = db.Column(db.DateTime)
    created_at = db.Column(db.DateTime, default=utc_now, nullable=False)

    user = db.relationship(
        "User",
        foreign_keys=[user_id],
        backref=db.backref("password_reset_tokens", lazy=True),
    )
    created_by = db.relationship("User", foreign_keys=[created_by_id])


class UploadRecord(db.Model):
    __tablename__ = "upload_records"

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
    original_filename = db.Column(db.String(255), nullable=False)
    saved_path = db.Column(db.String(500), nullable=False)
    prediction_path = db.Column(db.String(500), nullable=False)
    image_path = db.Column(db.String(500), nullable=False)
    upload_time = db.Column(db.DateTime, default=utc_now)

    user = db.relationship("User", backref=db.backref("uploads", lazy=True))


class AppSetting(db.Model):
    __tablename__ = "app_settings"

    id = db.Column(db.Integer, primary_key=True)
    key = db.Column(db.String(100), unique=True, nullable=False)
    value = db.Column(db.String(255), nullable=False)
    updated_at = db.Column(db.DateTime, default=utc_now, onupdate=utc_now)

    @classmethod
    def get_bool(cls, key: str, default: bool = False) -> bool:
        setting = cls.query.filter_by(key=key).first()
        if setting is None:
            return default
        return setting.value.strip().lower() in {"1", "true", "yes", "on"}

    @classmethod
    def set_bool(cls, key: str, value: bool) -> "AppSetting":
        setting = cls.query.filter_by(key=key).first()
        if setting is None:
            setting = cls(key=key, value="")
            db.session.add(setting)
        setting.value = "true" if value else "false"
        return setting