"""
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, SystemStatus
from .grpc import (
    GRPC_SERVICE,
    CancelRequest,
    CancelResponse,
    Empty,
    JobRequest,
    JobResponse,
    ProgressRequest,
    ProgressUpdate,
    WorkerStatus,
)
from .jobs import JobStatus, TranscodeJob
from .media import AssetStatus, MediaAsset
from .presets import BUILTIN_PRESETS, TranscodePreset

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

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

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

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

__all__ = [
    # Models
    "MediaAsset",
    "TranscodePreset",
    "TranscodeJob",
    # API Models
    "CreateJobRequest",
    "SystemStatus",
    # Enums
    "AssetStatus",
    "JobStatus",
    # gRPC
    "GRPC_SERVICE",
    "JobRequest",
    "JobResponse",
    "ProgressRequest",
    "ProgressUpdate",
    "CancelRequest",
    "CancelResponse",
    "WorkerStatus",
    "Empty",
    # For generator
    "DATACLASSES",
    "API_MODELS",
    "ENUMS",
    "GRPC_MESSAGES",
    "BUILTIN_PRESETS",
]