package dk.thoerup.traininfoservice;

import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import dk.thoerup.traininfoservice.banedk.TimeoutMap;

public class RequestPlotter extends HttpServlet {
	private static final long serialVersionUID = 1L;

	static final Logger log = Logger.getLogger(RequestPlotter.class.getName());

	static final String KML = "application/vnd.google-earth.kml";
	static final String KMZ = "application/vnd.google-earth.kmz";
	
	Map<String,String> cache = new TimeoutMap<String,String>(2*60*1000);

	class RequestPosition {
		public String ip;
		public Date time;
		public String lat;
		public String lng;
	}
	
	class PositionContainer {
		List<RequestPosition> green = new ArrayList<RequestPosition>();
		List<RequestPosition> yellow = new ArrayList<RequestPosition>();
		List<RequestPosition> red = new ArrayList<RequestPosition>();
	}


	boolean isGz(String fileStr) {
		return fileStr.substring(fileStr.length() - 3).equals(".gz");
	}

	protected PositionContainer getRequestsFromFileWorker(boolean multiple) throws IOException{
		PositionContainer positions = new PositionContainer();

		try {
			String files_single[] = {"/var/log/apache2/app_access.log"};
			String files_multi[] = {"/var/log/apache2/app_access.log.3.gz", "/var/log/apache2/app_access.log.2.gz", "/var/log/apache2/app_access.log.1", "/var/log/apache2/app_access.log"};
			
			String files[];
			
			if (multiple == false) {
				files = files_single;
			} else {
				files = files_multi;
			}

			SimpleDateFormat df = new SimpleDateFormat("dd/MMM/yyyy:HH:mm:ss");
		
			Date now = new Date();
			for (String fileStr : files ) {
				File f = new File(fileStr);
				if ( !f.exists() ) {
					continue;
				}
				
				FileInputStream fis = new FileInputStream(fileStr);
				
				InputStream input;
				if ( isGz(fileStr)) {
					input = new GZIPInputStream(fis);
				} else {
					input = fis;
				}
				
				BufferedReader in = new BufferedReader( new InputStreamReader(input) );

	
				String line;
				while ( (line=in.readLine()) != null) {
					if (line.indexOf("LocateStation") == -1 ){
						continue;
					}
	
					if (line.indexOf("latitude=") == -1 ) {
						continue;
					}
					
					if (line.indexOf("longitude=") == -1) {
						continue;
					}
					
					RequestPosition pos = new RequestPosition();
	
					String toks[] = line.split(" ");
					pos.ip = toks[0];
	
					pos.time = df.parse( toks[3].replace("[", "") );
	
					String argpart = toks[6].split("\\?")[1];
	
					String args[] = argpart.split("&");
	
					pos.lat = args[0].split("=")[1];
					pos.lng = args[1].split("=")[1];
					
					
					long timediff = now.getTime() - pos.time.getTime();					
					if ( timediff < (3*60*60*1000) ) {
						positions.red.add(pos); //RED
					} else if ( timediff < (24*60*60*1000)) {
						positions.yellow.add(pos); //YELLOW
					} else {
						positions.green.add(pos); //GREEN
					}
	
				}
				in.close();
				input.close();
				fis.close();
			}
		} catch (ParseException pe) {
			log.log(Level.SEVERE, "parseException", pe);
			throw new IOException(pe);
		} catch (IOException e) {
			log.log(Level.SEVERE, "getKml()", e);
			throw e;
		}

		return positions;
	}
	
	protected void formatPositions(StringBuilder sb, String color, List<RequestPosition> list) {
		sb.append( "<Folder>\n");
		sb.append( " <name>" ).append(color).append("</name>\n");
		sb.append( " <open>0</open>\n" );
		sb.append( " <description>Count=").append(list.size()).append("</description>\n");
		
		for(RequestPosition current : list) {			
			sb.append( " <Placemark>\n" );
			sb.append( "  <styleUrl>#").append(color).append("</styleUrl>\n" );
			sb.append( "  <description>IP=").append(current.ip).append("  Time=").append(current.time).append("</description>\n" );
			sb.append( "  <Point><coordinates>").append(current.lng).append(",").append(current.lat).append(",0</coordinates></Point>\n" );
			sb.append( " </Placemark>\n" );			
		}		
		
		sb.append("</Folder>\n");
	}

	protected String formatXml(PositionContainer positions) {
		StringBuilder sb = new StringBuilder(1024*1024);

		sb.append( "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" );
		sb.append( "<kml xmlns=\"http://www.opengis.net/kml/2.2\">\n" );
		sb.append( "<Document>\n" );		
		
		
		sb.append( " <Style id=\"red\">\n" );
		sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/red-circle.png</href></Icon></IconStyle>\n" );
		sb.append( " </Style>\n");

		sb.append( " <Style id=\"yellow\">\n" );
		sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/ylw-circle.png</href></Icon></IconStyle>\n" );
		sb.append( " </Style>\n" );

		sb.append( " <Style id=\"green\">\n" );
		sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/grn-circle.png</href></Icon></IconStyle>\n" );
		sb.append( " </Style>\n\n" );
		
/*
		String overlay = 
	      " <ScreenOverlay>" +
	        " <name>Absolute Positioning: Top left</name>" +
	        " <description>Absolute Positioning: Top left</description>" +
	        " <visibility>1</visibility>" +
	        "  <Icon>" +
	        "   <href>http://code.google.com/apis/kml/documentation/top_left.jpg</href>" +
	        "  </Icon>" +
	        "  <overlayXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
	        "  <screenXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
	        "  <rotationXY x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
	        "  <size x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
	      " </ScreenOverlay>" ;
		
		sb.append(overlay);
	*/	
		

		formatPositions(sb, "green", positions.green);
		formatPositions(sb, "yellow", positions.yellow);
		formatPositions(sb, "red", positions.red);
		

		sb.append( "</Document>\n" );
		sb.append( "</kml>\n" );

		return sb.toString(); 
	}
	
	protected String getRequestsFromFile(boolean multiple) throws IOException {		
		String kmlData = null;
		String key;
		
		if (multiple == false)  {
			key = "kmldata";
		} else {
			key = "kmldata-multi";
		}
					
		kmlData = cache.get(key);
			
		if (kmlData == null) {
			kmlData = formatXml( getRequestsFromFileWorker(multiple) );
			cache.put(key, kmlData);
			kmlData += "<!-- from source -->";			
		} else {
			kmlData += "<!-- cached -->";
		}
					
		return kmlData;
	}
	
	boolean enabled(String param) {
		if (param == null || param.equals("")) {
			return false;
		}
		
		int p = 0;
		try { 
			p = Integer.parseInt(param);
		} catch (Exception e) {}
		
		return (p != 0);
	}

	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		
		boolean multiple = enabled( req.getParameter("multi") );
		
		String kmlData = getRequestsFromFile(multiple);
		
		if ( enabled(req.getParameter("zip")) ) {

			ByteArrayOutputStream baos = new ByteArrayOutputStream();

			ZipOutputStream zip = new ZipOutputStream(baos);
			zip.putNextEntry( new ZipEntry("trains.kml") );
			zip.write( kmlData.getBytes() );
			zip.closeEntry();
			zip.close();

			byte bytes[] = baos.toByteArray();

			resp.setContentType(KMZ);
			resp.setContentLength( bytes.length );
			resp.getOutputStream().write(bytes);

		} else {
			resp.setContentType(KML);
			resp.getWriter().print( kmlData );
		}
	}
}