"""
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 .stage_output import StageOutput
from .brand import BrandSource, Brand
from .media import AssetStatus, MediaAsset
from .profile import Profile
from .preset import BUILTIN_PRESETS, TranscodePreset
from .event import DETECT_VIEWS  # noqa: F401
from .inference import INFERENCE_VIEWS  # noqa: F401
from .ui_state import UI_STATE_VIEWS  # noqa: F401
from .stage import STAGE_VIEWS  # noqa: F401
from .view import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent
from .source import ChunkInfo, SourceJob, SourceType

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

# 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,
]