from sqlalchemy import Column, String, Text, Float, DateTime, Integer, Boolean, func, ForeignKey, JSON
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.schema import UniqueConstraint
import datetime

Base = declarative_base()

class DeviceConfig(Base):
    __tablename__ = "device_configs"
    
    id = Column(Integer, primary_key=True, autoincrement=True)
    device_id = Column(String(64), unique=True, index=True, nullable=False)
    selected_role_key = Column(String(64), nullable=False)
    preferred_language = Column(String(10), nullable=True)
    volume = Column(Integer, nullable=True)
    last_update_time = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    
    def to_dict(self):
        """将配置转换为可序列化的字典"""
        return {
            "selected_role_key": self.selected_role_key,
            "preferred_language": self.preferred_language,
            "volume": self.volume if hasattr(self, 'volume') else None,
        }

class ConversationHistory(Base):
    __tablename__ = "conversation_histories"
    id = Column(Integer, primary_key=True, autoincrement=True)
    device_id = Column(String(64), nullable=False, index=True)
    role_key = Column(String(64), nullable=False, index=True)
    last_interaction_time = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class ConversationMessage(Base):
    __tablename__ = "conversation_messages"
    id = Column(Integer, primary_key=True, autoincrement=True)
    conversation_id = Column(Integer, ForeignKey("conversation_histories.id", ondelete="CASCADE"), nullable=False, index=True)
    is_user = Column(Boolean, nullable=False, default=False)  # True表示用户消息，False表示助手消息
    content = Column(Text, nullable=False)
    timestamp = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class Role(Base):
    __tablename__ = "roles"
    id = Column(Integer, primary_key=True, autoincrement=True)
    role_key = Column(String(64), unique=True, nullable=False, index=True)
    name = Column(String(128), nullable=False)
    description = Column(Text, nullable=True)
    content = Column(Text, nullable=False)
    default_language = Column(String(10), nullable=True)
    volcano_model_id = Column(String(64), nullable=True)
    minimax_voice_id = Column(String(64), nullable=True)
    url = Column(String(255), nullable=True)
    homophones = Column(JSON, nullable=True)
    enabled = Column(Boolean, default=True, nullable=False)
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    
    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class RoleLanguage(Base):
    __tablename__ = "role_languages"
    id = Column(Integer, primary_key=True, autoincrement=True)
    role_id = Column(Integer, ForeignKey("roles.id", ondelete="CASCADE"), nullable=False)
    language_code = Column(String(10), nullable=False)
    name = Column(String(128), nullable=True)
    content = Column(Text, nullable=True)
    minimax_voice_id = Column(String(64), nullable=True)
    url = Column(String(255), nullable=True)
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    
    __table_args__ = (
        UniqueConstraint('role_id', 'language_code', name='uix_role_language'),
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class DeviceAuth(Base):
    __tablename__ = "device_auth"
    id = Column(Integer, primary_key=True, autoincrement=True)
    device_id = Column(String(64), nullable=False, unique=True, index=True)
    serial_number = Column(String(64), nullable=False)
    batch_id = Column(String(20), nullable=True, index=True)
    is_active = Column(Boolean, default=True, nullable=False)
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    
    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class DeviceFirmwareUpdate(Base):
    __tablename__ = "device_firmware_update"
    id = Column(Integer, primary_key=True, autoincrement=True)
    device_id = Column(String(64), nullable=False, unique=True, index=True)
    firmware_version = Column(String(64), nullable=False)
    update_status = Column(String(32), nullable=False, default="success")     # 更新状态，如 updating/success/failed
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
    progress = Column(Float, nullable=True, default=0.0)  # 更新进度

    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )

class SystemConfig(Base):
    __tablename__ = "system_config"
    id = Column(Integer, primary_key=True, autoincrement=True)
    config_key = Column(String(128), unique=True, nullable=False, index=True)
    config_value = Column(Text, nullable=True)
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())

    __table_args__ = (
        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
    )