--- android/TrainInfoService/src/dk/thoerup/traininfoservice/RequestPlotter.java	2010/06/03 14:21:35	796
+++ android/TrainInfoService/src/dk/thoerup/traininfoservice/RequestPlotter.java	2010/07/11 17:21:34	989
@@ -1,78 +1,293 @@
 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.annotation.WebListener;
+import javax.servlet.annotation.WebServlet;
 import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import dk.thoerup.traininfoservice.banedk.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";
 	
-	protected String getKml() {
-		StringBuilder sb = new StringBuilder();
+	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 PositionContainer getRequestsFromFileWorker(boolean multiple) throws IOException{
+		PositionContainer positions = new PositionContainer();
+
 		try {
-			FileInputStream fis = new FileInputStream("/var/log/apache2/access.log");
-			BufferedReader in = new BufferedReader( new InputStreamReader(fis) );
+			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"};
 			
-		
-			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" );
+			String files[];
 			
-			String line;
-			while ( (line=in.readLine()) != null) {
-				if (line.indexOf("LocateStation") == -1 ){
-					continue;
-				}
-				
-				if (line.indexOf("latitude") == -1 ) {
+			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;
 				}
 				
-				String toks[] = line.split(" ");
-				String ip = toks[0];
-				String time = toks[3].replace("[", "");
-				String argpart = toks[6].split("?")[1];
-				
-				String args[] = argpart.split("&");
+				FileInputStream fis = new FileInputStream(fileStr);
 				
-				String lat = args[0].split("=")[1];
-				String lng = args[0].split("=")[1];
-				
-		        sb.append( " <Placemark>" );
-		        sb.append( "  <description>IP=" + ip + "  Time=" + time + "</description>" );
-		        sb.append( "  <Point><coordinates>" + lng + "," + lat + ",0</coordinates></Point>" );
-		        sb.append( " </Placemark>" );
+				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();
 			}
-			
-			sb.append(  "</Document>\n" );
-			sb.append( "</kml>\n" );
-			
-		} catch (Exception e) {
-			sb.append("<!-- error -->");
+		} 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" );
+		
+		for(RequestPosition current : list) {			
+			sb.append( " <Placemark>\n" );
+			sb.append( "  <styleUrl>#").append(color).append("</styleUrl>\n" );
+			sb.append( "  <description>IP=").append(current.ip).append("&lt;br/&gt;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" );			
+		}		
 		
-		return sb.toString();
+		sb.append("</Folder>\n");
 	}
 
-	@Override
-	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
-		 resp.setContentType("application/vnd.google-earth.kml");
-		 resp.getWriter().print( getKml() );
+	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(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 );
+		}
+	}
 }