import uuid
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy import (
    Column,
    Integer,
    String,
    BigInteger,
    TIMESTAMP,
    ForeignKey,
    Boolean,
)
from sqlalchemy.orm import relationship, Mapped, mapped_column
from app.db.database import Base
from datetime import datetime, timezone


class UserFile(Base):
    __tablename__ = "user_files"
    id = Column(Integer, primary_key=True, index=True)
    tenant_id = Column(
        UUID(as_uuid=True), ForeignKey("tenants.id"), nullable=False, index=True
    )
    user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
    filename = Column(String, nullable=False)
    storage_provider = Column(String(50), nullable=False, default="b2")
    s3_key = Column(String, nullable=False)
    size = Column(BigInteger, default=0)
    content_type = Column(String, nullable=True)
    status = Column(String, default="uploaded")
    is_deleted = Column(Boolean, default=False)
    deleted_at = Column(TIMESTAMP, nullable=True)
    created_at = Column(TIMESTAMP, default=lambda: datetime.now(timezone.utc))

    user = relationship("User", back_populates="user_files")


class UserStorageUsage(Base):
    __tablename__ = "user_storage_usage"
    user_id = Column(Integer, ForeignKey("users.id"), primary_key=True, nullable=False)
    total_bytes_used = Column(BigInteger, default=0)
    total_files_count = Column(Integer, default=0)
    max_bytes_quota = Column(BigInteger, default=1073741824)
    last_updated = Column(
        TIMESTAMP,
        default=lambda: datetime.now(timezone.utc),
        onupdate=lambda: datetime.now(timezone.utc),
    )

    user = relationship("User", back_populates="storage_usage")