from __future__ import annotations

from typing import Literal

from pydantic import BaseModel, ConfigDict, Field

TerminationReason = Literal["target_hit", "max_iters", "stuck", "in_progress"]
IterationReason = Literal[
    "score_improved",
    "score_regressed",
    "score_plateau",
    "target_hit",
    "sandbox_error",
    "synthesizer_error",
]


class LoopConfig(BaseModel):
    model_config = ConfigDict(frozen=True)
    max_iters: int = Field(default=5, gt=0, le=100)
    target_score: float = Field(default=0.95, gt=0.0, le=2.0)
    tools: tuple[str, ...] = ("ruff", "imports")
    timeout_per_tool: float = 60.0
    top_k_citations: int = Field(default=5, gt=0, le=50)


class SynthesisResult(BaseModel):
    model_config = ConfigDict(frozen=True)
    proposed_files: dict[str, str]
    rationale: str
    cited_node_ids: tuple[str, ...]


class Iteration(BaseModel):
    model_config = ConfigDict(frozen=True)
    index: int
    cited_node_ids: tuple[str, ...]
    rationale: str
    proposed_files: dict[str, str]
    sandbox_score_before: float
    sandbox_score_after: float
    kept: bool
    reason: IterationReason


class RunResult(BaseModel):
    model_config = ConfigDict(frozen=True)
    run_id: str
    spec: str
    started_at: str
    ended_at: str
    final_score: float
    final_files: dict[str, str]
    iterations: tuple[Iteration, ...]
    terminated_by: TerminationReason