Source code for kittycad.models.kcl_migration_client_message

from typing import Dict, Literal, Union

from pydantic import Field, RootModel
from typing_extensions import Annotated

from ..models.kcl_migration_application_status import KclMigrationApplicationStatus
from ..models.kcl_migration_request import KclMigrationRequest
from ..models.uuid import Uuid
from .base import KittyCadBaseModel




class OptionHeaders(KittyCadBaseModel):
    """Authenticate the connection."""

    headers: Dict[str, str]

    type: Literal["headers"] = "headers"





class OptionStart(KittyCadBaseModel):
    """Start or retrieve the same attempt after a delivery retry."""

    request: KclMigrationRequest

    type: Literal["start"] = "start"





class OptionStatus(KittyCadBaseModel):
    """Read an existing attempt without resuming it or changing its deadline."""

    operation_id: Uuid

    type: Literal["status"] = "status"





class OptionCancel(KittyCadBaseModel):
    """Cancel an existing attempt. Cancellation is idempotent."""

    operation_id: Uuid

    type: Literal["cancel"] = "cancel"





class OptionHistory(KittyCadBaseModel):
    """Read linked migration summaries, newest first, without loading or applying files."""

    conversation_id: Uuid

    type: Literal["history"] = "history"





class OptionApplication(KittyCadBaseModel):
    """Acknowledge a successful local apply/undo/redo. Does not execute or charge work."""

    expected_revision: int

    operation_id: Uuid

    status: KclMigrationApplicationStatus

    type: Literal["application"] = "application"





class OptionPing(KittyCadBaseModel):
    """Application heartbeat."""

    type: Literal["ping"] = "ping"



KclMigrationClientMessage = RootModel[
    Annotated[
        Union[
            OptionHeaders,
            OptionStart,
            OptionStatus,
            OptionCancel,
            OptionHistory,
            OptionApplication,
            OptionPing,
        ],
        Field(discriminator="type"),
    ]
]