import type { RequestHandler } from '@sveltejs/kit';
import { Payment } from '../../../../../models/Payment';
import { PaymentSplit } from '../../../../../models/PaymentSplit';
import { dbConnect, dbDisconnect } from '../../../../../utils/db';
import { error, json } from '@sveltejs/kit';

export const GET: RequestHandler = async ({ params, locals }) => {
  const auth = await locals.auth();
  if (!auth || !auth.user?.nickname) {
    throw error(401, 'Not logged in');
  }

  const { id } = params;

  await dbConnect();
  
  try {
    const payment = await Payment.findById(id).populate('splits').lean();
    
    if (!payment) {
      throw error(404, 'Payment not found');
    }

    return json({ payment });
  } catch (e) {
    if (e.status === 404) throw e;
    throw error(500, 'Failed to fetch payment');
  } finally {
    await dbDisconnect();
  }
};

export const PUT: RequestHandler = async ({ params, request, locals }) => {
  const auth = await locals.auth();
  if (!auth || !auth.user?.nickname) {
    throw error(401, 'Not logged in');
  }

  const { id } = params;
  const data = await request.json();

  await dbConnect();
  
  try {
    const payment = await Payment.findById(id);
    
    if (!payment) {
      throw error(404, 'Payment not found');
    }

    if (payment.createdBy !== auth.user.nickname) {
      throw error(403, 'Not authorized to edit this payment');
    }

    const updatedPayment = await Payment.findByIdAndUpdate(
      id,
      { 
        title: data.title,
        description: data.description,
        amount: data.amount,
        paidBy: data.paidBy,
        date: data.date ? new Date(data.date) : payment.date,
        image: data.image,
        category: data.category || payment.category,
        splitMethod: data.splitMethod
      },
      { new: true }
    );

    if (data.splits) {
      await PaymentSplit.deleteMany({ paymentId: id });
      
      const splitPromises = data.splits.map((split: any) => {
        return PaymentSplit.create({
          paymentId: id,
          username: split.username,
          amount: split.amount,
          proportion: split.proportion,
          personalAmount: split.personalAmount
        });
      });

      await Promise.all(splitPromises);
    }

    return json({ success: true, payment: updatedPayment });
  } catch (e) {
    if (e.status) throw e;
    throw error(500, 'Failed to update payment');
  } finally {
    await dbDisconnect();
  }
};

export const DELETE: RequestHandler = async ({ params, locals }) => {
  const auth = await locals.auth();
  if (!auth || !auth.user?.nickname) {
    throw error(401, 'Not logged in');
  }

  const { id } = params;

  await dbConnect();
  
  try {
    const payment = await Payment.findById(id);
    
    if (!payment) {
      throw error(404, 'Payment not found');
    }

    if (payment.createdBy !== auth.user.nickname) {
      throw error(403, 'Not authorized to delete this payment');
    }

    await PaymentSplit.deleteMany({ paymentId: id });
    await Payment.findByIdAndDelete(id);

    return json({ success: true });
  } catch (e) {
    if (e.status) throw e;
    throw error(500, 'Failed to delete payment');
  } finally {
    await dbDisconnect();
  }
};