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


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


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


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


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


# Дескриптор формы для узла Run
RUN_FORM_DESCRIPTOR = FormDescriptor(
    elements=[
        RowElement(
            name="task_row",
            type="row",
            label="Task",
            elements=[
                SelectElement(
                    name="task",
                    type="select",
                    label="Task",
                    data=["send_email", "process_data", "generate_report", "validate_input", "transform_data"],
                    default="send_email",
                )
            ],
        ),
        RowElement(
            name="context_row",
            type="row",
            label="Arguments",
            elements=[
                AreaElement(
                    name="context",
                    type="area",
                    label="Arguments",
                    placeholder="Build request JSON from provided environment variables",
                )
            ],
        ),
        RowElement(
            name="then_row",
            type="row",
            label="Then",
            link_port=LinkPort(id="then_output", label="Then"),
        ),
    ]
)