from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, Text, JSON
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
from app.db.database import Base
import uuid

class License(Base):
    __tablename__ = "licenses"
    
    id = Column(Integer, primary_key=True, index=True)
    license_key = Column(String, unique=True, index=True, default=lambda: str(uuid.uuid4()))
    product_id = Column(String, nullable=False)
    customer_email = Column(String, nullable=False)
    customer_name = Column(String)
    
    device_limit = Column(Integer, default=3)
    concurrent_sessions_limit = Column(Integer, default=1)
    is_active = Column(Boolean, default=True)
    expires_at = Column(DateTime, nullable=True)
    
    created_at = Column(DateTime, server_default=func.now())
    updated_at = Column(DateTime, onupdate=func.now())
    
    device_registrations = relationship("DeviceRegistration", back_populates="license")
    sessions = relationship("LicenseSession", back_populates="license")
    
class DeviceRegistration(Base):
    __tablename__ = "device_registrations"
    
    id = Column(Integer, primary_key=True, index=True)
    license_id = Column(Integer, ForeignKey("licenses.id"))
    hardware_fingerprint = Column(String, nullable=False)
    device_name = Column(String, nullable=False)
    device_type = Column(String, default="unknown")
    operating_system = Column(String)
    
    first_activated_at = Column(DateTime, server_default=func.now())
    last_seen_at = Column(DateTime, server_default=func.now())
    is_active = Column(Boolean, default=True)
    
    deactivated_at = Column(DateTime, nullable=True)
    deactivated_by = Column(String, nullable=True)
    
    ip_address = Column(String)
    user_agent = Column(String)
    app_version = Column(String)
    device_metadata = Column("metadata", JSON, default={})
    
    license = relationship("License", back_populates="device_registrations")
    sessions = relationship("LicenseSession", back_populates="device_registration")


class LicenseSession(Base):
    __tablename__ = "license_sessions"
    
    id = Column(Integer, primary_key=True, index=True)
    license_id = Column(Integer, ForeignKey("licenses.id"))
    device_registration_id = Column(Integer, ForeignKey("device_registrations.id"))
    
    session_token = Column(String, unique=True, nullable=False)
    hardware_fingerprint = Column(String, nullable=False)
    
    started_at = Column(DateTime, server_default=func.now())
    last_heartbeat = Column(DateTime, server_default=func.now())
    ended_at = Column(DateTime, nullable=True)
    end_reason = Column(String, nullable=True)
    
    ip_address = Column(String)
    client_version = Column(String)
    user_agent = Column(String)
    
    license = relationship("License", back_populates="sessions")
    device_registration = relationship("DeviceRegistration", back_populates="sessions")
    
class Version(Base):
    __tablename__ = "versions"
    
    id = Column(Integer, primary_key=True, index=True)
    version_number = Column(String, unique=True, nullable=False)
    release_date = Column(DateTime, server_default=func.now())
    is_mandatory = Column(Boolean, default=False)
    min_version = Column(String)
    
    download_url = Column(String)
    release_notes = Column(Text)
    
    created_at = Column(DateTime, server_default=func.now())
    
class ApiKey(Base):
    __tablename__ = "api_keys"
    
    id = Column(Integer, primary_key=True, index=True)
    key = Column(String, unique=True, index=True, default=lambda: str(uuid.uuid4()))
    name = Column(String, nullable=False)
    is_active = Column(Boolean, default=True)
    
    created_at = Column(DateTime, server_default=func.now())
    last_used = Column(DateTime)