from sqlalchemy import Column, String, Float, DateTime, ForeignKey, JSON
from sqlalchemy.dialects.postgresql import UUID
from uuid import uuid4
from datetime import datetime
from ..database import Base


class ChatMessage(Base):
    __tablename__ = "chat_messages"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid4)
    stream_id = Column(UUID(as_uuid=True), ForeignKey("streams.id", ondelete="CASCADE"), nullable=False, index=True)
    username = Column(String(100), nullable=False)
    message = Column(String(500), nullable=False)
    timestamp_offset = Column(Float, nullable=False)  # Offset in seconds from video start
    color = Column(String(7), nullable=True)  # Hex color code for username
    badges = Column(JSON, nullable=True)  # JSON array of badges
    emotes = Column(JSON, nullable=True)  # JSON array of emotes
    user_avatar_url = Column(String(500), nullable=True)  # URL to user avatar image
    created_at = Column(DateTime(timezone=True), default=datetime.utcnow)