com.havoc.hotel.admin.controller.BookingAdminController.java Source code

Introduction

Here is the source code for com.havoc.hotel.admin.controller.BookingAdminController.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.havoc.hotel.admin.controller;

import com.havoc.hotel.admin.dao.BookingDAO;
import com.havoc.hotel.admin.dao.CustomerDAO;
import com.havoc.hotel.admin.dao.RoomDAO;
import com.havoc.hotel.admin.entity.Booking;
import com.havoc.hotel.admin.entity.Customer;
import com.havoc.hotel.admin.entity.Room;
import com.havoc.hotel.util.BookingPdf;
import com.itextpdf.text.DocumentException;
import java.io.IOException;
import java.sql.SQLException;
import java.text.ParseException;
import javax.servlet.http.HttpServletRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.ModelAndView;

/**
 *
 * @author HYOJU
 */
@Controller
@RequestMapping(value = "/admin/booking")
public class BookingAdminController {

    @Autowired
    private BookingDAO bookingDAO;
    @Autowired
    private CustomerDAO customerDAO;
    @Autowired
    private RoomDAO roomDAO;

    @RequestMapping(method = RequestMethod.GET)
    public String index(ModelMap map) throws SQLException {
        map.addAttribute("Booking", bookingDAO.getALL());
        return "admin/booking/index";
    }

    @RequestMapping(value = "/add", method = RequestMethod.GET)
    public ModelAndView add(HttpServletRequest request) throws SQLException {
        ModelAndView mv = new ModelAndView("/admin/booking/add");
        String customer_id = null;
        String room_id = null;
        if (request.getParameter("customerId") != null && !request.getParameter("customerId").equals("")) {
            customer_id = request.getParameter("customerId");
        }
        if (request.getParameter("roomId") != null && !request.getParameter("roomId").equals("")) {
            room_id = request.getParameter("roomId");
        }
        mv.addObject("Booking", new Booking());
        mv.addObject("Customer", customerDAO.getALL());
        mv.addObject("Room", roomDAO.getALL());

        return mv;
    }

    @RequestMapping(value = "/edit/{bookingId}", method = RequestMethod.GET)
    public ModelAndView edit(@PathVariable("bookingId") int bookingId) throws SQLException {
        ModelAndView mv = new ModelAndView("admin/booking/edit");
        mv.addObject("Booking", bookingDAO.getById(bookingId));
        return mv;
    }

    @RequestMapping(value = "/checkin/{bookingId}", method = RequestMethod.GET)
    public ModelAndView checkin(@PathVariable("bookingId") int bookingId) throws SQLException {
        ModelAndView mv = new ModelAndView("admin/checkin/index");
        mv.addObject("booking", bookingDAO.checkin(bookingId));
        return mv;
    }

    @RequestMapping(value = "/new", method = RequestMethod.GET)
    public ModelAndView newbooking() throws SQLException {
        ModelAndView mv = new ModelAndView("admin/booking/newbooking");
        mv.addObject("Booking", bookingDAO.getLast());
        return mv;
    }

    @RequestMapping(value = "/delete/{bookingId}", method = RequestMethod.GET)
    public String delete(@PathVariable("bookingId") int bookingId) throws SQLException {
        bookingDAO.delete(bookingId);
        return "redirect:/admin/booking";
    }

    @RequestMapping(value = "/save", method = RequestMethod.POST)
    public String save(@ModelAttribute("Booking") Booking booking, BindingResult result, HttpServletRequest request)
            throws SQLException, ParseException, DocumentException, IOException {
        Customer customer = customerDAO.getById(booking.getCustomer().getCustomerId());
        Room room = roomDAO.getById(booking.getRoom().getRoomId());
        long totalNights = (booking.getCheckoutDate().getTime() - booking.getCheckinDate().getTime())
                / (24 * 60 * 60 * 1000);
        System.out.println(totalNights);
        booking.setTotalNights((int) totalNights);
        int totalPrice = (booking.getTotalNights() * booking.getRoom().getRoomPrice());
        booking.setTotalPrice(totalPrice);
        System.out.println(totalPrice);

        System.out.println("check" + booking);
        try {
            booking.setCustomer(customer);
            booking.setRoom(room);
            if (booking.getBookingId() == 0) {

                booking.setPdf(BookingPdf.generateBookingPDF(booking));
                bookingDAO.insert(booking);

            } else {
                bookingDAO.update(booking);
            }
        } catch (SQLException ex) {
        }
        return "redirect:/admin/booking";
    }
}