import datetime
from typing import Optional, Union

from pydantic import BaseModel


class IDData(BaseModel):
    id: int


class IDsData(BaseModel):
    ids: list[int]


class ShareItem(BaseModel):
    expire_value: int
    expire_style: str = "day"
    filename: str


class DeleteItem(BaseModel):
    filename: str


class LoginData(BaseModel):
    password: str


class UpdateFileData(BaseModel):
    id: int
    code: Optional[str] = None
    prefix: Optional[str] = None
    suffix: Optional[str] = None
    expired_at: Optional[Union[datetime.datetime, str]] = None
    expired_count: Optional[int] = None


class BatchUpdateFileData(BaseModel):
    ids: list[int]
    expired_at: Optional[Union[datetime.datetime, str]] = None
    expired_count: Optional[int] = None
    clearExpiredAt: Optional[bool] = None
    clear_expired_at: Optional[bool] = None


class FilePolicyActionData(BaseModel):
    id: int
    action: str
    downloadLimit: Optional[int] = None
    download_limit: Optional[int] = None


class BatchFilePolicyActionData(BaseModel):
    ids: list[int]
    action: str
    downloadLimit: Optional[int] = None
    download_limit: Optional[int] = None