BookingInfoServlet.java

package Servlet;

import jakarta.servlet.ServletException;
import jakarta.servlet.annotation.WebServlet;
import jakarta.servlet.http.*;
import model.Booking;
import Service.BookingService;
import Service.BookingInfoService;

import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;

@WebServlet("/bookingInfo")
public class BookingInfoServlet extends HttpServlet {
    private BookingService bookingService;
    private BookingInfoService bookingInfoService;

    @Override
    public void init() throws ServletException {
        System.out.println("**************** Test output from BookingInfoServlet init...");
        try {
            bookingService = new BookingService(getServletContext());
            bookingInfoService = new BookingInfoService(bookingService);
            String filePath = bookingService.getFilePath();
            if (filePath == null) {
                System.err.println("Error: BookingService filePath is null. Ensure WEB-INF/bookingData/bookings.txt is accessible.");
                throw new ServletException("Initialization failed: File path is null");
            }
            System.out.println("BookingInfoServlet filePath: " + filePath);
            System.out.println("**************** BookingInfoServlet initialized successfully");
        } catch (Exception e) {
            System.err.println("Failed to initialize BookingInfoServlet: " + e.getMessage());
            e.printStackTrace();
            throw new ServletException("Initialization failed: " + e.getMessage(), e);
        }
    }

    @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 view your bookings.");
            request.getRequestDispatcher("/event/SignIN/signin.jsp").forward(request, response);
            return;
        }

        System.out.println("Handling GET request for BookingInfoServlet");
        try {
            List<Booking> allBookings = bookingService.readBookings();
            List<Booking> userBookings = allBookings.stream()
                    .filter(booking -> username.equalsIgnoreCase(booking.getUsername()))
                    .collect(Collectors.toList());
            System.out.println("Read " + allBookings.size() + " total bookings, filtered to " + userBookings.size() + " for user: " + username);
            System.out.println("User bookings: " + userBookings);
            request.setAttribute("bookings", userBookings);
            System.out.println("Forwarding to bookingInfo.jsp with bookings attribute set");
            request.getRequestDispatcher("/event/BookingInfo/bookingInfo.jsp").forward(request, response);
        } catch (IOException e) {
            System.err.println("Error reading bookings: " + e.getMessage());
            request.setAttribute("errorMessage", "Failed to load bookings: " + e.getMessage());
            request.getRequestDispatcher("/event/BookingInfo/bookingInfo.jsp").forward(request, response);
        } catch (Exception e) {
            System.err.println("Unexpected error in doGet: " + e.getMessage());
            e.printStackTrace();
            throw new ServletException("Error processing GET request", e);
        }
    }

    @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 manage your bookings.");
            request.getRequestDispatcher("/event/SignIN/signin.jsp").forward(request, response);
            return;
        }

        System.out.println("Handling POST request for BookingInfoServlet");
        String action = request.getParameter("action");
        System.out.println("Action received: " + action);

        int bookingId;
        try {
            bookingId = Integer.parseInt(request.getParameter("id"));
        } catch (NumberFormatException e) {
            request.setAttribute("errorMessage", "Invalid booking ID.");
            forwardWithUserBookings(request, response, username);
            return;
        }

        List<Booking> allBookings = bookingService.readBookings();
        Booking targetBooking = allBookings.stream()
                .filter(b -> b.getId() == bookingId)
                .findFirst()
                .orElse(null);

        if (targetBooking == null || (targetBooking.getUsername() != null && !username.equalsIgnoreCase(targetBooking.getUsername()))) {
            request.setAttribute("errorMessage", "You are not authorized to perform this action.");
            forwardWithUserBookings(request, response, username);
            return;
        }

        if ("update".equals(action)) {
            handleUpdateBooking(request, response, username);
        } else if ("cancel".equals(action)) {
            handleCancelBooking(request, response, username);
        } else if ("accept".equals(action)) {
            handleStatusChange(request, response, "Accepted", username);
        } else if ("decline".equals(action)) {
            handleStatusChange(request, response, "Declined", username);
        } else if ("filter".equals(action)) {
            handleFilterBookings(request, response, username);
        } else {
            System.out.println("Invalid action received");
            request.setAttribute("errorMessage", "Invalid action.");
            forwardWithUserBookings(request, response, username);
        }
    }

    private void handleUpdateBooking(HttpServletRequest request, HttpServletResponse response, String username)
            throws ServletException, IOException {
        System.out.println("Handling update booking request");
        try {
            int id = Integer.parseInt(request.getParameter("id"));
            String name = request.getParameter("name");
            String phone = request.getParameter("phone");
            String email = request.getParameter("email");
            String address = request.getParameter("address");
            String city = request.getParameter("city");
            String state = request.getParameter("state");
            String postal = request.getParameter("postal");
            String sessionType = request.getParameter("sessionType");
            String sessionDate = request.getParameter("sessionDate");
            String startTime = request.getParameter("startTime");
            String endTime = request.getParameter("endTime");
            String numPeopleStr = request.getParameter("numPeople");
            String ages = request.getParameter("ages");
            String notes = request.getParameter("notes");
            String extras = request.getParameter("extras");
            String status = request.getParameter("status");

            // Log all received parameters
            System.out.println("Received parameters for update - ID: " + id + ", Name: " + name + ", Email: " + email +
                    ", SessionType: " + sessionType + ", NumPeople: " + numPeopleStr + ", Status: " + status);

            // Validate required fields
            if (name == null || name.trim().isEmpty() ||
                    email == null || email.trim().isEmpty() ||
                    sessionType == null || sessionType.trim().isEmpty() ||
                    numPeopleStr == null || numPeopleStr.trim().isEmpty()) {
                System.out.println("Validation failed: Required fields missing or invalid - Name: " + name +
                        ", Email: " + email + ", SessionType: " + sessionType + ", NumPeople: " + numPeopleStr);
                request.setAttribute("errorMessage", "Required fields (name, email, session type, number of people) are missing or invalid.");
                forwardWithUserBookings(request, response, username);
                return;
            }

            int numPeople;
            try {
                numPeople = Integer.parseInt(numPeopleStr);
                if (numPeople < 1) {
                    System.out.println("Validation failed: Number of people must be at least 1, got: " + numPeople);
                    request.setAttribute("errorMessage", "Number of people must be at least 1.");
                    forwardWithUserBookings(request, response, username);
                    return;
                }
            } catch (NumberFormatException e) {
                System.err.println("Invalid number format for numPeople: " + numPeopleStr);
                request.setAttribute("errorMessage", "Invalid number of people.");
                forwardWithUserBookings(request, response, username);
                return;
            }

            // Create updated booking object
            Booking updatedBooking = new Booking(id, name, phone, email, address, city, state, postal,
                    sessionType, sessionDate, startTime, endTime, numPeople, ages, notes, extras, username);
            updatedBooking.setStatus(status != null ? status : "Pending");
            System.out.println("Created updated booking: " + updatedBooking);

            // Update the booking
            boolean updated = bookingInfoService.updateBooking(updatedBooking);
            if (updated) {
                System.out.println("Booking updated successfully for ID: " + id);
                request.setAttribute("successMessage", "Booking updated successfully!");
            } else {
                System.out.println("Booking not found for update, ID: " + id);
                request.setAttribute("errorMessage", "Booking not found.");
            }
        } catch (Exception e) {
            System.err.println("Error updating booking: " + e.getMessage() + " - Stack trace: " + e);
            request.setAttribute("errorMessage", "Failed to update booking: " + e.getMessage());
        }
        forwardWithUserBookings(request, response, username);
    }

    private void handleCancelBooking(HttpServletRequest request, HttpServletResponse response, String username)
            throws ServletException, IOException {
        System.out.println("Handling cancel booking request");
        try {
            int id = Integer.parseInt(request.getParameter("id"));
            System.out.println("Cancelling booking ID: " + id);
            boolean removed = bookingInfoService.cancelBooking(id);
            if (removed) {
                System.out.println("Booking cancelled successfully for ID: " + id);
                request.setAttribute("successMessage", "Booking cancelled successfully!");
            } else {
                System.out.println("Booking not found for cancellation, ID: " + id);
                request.setAttribute("errorMessage", "Booking not found.");
            }
        } catch (Exception e) {
            System.err.println("Error cancelling booking: " + e.getMessage());
            request.setAttribute("errorMessage", "Failed to cancel booking: " + e.getMessage());
        }
        forwardWithUserBookings(request, response, username);
    }

    private void handleStatusChange(HttpServletRequest request, HttpServletResponse response, String status, String username)
            throws ServletException, IOException {
        System.out.println("Handling status change to: " + status);
        try {
            int id = Integer.parseInt(request.getParameter("id"));
            System.out.println("Changing status for booking ID: " + id);
            boolean updated = bookingInfoService.setBookingStatus(id, status);
            if (updated) {
                System.out.println("Booking status updated to " + status + " for ID: " + id);
                request.setAttribute("successMessage", "Booking " + status.toLowerCase() + " successfully!");
            } else {
                System.out.println("Booking not found for status change, ID: " + id);
                request.setAttribute("errorMessage", "Booking not found.");
            }
        } catch (Exception e) {
            System.err.println("Error changing booking status: " + e.getMessage());
            request.setAttribute("errorMessage", "Failed to change booking status: " + e.getMessage());
        }
        forwardWithUserBookings(request, response, username);
    }

    private void handleFilterBookings(HttpServletRequest request, HttpServletResponse response, String username)
            throws ServletException, IOException {
        System.out.println("Handling filter bookings request");
        try {
            String filterType = request.getParameter("filterType");
            String filterValue = request.getParameter("filterValue");
            String sessionType = request.getParameter("sessionType");
            List<Booking> allBookings = bookingService.readBookings();
            List<Booking> userBookings = allBookings.stream()
                    .filter(booking -> username.equalsIgnoreCase(booking.getUsername()))
                    .collect(Collectors.toList());
            List<Booking> filteredBookings;

            if ("name".equals(filterType)) {
                filteredBookings = userBookings.stream()
                        .filter(booking -> booking.filterBookings(filterValue))
                        .collect(Collectors.toList());
            } else if ("dateAndType".equals(filterType)) {
                filteredBookings = userBookings.stream()
                        .filter(booking -> booking.filterBookings(filterValue, sessionType))
                        .collect(Collectors.toList());
            } else {
                filteredBookings = userBookings;
            }

            request.setAttribute("bookings", filteredBookings);
            request.getRequestDispatcher("/event/BookingInfo/bookingInfo.jsp").forward(request, response);
        } catch (Exception e) {
            System.err.println("Error filtering bookings: " + e.getMessage());
            request.setAttribute("errorMessage", "Failed to filter bookings: " + e.getMessage());
            forwardWithUserBookings(request, response, username);
        }
    }

    private void forwardWithUserBookings(HttpServletRequest request, HttpServletResponse response, String username)
            throws ServletException, IOException {
        List<Booking> allBookings = bookingService.readBookings();
        List<Booking> userBookings = allBookings.stream()
                .filter(booking -> username.equalsIgnoreCase(booking.getUsername()))
                .collect(Collectors.toList());
        System.out.println("Forwarding with user bookings: " + userBookings.size() + " bookings for user: " + username);
        request.setAttribute("bookings", userBookings);
        request.getRequestDispatcher("/event/BookingInfo/bookingInfo.jsp").forward(request, response);
    }
}