/*
    * Copyright (C) 2007 Alf Mikula
    * 
    * This file is part of PromoteGo.
    *
    * PromoteGo is free software: you can redistribute it and/or modify
    * it under the terms of the GNU General Public License as published by
    * the Free Software Foundation, either version 3 of the License, or
    * (at your option) any later version.
   *
   * PromoteGo is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with PromoteGo.  If not, see <http://www.gnu.org/licenses/>.
   */
  package org.promotego.controllers;
  
  import java.util.Collections;
  import java.util.HashMap;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  
  import org.promotego.api.geocoder.util.DistanceUtil;
  import org.promotego.beans.Address;
  import org.promotego.beans.Location;
  import org.promotego.beans.LocationType;
  import org.promotego.beans.User;
  import org.promotego.beans.UserHolder;
  import org.promotego.dao.interfaces.AddressDao;
  import org.promotego.dao.interfaces.LocationDao;
  import org.promotego.dao.interfaces.LocationTypeDao;
  import org.promotego.editors.BeanIdEditor;
  import org.promotego.formbackingbeans.RadiusSearchBean;
  import org.promotego.viewbeans.LocationViewBean;
  import org.springframework.beans.factory.annotation.Required;
  import org.springframework.transaction.annotation.Transactional;
  import org.springframework.validation.BindException;
  import org.springframework.web.bind.ServletRequestDataBinder;
  import org.springframework.web.servlet.ModelAndView;
  import org.springframework.web.servlet.mvc.SimpleFormController;
  
  public class LocationSearchController extends SimpleFormController
  {
      // TODO support search Radii in miles and kilometers
      private static int [] s_searchRadii = {1,5,10,25,50,100};
      
      private LocationDao m_locationDao;
      private AddressDao m_addressDao;
      private LocationTypeDao m_locationTypeDao;
      private UserHolder m_userHolder;
      
      @Transactional
      @Override
      public ModelAndView onSubmit(Object command)
      {
          RadiusSearchBean searchParams = (RadiusSearchBean)command;
          assert m_userHolder.getUser() != null : "User object may not be null";
  
          Address address = searchParams.getAddress();
          int searchRadius = searchParams.getMiles();
          
          List<Location> locations = m_locationDao.findNear(address.getGeolocation(), searchRadius);
          
          List<LocationViewBean> viewLocations = new LinkedList<LocationViewBean>();
          
          // Calculate distance for all locations
          for(Location thisLocation : locations)
          {
              LocationViewBean viewLocation = new LocationViewBean();
              viewLocation.setLocation(thisLocation);
              double distance = DistanceUtil.calculateDistance(thisLocation.getAddress().getGeolocation(), address.getGeolocation());
              distance = Math.floor(distance*10+0.5)/10.0;
              viewLocation.setDistance(distance);
              viewLocations.add(viewLocation);
          }
          
          // Sort locations by distance
          Collections.sort(viewLocations);
          
          Map<String,Object> myModel = new HashMap<String, Object>();
          
          myModel.put("locations", viewLocations);
          myModel.put("address", address);
          myModel.put("searchRadius", searchRadius);
          
          return new ModelAndView(getSuccessView(), myModel);
      }
  
      @SuppressWarnings("unchecked")
      @Transactional(readOnly=true)
      @Override
      protected ModelAndView showForm(HttpServletRequest request, HttpServletResponse response, BindException errors, Map controlModel) throws Exception
     {
         Map <String,Object>myControlModel;
         
         // Create a new Map, leaving the original untouched.
         if (controlModel == null)
         {
             myControlModel = new HashMap<String,Object>();
         }
         else
         {
             myControlModel = new HashMap<String,Object>(controlModel);
         }
         
         // Populate location types for filtering functionality
         List<LocationType> locationTypes = m_locationTypeDao.findAll();
         Collections.sort(locationTypes);
         myControlModel.put("locationTypes", locationTypes);
         
         // Populate user's addresses for searches near an address
         User theUser = m_userHolder.getUser();
         assert theUser != null : "User object may not be null";
         myControlModel.put("addresses", theUser.getAddresses());
         
         // Populate allowed search radii
         myControlModel.put("searchRadii", s_searchRadii);
         
         return super.showForm(request, response, errors, myControlModel);
     }
     
     @Override
     protected void initBinder(HttpServletRequest request, ServletRequestDataBinder binder) throws Exception
     {
         binder.registerCustomEditor(LocationType.class, new BeanIdEditor<LocationType>(m_locationTypeDao));
         binder.registerCustomEditor(Address.class, new BeanIdEditor<Address>(m_addressDao));
         
         super.initBinder(request, binder);
     }
     
     @Required
     public void setUserHolder(UserHolder theUserHolder)
     {
         m_userHolder = theUserHolder;
     }
     
     @Required
     public void setLocationDao(LocationDao locationDao)
     {
         m_locationDao = locationDao;
     }
 
     @Required
     public void setLocationTypeDao(LocationTypeDao locationTypeDao)
     {
         m_locationTypeDao = locationTypeDao;
     }
 
     @Required
     public void setAddressDao(AddressDao addressDao)
     {
         m_addressDao = addressDao;
     }
 }