"""
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 .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, 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,
]