import { NextRequest, NextResponse } from 'next/server';
import { getBannedSymbols, setBannedSymbol } from '@/lib/db';

type Params = { params: Promise<{ id: string }> };

/** GET /api/firms/[id]/banned-symbols — list banned symbols for a firm */
export async function GET(_req: NextRequest, { params }: Params) {
    const { id: idStr } = await params;
    const firmId = parseInt(idStr, 10);
    if (isNaN(firmId)) return NextResponse.json({ error: 'Invalid id' }, { status: 400 });
    return NextResponse.json(getBannedSymbols(firmId));
}

/** PATCH /api/firms/[id]/banned-symbols — { symbol, banned } to ban or unban */
export async function PATCH(req: NextRequest, { params }: Params) {
    const { id: idStr } = await params;
    const firmId = parseInt(idStr, 10);
    if (isNaN(firmId)) return NextResponse.json({ error: 'Invalid id' }, { status: 400 });

    const { symbol, banned } = await req.json() as { symbol: string; banned: boolean };
    if (!symbol) return NextResponse.json({ error: 'symbol required' }, { status: 400 });

    setBannedSymbol(firmId, symbol, banned);
    return NextResponse.json({ firmId, symbol, banned });
}