from typing import Optional
from pydantic import BaseModel, Field
from .form_base_descriptor_models import FormDescriptor, RowElement, LineElement, LinkPort


class EachNodeData(BaseModel):
    """
    Pydantic модель для валидации данных узла each
    """
    ps_id: Optional[int] = Field(default=None, description="ID процесса")
    node_type: Optional[str] = Field(default=None, description="Тип узла")


class EachNodeLinks(BaseModel):
    """
    Pydantic модель для валидации связей узла each
    """
    parent_id: Optional[int] = Field(default=None, description="ID родительского узла")
    parent_port_number: Optional[int] = Field(default=None, description="Номер порта родительского узла")


class EachNodeCoreSchemaData(BaseModel):
    """
    Pydantic модель для данных портов узла each
    """
    then_port_number: Optional[int] = Field(default=0, description="Номер порта для перехода по Связи Then (LINK)")
    else_port_number: Optional[int] = Field(default=1, description="Номер порта для перехода по Связи Else (LINK)")


class EachNodeCoreSchema(BaseModel):
    """
    Pydantic модель для схемы узла each
    """
    ps_id: Optional[int] = Field(default=None, description="ID процесса")
    node_type: Optional[str] = Field(default=None, description="Тип узла")
    parent_id: Optional[int] = Field(default=None, description="ID родительского узла")
    parent_port_number: Optional[int] = Field(default=None, description="Номер порта родительского узла")
    data: Optional[EachNodeCoreSchemaData] = Field(default=None, description="Данные узла")


# Дескриптор формы узла EACH
EACH_FORM_DESCRIPTOR = FormDescriptor(
    elements=[
        RowElement(
            name="list_row",
            type="row",
            label="List",
            link_port=LinkPort(id="then_output", label="Then"),
            elements=[
                LineElement(
                    name="list",
                    type="line",
                    label="List",
                    placeholder="Validate list and extract next value from list",
                )
            ],
        ),
        RowElement(
            name="else_row",
            type="row",
            label="Alternative path",
            link_port=LinkPort(id="else_output", label="Else"),
        ),
    ]
)