ai-agent-admin/backend-fastapi/core/message/schema.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
消息中心 Schema
"""
from datetime import datetime
from typing import Optional, List, Dict

from pydantic import BaseModel, Field, ConfigDict

from app.base_schema import CSTDatetime


# ============ 消息相关 Schema ============
class MessageOut(BaseModel):
    """消息输出"""
    id: str
    title: str
    content: str
    msg_type: str
    status: str
    link_type: str = ""
    link_id: str = ""
    sender_name: str = ""
    read_at: Optional[CSTDatetime] = None
    created_at: Optional[CSTDatetime] = Field(None, alias="sys_create_datetime")

    model_config = ConfigDict(from_attributes=True, populate_by_name=True)


class MessageListOut(BaseModel):
    """消息列表输出"""
    id: str
    title: str
    content: str
    msg_type: str
    status: str
    link_type: str = ""
    link_id: str = ""
    sender_id: Optional[str] = None
    sender_name: str = ""
    created_at: Optional[CSTDatetime] = Field(None, alias="sys_create_datetime")

    model_config = ConfigDict(from_attributes=True, populate_by_name=True)


class UnreadCountOut(BaseModel):
    """未读数量输出"""
    total: int
    by_type: Dict[str, int]


class MarkReadInput(BaseModel):
    """标记已读输入"""
    msg_type: Optional[str] = Field(None, description="消息类型，不传则标记全部")


# ============ 公告相关 Schema ============
class AnnouncementCreate(BaseModel):
    """创建公告"""
    title: str = Field(..., max_length=200, description="公告标题")
    content: str = Field(..., description="公告内容")
    summary: str = Field(default="", max_length=500, description="摘要")
    status: str = Field(default="draft", description="状态: draft/published")
    priority: int = Field(default=0, description="优先级: 0普通/1重要/2紧急")
    is_top: bool = Field(default=False, description="是否置顶")
    target_type: str = Field(default="all", description="接收范围: all/dept/role/user")
    target_ids: List[str] = Field(default=[], description="接收目标ID列表")
    publish_time: Optional[datetime] = Field(None, description="发布时间")
    expire_time: Optional[datetime] = Field(None, description="过期时间")


class AnnouncementUpdate(BaseModel):
    """更新公告"""
    title: Optional[str] = Field(None, max_length=200, description="公告标题")
    content: Optional[str] = Field(None, description="公告内容")
    summary: Optional[str] = Field(None, max_length=500, description="摘要")
    status: Optional[str] = Field(None, description="状态")
    priority: Optional[int] = Field(None, description="优先级")
    is_top: Optional[bool] = Field(None, description="是否置顶")
    target_type: Optional[str] = Field(None, description="接收范围")
    target_ids: Optional[List[str]] = Field(None, description="接收目标ID列表")
    publish_time: Optional[datetime] = Field(None, description="发布时间")
    expire_time: Optional[datetime] = Field(None, description="过期时间")


class AnnouncementOut(BaseModel):
    """公告输出"""
    id: str
    title: str
    content: str
    summary: str = ""
    status: str
    priority: int
    is_top: bool
    target_type: str
    target_ids: List[str] = []
    publish_time: Optional[CSTDatetime] = None
    expire_time: Optional[CSTDatetime] = None
    publisher_id: Optional[str] = None
    publisher_name: str = ""
    read_count: int = 0
    created_at: Optional[CSTDatetime] = Field(None, alias="sys_create_datetime")

    model_config = ConfigDict(from_attributes=True, populate_by_name=True)


class AnnouncementListOut(BaseModel):
    """公告列表输出"""
    id: str
    title: str
    summary: str = ""
    status: str
    priority: int
    is_top: bool
    target_type: str
    publisher_name: str = ""
    read_count: int = 0
    publish_time: Optional[CSTDatetime] = None
    created_at: Optional[CSTDatetime] = Field(None, alias="sys_create_datetime")

    model_config = ConfigDict(from_attributes=True, populate_by_name=True)


class UserAnnouncementOut(BaseModel):
    """用户公告输出"""
    id: str
    title: str
    summary: str = ""
    content: str
    priority: int
    is_top: bool
    is_read: bool = False
    publisher_name: str = ""
    publish_time: Optional[CSTDatetime] = None

    model_config = ConfigDict(from_attributes=True, populate_by_name=True)


class ReadStatsOut(BaseModel):
    """阅读统计输出"""
    total_read: int
    readers: List[dict]


# ============ 发送消息 Schema ============
class SendMessageInput(BaseModel):
    """发送站内消息"""
    recipient_ids: List[str] = Field(..., description="接收人用户ID列表")
    title: str = Field(..., max_length=200, description="消息标题")
    content: str = Field(..., description="消息内容")
    msg_type: str = Field(default="system", description="消息类型: system/workflow/todo")
    channels: List[str] = Field(default=["site"], description="发送渠道: site/email/dingtalk/dingtalk_todo/feishu/wechat/wechat_mp/sms/chat")


# ============ 邮件相关 Schema ============
class EmailTestInput(BaseModel):
    """测试邮件输入"""
    to_email: str = Field(..., description="收件人邮箱")


class EmailSendInput(BaseModel):
    """发送邮件输入"""
    recipient_ids: List[str] = Field(..., description="收件人用户ID列表")
    title: str = Field(..., max_length=200, description="邮件主题")
    content: str = Field(..., description="邮件内容")


# ============ 钉钉相关 Schema ============
class DingTalkWebhookTestInput(BaseModel):
    """钉钉 Webhook 测试输入"""
    content: str = Field(default="这是一条测试消息", description="消息内容")


class DingTalkWorkNoticeTestInput(BaseModel):
    """钉钉工作通知测试输入"""
    user_id: str = Field(..., description="系统用户ID（需已绑定钉钉）")


# ============ 钉钉待办相关 Schema ============
class DingTalkTodoTestInput(BaseModel):
    """钉钉待办测试输入"""
    user_id: str = Field(..., description="系统用户ID（需已绑定钉钉）")
    title: str = Field(default="测试待办任务", description="待办标题")
    content: str = Field(default="这是一条测试钉钉待办，用于验证钉钉待办推送功能。", description="待办内容")


class DingTalkTodoCompleteInput(BaseModel):
    """钉钉待办完成输入"""
    source_type: str = Field(..., description="来源类型: workflow_task/workflow_instance/message")
    source_id: str = Field(..., description="业务ID")


# ============ 飞书相关 Schema ============
class FeishuWebhookTestInput(BaseModel):
    """飞书 Webhook 测试输入"""
    content: str = Field(default="这是一条测试消息", description="消息内容")


class FeishuAppMessageTestInput(BaseModel):
    """飞书应用消息测试输入"""
    user_id: str = Field(..., description="系统用户ID（需已绑定飞书）")


# ============ 企业微信相关 Schema ============
class WecomWebhookTestInput(BaseModel):
    """企业微信 Webhook 测试输入"""
    content: str = Field(default="这是一条测试消息", description="消息内容")


class WecomAppMessageTestInput(BaseModel):
    """企业微信应用消息测试输入"""
    user_id: str = Field(..., description="系统用户ID（需已绑定企业微信）")


# ============ 微信公众号相关 Schema ============
class WechatMPTestInput(BaseModel):
    """微信公众号模板消息测试输入"""
    user_id: str = Field(..., description="系统用户ID（需已绑定微信）")