"""
Demo Session Models
Tracks ephemeral demo sessions for prospect users
"""

from sqlalchemy import Column, String, Boolean, DateTime, Integer, Enum as SQLEnum
from sqlalchemy.dialects.postgresql import UUID, JSONB
from datetime import datetime, timezone
import uuid
import enum

from shared.database.base import Base


class DemoSessionStatus(enum.Enum):
    """Demo session status"""
    ACTIVE = "active"
    EXPIRED = "expired"
    DESTROYED = "destroyed"


class DemoSession(Base):
    """Demo Session tracking model"""
    __tablename__ = "demo_sessions"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    session_id = Column(String(100), unique=True, nullable=False, index=True)

    # Session ownership
    user_id = Column(UUID(as_uuid=True), nullable=True)
    ip_address = Column(String(45), nullable=True)
    user_agent = Column(String(500), nullable=True)

    # Demo tenant linking
    base_demo_tenant_id = Column(UUID(as_uuid=True), nullable=False, index=True)
    virtual_tenant_id = Column(UUID(as_uuid=True), nullable=False, index=True)
    demo_account_type = Column(String(50), nullable=False)  # 'individual_bakery', 'central_baker'

    # Session lifecycle
    status = Column(SQLEnum(DemoSessionStatus, values_callable=lambda obj: [e.value for e in obj]), default=DemoSessionStatus.ACTIVE, index=True)
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), index=True)
    expires_at = Column(DateTime(timezone=True), nullable=False, index=True)
    last_activity_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
    destroyed_at = Column(DateTime(timezone=True), nullable=True)

    # Session metrics
    request_count = Column(Integer, default=0)
    data_cloned = Column(Boolean, default=False)
    redis_populated = Column(Boolean, default=False)

    # Session metadata
    session_metadata = Column(JSONB, default=dict)

    def __repr__(self):
        return f"<DemoSession(session_id={self.session_id}, status={self.status.value})>"

    def to_dict(self):
        """Convert to dictionary"""
        return {
            "id": str(self.id),
            "session_id": self.session_id,
            "user_id": str(self.user_id) if self.user_id else None,
            "virtual_tenant_id": str(self.virtual_tenant_id),
            "base_demo_tenant_id": str(self.base_demo_tenant_id),
            "demo_account_type": self.demo_account_type,
            "status": self.status.value,
            "created_at": self.created_at.isoformat() if self.created_at else None,
            "expires_at": self.expires_at.isoformat() if self.expires_at else None,
            "last_activity_at": self.last_activity_at.isoformat() if self.last_activity_at else None,
            "request_count": self.request_count,
            "metadata": self.session_metadata
        }