Unified_Authentication_Platform/backend/app/schemas/message.py

from pydantic import BaseModel, Field, model_validator
from typing import Optional, Union, Any
from datetime import datetime
from enum import Enum

class MessageType(str, Enum):
    MESSAGE = "MESSAGE"
    NOTIFICATION = "NOTIFICATION"

class ContentType(str, Enum):
    TEXT = "TEXT"
    IMAGE = "IMAGE"
    VIDEO = "VIDEO"
    FILE = "FILE"

class MessageBase(BaseModel):
    title: str = Field(..., max_length=255)
    content: Union[str, dict, Any] = Field(..., description="内容")
    content_type: ContentType = ContentType.TEXT
    
    type: MessageType = MessageType.MESSAGE
    app_id: Optional[int] = None
    
    action_url: Optional[str] = None
    action_text: Optional[str] = None
    
    # SSO 扩展
    target_url: Optional[str] = None
    auto_sso: bool = False

class MessageCreate(MessageBase):
    receiver_id: Optional[int] = None
    app_user_id: Optional[str] = None
    
    @model_validator(mode='after')
    def check_receiver(self):
        if not self.receiver_id and not (self.app_user_id and self.app_id):
            raise ValueError("必须提供 receiver_id，或者同时提供 app_user_id 和 app_id")
        return self

class MessageUpdate(BaseModel):
    is_read: Optional[bool] = None

class MessageResponse(BaseModel):
    id: int
    sender_id: Optional[int]
    receiver_id: int
    app_id: Optional[int]
    
    type: MessageType
    content_type: ContentType
    
    title: str
    content: str  # DB 中存的是字符串
    
    action_url: Optional[str]
    action_text: Optional[str]
    
    is_read: bool
    created_at: datetime
    read_at: Optional[datetime]

    class Config:
        from_attributes = True

class ConversationResponse(BaseModel):
    user_id: int
    username: str
    full_name: Optional[str]
    unread_count: int
    last_message: str
    last_message_type: ContentType
    updated_at: datetime

    class Config:
        from_attributes = True

# Device Schema
class DeviceRegister(BaseModel):
    device_token: str
    platform: str
    device_name: Optional[str] = None