from enum import Enum
from typing import List, Optional

from pydantic import BaseModel, Field


class CodeTaskType(str, Enum):
    GENERATE = "generate"
    FIX = "fix"
    EXPLAIN = "explain"
    REFACTOR = "refactor"
    REVIEW = "review"
    UNKNOWN = "unknown"


class SourceType(str, Enum):
    STACKOVERFLOW = "stackoverflow"
    GITHUB = "github"
    MODEL = "model"


class RetrievedEvidence(BaseModel):
    source_type: SourceType
    title: str = Field(..., min_length=1)
    snippet: str = Field(..., min_length=1)
    url: Optional[str] = None
    score: Optional[float] = None


class ResponseMeta(BaseModel):
    used_model: str = Field(..., min_length=1)
    fallback_used: bool = False
    retrieval_used: bool = False
    source_count: int = 0
    processing_time_ms: Optional[int] = None


class CodeXRequest(BaseModel):
    message: str = Field(..., min_length=1)
    session_id: Optional[str] = None
    mode: Optional[CodeTaskType] = None
    language: Optional[str] = None
    code: Optional[str] = None
    error_message: Optional[str] = None
    framework: Optional[str] = None
    file_name: Optional[str] = None
    previous_context: Optional[str] = None
    use_retrieval: bool = True


class CodeXResponse(BaseModel):
    answer: str = Field(..., min_length=1)
    task_type: CodeTaskType
    code_output: Optional[str] = None
    explanation: Optional[str] = None
    warnings: List[str] = Field(default_factory=list)
    sources: List[RetrievedEvidence] = Field(default_factory=list)
    needs_clarification: bool = False
    meta: ResponseMeta