CartDeliveryOrderInformationController.java

/**
 * Most of the code in the Qalingo project is copyrighted Hoteia and licensed
 * under the Apache License Version 2.0 (release version 0.8.0)
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *                   Copyright (c) Hoteia, 2012-2014
 * http://www.hoteia.com - http://twitter.com/hoteia - contact@hoteia.com
 *
 */
package org.hoteia.qalingo.web.mvc.controller.eco;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;

import org.hoteia.qalingo.core.ModelConstants;
import org.hoteia.qalingo.core.domain.Cart;
import org.hoteia.qalingo.core.domain.Customer;
import org.hoteia.qalingo.core.domain.CustomerAddress;
import org.hoteia.qalingo.core.domain.DeliveryMethod;
import org.hoteia.qalingo.core.domain.enumtype.FoUrls;
import org.hoteia.qalingo.core.web.mvc.form.CartForm;
import org.hoteia.qalingo.core.web.mvc.viewbean.CartViewBean;
import org.hoteia.qalingo.core.web.mvc.viewbean.CustomerAddressViewBean;
import org.hoteia.qalingo.core.web.resolver.RequestData;
import org.hoteia.qalingo.core.web.servlet.ModelAndViewThemeDevice;
import org.hoteia.qalingo.core.web.servlet.view.RedirectView;
import org.hoteia.qalingo.web.mvc.controller.AbstractMCommerceController;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.ModelAndView;

/**
 * 
 */
@Controller("cartDeliveryOrderInformationController")
public class CartDeliveryOrderInformationController extends AbstractMCommerceController {

    private final Logger logger = LoggerFactory.getLogger(getClass());

    @RequestMapping(value = FoUrls.CART_DELIVERY_URL, method = RequestMethod.GET)
    public ModelAndView displayOrderDelivery(final HttpServletRequest request, final HttpServletResponse response) throws Exception {
        ModelAndViewThemeDevice modelAndView = new ModelAndViewThemeDevice(getCurrentVelocityPath(request), FoUrls.CART_DELIVERY.getVelocityPage());

        // SANITY CHECK
        final RequestData requestData = requestUtil.getRequestData(request);
        final Cart currentCart = requestData.getCart();
        if (currentCart == null || currentCart.getTotalCartItems() == 0) {
            return new ModelAndView(new RedirectView(urlService.generateRedirectUrl(FoUrls.CART_DETAILS, requestUtil.getRequestData(request))));
        }

        final CartViewBean cartViewBean = frontofficeViewBeanFactory.buildViewBeanCart(requestUtil.getRequestData(request), currentCart);
        modelAndView.addObject(ModelConstants.CART_VIEW_BEAN, cartViewBean);

        modelAndView.addObject(ModelConstants.CHECKOUT_STEP, 3);

        modelAndView.addObject(ModelConstants.CART_FORM, formFactory.buildCartForm(requestData));

        overrideDefaultPageTitle(request, modelAndView, FoUrls.CART_DELIVERY.getKey());

        return modelAndView;
    }

    @RequestMapping(value = FoUrls.CART_DELIVERY_URL, method = RequestMethod.POST)
    public ModelAndView submitOrderDelivery(HttpServletRequest request, HttpServletResponse response, CartForm cartForm, BindingResult result, Model model) throws Exception {

        final RequestData requestData = requestUtil.getRequestData(request);

        // SANITY CHECK
        final Cart currentCart = requestData.getCart();
        if(currentCart == null){
            return new ModelAndView(new RedirectView(urlService.generateRedirectUrl(FoUrls.HOME, requestUtil.getRequestData(request))));
        }

        if (currentCart.getTotalCartItems() == 0) {
            return new ModelAndView(new RedirectView(urlService.generateRedirectUrl(FoUrls.CART_DETAILS, requestUtil.getRequestData(request))));
        }

        if (result.hasErrors()) {
            return displayOrderDelivery(request, response);
        }

        Set<DeliveryMethod> deliveryMethods = currentCart.getDeliveryMethods();
        if (deliveryMethods == null || deliveryMethods.size() == 0 || cartForm.getShippingAddressId() == null) {
            addSessionErrorMessage(request, "DELIVERY");
            return displayOrderDelivery(request, response);
        }

        String shippingAddressId = cartForm.getShippingAddressId();
        String billingAddressId = cartForm.getBillingAddressId();
        if (billingAddressId == null) {
            billingAddressId = shippingAddressId;
        }
        webManagementService.updateCartAddresses(requestData, Long.parseLong(billingAddressId), Long.parseLong(shippingAddressId));

        final String urlRedirect = urlService.generateRedirectUrl(FoUrls.CART_ORDER_PAYMENT, requestUtil.getRequestData(request));
        return new ModelAndView(new RedirectView(urlRedirect));
    }

    @ModelAttribute(ModelConstants.BILLING_ADDRESSES_VIEW_BEAN)
    public List<CustomerAddressViewBean> getBillingAddresses(HttpServletRequest request) {
        List<CustomerAddressViewBean> addressesValues = new ArrayList<CustomerAddressViewBean>();
        try {
            final RequestData requestData = requestUtil.getRequestData(request);
            final Customer customer = requestData.getCustomer();
            Set<CustomerAddress> addresses = customer.getAddresses();
            for (final CustomerAddress customerAddress : addresses) {
                addressesValues.add(frontofficeViewBeanFactory.buildViewBeanCustomerAddress(requestData, customerAddress));
            }

            Collections.sort(addressesValues, new Comparator<CustomerAddressViewBean>() {
                @Override
                public int compare(CustomerAddressViewBean o1, CustomerAddressViewBean o2) {
                    return o1.getAddressName().compareTo(o2.getAddressName());
                }
            });
        } catch (Exception e) {
            logger.error("", e);
        }
        return addressesValues;
    }
    
    @ModelAttribute(ModelConstants.SHIPPING_ADDRESSES_VIEW_BEAN)
    public List<CustomerAddressViewBean> getshippingAddresses(HttpServletRequest request) {
        List<CustomerAddressViewBean> addressesValues = new ArrayList<CustomerAddressViewBean>();
        try {
            final RequestData requestData = requestUtil.getRequestData(request);
            final Customer customer = requestData.getCustomer();
            Set<CustomerAddress> addresses = customer.getAddresses();
            for (final CustomerAddress customerAddress : addresses) {
                addressesValues.add(frontofficeViewBeanFactory.buildViewBeanCustomerAddress(requestData, customerAddress));
            }

            Collections.sort(addressesValues, new Comparator<CustomerAddressViewBean>() {
                @Override
                public int compare(CustomerAddressViewBean o1, CustomerAddressViewBean o2) {
                    return o1.getAddressName().compareTo(o2.getAddressName());
                }
            });
        } catch (Exception e) {
            logger.error("", e);
        }
        return addressesValues;
    }

}