"""
MPR Schema Models

This module exports all dataclasses, enums, and constants that the generator
should process. Add new models here to have them included in generation.
"""

from .api import (
    CreateJobRequest,
    DeleteResult,
    ScanResult,
    SystemStatus,
    UpdateAssetRequest,
)
from .grpc import (
    GRPC_SERVICE,
    CancelRequest,
    CancelResponse,
    ChunkPipelineEvent,
    ChunkStreamRequest,
    Empty,
    JobRequest,
    JobResponse,
    ProgressRequest,
    ProgressUpdate,
    WorkerStatus,
)
from .job import Job, JobStatus, RunType
from .timeline import Timeline
from .checkpoint import Checkpoint
from .brand import BrandSource, Brand
from .media import AssetStatus, MediaAsset
from .presets import BUILTIN_PRESETS, TranscodePreset
from .detect import DETECT_VIEWS  # noqa: F401 — discovered by modelgen generic loader
from .inference import INFERENCE_VIEWS  # noqa: F401 — GPU inference server API types
from .ui_state import UI_STATE_VIEWS  # noqa: F401 — UI store state types
from .stages import StageConfigField, StageIO, StageDefinition, STAGE_VIEWS  # noqa: F401
from .views import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent
from .sources import ChunkInfo, SourceJob, SourceType

# Core domain models - generates SQLModel, TypeScript
DATACLASSES = [MediaAsset, TranscodePreset,
               Job, Timeline, Checkpoint, Brand]

# API request/response models
API_MODELS = [
    CreateJobRequest,
    UpdateAssetRequest,
    SystemStatus,
    ScanResult,
    DeleteResult,
    WorkerStatus,
    SourceJob,
    ChunkInfo,
]

# Status enums
ENUMS = [AssetStatus, JobStatus, RunType, BrandSource, SourceType]

# View/event models
VIEWS = [ChunkEvent, WorkerEvent, PipelineStats, ChunkOutputFile]

# gRPC messages
GRPC_MESSAGES = [
    JobRequest,
    JobResponse,
    ProgressRequest,
    ProgressUpdate,
    CancelRequest,
    CancelResponse,
    WorkerStatus,
    Empty,
    ChunkStreamRequest,
    ChunkPipelineEvent,
]

__all__ = [
    # Models
    "MediaAsset",
    "TranscodePreset",
    "Job",
    "Timeline",
    "Checkpoint",
    # Enums
    "AssetStatus",
    "JobStatus",
    "RunType",
    "BrandSource",
    "SourceType",
    # Stages
    "StageConfigField",
    "StageIO",
    "StageDefinition",
    # API
    "CreateJobRequest",
    "UpdateAssetRequest",
    "DeleteResult",
    "ScanResult",
    "SystemStatus",
    # gRPC
    "GRPC_SERVICE",
    "JobRequest",
    "JobResponse",
    "ProgressRequest",
    "ProgressUpdate",
    "CancelRequest",
    "CancelResponse",
    "WorkerStatus",
    "Empty",
    "ChunkStreamRequest",
    "ChunkPipelineEvent",
    # Views
    "ChunkEvent",
    "WorkerEvent",
    "PipelineStats",
    "ChunkOutputFile",
    # Sources
    "SourceJob",
    "ChunkInfo",
    # For generator
    "DATACLASSES",
    "API_MODELS",
    "ENUMS",
    "VIEWS",
    "GRPC_MESSAGES",
    "BUILTIN_PRESETS",
]