import { NextResponse } from 'next/server';
import { getServerSession } from 'next-auth';
import prisma from '@/lib/prisma';

// GET - Fetch training programs
export async function GET(request: Request) {
  try {
    const session = await getServerSession();
    if (!session) {
      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
    }

    const { searchParams } = new URL(request.url);
    const status = searchParams.get('status');

    const where: any = {};
    if (status && status !== 'All') where.status = status;

    const trainingPrograms = await prisma.trainingProgram.findMany({
      where,
      orderBy: { startDate: 'asc' }
    });

    return NextResponse.json(trainingPrograms);
  } catch (error) {
    console.error('Error fetching training programs:', error);
    return NextResponse.json({ error: 'Internal Server Error' }, { status: 500 });
  }
}

// POST - Create training program
export async function POST(request: Request) {
  try {
    const session = await getServerSession();
    if (!session || ((session.user as any)?.role !== 'admin' && (session.user as any)?.role !== 'hr')) {
      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
    }

    const body = await request.json();
    const { title, description, startDate, endDate, capacity, location, dept, trainer, cost } = body;

    const trainingProgram = await prisma.trainingProgram.create({
      data: {
        title,
        dept: dept || 'General',
        trainer: trainer || (session.user as any)?.name || 'TBD',
        cost: cost || 0,
        description,
        startDate: new Date(startDate),
        endDate: new Date(endDate),
        capacity,
        venue: location,
        status: 'Upcoming'
      }
    });

    return NextResponse.json(trainingProgram, { status: 201 });
  } catch (error) {
    console.error('Error creating training program:', error);
    return NextResponse.json({ error: 'Internal Server Error' }, { status: 500 });
  }
}