import { NextRequest, NextResponse } from "next/server";
import {
  getCuratedModels,
  getEnabledModels,
  upsertCuratedModel,
  removeCuratedModel,
  toggleModelEnabled,
  updateModelCost,
  CuratedModel,
} from "@/lib/model-store";

export async function GET(request: NextRequest) {
  const enabledOnly = request.nextUrl.searchParams.get("enabled") === "true";
  return NextResponse.json(enabledOnly ? await getEnabledModels() : await getCuratedModels());
}

export async function POST(request: NextRequest) {
  const body = await request.json();

  if (body.action === "toggle" && body.id) {
    const result = await toggleModelEnabled(body.id);
    if (!result) return NextResponse.json({ error: "not found" }, { status: 404 });
    return NextResponse.json(result);
  }

  if (body.action === "update-cost" && body.id) {
    const result = await updateModelCost(body.id, body.costPer1kInput, body.costPer1kOutput);
    if (!result) return NextResponse.json({ error: "not found" }, { status: 404 });
    return NextResponse.json(result);
  }

  if (!body.id || !body.provider) {
    return NextResponse.json({ error: "id and provider are required" }, { status: 400 });
  }

  const model: CuratedModel = {
    id: body.id,
    name: body.name || body.id,
    provider: body.provider,
    enabled: body.enabled ?? true,
    contextWindow: body.contextWindow,
    costPer1kInput: body.costPer1kInput,
    costPer1kOutput: body.costPer1kOutput,
  };

  return NextResponse.json(await upsertCuratedModel(model), { status: 201 });
}

export async function DELETE(request: NextRequest) {
  const id = request.nextUrl.searchParams.get("id");
  if (!id) return NextResponse.json({ error: "id required" }, { status: 400 });
  await removeCuratedModel(id);
  return NextResponse.json({ ok: true });
}