"""上传相关 Schema"""

from apiflask.fields import String, File, Integer, Dict as DictField, Enum
from iti.applications.common.enums.sys import StorageTypeEnum
from iti.applications.common.utils import BaseSchema


class UploadIn(BaseSchema):
    """文件直接上传输入"""

    file = File(required=True)
    directory_id = String(required=False, load_default=None, data_key="directoryId")
    storage_type = Enum(StorageTypeEnum, required=False, load_default=None, data_key="storageType")


class ChunkUploadInitIn(BaseSchema):
    """分片上传初始化输入"""

    filename = String(required=True, metadata={"description": "文件名"})
    file_size = Integer(required=True, metadata={"description": "文件总大小（字节）"})
    file_hash = String(
        required=False,
        load_default=None,
        metadata={"description": "文件MD5哈希（用于秒传）"},
    )
    chunk_size = Integer(
        required=False,
        load_default=2097152,
        metadata={"description": "分片大小（字节），默认2MB"},
    )
    total_chunks = Integer(
        required=False, load_default=None, metadata={"description": "总分片数"}
    )
    directory_id = String(
        required=False, load_default=None, metadata={"description": "目录ID"}
    )
    storage_type = Enum(
        StorageTypeEnum,
        required=False,
        load_default=None,
        metadata={"description": "存储类型"},
    )
    metadata = DictField(
        required=False, load_default=dict, metadata={"description": "扩展元数据"}
    )


class ChunkUploadIn(BaseSchema):
    """分片上传输入"""

    upload_id = String(required=True, metadata={"description": "上传任务ID"})
    chunk_index = Integer(
        required=True, metadata={"description": "分片索引（从0开始）"}
    )
    chunk_hash = String(
        required=False,
        load_default=None,
        metadata={"description": "分片MD5哈希（用于校验）"},
    )
    file = File(required=True, metadata={"description": "分片数据"})


class ChunkUploadMergeIn(BaseSchema):
    """分片合并输入"""

    upload_id = String(required=True, metadata={"description": "上传任务ID"})
    file_hash = String(
        required=False,
        load_default=None,
        metadata={"description": "文件MD5哈希（用于最终校验）"},
    )