/*

  * Copyright (c) 2003-2011, Simon Brown

  * All rights reserved.

  *

  * Redistribution and use in source and binary forms, with or without

  * modification, are permitted provided that the following conditions are met:

  *

  *   - Redistributions of source code must retain the above copyright

  *     notice, this list of conditions and the following disclaimer.

  *

  *   - Redistributions in binary form must reproduce the above copyright

  *     notice, this list of conditions and the following disclaimer in

  *     the documentation and/or other materials provided with the

  *     distribution.

  *

  *   - Neither the name of Pebble nor the names of its contributors may

  *     be used to endorse or promote products derived from this software

  *     without specific prior written permission.

  *

  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE

  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

  * POSSIBILITY OF SUCH DAMAGE.

  */

 package net.sourceforge.pebble.web.action;

 import com.maxmind.geoip.LookupService;

 import net.sourceforge.pebble.logging.Log;

 import net.sourceforge.pebble.logging.LogEntry;

 import net.sourceforge.pebble.logging.Request;

 import net.sourceforge.pebble.web.view.View;

 import net.sourceforge.pebble.web.view.impl.CountriesView;

 import net.sourceforge.pebble.domain.Blog;

 import net.sourceforge.pebble.Constants;

 import javax.servlet.ServletException;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import java.io.IOException;

 import java.util.*;

 /**

  * Gets the visitor country information for the specified time period.

  *

  * @author    Simon Brown

  */

 public class ViewCountriesAction extends AbstractLogAction {

   /**

    * Peforms the processing associated with this action.

    *

    * @param request  the HttpServletRequest instance

    * @param response the HttpServletResponse instance

    * @return the name of the next view

    */

   public View process(HttpServletRequest request, HttpServletResponse response) throws ServletException {

     Blog blog = (Blog)getModel().get(Constants.BLOG_KEY);

     Log log = getLog(request, response);

     Set<String> countries = new TreeSet<String>(new Comparator<String>() {

       public int compare(String s1, String s2) {

         return s1 != null ? s1.compareToIgnoreCase(s2) : -1;

       }

     });

     Map<String,Integer> consolidatedCountries = new HashMap<String,Integer>();

     Map<String,Integer> countriesForNewsFeeds = new HashMap<String,Integer>();

     Map<String,Integer> countriesForPageViews = new HashMap<String,Integer>();

     Map<String,Integer> countriesForFileDownloads = new HashMap<String,Integer>();

     LookupService lookupService = null;

     try {

       String filename = getClass().getResource("/geo-ip.dat").toExternalForm().substring(5);

       lookupService = new LookupService(filename, LookupService.GEOIP_MEMORY_CACHE);

       for (LogEntry logEntry : log.getLogEntries()) {

         String country = lookupService.getCountry(logEntry.getHost()).getName();

         countries.add(country);

         register(country, countriesForNewsFeeds);

         register(country, countriesForPageViews);

         register(country, countriesForFileDownloads);

         register(country, consolidatedCountries);

         Request req = new Request(logEntry.getRequestUri(), blog);

         if (req.isNewsFeed()) {

           increment(country, countriesForNewsFeeds);

           increment(country, consolidatedCountries);

         } else if (req.isPageView()) {

           increment(country, countriesForPageViews);

           increment(country, consolidatedCountries);

         } else if (req.isFileDownload()) {

           increment(country, countriesForFileDownloads);

           increment(country, consolidatedCountries);

         }

       }

     } catch (IOException ioe) {

       throw new ServletException(ioe);

     } finally {

       if (lookupService != null) {

         lookupService.close();

       }

     }

     getModel().put("logAction", "viewCountries");

     getModel().put("countries", countries);

     getModel().put("consolidatedCountries", consolidatedCountries);

     getModel().put("countriesForNewsFeeds", countriesForNewsFeeds);

     getModel().put("countriesForPageViews", countriesForPageViews);

     getModel().put("countriesForFileDownloads", countriesForFileDownloads);

     return new CountriesView();

   }

   private void register(String country, Map<String,Integer> map) {

     Integer count = map.get(country);

     if (count == null) {

       count = 0;

     }

     map.put(country, count);

   }

   private void increment(String country, Map<String,Integer> map) {

     Integer count = map.get(country);

     count = count + 1;

     map.put(country, count);

   }

 }