ax-backend-v1/app/schemas/block.py

"""block.py — Block 相关 Schema"""

from typing import Optional

from pydantic import BaseModel, Field


class BlockBase(BaseModel):
    """Block 基础信息"""
    id: str = Field(..., validation_alias="id", serialization_alias="id")
    block_order: int = Field(..., validation_alias="blockOrder", serialization_alias="blockOrder")
    type: str = Field(..., validation_alias="type", serialization_alias="type")
    level: int = Field(..., validation_alias="level", serialization_alias="level")
    index: int = Field(..., validation_alias="index", serialization_alias="index")
    content: str | dict | list = Field(..., validation_alias="content", serialization_alias="content")
    word_style: str = Field(..., validation_alias="wordStyle", serialization_alias="wordStyle")
    style: dict = Field(..., validation_alias="style", serialization_alias="style")
    metadata: dict = Field(..., validation_alias="metadata", serialization_alias="metadata")
    
    class Config:
        populate_by_name = True


class BlockUpdate(BaseModel):
    """Block 更新请求"""
    content: Optional[str | dict | list] = None
    style: Optional[dict] = None
    word_style: Optional[str] = None
    metadata: Optional[dict] = None


class BlockCreate(BaseModel):
    """Block 创建请求"""
    type: str
    level: int = 0
    index: int = 0
    content: str | dict | list
    word_style: str = ""
    style: dict = {}
    metadata: dict = {}
    after_block_id: Optional[str] = None  # 在哪个 block 后插入


class TOCContent(BaseModel):
    """TOC Block content 结构"""
    title: str = "目录"


class TOCConfig(BaseModel):
    """TOC 配置"""
    levels: str = "1-1"  # 包含的标题层级
    use_hyperlinks: bool = True
    use_page_numbers: bool = True
    hide_page_numbers_in_web: bool = True
    use_outline_levels: bool = True
    show_leader_dots: bool = True
    leader_char: str = "."


class TOCMetadata(BaseModel):
    """TOC Block metadata 结构"""
    toc_config: TOCConfig
    is_auto_generated: bool = True
    readonly: bool = True
    deletable: bool = True


class TOCBlock(BaseModel):
    """完整的 TOC Block"""
    id: str
    block_order: int = Field(..., serialization_alias="blockOrder")
    type: str = "toc"
    level: int = 0
    index: int = 0
    content: TOCContent
    word_style: str = "TOC"
    metadata: TOCMetadata
    
    class Config:
        populate_by_name = True


class TOCItem(BaseModel):
    """目录树节点"""
    id: str
    level: int
    content: str
    children: list['TOCItem'] = []


class BlockSearchResult(BaseModel):
    """Block 搜索结果"""
    blocks: list[BlockBase]
    total: int


class BlockStats(BaseModel):
    """Block 统计信息"""
    total: int
    by_type: dict[str, int] = Field(..., alias="byType")
    
    class Config:
        populate_by_name = True