"""
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 .jobs import ChunkJob, ChunkJobStatus, JobStatus, TranscodeJob
from .media import AssetStatus, MediaAsset
from .presets import BUILTIN_PRESETS, TranscodePreset
from .views import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent

# Core domain models - generates Django, Pydantic, TypeScript
DATACLASSES = [MediaAsset, TranscodePreset, TranscodeJob, ChunkJob]

# API request/response models - generates TypeScript only (no Django)
# WorkerStatus from grpc.py is reused here
API_MODELS = [
    CreateJobRequest,
    UpdateAssetRequest,
    SystemStatus,
    ScanResult,
    DeleteResult,
    WorkerStatus,
]

# Status enums - included in generated code
ENUMS = [AssetStatus, JobStatus, ChunkJobStatus]

# View/event models - generates TypeScript for UI consumption
VIEWS = [ChunkEvent, WorkerEvent, PipelineStats, ChunkOutputFile]

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

__all__ = [
    # Models
    "MediaAsset",
    "TranscodePreset",
    "TranscodeJob",
    "ChunkJob",
    # API Models
    "CreateJobRequest",
    "UpdateAssetRequest",
    "DeleteResult",
    "ScanResult",
    "SystemStatus",
    # Enums
    "AssetStatus",
    "JobStatus",
    "ChunkJobStatus",
    # gRPC
    "GRPC_SERVICE",
    "JobRequest",
    "JobResponse",
    "ProgressRequest",
    "ProgressUpdate",
    "CancelRequest",
    "CancelResponse",
    "WorkerStatus",
    "Empty",
    "ChunkStreamRequest",
    "ChunkPipelineEvent",
    # Views
    "ChunkEvent",
    "WorkerEvent",
    "PipelineStats",
    "ChunkOutputFile",
    # For generator
    "DATACLASSES",
    "API_MODELS",
    "ENUMS",
    "VIEWS",
    "GRPC_MESSAGES",
    "BUILTIN_PRESETS",
]