"""大屏规划协议，描述设计方案，不直接执行动作"""

from typing import Literal, Self

from pydantic import Field, model_validator

from app.schemas.base import ExtensibleModel, JsonObject, NonBlankString
from app.schemas.screen import Layout


class PlanRegion(ExtensibleModel):
    region_id: NonBlankString = Field(description="规划内的区域编号，不是前端组件 ID")
    name: NonBlankString = Field(description="区域名称，例如顶部指标区")
    purpose: NonBlankString = Field(description="区域用途和展示重点")
    layout_description: NonBlankString = Field(
        description="区域位置、排列和空间分配说明"
    )
    layout: Layout | None = Field(
        default=None, description="可选的建议位置和尺寸，画布绝对坐标，单位 px"
    )


class PlanComponent(ExtensibleModel):
    plan_id: NonBlankString = Field(
        description="规划内的组件编号，用于关联资源，不作为新增组件的实际 ID"
    )
    disposition: Literal["keep", "modify", "add", "remove"] = Field(
        description="保留、调整、新增或移除，只表达规划意图，不代表执行能力已经支持"
    )
    target_id: NonBlankString | None = Field(
        default=None, description="已有组件 ID，新增时必须为空"
    )
    region_id: NonBlankString | None = Field(
        default=None, description="所属规划区域，可为空，例如移除组件或未分区"
    )
    purpose: NonBlankString = Field(description="展示目的或调整原因")
    component_type: NonBlankString | None = Field(
        default=None, description="建议组件类型，不代表已在组件库中确认可用"
    )
    metrics: list[NonBlankString] = Field(
        default_factory=list, description="展示的指标和维度描述，不是数据绑定"
    )
    design: JsonObject = Field(
        default_factory=dict, description="样式、交互及内容等设计要求，保留扩展字段"
    )
    layout: Layout | None = Field(
        default=None, description="可选的建议布局，画布绝对坐标，不是区域内相对坐标"
    )

    @model_validator(mode="after")
    def validate_target(self) -> Self:
        if self.disposition == "add":
            if self.target_id is not None:
                raise ValueError("新增组件不能指定已有 target_id")
        elif self.target_id is None:
            raise ValueError("保留、调整或移除组件必须提供 target_id")
        return self


class PlanResourceNeed(ExtensibleModel):
    resource_id: NonBlankString = Field(
        description="本次规划中的资源需求编号，不是已获取资源的 ID"
    )
    kind: NonBlankString = Field(
        description="资源类别，例如 data、component、template、asset、knowledge，允许扩展"
    )
    description: NonBlankString = Field(description="需要查询或补充的内容")
    purpose: NonBlankString = Field(description="资源用途和查询原因")
    component_ids: list[NonBlankString] = Field(
        default_factory=list, description="关联的 plan_id 列表，为空表示整个规划使用"
    )


class ScreenPlan(ExtensibleModel):
    goal: NonBlankString = Field(description="规划目标，遵循已确认需求")
    design_summary: NonBlankString = Field(description="整体或局部设计思路")
    layout_description: NonBlankString = Field(
        description="区域关系、布局安排及保留内容"
    )
    canvas_changes: JsonObject = Field(
        default_factory=dict, description="建议的画布属性变化，为空表示不改画布"
    )
    regions: list[PlanRegion] = Field(default_factory=list)
    components: list[PlanComponent] = Field(default_factory=list)
    resource_needs: list[PlanResourceNeed] = Field(
        default_factory=list, description="规划细化后的待查询资源，不能声称已经获取"
    )
    constraints: list[NonBlankString] = Field(
        default_factory=list, description="沿用用户约束，不擅自放宽"
    )
    assumptions: list[NonBlankString] = Field(
        default_factory=list, description="非阻塞设计偏好的默认假设"
    )

    @model_validator(mode="after")
    def validate_references(self) -> Self:
        region_ids = [region.region_id for region in self.regions]
        plan_ids = [component.plan_id for component in self.components]
        target_ids = [
            component.target_id
            for component in self.components
            if component.target_id is not None
        ]
        resource_ids = [resource.resource_id for resource in self.resource_needs]
        for name, values in (
            ("region_id", region_ids),
            ("plan_id", plan_ids),
            ("target_id", target_ids),
            ("resource_id", resource_ids),
        ):
            if len(values) != len(set(values)):
                raise ValueError(f"规划中存在重复的 {name}")
        for component in self.components:
            if (
                component.region_id is not None
                and component.region_id not in region_ids
            ):
                raise ValueError(f"组件引用了不存在的区域: {component.region_id}")
        for resource in self.resource_needs:
            if len(resource.component_ids) != len(set(resource.component_ids)):
                raise ValueError(
                    f"资源 {resource.resource_id} 的 component_ids 不能重复"
                )
            if unknown := set(resource.component_ids) - set(plan_ids):
                raise ValueError(f"资源引用了不存在的规划组件: {sorted(unknown)}")
        return self