package at.meikel.dmrl.server.httpclient;
   
   import java.io.IOException;
   import java.io.InputStream;
   import java.util.Date;
   
   import org.apache.http.HttpEntity;
   import org.apache.http.HttpHost;
   import org.apache.http.StatusLine;
  import org.apache.http.client.ClientProtocolException;
  import org.apache.http.client.config.RequestConfig;
  import org.apache.http.client.methods.CloseableHttpResponse;
  import org.apache.http.client.methods.HttpGet;
  import org.apache.http.impl.client.CloseableHttpClient;
  import org.apache.http.impl.client.HttpClients;
  import org.apache.log4j.Logger;
  
  import at.meikel.dmrl.server.server.HttpDataRetriever;
  
  public class HttpDataRetrieverImpl implements HttpDataRetriever {
  
      private String proxyHost;
      private int proxyPort;
      private String proxyScheme;
      private String currentName = null;
      private Date currentDate = null;
      private CloseableHttpClient httpClient;
      private CloseableHttpResponse response;
  
  
      private static final String URL = "http://www.minigolfsport.de/download/";
      private final static Logger LOGGER = Logger.getLogger(HttpDataRetrieverImpl.class);
  
      public HttpDataRetrieverImpl() {
          this(null, 0);
      }
  
      public HttpDataRetrieverImpl(String proxyHost, int proxyPort) {
          // See: http://hc.apache.org/httpcomponents-client/tutorial/html/
          this.proxyHost = proxyHost;
          this.proxyPort = proxyPort;
          this.proxyScheme = "http";
      }
  
      @Override
      public String getProxyHost() {
          return proxyHost;
      }
  
      @Override
      public void setProxyHost(String proxyHost) {
          this.proxyHost = proxyHost;
      }
  
      @Override
      public int getProxyPort() {
          return proxyPort;
      }
  
      @Override
      public void setProxyPort(int proxyPort) {
          this.proxyPort = proxyPort;
      }
  
      @Override
      public String getUrl() {
          return URL;
      }
  
      @Override
      public InputStream retrieveInputStream() {
          // See: http://hc.apache.org/httpcomponents-client/tutorial/html/
  
          assert currentName != null;
  
          InputStream result = null;
  
          HttpGet httpGet = new HttpGet(URL + currentName);
          if (proxyHost != null) {
              HttpHost proxy = new HttpHost(proxyHost, proxyPort, proxyScheme);
              RequestConfig config = RequestConfig.custom().setProxy(proxy).build();
              httpGet.setConfig(config);
              LOGGER.debug("Using proxy (" + proxyHost + "," + proxyPort + "," + proxyScheme + ").");
          }
          httpClient = HttpClients.createDefault();
          try {
          	response = httpClient.execute(httpGet);
          	HttpEntity entity = response.getEntity();
          	if (entity != null) {
          		result = entity.getContent();
          	}
          } catch (ClientProtocolException e) {
          	LOGGER.warn("Failed to retrieve data from '" + URL + currentName + "'.", e);
          } catch (IOException e) {
          	LOGGER.warn("Failed to retrieve data from '" + URL + currentName + "'.", e);
          }
  
          return result;
      }
 
     @Override
     public void freeAllResources() {
     	if (response != null) {
     		try {
 				response.close();
 			} catch (IOException e) {
 				LOGGER.warn("Failed to close http reponse.", e);
 			}
     	}
     	
     	if (httpClient != null) {
     		try {
 				httpClient.close();
 			} catch (IOException e) {
 				LOGGER.warn("Failed to close http reponse.", e);
 			}
     	}
     }
     
     @Override
     public boolean findCurrentFile() {
         // See: http://code.google.com/p/selenium/wiki/GettingStarted
 
         currentName = null;
         currentDate = null;
 
         CloseableHttpClient httpClient = HttpClients.createDefault();
         try {
         	for (int i=99; i>=45; i--) {
         		String currentNameInLoop = "rangliste" + Integer.toString(i) + ".xls";
         		String currentUrl = URL + currentNameInLoop;
         		HttpGet httpGet = new HttpGet(currentUrl);
         		if (proxyHost != null) {
         			HttpHost proxy = new HttpHost(proxyHost, proxyPort, "http");
         			RequestConfig config = RequestConfig.custom().setProxy(proxy).build();
         			httpGet.setConfig(config);
                     LOGGER.debug("Using proxy (" + proxyHost + "," + proxyPort + "," + proxyScheme + ").");
         		}
         		try {
         			CloseableHttpResponse response = httpClient.execute(httpGet);
         			try {
         				StatusLine statusLine = response.getStatusLine();
         				int statusCode = statusLine.getStatusCode();
         				LOGGER.debug("Status code for URL '" + currentUrl + "' is " + Integer.toString(statusCode));
         				if (statusCode == 200) {
         					currentName = currentNameInLoop;
         					currentDate = new Date(0);
         					break;
         				}
         			} finally {
         				response.close();
         			}
         		} catch (ClientProtocolException e) {
         			LOGGER.warn("Failed to check URL '" + currentUrl + "'.", e);
         		} catch (IOException e) {
         			LOGGER.warn("Failed to check URL '" + currentUrl + "'.", e);
         		}
         	}
         } finally {
         	try {
 				httpClient.close();
 			} catch (IOException e) {
 				LOGGER.warn("Error while closing http client.", e);
 			}
         }
 
         if (LOGGER.isDebugEnabled()) {
             LOGGER.debug("Found file '" + currentName + "' (" + currentDate
                     + ")");
         }
         
         return currentName != null;
     }
 
     @Override
     public String getCurrentFileName() {
         return currentName;
     }
 
     @Override
     public Date getCurrentFileDate() {
         return currentDate;
     }
 }