PaymentServlet.java

package Servlet;

import model.*;
import Service.BookingService;
import Service.BookingInfoService;
import model.Booking;
import jakarta.servlet.ServletException;
import jakarta.servlet.annotation.WebServlet;
import jakarta.servlet.http.HttpServlet;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import java.io.IOException;

@WebServlet("/event/Payment/process")
public class PaymentServlet extends HttpServlet {
    private BookingService bookingService;

    @Override
    public void init() throws ServletException {
        bookingService = new BookingService(getServletContext());
    }

    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        String username = (String) request.getSession().getAttribute("loggedInUser");
        if (username == null) {
            request.setAttribute("errorMessage", "Please sign in to make a payment.");
            request.getRequestDispatcher("/event/SignIN/signin.jsp").forward(request, response);
            return;
        }

        String bookingId = request.getParameter("bookingId");
        if (bookingId == null || bookingId.trim().isEmpty()) {
            request.setAttribute("errorMessage", "Booking ID is missing.");
            request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
            return;
        }

        try {
            Booking booking = bookingService.readBookings().stream()
                    .filter(b -> b.getId() == Integer.parseInt(bookingId))
                    .findFirst()
                    .orElse(null);
            if (booking == null) {
                request.setAttribute("errorMessage", "Booking not found for ID: " + bookingId);
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                return;
            }
            if (!username.equals(booking.getUsername())) {
                request.setAttribute("errorMessage", "You are not authorized to pay for this booking.");
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                return;
            }

            PaymentDAO dao = new PaymentDAO(getServletContext());
            if (!dao.getPaymentsByBookingId(bookingId).isEmpty()) {
                request.setAttribute("errorMessage", "Payment already exists for this booking.");
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                return;
            }

            request.setAttribute("bookingId", bookingId);
            request.setAttribute("sessionType", booking.getSessionType());
            request.setAttribute("extras", booking.getExtras());
            request.setAttribute("numPeople", booking.getNumPeople());
            request.setAttribute("sessionDate", booking.getSessionDate());
            request.setAttribute("startTime", booking.getStartTime());
            request.setAttribute("endTime", booking.getEndTime());
            request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
        } catch (Exception e) {
            request.setAttribute("errorMessage", "Error retrieving booking: " + e.getMessage());
            request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
        }
    }

    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        String username = (String) request.getSession().getAttribute("loggedInUser");
        if (username == null) {
            request.setAttribute("errorMessage", "Please sign in to make a payment.");
            request.getRequestDispatcher("/event/SignIN/signin.jsp").forward(request, response);
            return;
        }

        try {
            String bookingId = request.getParameter("bookingId");
            String paymentType = request.getParameter("paymentType");
            Booking booking = bookingService.readBookings().stream()
                    .filter(b -> b.getId() == Integer.parseInt(bookingId))
                    .findFirst()
                    .orElse(null);
            if (booking == null) {
                request.setAttribute("errorMessage", "Booking not found.");
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                return;
            }

            PaymentDAO dao = new PaymentDAO(getServletContext());
            if (!dao.getPaymentsByBookingId(bookingId).isEmpty()) {
                request.setAttribute("errorMessage", "Payment already exists for this booking.");
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                return;
            }

            double amount = Payment.calculateFees(booking.getSessionType(), booking.getExtras(), paymentType);
            String transactionId = "TXN" + System.currentTimeMillis();
            System.out.println("Transaction ID: " + transactionId);

            Payment payment;
            if ("online".equals(paymentType)) {
                String cardNumber = request.getParameter("cardNumber");
                String cardHolder = request.getParameter("cardHolder");
                if (cardNumber == null || cardHolder == null || cardNumber.trim().isEmpty() || cardHolder.trim().isEmpty()) {
                    request.setAttribute("errorMessage", "Card details required.");
                    request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                    return;
                }
                payment = new OnlinePayment(transactionId, amount, bookingId, username, cardNumber, cardHolder);
            } else {
                String receivedBy = request.getParameter("receivedBy");
                if (receivedBy == null || receivedBy.trim().isEmpty()) {
                    request.setAttribute("errorMessage", "Receiver name required.");
                    request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
                    return;
                }
                payment = new CashPayment(transactionId, amount, bookingId, username, receivedBy);
            }

            if (payment.processPayment()) {
                dao.savePayment(payment);
                booking.setStatus("Paid");
                new BookingInfoService(bookingService).updateBooking(booking);
                request.setAttribute("payment", payment);
                request.getRequestDispatcher("/event/Payment/invoice.jsp").forward(request, response);
            } else {
                request.setAttribute("errorMessage", "Payment failed.");
                request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
            }
        } catch (Exception e) {
            request.setAttribute("errorMessage", "Error: " + e.getMessage());
            request.getRequestDispatcher("/event/Payment/payment.jsp").forward(request, response);
        }
    }
}