from datetime import datetime

from pydantic import BaseModel, Field


class EditTemplateResponse(BaseModel):
    id: str
    workspace_id: str
    project_id: str
    name: str
    description: str
    target_duration: float
    clip_count: int
    is_active: bool
    created_at: datetime | None = None


class EditPlanClipResponse(BaseModel):
    id: str
    asset_id: str
    asset_name: str
    sequence: int
    start_time: float
    duration: float
    reason: str


class EditPlanResponse(BaseModel):
    id: str
    workspace_id: str
    project_id: str
    template_id: str
    asset_library_id: str
    title_id: str = ""
    status: str
    summary: str
    clips: list[EditPlanClipResponse] = Field(default_factory=list)
    created_at: datetime | None = None
    updated_at: datetime | None = None


class CreateEditPlanRequest(BaseModel):
    workspace_id: str
    asset_library_id: str
    template_id: str = ""
    title_id: str = ""