# models.py — UploadLog table (SQLModel) and related models

from sqlmodel import SQLModel, Field, Session, select
from typing import Optional
from datetime import datetime
from database import engine

class User(SQLModel, table=True):
    token_created: datetime = Field(default_factory=datetime.utcnow)
    email: str = Field(primary_key=True)
    username: str = Field(unique=True, index=True)
    token: str
    confirmed: bool = False
    ip: str = Field(default="")


class UserQuota(SQLModel, table=True):
    uid: str = Field(primary_key=True)
    storage_bytes: int = 0


class UploadLog(SQLModel, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)
    uid: str
    ip: str
    filename: Optional[str]  # Original filename
    processed_filename: Optional[str]  # Processed filename (UUID.opus)
    size_bytes: int
    created_at: datetime = Field(default_factory=datetime.utcnow)


class DBSession(SQLModel, table=True):
    token: str = Field(primary_key=True)
    uid: str = Field(foreign_key="user.email")  # This references User.email (primary key)
    ip_address: str
    user_agent: str
    created_at: datetime = Field(default_factory=datetime.utcnow)
    expires_at: datetime
    is_active: bool = True
    last_activity: datetime = Field(default_factory=datetime.utcnow)


class PublicStream(SQLModel, table=True):
    """Stores public stream metadata for all users"""
    uid: str = Field(primary_key=True)
    username: Optional[str] = Field(default=None, index=True)
    storage_bytes: int = 0
    mtime: int = Field(default_factory=lambda: int(datetime.utcnow().timestamp()))
    last_updated: Optional[datetime] = Field(default_factory=datetime.utcnow)
    created_at: datetime = Field(default_factory=datetime.utcnow)
    updated_at: datetime = Field(default_factory=datetime.utcnow)


def get_user_by_uid(uid: str) -> Optional[User]:
    """
    Retrieve a user by their UID (email).
    
    Note: In this application, UIDs are consistently email-based.
    The User model uses email as primary key, and all user references
    throughout the system use email format.
    
    Args:
        uid: The email to look up
        
    Returns:
        User object if found, None otherwise
    """
    with Session(engine) as session:
        # Primary lookup by email (which is what we're using as UID)
        statement = select(User).where(User.email == uid)
        user = session.exec(statement).first()
        
        # Fallback: try by username for legacy compatibility
        if not user and '@' not in uid:
            statement = select(User).where(User.username == uid)
            user = session.exec(statement).first()
            
        return user


def verify_session(db: Session, token: str) -> DBSession:
    """Verify a session token and return the session if valid"""
    from datetime import datetime
    
    # Find the session
    session = db.query(DBSession).filter(
        DBSession.token == token,
        DBSession.is_active == True,  # noqa: E712
        DBSession.expires_at > datetime.utcnow()
    ).first()
    
    if not session:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail="Invalid or expired session",
            headers={"WWW-Authenticate": "Bearer"},
        )
    
    # Update last activity
    session.last_activity = datetime.utcnow()
    db.add(session)
    db.commit()
    db.refresh(session)
    
    return session