package dk.thoerup.traininfoservice;

import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FilenameFilter;
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.Arrays;
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.ZipEntry;
import java.util.zip.ZipOutputStream;

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

import dk.thoerup.genericjavautils.TimeoutMap;

@WebServlet(urlPatterns={"/RequestPlotter"})
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>(30*60*1000);

	class RequestPosition {
		public String ip;
		public Date time;
		public String lat;
		public String lng;
	}
	
	class PositionContainer {
		List<RequestPosition> blue = new ArrayList<RequestPosition>();
		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 File[] getFiles(int count) {
		File accessLogDir = new File("/home/app/domain1/logs/access/");
		//File accessLogDir = new File("/home/torben/inst/glassfishv3/glassfish/domains/domain1/logs/access/");
		
		File logFiles[] = accessLogDir.listFiles( new FilenameFilter() {
			@Override
			public boolean accept(File dir, String name) {
				//log.info("name:" + name);
				return name.startsWith("server_access_log");
			}			
		});
		
		Arrays.sort(logFiles);
		
		if (logFiles == null) {			
			File[] empty = {};
			log.info("file array was empty");
			return empty;
		}
		
		int from = (count>logFiles.length) ? 0 : logFiles.length - (count);
		int to = logFiles.length;
				
		return Arrays.copyOfRange(logFiles, from, to);	
	}
	
	protected PositionContainer getRequestsFromFileWorker(int  count) throws IOException{
		PositionContainer positions = new PositionContainer();

		try {			
			
			File files[] = getFiles(count);

			SimpleDateFormat df = new SimpleDateFormat("dd/MMM/yyyy:HH:mm:ss");
		
			Date now = new Date();
			for (File f : files ) {
				log.info("Parsing file: "  + f.getName() );
				
				if ( !f.exists() ) {
					continue;
				}
				
				
				InputStream input  = new FileInputStream(f);

				
				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].replaceAll("\"", "");
	
					pos.time = df.parse( toks[2].replace("\"", "") );
	
					String argpart = toks[5].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 if ( timediff < (7*24*60*60*1000)) {
						positions.green.add(pos); //GREEN
					} else {
						positions.blue.add(pos); //BLUE
					}
	
				}
				in.close();
				input.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;
	}

	
	/* old apache code
	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 if ( timediff < (7*24*60*60*1000)) {
						positions.green.add(pos); //GREEN
					} else {
						positions.blue.add(pos); //BLUE
					}
	
				}
				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" );
		
		int count=0;
		for(RequestPosition current : list) {
			String id = color + count++; 
			sb.append( " <Placemark id=\"" + id + "\">\n" );
			sb.append( "  <name>").append(current.time).append("</name>\n" );
			sb.append( "  <styleUrl>#").append(color).append("</styleUrl>\n" );
			sb.append( "  <description><![CDATA[IP=").append(current.ip).append("<br/> ]]></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( " <description><![CDATA[");
		sb.append( "  Red:").append(positions.red.size()).append(" (whithin 3 hours)<br/>\n");
		sb.append( "  Yellow:").append(positions.yellow.size()).append(" (within 24 hours)<br/>\n");
		sb.append( "  Green:").append(positions.green.size()).append(" (within one week)<br/>\n");
		sb.append( "  Blue:").append(positions.blue.size()).append(" (older)<br/>\n");		
		sb.append( " ]]></description>");
		
		
		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" );
		
		sb.append( " <Style id=\"blue\">\n" );
		sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/blu-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, "blue", positions.blue);
		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(int count) throws IOException {		
		String kmlData = null;
		String key;
		
		key = "kmldata-" + count;
					
		kmlData = cache.get(key);
			
		if (kmlData == null) {
			kmlData = formatXml( getRequestsFromFileWorker(count) );
			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);
	}

	int getCount(String param) {
		if (param == null || param.equals("")) {
			return 1;
		}
				
		return Integer.parseInt(param);
	}

	
	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		
		
		int count = getCount( req.getParameter("count") );
		
		if (count > 30) //limit to 30 days
			count = 30;
		if (count < 0) //negative count is not allowed
			count = 0;
		
		String kmlData = getRequestsFromFile(count);
		
		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.setHeader("Content-disposition", "attachment; filename=Traininfo-requestplotter.kmz");
			resp.setContentLength( bytes.length );			
			resp.getOutputStream().write(bytes);

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