"""
Generator - Stack-specific code generators for modelgen.

Supported generators:
- PydanticGenerator: Pydantic BaseModel classes
- DjangoGenerator: Django ORM models
- TypeScriptGenerator: TypeScript interfaces
- ProtobufGenerator: Protocol Buffer definitions
- PrismaGenerator: Prisma schema
"""

from typing import Dict, Type

from .base import BaseGenerator
from .django import DjangoGenerator
from .graphene import GrapheneGenerator
from .prisma import PrismaGenerator
from .protobuf import ProtobufGenerator
from .pydantic import PydanticGenerator
from .typescript import TypeScriptGenerator

# Registry of available generators
GENERATORS: Dict[str, Type[BaseGenerator]] = {
    "pydantic": PydanticGenerator,
    "django": DjangoGenerator,
    "typescript": TypeScriptGenerator,
    "ts": TypeScriptGenerator,  # Alias
    "protobuf": ProtobufGenerator,
    "proto": ProtobufGenerator,  # Alias
    "prisma": PrismaGenerator,
    "graphene": GrapheneGenerator,
}

__all__ = [
    "BaseGenerator",
    "PydanticGenerator",
    "DjangoGenerator",
    "GrapheneGenerator",
    "TypeScriptGenerator",
    "ProtobufGenerator",
    "PrismaGenerator",
    "GENERATORS",
]