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

// GET - Fetch disciplinary cases
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 disciplinaryCases = await prisma.disciplinaryCase.findMany({
      where,
      orderBy: { createdAt: 'desc' }
    });

    return NextResponse.json(disciplinaryCases);
  } catch (error) {
    console.error('Error fetching disciplinary cases:', error);
    return NextResponse.json({ error: 'Internal Server Error' }, { status: 500 });
  }
}

// POST - Create disciplinary case
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 { employeeId, employeeName, dept, offenceType, description, offenceDate } = body;

    const caseNumber = `DISC-${Date.now()}-${Math.floor(Math.random() * 1000)}`;

    const disciplinaryCase = await prisma.disciplinaryCase.create({
      data: {
        caseNumber,
        employeeId,
        employeeName: employeeName || '',
        dept: dept || '',
        offenceType,
        description,
        offenceDate: new Date(offenceDate),
        status: 'Open',
        createdBy: (session.user as any)?.name || (session.user as any)?.email || 'System'
      }
    });

    return NextResponse.json(disciplinaryCase, { status: 201 });
  } catch (error) {
    console.error('Error creating disciplinary case:', error);
    return NextResponse.json({ error: 'Internal Server Error' }, { status: 500 });
  }
}