/*
    * Licensed under the GPL License. You may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *   https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
    *
    * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
    * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
    * PURPOSE.
   */
  package psiprobe.controllers;
  
  import java.io.BufferedReader;
  import java.io.ByteArrayInputStream;
  import java.io.IOException;
  import java.io.InputStreamReader;
  import java.net.InetAddress;
  import java.net.UnknownHostException;
  import java.nio.charset.StandardCharsets;
  import java.util.ArrayList;
  import java.util.List;
  
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.beans.factory.annotation.Value;
  import org.springframework.stereotype.Controller;
  import org.springframework.web.bind.ServletRequestUtils;
  import org.springframework.web.bind.annotation.RequestMapping;
  import org.springframework.web.servlet.ModelAndView;
  import org.springframework.web.servlet.mvc.ParameterizableViewController;
  
  import psiprobe.tools.Whois;
  
  /**
   * The Class WhoisController.
   */
  @Controller
  public class WhoisController extends ParameterizableViewController {
  
    /** The Constant logger. */
    private static final Logger logger = LoggerFactory.getLogger(WhoisController.class);
  
    /** The lookup timeout. */
    private long lookupTimeout;
  
    /** The default server. */
    private String defaultServer;
  
    /** The default port. */
    private int defaultPort;
  
    /**
     * Gets the lookup timeout.
     *
     * @return the lookup timeout
     */
    public long getLookupTimeout() {
      return lookupTimeout;
    }
  
    /**
     * Sets the lookup timeout.
     *
     * @param lookupTimeout the new lookup timeout in seconds
     */
    @Value("5")
    public void setLookupTimeout(long lookupTimeout) {
      this.lookupTimeout = lookupTimeout;
    }
  
    /**
     * Gets the default server.
     *
     * @return the default server
     */
    public String getDefaultServer() {
      return defaultServer;
    }
  
    /**
     * Sets the default server.
     *
     * @param defaultServer the new default server
     */
    @Value("whois.arin.net")
    public void setDefaultServer(String defaultServer) {
      this.defaultServer = defaultServer;
    }
  
    /**
     * Gets the default port.
     *
     * @return the default port
     */
    public int getDefaultPort() {
      return defaultPort;
   }
 
   /**
    * Sets the default port.
    *
    * @param defaultPort the new default port
    */
   @Value("43")
   public void setDefaultPort(int defaultPort) {
     this.defaultPort = defaultPort;
   }
 
   @RequestMapping(path = "/whois.ajax")
   @Override
   public ModelAndView handleRequest(HttpServletRequest request, HttpServletResponse response)
       throws Exception {
     return super.handleRequest(request, response);
   }
 
   @Override
   protected ModelAndView handleRequestInternal(HttpServletRequest request,
       HttpServletResponse response) throws Exception {
 
     boolean timeout = false;
 
     String ipAddress = ServletRequestUtils.getStringParameter(request, "ip");
 
     Whois.Response wh = null;
     try {
       wh = Whois.lookup(getDefaultServer(), getDefaultPort(), ipAddress, getLookupTimeout());
     } catch (IOException e) {
       timeout = true;
       logger.trace("", e);
     }
 
     List<String> lines = null;
     if (wh != null) {
       lines = new ArrayList<>(50);
       try (BufferedReader br = new BufferedReader(new InputStreamReader(
           new ByteArrayInputStream(wh.getSummary().getBytes(StandardCharsets.UTF_8)),
           StandardCharsets.UTF_8))) {
         String line;
         while ((line = br.readLine()) != null) {
           lines.add(line);
         }
       }
     }
 
     String reverseName = null;
     if (ipAddress != null) {
       try {
         reverseName = InetAddress.getByName(ipAddress).getCanonicalHostName();
       } catch (UnknownHostException e) {
         logger.error("could not run a DNS query on {}", ipAddress);
         logger.trace("", e);
       }
     }
     return new ModelAndView(getViewName(), "result", lines).addObject("timeout", timeout)
         .addObject("whoisServer",
             wh != null ? wh.getServer() + ":" + wh.getPort() : defaultServer + ":" + defaultPort)
         .addObject("domainName", reverseName);
   }
 
   @Value("ajax/whois")
   @Override
   public void setViewName(String viewName) {
     super.setViewName(viewName);
   }
 
 }