from datetime import datetime

from pydantic import BaseModel

from api.schemas.decision import DecisionRead
from api.schemas.progress_event import ProgressEventRead
from api.schemas.task import TaskRead
from api.schemas.topic import TopicWithWorkstreams
from api.schemas.workstream import WorkstreamRead


class TopicTotals(BaseModel):
    active: int = 0
    paused: int = 0
    archived: int = 0
    total: int = 0


class WorkstreamTotals(BaseModel):
    active: int = 0
    blocked: int = 0
    completed: int = 0
    archived: int = 0
    total: int = 0


class TaskTotals(BaseModel):
    todo: int = 0
    in_progress: int = 0
    blocked: int = 0
    done: int = 0
    cancelled: int = 0
    total: int = 0


class DecisionTotals(BaseModel):
    open: int = 0
    resolved: int = 0
    escalated: int = 0
    superseded: int = 0
    total: int = 0


class Totals(BaseModel):
    topics: TopicTotals
    workstreams: WorkstreamTotals
    tasks: TaskTotals
    decisions: DecisionTotals


class StateSummary(BaseModel):
    generated_at: datetime
    totals: Totals
    topics: list[TopicWithWorkstreams]
    blocking_decisions: list[DecisionRead]
    blocked_tasks: list[TaskRead]
    recent_progress: list[ProgressEventRead]
    open_workstreams: list[WorkstreamRead]