from __future__ import annotations

from dataclasses import dataclass, field
from typing import TypeAlias


@dataclass(frozen=True)
class AnyAnnotation:
    pass


@dataclass(frozen=True)
class DictAnnotation:
    key: TypeAnnotation
    value: TypeAnnotation


@dataclass(frozen=True)
class ListAnnotation:
    item: TypeAnnotation


@dataclass(frozen=True)
class MappingAnnotation:
    key: TypeAnnotation
    value: TypeAnnotation


@dataclass(frozen=True)
class LiteralAnnotation:
    values: tuple[object, ...]


@dataclass(frozen=True)
class NamedAnnotation:
    name: str


@dataclass(frozen=True)
class TupleAnnotation:
    items: tuple[TypeAnnotation, ...]


@dataclass(frozen=True)
class UnionAnnotation:
    items: tuple[TypeAnnotation, ...]


TypeAnnotation: TypeAlias = (
    AnyAnnotation
    | DictAnnotation
    | ListAnnotation
    | MappingAnnotation
    | LiteralAnnotation
    | NamedAnnotation
    | TupleAnnotation
    | UnionAnnotation
)


@dataclass(frozen=True)
class FieldDef:
    name: str
    annotation: TypeAnnotation
    required: bool
    description: str | None = None


@dataclass(frozen=True)
class TypedDictDef:
    name: str
    fields: tuple[FieldDef, ...]
    description: str | None = None


@dataclass(frozen=True)
class TypeAliasDef:
    name: str
    annotation: TypeAnnotation


@dataclass(frozen=True)
class EnumDef:
    name: str
    values: tuple[object, ...]


@dataclass(frozen=True)
class OperationDef:
    method: str
    route_literal: str
    symbol: str
    protocol_name: str
    params_type: TypeAnnotation
    params_required: bool
    query_type: TypeAnnotation
    query_required: bool
    headers_type: TypeAnnotation
    headers_required: bool
    body_type: TypeAnnotation | None
    body_required: bool
    response_type: TypeAnnotation


@dataclass(frozen=True)
class NormalizedSpec:
    package_name: str
    typed_dicts: tuple[TypedDictDef, ...]
    aliases: tuple[TypeAliasDef, ...]
    enums: tuple[EnumDef, ...]
    operations: tuple[OperationDef, ...]


@dataclass(frozen=True)
class GeneratedArtifact:
    relative_path: str
    content: str


@dataclass
class RenderContext:
    package_name: str
    import_typing: set[str] = field(default_factory=set)
    import_typeddict: bool = False