import uuid
from datetime import datetime
from typing import Optional, TYPE_CHECKING
from sqlalchemy import ForeignKey, String, Integer, DateTime, func, Boolean, Text, text
from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy.dialects.postgresql import UUID as PG_UUID
from app.db.base_class import Base 


if TYPE_CHECKING:
    from app.models.identity import User

SCHEMA_ARGS = {"schema": "data"}

class PointRule(Base):
    __tablename__ = "point_rules"
    __table_args__ = SCHEMA_ARGS
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    action_key: Mapped[str] = mapped_column(String, unique=True, index=True)
    points: Mapped[int] = mapped_column(Integer, default=0)
    description: Mapped[Optional[str]] = mapped_column(String)
    is_active: Mapped[bool] = mapped_column(Boolean, default=True)

class LevelConfig(Base):
    __tablename__ = "level_configs"
    __table_args__ = SCHEMA_ARGS
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    level_number: Mapped[int] = mapped_column(Integer, unique=True)
    min_points: Mapped[int] = mapped_column(Integer)
    rank_name: Mapped[str] = mapped_column(String)

class PointsLedger(Base):
    __tablename__ = "points_ledger"
    __table_args__ = SCHEMA_ARGS
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    user_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.users.id"))
    points: Mapped[int] = mapped_column(Integer, default=0)
    # JAVÍTÁS: Itt is server_default-ot használunk
    penalty_change: Mapped[int] = mapped_column(Integer, server_default=text("0"), default=0)
    reason: Mapped[str] = mapped_column(String)
    created_at: Mapped[datetime] = mapped_column(DateTime, default=func.now())
    
    user: Mapped["User"] = relationship("User")

class UserStats(Base):
    __tablename__ = "user_stats"
    __table_args__ = {"schema": "data", "extend_existing": True} # Biztosítjuk a sémát

    # A ForeignKey-nek látnia kell a data sémát!
    user_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.users.id"), primary_key=True)
    
    total_xp: Mapped[int] = mapped_column(Integer, default=0)
    social_points: Mapped[int] = mapped_column(Integer, default=0)
    current_level: Mapped[int] = mapped_column(Integer, default=1)
    
    # --- BÜNTETŐ RENDSZER ---
    penalty_points: Mapped[int] = mapped_column(Integer, server_default=text("0"), default=0)
    restriction_level: Mapped[int] = mapped_column(Integer, server_default=text("0"), default=0)
    
    updated_at: Mapped[datetime] = mapped_column(DateTime, default=func.now(), onupdate=func.now())

    # VISSZAMUTATÁS A USER-RE: a back_populates értéke meg kell egyezzen a User osztály 'stats' mezőjével!
    user: Mapped["User"] = relationship("User", back_populates="stats")


class Badge(Base):
    __tablename__ = "badges"
    __table_args__ = SCHEMA_ARGS
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    name: Mapped[str] = mapped_column(String, unique=True)
    description: Mapped[str] = mapped_column(String)
    icon_url: Mapped[Optional[str]] = mapped_column(String)

class UserBadge(Base):
    __tablename__ = "user_badges"
    __table_args__ = SCHEMA_ARGS
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    user_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.users.id"))
    badge_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.badges.id"))
    earned_at: Mapped[datetime] = mapped_column(DateTime, default=func.now())
    
    user: Mapped["User"] = relationship("User")