# generated by datamodel-codegen:
#   filename:  openapi.yaml
#   timestamp: 2025-06-17T19:50:44+00:00

from __future__ import annotations

from datetime import datetime
from enum import Enum
from typing import Any, Dict, List, Optional, Union

from pydantic import BaseModel, Field, RootModel, conint


class OperationType(Enum):
    install = 'install'
    uninstall = 'uninstall'
    update = 'update'
    update_all = 'update-all'
    update_comfyui = 'update-comfyui'
    fix = 'fix'
    disable = 'disable'
    enable = 'enable'
    install_model = 'install-model'


class OperationResult(Enum):
    success = 'success'
    failed = 'failed'
    skipped = 'skipped'
    error = 'error'
    skip = 'skip'


class TaskExecutionStatus(BaseModel):
    status_str: OperationResult
    completed: bool = Field(..., description='Whether the task completed')
    messages: List[str] = Field(..., description='Additional status messages')


class ManagerMessageName(Enum):
    cm_task_completed = 'cm-task-completed'
    cm_task_started = 'cm-task-started'
    cm_queue_status = 'cm-queue-status'


class ManagerPackInfo(BaseModel):
    id: str = Field(
        ...,
        description='Either github-author/github-repo or name of pack from the registry',
    )
    version: str = Field(..., description='Semantic version or Git commit hash')
    ui_id: Optional[str] = Field(None, description='Task ID - generated internally')


class ManagerPackInstalled(BaseModel):
    ver: str = Field(
        ...,
        description='The version of the pack that is installed (Git commit hash or semantic version)',
    )
    cnr_id: Optional[str] = Field(
        None, description='The name of the pack if installed from the registry'
    )
    aux_id: Optional[str] = Field(
        None,
        description='The name of the pack if installed from github (author/repo-name format)',
    )
    enabled: bool = Field(..., description='Whether the pack is enabled')


class SelectedVersion(Enum):
    latest = 'latest'
    nightly = 'nightly'


class ManagerChannel(Enum):
    default = 'default'
    recent = 'recent'
    legacy = 'legacy'
    forked = 'forked'
    dev = 'dev'
    tutorial = 'tutorial'


class ManagerDatabaseSource(Enum):
    remote = 'remote'
    local = 'local'
    cache = 'cache'


class ManagerPackState(Enum):
    installed = 'installed'
    disabled = 'disabled'
    not_installed = 'not_installed'
    import_failed = 'import_failed'
    needs_update = 'needs_update'


class ManagerPackInstallType(Enum):
    git_clone = 'git-clone'
    copy = 'copy'
    cnr = 'cnr'


class SecurityLevel(Enum):
    strong = 'strong'
    normal = 'normal'
    normal_ = 'normal-'
    weak = 'weak'


class RiskLevel(Enum):
    block = 'block'
    high = 'high'
    middle = 'middle'


class UpdateState(Enum):
    false = 'false'
    true = 'true'


class ManagerPack(ManagerPackInfo):
    author: Optional[str] = Field(
        None, description="Pack author name or 'Unclaimed' if added via GitHub crawl"
    )
    files: Optional[List[str]] = Field(
        None,
        description='Repository URLs for installation (typically contains one GitHub URL)',
    )
    reference: Optional[str] = Field(
        None, description='The type of installation reference'
    )
    title: Optional[str] = Field(None, description='The display name of the pack')
    cnr_latest: Optional[SelectedVersion] = None
    repository: Optional[str] = Field(None, description='GitHub repository URL')
    state: Optional[ManagerPackState] = None
    update_state: Optional[UpdateState] = Field(
        None, alias='update-state', description='Update availability status'
    )
    stars: Optional[int] = Field(None, description='GitHub stars count')
    last_update: Optional[datetime] = Field(None, description='Last update timestamp')
    health: Optional[str] = Field(None, description='Health status of the pack')
    description: Optional[str] = Field(None, description='Pack description')
    trust: Optional[bool] = Field(None, description='Whether the pack is trusted')
    install_type: Optional[ManagerPackInstallType] = None


class InstallPackParams(ManagerPackInfo):
    selected_version: Union[str, SelectedVersion] = Field(
        ..., description='Semantic version, Git commit hash, latest, or nightly'
    )
    repository: Optional[str] = Field(
        None,
        description='GitHub repository URL (required if selected_version is nightly)',
    )
    pip: Optional[List[str]] = Field(None, description='PyPi dependency names')
    mode: ManagerDatabaseSource
    channel: ManagerChannel
    skip_post_install: Optional[bool] = Field(
        None, description='Whether to skip post-installation steps'
    )


class UpdateAllPacksParams(BaseModel):
    mode: Optional[ManagerDatabaseSource] = None
    ui_id: Optional[str] = Field(None, description='Task ID - generated internally')


class UpdatePackParams(BaseModel):
    node_name: str = Field(..., description='Name of the node package to update')
    node_ver: Optional[str] = Field(
        None, description='Current version of the node package'
    )


class UpdateComfyUIParams(BaseModel):
    is_stable: Optional[bool] = Field(
        True,
        description='Whether to update to stable version (true) or nightly (false)',
    )
    target_version: Optional[str] = Field(
        None,
        description='Specific version to switch to (for version switching operations)',
    )


class FixPackParams(BaseModel):
    node_name: str = Field(..., description='Name of the node package to fix')
    node_ver: str = Field(..., description='Version of the node package')


class UninstallPackParams(BaseModel):
    node_name: str = Field(..., description='Name of the node package to uninstall')
    is_unknown: Optional[bool] = Field(
        False, description='Whether this is an unknown/unregistered package'
    )


class DisablePackParams(BaseModel):
    node_name: str = Field(..., description='Name of the node package to disable')
    is_unknown: Optional[bool] = Field(
        False, description='Whether this is an unknown/unregistered package'
    )


class EnablePackParams(BaseModel):
    cnr_id: str = Field(
        ..., description='ComfyUI Node Registry ID of the package to enable'
    )


class QueueStatus(BaseModel):
    total_count: int = Field(
        ..., description='Total number of tasks (pending + running)'
    )
    done_count: int = Field(..., description='Number of completed tasks')
    in_progress_count: int = Field(..., description='Number of tasks currently running')
    pending_count: Optional[int] = Field(
        None, description='Number of tasks waiting to be executed'
    )
    is_processing: bool = Field(..., description='Whether the task worker is active')
    client_id: Optional[str] = Field(
        None, description='Client ID (when filtered by client)'
    )


class ManagerMappings1(BaseModel):
    title_aux: Optional[str] = Field(None, description='The display name of the pack')


class ManagerMappings(
    RootModel[Optional[Dict[str, List[Union[List[str], ManagerMappings1]]]]]
):
    root: Optional[Dict[str, List[Union[List[str], ManagerMappings1]]]] = Field(
        None, description='Tuple of [node_names, metadata]'
    )


class ModelMetadata(BaseModel):
    name: str = Field(..., description='Name of the model')
    type: str = Field(..., description='Type of model')
    base: Optional[str] = Field(None, description='Base model type')
    save_path: Optional[str] = Field(None, description='Path for saving the model')
    url: str = Field(..., description='Download URL')
    filename: str = Field(..., description='Target filename')
    ui_id: Optional[str] = Field(None, description='ID for UI reference')


class InstallType(Enum):
    git = 'git'
    copy = 'copy'
    pip = 'pip'


class NodePackageMetadata(BaseModel):
    title: Optional[str] = Field(None, description='Display name of the node package')
    name: Optional[str] = Field(None, description='Repository/package name')
    files: Optional[List[str]] = Field(None, description='Source URLs for the package')
    description: Optional[str] = Field(
        None, description='Description of the node package functionality'
    )
    install_type: Optional[InstallType] = Field(None, description='Installation method')
    version: Optional[str] = Field(None, description='Version identifier')
    id: Optional[str] = Field(
        None, description='Unique identifier for the node package'
    )
    ui_id: Optional[str] = Field(None, description='ID for UI reference')
    channel: Optional[str] = Field(None, description='Source channel')
    mode: Optional[str] = Field(None, description='Source mode')


class SnapshotItem(RootModel[str]):
    root: str = Field(..., description='Name of the snapshot')


class Error(BaseModel):
    error: str = Field(..., description='Error message')


class InstalledPacksResponse(RootModel[Optional[Dict[str, ManagerPackInstalled]]]):
    root: Optional[Dict[str, ManagerPackInstalled]] = None


class HistoryListResponse(BaseModel):
    ids: Optional[List[str]] = Field(
        None, description='List of available batch history IDs'
    )


class InstalledNodeInfo(BaseModel):
    name: str = Field(..., description='Node package name')
    version: str = Field(..., description='Installed version')
    repository_url: Optional[str] = Field(None, description='Git repository URL')
    install_method: str = Field(
        ..., description='Installation method (cnr, git, pip, etc.)'
    )
    enabled: Optional[bool] = Field(
        True, description='Whether the node is currently enabled'
    )
    install_date: Optional[datetime] = Field(
        None, description='ISO timestamp of installation'
    )


class InstalledModelInfo(BaseModel):
    name: str = Field(..., description='Model filename')
    path: str = Field(..., description='Full path to model file')
    type: str = Field(..., description='Model type (checkpoint, lora, vae, etc.)')
    size_bytes: Optional[conint(ge=0)] = Field(None, description='File size in bytes')
    hash: Optional[str] = Field(None, description='Model file hash for verification')
    install_date: Optional[datetime] = Field(
        None, description='ISO timestamp when added'
    )


class ComfyUIVersionInfo(BaseModel):
    version: str = Field(..., description='ComfyUI version string')
    commit_hash: Optional[str] = Field(None, description='Git commit hash')
    branch: Optional[str] = Field(None, description='Git branch name')
    is_stable: Optional[bool] = Field(
        False, description='Whether this is a stable release'
    )
    last_updated: Optional[datetime] = Field(
        None, description='ISO timestamp of last update'
    )


class BatchOperation(BaseModel):
    operation_id: str = Field(..., description='Unique operation identifier')
    operation_type: OperationType
    target: str = Field(
        ..., description='Target of the operation (node name, model name, etc.)'
    )
    target_version: Optional[str] = Field(
        None, description='Target version for the operation'
    )
    result: OperationResult
    error_message: Optional[str] = Field(
        None, description='Error message if operation failed'
    )
    start_time: datetime = Field(
        ..., description='ISO timestamp when operation started'
    )
    end_time: Optional[datetime] = Field(
        None, description='ISO timestamp when operation completed'
    )
    client_id: Optional[str] = Field(
        None, description='Client that initiated the operation'
    )


class ComfyUISystemState(BaseModel):
    snapshot_time: datetime = Field(
        ..., description='ISO timestamp when snapshot was taken'
    )
    comfyui_version: ComfyUIVersionInfo
    frontend_version: Optional[str] = Field(
        None, description='ComfyUI frontend version if available'
    )
    python_version: str = Field(..., description='Python interpreter version')
    platform_info: str = Field(
        ..., description='Operating system and platform information'
    )
    installed_nodes: Optional[Dict[str, InstalledNodeInfo]] = Field(
        None, description='Map of installed node packages by name'
    )
    installed_models: Optional[Dict[str, InstalledModelInfo]] = Field(
        None, description='Map of installed models by name'
    )
    manager_config: Optional[Dict[str, Any]] = Field(
        None, description='ComfyUI Manager configuration settings'
    )
    comfyui_root_path: Optional[str] = Field(
        None, description='ComfyUI root installation directory'
    )
    model_paths: Optional[Dict[str, List[str]]] = Field(
        None, description='Map of model types to their configured paths'
    )
    manager_version: Optional[str] = Field(None, description='ComfyUI Manager version')
    security_level: Optional[SecurityLevel] = None
    network_mode: Optional[str] = Field(
        None, description='Network mode (online, offline, private)'
    )
    cli_args: Optional[Dict[str, Any]] = Field(
        None, description='Selected ComfyUI CLI arguments'
    )
    custom_nodes_count: Optional[conint(ge=0)] = Field(
        None, description='Total number of custom node packages'
    )
    failed_imports: Optional[List[str]] = Field(
        None, description='List of custom nodes that failed to import'
    )
    pip_packages: Optional[Dict[str, str]] = Field(
        None, description='Map of installed pip packages to their versions'
    )


class BatchExecutionRecord(BaseModel):
    batch_id: str = Field(..., description='Unique batch identifier')
    start_time: datetime = Field(..., description='ISO timestamp when batch started')
    end_time: Optional[datetime] = Field(
        None, description='ISO timestamp when batch completed'
    )
    state_before: ComfyUISystemState
    state_after: Optional[ComfyUISystemState] = Field(
        None, description='System state after batch execution'
    )
    operations: Optional[List[BatchOperation]] = Field(
        None, description='List of operations performed in this batch'
    )
    total_operations: Optional[conint(ge=0)] = Field(
        0, description='Total number of operations in batch'
    )
    successful_operations: Optional[conint(ge=0)] = Field(
        0, description='Number of successful operations'
    )
    failed_operations: Optional[conint(ge=0)] = Field(
        0, description='Number of failed operations'
    )
    skipped_operations: Optional[conint(ge=0)] = Field(
        0, description='Number of skipped operations'
    )


class QueueTaskItem(BaseModel):
    ui_id: str = Field(..., description='Unique identifier for the task')
    client_id: str = Field(..., description='Client identifier that initiated the task')
    kind: OperationType
    params: Union[
        InstallPackParams,
        UpdatePackParams,
        UpdateAllPacksParams,
        UpdateComfyUIParams,
        FixPackParams,
        UninstallPackParams,
        DisablePackParams,
        EnablePackParams,
        ModelMetadata,
    ]


class TaskHistoryItem(BaseModel):
    ui_id: str = Field(..., description='Unique identifier for the task')
    client_id: str = Field(..., description='Client identifier that initiated the task')
    kind: str = Field(..., description='Type of task that was performed')
    timestamp: datetime = Field(..., description='ISO timestamp when task completed')
    result: str = Field(..., description='Task result message or details')
    status: Optional[TaskExecutionStatus] = None


class TaskStateMessage(BaseModel):
    history: Dict[str, TaskHistoryItem] = Field(
        ..., description='Map of task IDs to their history items'
    )
    running_queue: List[QueueTaskItem] = Field(
        ..., description='Currently executing tasks'
    )
    pending_queue: List[QueueTaskItem] = Field(
        ..., description='Tasks waiting to be executed'
    )
    installed_packs: Dict[str, ManagerPackInstalled] = Field(
        ..., description='Map of currently installed node packages by name'
    )


class MessageTaskDone(BaseModel):
    ui_id: str = Field(..., description='Task identifier')
    result: str = Field(..., description='Task result message')
    kind: str = Field(..., description='Type of task')
    status: Optional[TaskExecutionStatus] = None
    timestamp: datetime = Field(..., description='ISO timestamp when task completed')
    state: TaskStateMessage


class MessageTaskStarted(BaseModel):
    ui_id: str = Field(..., description='Task identifier')
    kind: str = Field(..., description='Type of task')
    timestamp: datetime = Field(..., description='ISO timestamp when task started')
    state: TaskStateMessage


class MessageTaskFailed(BaseModel):
    ui_id: str = Field(..., description='Task identifier')
    error: str = Field(..., description='Error message')
    kind: str = Field(..., description='Type of task')
    timestamp: datetime = Field(..., description='ISO timestamp when task failed')
    state: TaskStateMessage


class MessageUpdate(
    RootModel[Union[MessageTaskDone, MessageTaskStarted, MessageTaskFailed]]
):
    root: Union[MessageTaskDone, MessageTaskStarted, MessageTaskFailed] = Field(
        ..., description='Union type for all possible WebSocket message updates'
    )


class HistoryResponse(BaseModel):
    history: Optional[Dict[str, TaskHistoryItem]] = Field(
        None, description='Map of task IDs to their history items'
    )