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

# 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",
    # Enums
    "AssetStatus",
    "JobStatus",
    # gRPC
    "GRPC_SERVICE",
    "JobRequest",
    "JobResponse",
    "ProgressRequest",
    "ProgressUpdate",
    "CancelRequest",
    "CancelResponse",
    "WorkerStatus",
    "Empty",
    # For generator
    "DATACLASSES",
    "ENUMS",
    "GRPC_MESSAGES",
    "BUILTIN_PRESETS",
]