ctms/backend/app/schemas/collaboration.py

import uuid
from datetime import datetime
from typing import Literal, Optional

from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator


CollaborationFileType = Literal["word", "cell", "slide"]
CollaborationMemberRole = Literal["EDITOR", "MANAGER"]
CollaborationFileStatus = Literal["ACTIVE", "ARCHIVED", "DELETED"]
CollaborationShareAccessMode = Literal["VIEW", "EDIT"]
CollaborationShareExpiryPolicy = Literal["ONE_DAY", "SEVEN_DAYS", "THIRTY_DAYS", "PERMANENT"]
CollaborationEditRequestStatus = Literal["PENDING", "APPROVED", "REJECTED"]


class CollaborationFolderCreate(BaseModel):
    name: str = Field(min_length=1, max_length=120)
    parent_id: Optional[uuid.UUID] = None
    sort_order: int = 0

    @field_validator("name")
    @classmethod
    def normalize_name(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("文件夹名称不能为空")
        return value


class CollaborationFolderUpdate(BaseModel):
    name: Optional[str] = Field(default=None, min_length=1, max_length=120)
    parent_id: Optional[uuid.UUID] = None
    sort_order: Optional[int] = None


class CollaborationFolderRead(BaseModel):
    id: uuid.UUID
    study_id: uuid.UUID
    parent_id: Optional[uuid.UUID]
    name: str
    sort_order: int
    created_by: uuid.UUID
    deleted_at: Optional[datetime]
    created_at: datetime
    updated_at: datetime

    model_config = ConfigDict(from_attributes=True)


class CollaborationFileCreate(BaseModel):
    title: str = Field(min_length=1, max_length=240)
    file_type: CollaborationFileType
    folder_id: Optional[uuid.UUID] = None


class CollaborationFileUpdate(BaseModel):
    title: Optional[str] = Field(default=None, min_length=1, max_length=240)
    folder_id: Optional[uuid.UUID] = None
    status: Optional[Literal["ACTIVE", "ARCHIVED"]] = None
    allow_export: Optional[bool] = None
    allow_edit_request: Optional[bool] = None
    allow_sheet_structure_edit: Optional[bool] = None


class CollaborationRevisionRead(BaseModel):
    id: uuid.UUID
    file_id: uuid.UUID
    revision_no: int
    parent_revision_id: Optional[uuid.UUID]
    original_filename: str
    file_hash: str
    file_size: int
    mime_type: str
    source: str
    change_summary: Optional[str]
    created_by: Optional[uuid.UUID]
    created_by_name: Optional[str] = None
    created_by_avatar_url: Optional[str] = None
    created_at: datetime

    model_config = ConfigDict(from_attributes=True)


class CollaborationRevisionUpdate(BaseModel):
    change_summary: str = Field(min_length=1, max_length=240)

    @field_validator("change_summary")
    @classmethod
    def normalize_change_summary(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("版本名称不能为空")
        return value


class CollaborationRevisionCopyRequest(BaseModel):
    title: str = Field(min_length=1, max_length=240)
    folder_id: Optional[uuid.UUID] = None

    @field_validator("title")
    @classmethod
    def normalize_title(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("文件名不能为空")
        return value


class CollaborationFileCollaboratorRead(BaseModel):
    user_id: uuid.UUID
    full_name: str
    role: CollaborationMemberRole
    avatar_url: Optional[str] = None


class CollaborationFileRead(BaseModel):
    id: uuid.UUID
    study_id: uuid.UUID
    folder_id: Optional[uuid.UUID]
    title: str
    file_type: CollaborationFileType
    extension: str
    status: CollaborationFileStatus
    owner_id: uuid.UUID
    current_revision_id: Optional[uuid.UUID]
    generation: int
    allow_export: bool = False
    allow_edit_request: bool = False
    allow_sheet_structure_edit: bool = True
    deleted_at: Optional[datetime]
    created_at: datetime
    updated_at: datetime
    owner_name: Optional[str] = None
    folder_name: Optional[str] = None
    current_revision_no: Optional[int] = None
    current_revision_file_size: Optional[int] = None
    current_revision_mime_type: Optional[str] = None
    current_revision_created_at: Optional[datetime] = None
    collaboration_role: Optional[CollaborationMemberRole] = None
    collaborators: list[CollaborationFileCollaboratorRead] = Field(default_factory=list)
    can_edit: bool = False
    can_manage: bool = False
    can_export: bool = False
    can_request_edit: bool = False
    edit_request_status: Optional[CollaborationEditRequestStatus] = None
    can_transfer_ownership: bool = False

    model_config = ConfigDict(from_attributes=True)


class CollaborationMemberUpsert(BaseModel):
    user_id: uuid.UUID
    role: CollaborationMemberRole


class CollaborationMemberRead(BaseModel):
    id: uuid.UUID
    file_id: uuid.UUID
    user_id: uuid.UUID
    role: CollaborationMemberRole
    invited_by: uuid.UUID
    full_name: str
    email: str
    created_at: datetime


class CollaborationEditRequestRead(BaseModel):
    id: uuid.UUID
    file_id: uuid.UUID
    requester_id: uuid.UUID
    requester_name: str
    requester_email: str
    status: CollaborationEditRequestStatus
    resolved_by: Optional[uuid.UUID]
    resolved_at: Optional[datetime]
    created_at: datetime


class CollaborationEditRequestResolve(BaseModel):
    status: Literal["APPROVED", "REJECTED"]


class CollaborationOwnershipTransferRequest(BaseModel):
    new_owner_id: uuid.UUID


class CollaborationShareLinkUpdate(BaseModel):
    enabled: bool
    access_mode: CollaborationShareAccessMode = "VIEW"
    expiry_policy: CollaborationShareExpiryPolicy = "SEVEN_DAYS"
    password_mode: Literal["KEEP", "SET", "CLEAR"] = "KEEP"
    password: Optional[str] = Field(default=None, min_length=4, max_length=64)

    @model_validator(mode="after")
    def validate_password_change(self):
        if self.password_mode == "SET" and not self.password:
            raise ValueError("设置链接密码时必须提供密码")
        if self.password_mode != "SET" and self.password is not None:
            raise ValueError("仅在设置链接密码时允许提交密码")
        return self


class CollaborationShareLinkRead(BaseModel):
    id: uuid.UUID
    file_id: uuid.UUID
    enabled: bool
    access_mode: CollaborationShareAccessMode
    expiry_policy: CollaborationShareExpiryPolicy
    expires_at: Optional[datetime]
    has_password: bool
    share_path: str = "/collaboration/share"
    share_token: Optional[str] = None
    created_at: datetime
    updated_at: datetime


class CollaborationPublicShareMetadata(BaseModel):
    file_name: str
    file_type: CollaborationFileType
    access_mode: Literal["view", "edit"]
    allow_export: bool
    requires_password: bool
    expires_at: Optional[datetime]


class CollaborationSharePasswordRequest(BaseModel):
    password: str = Field(min_length=1, max_length=64)


class CollaborationShareAccessGrant(BaseModel):
    access_token: str
    expires_at: datetime


class CollaborationPublicEditorConfigRequest(BaseModel):
    access_token: Optional[str] = Field(default=None, max_length=2048)
    client_id: str = Field(min_length=8, max_length=64, pattern=r"^[A-Za-z0-9_-]+$")
    display_name: str = Field(default="链接访客", min_length=1, max_length=40)

    @field_validator("display_name")
    @classmethod
    def normalize_display_name(cls, value: str) -> str:
        return value.strip() or "链接访客"


class CollaborationCandidateRead(BaseModel):
    user_id: uuid.UUID
    full_name: str
    email: str
    role_in_study: str
    can_be_editor: bool
    can_be_manager: bool


class CollaborationEditorConfigRead(BaseModel):
    file_id: uuid.UUID
    file_name: str
    access_mode: Literal["view", "edit"]
    can_save_as: bool = False
    can_download: bool = False
    can_request_edit: bool = False
    host_path: str = "/onlyoffice-host.html"
    expires_at: datetime
    config: dict


class CollaborationExportRecord(BaseModel):
    file_type: str = Field(min_length=1, max_length=16, pattern=r"^[a-z0-9]+$")


class CollaborationCallbackPayload(BaseModel):
    key: str
    status: int
    url: Optional[str] = None
    changesurl: Optional[str] = None
    filetype: Optional[str] = None
    forcesavetype: Optional[int] = None
    userdata: Optional[str] = None
    users: list[str] = Field(default_factory=list)
    actions: list[dict] = Field(default_factory=list)
    history: Optional[dict] = None


class CollaborationRestoreRequest(BaseModel):
    change_summary: Optional[str] = Field(default=None, max_length=500)