from typing import Optional
from pydantic import BaseModel, Field
from .form_base_descriptor_models import FormDescriptor, RowElement, SelectElement, NumberElement, DateElement, LinkPort


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


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


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


class WaitNodeCoreSchema(BaseModel):
    """
    Pydantic модель для схемы узла wait
    """
    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[WaitNodeCoreSchemaData] = Field(default=None, description="Данные узла")


# Дескриптор формы узла WAIT
WAIT_FORM_DESCRIPTOR = FormDescriptor(
    elements=[
        RowElement(
            name="type_row",
            type="row",
            label="Parameter",
            elements=[
                SelectElement(
                    name="type",
                    type="select",
                    label="Parameter",
                    data=["seconds", "datetime"],
                    placeholder="Selected option",
                )
            ],
        ),
        RowElement(
            name="wait_for_row",
            type="row",
            label="Time",
            elements=[
                NumberElement(
                    name="wait_for",
                    type="number",
                    label="Time",
                    placeholder="Compute wait time in seconds",
                )
            ],
        ),
        RowElement(
            name="wait_datetime_row",
            type="row",
            label="Wait until",
            elements=[
                DateElement(
                    name="wait_datetime",
                    type="date",
                    label="Wait until",
                    placeholder="Compute value until specific datetime",
                )
            ],
        ),
        RowElement(
            name="then_row",
            type="row",
            label="Transition",
            link_port=LinkPort(id="then_output", label="then"),
        ),
    ]
)