import datetime
import uuid
from enum import IntEnum

from sqlalchemy import Column, ForeignKey, Integer, DateTime, String
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.dialects.postgresql import JSONB

from app.db import Base
from app.schemas import FileType


class PerceptualType(IntEnum):
    VISION = 1
    AUDIO = 2
    TEXT = 3
    CONVERSATION = 4

    @staticmethod
    def trans_from_file_type(file_type: FileType | str):
        type_map = {
            FileType.IMAGE: PerceptualType.VISION,
            FileType.AUDIO: PerceptualType.AUDIO,
            FileType.VIDEO: PerceptualType.VISION,
            FileType.DOCUMENT: PerceptualType.TEXT
        }
        return type_map.get(file_type, PerceptualType.TEXT)


class FileStorageService(IntEnum):
    LOCAL = 1
    REMOTE = 2


class MemoryPerceptualModel(Base):
    __tablename__ = "memory_perceptual"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    end_user_id = Column(UUID(as_uuid=True), ForeignKey("end_users.id"), index=True)

    perceptual_type = Column(Integer, index=True, nullable=False, comment="感知类型")

    storage_service = Column(Integer, default=0, comment="存储服务类型")
    file_path = Column(String, nullable=False, comment="文件路径")
    file_name = Column(String, nullable=False, comment="文件名称")
    file_ext = Column(String, nullable=False, comment="文件后缀名")

    summary = Column(String, comment="摘要")
    meta_data = Column(JSONB, comment="元信息")

    created_time = Column(DateTime, default=datetime.datetime.now, comment="创建时间")