from __future__ import annotations

from datetime import datetime

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

from .extensions import db


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=datetime.utcnow)

    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 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=datetime.utcnow)

    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=datetime.utcnow, onupdate=datetime.utcnow)

    @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