from database import db
from datetime import datetime, timezone


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

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=True)
    password_hash = db.Column(db.String(256), nullable=False)
    avatar_url = db.Column(db.String(500), default="")
    theme = db.Column(db.String(20), default="dark")
    created_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc))

    chats = db.relationship("Chat", backref="user", lazy="dynamic", cascade="all, delete-orphan")
    memories = db.relationship("Memory", backref="user", lazy="dynamic", cascade="all, delete-orphan")
    settings = db.relationship("Setting", backref="user", lazy="dynamic", cascade="all, delete-orphan")
    images = db.relationship("GeneratedImage", backref="user", lazy="dynamic", cascade="all, delete-orphan")
    uploads = db.relationship("Upload", backref="user", lazy="dynamic", cascade="all, delete-orphan")
    scenes = db.relationship("Scene", backref="user", lazy="dynamic", cascade="all, delete-orphan")

    def to_dict(self):
        return {
            "id": self.id,
            "username": self.username,
            "email": self.email,
            "avatar_url": self.avatar_url,
            "theme": self.theme,
            "created_at": self.created_at.isoformat() if self.created_at else None
        }