from fastapi import APIRouter
from pydantic import BaseModel

from rag.rag_chain import run_rag_query
from services.code_analysis import detect_bugs, explain_function, analyze_complexity
from services.documentation import generate_module_docs, generate_readme
from services.file_creator import propose_new_file, apply_approved_file, FileProposal

router = APIRouter()


class QueryRequest(BaseModel):
    query: str
    k: int = 5


class ExplainRequest(BaseModel):
    file_path: str
    function_name: str


class FileProposalRequest(BaseModel):
    description: str
    context_query: str | None = None


class FileApprovalRequest(BaseModel):
    proposal: dict
    approved: bool


@router.post("/query")
def query(req: QueryRequest):
    return run_rag_query(req.query, req.k)


@router.post("/analyze/bugs")
def bugs(file_path: str):
    return {"bugs": detect_bugs(file_path)}


@router.post("/analyze/complexity")
def complexity(file_path: str):
    return analyze_complexity(file_path)


@router.post("/analyze/explain")
def explain(req: ExplainRequest):
    return {"explanation": explain_function(req.file_path, req.function_name)}


@router.post("/docs/module")
def module_docs(file_path: str):
    return {"docs": generate_module_docs(file_path)}


@router.post("/docs/readme")
def readme(root_path: str):
    return {"readme": generate_readme(root_path)}


@router.post("/files/propose")
def propose(req: FileProposalRequest):
    proposal = propose_new_file(req.description, req.context_query)
    return proposal


@router.post("/files/approve")
async def approve(req: FileApprovalRequest):
    proposal = FileProposal(**req.proposal)
    return await apply_approved_file(proposal, req.approved)