src/main/java/org/apache/sling/reqanalyzer/impl/RequestAnalyzerWebConsole.java - sling-org-apache-sling-reqanalyzer - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you under the Apache License, Version 2.0 (the
  * "License"); you may not use this file except in compliance
  * with the License.  You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
  */
 package org.apache.sling.reqanalyzer.impl;

 import java.awt.AWTEvent;
 import java.awt.Dimension;
 import java.awt.GraphicsEnvironment;
 import java.awt.Toolkit;
 import java.awt.event.WindowAdapter;
 import java.awt.event.WindowEvent;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.io.PrintWriter;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.net.InetAddress;
 import java.net.UnknownHostException;
 import java.util.Date;
 import java.util.zip.Deflater;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipOutputStream;

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

 import org.apache.commons.io.IOUtils;
 import org.apache.sling.reqanalyzer.impl.gui.MainFrame;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;

 @SuppressWarnings("serial")
 public class RequestAnalyzerWebConsole extends HttpServlet {

     private static final String WINDOW_MARKER = ".showWindow";

     private static final String RAW_FILE_MARKER = ".txt";

     private static final String ZIP_FILE_MARKER = ".txt.zip";

     /** default log */
     private final Logger log = LoggerFactory.getLogger(getClass());

     private final File logFile;

     private MainFrame frame;

     RequestAnalyzerWebConsole(final File logFile) {
         this.logFile = logFile;
     }

     void dispose() {
         if (this.frame != null) {
             MainFrame frame = this.frame;
             this.frame = null;
             AWTEvent e = new WindowEvent(frame, WindowEvent.WINDOW_CLOSING);
             Toolkit.getDefaultToolkit().getSystemEventQueue().postEvent(e);
         }
     }

     @Override
     protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
         if (req.getRequestURI().endsWith(RAW_FILE_MARKER) || req.getRequestURI().endsWith(ZIP_FILE_MARKER)) {

             InputStream input = null;
             OutputStream output = null;
             try {
                 input = new FileInputStream(this.logFile);
                 output = resp.getOutputStream();

                 if (req.getRequestURI().endsWith(ZIP_FILE_MARKER)) {
                     ZipOutputStream zip = new ZipOutputStream(output);
                     zip.setLevel(Deflater.BEST_SPEED);

                     ZipEntry entry = new ZipEntry(this.logFile.getName());
                     entry.setTime(this.logFile.lastModified());
                     entry.setMethod(ZipEntry.DEFLATED);

                     zip.putNextEntry(entry);

                     output = zip;
                     resp.setContentType("application/zip");
                 } else {
                     resp.setContentType("text/plain");
                     resp.setCharacterEncoding("UTF-8");
                     resp.setHeader("Content-Length", String.valueOf(this.logFile.length())); // might be bigger than
                 }
                 resp.setDateHeader("Last-Modified", this.logFile.lastModified());

                 IOUtils.copy(input, output);
             } catch (IOException ioe) {
                 throw new ServletException("Cannot create copy of log file", ioe);
             } finally {
                 IOUtils.closeQuietly(input);

                 if (output instanceof ZipOutputStream) {
                     ((ZipOutputStream) output).closeEntry();
                     ((ZipOutputStream) output).finish();
                 }
             }

             resp.flushBuffer();

         } else if (req.getRequestURI().endsWith(WINDOW_MARKER)) {

             if (canOpenSwingGui(req)) {
                 showWindow();
             }

             String target = req.getRequestURI();
             target = target.substring(0, target.length() - WINDOW_MARKER.length());
             resp.sendRedirect(target);
             resp.flushBuffer();

         } else {

             super.service(req, resp);

         }
     }

     @Override
     protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
         PrintWriter pw = resp.getWriter();

         final String fileSize = this.formatByteSize(this.logFile.length());
         pw.printf("<p class='statline ui-state-highlight'>Request Log Size %s</p>%n", fileSize);

         pw.println("<table class='nicetable ui-widget'>");
         pw.println("  <thead>");
         pw.println("    <tr>");
         pw.println("        <th class='ui-widget-header'>Name</th>");
         pw.println("        <th class='ui-widget-header'>Last Update</th>");
         pw.println("        <th class='ui-widget-header'>Size</th>");
         pw.println("        <th class='ui-widget-header'>Action</th>");
         pw.println("    </tr>");
         pw.println("</thead>");
         pw.println("<tbody>");

         pw.println("<tr>");

         pw.printf("<td><a href='${pluginRoot}%s'>%s</a> (<a href='${pluginRoot}%s'>zipped</a>)</td>%n", RAW_FILE_MARKER, this.logFile.getName(), ZIP_FILE_MARKER);
         pw.printf("<td>%s</td><td>%s</td>", new Date(this.logFile.lastModified()), fileSize);

         pw.println("<td><ul class='icons ui-widget'>");
         if (canOpenSwingGui(req)) {
             pw.printf(
                     "<li title='Analyze Now' class='dynhover ui-state-default ui-corner-all'><a href='${pluginRoot}%s'><span class='ui-icon ui-icon-wrench'></span></a></li>%n",
                     WINDOW_MARKER);
             // pw.printf(" (<a href='${pluginRoot}%s'>Analyze Now</a>)", WINDOW_MARKER);
         }
         // pw.println("<li title='Remove File' class='dynhover ui-state-default ui-corner-all'><span class='ui-icon ui-icon-trash'></span></li>");
         pw.println("</ul></td>");

         pw.println("</tr></tbody>");
         pw.println("</table>");

     }

     private synchronized void showWindow() throws ServletException, IOException {
         if (this.frame == null) {
             final File toAnalyze = this.getLogFileCopy();

             final Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
             MainFrame frame = new MainFrame(toAnalyze, Integer.MAX_VALUE, screenSize);

             // exit the application if the main frame is closed
             frame.addWindowListener(new WindowAdapter() {
                 @Override
                 public void windowClosing(WindowEvent e) {
                     // remove the tmp file we showed
                     if (toAnalyze.exists()) {
                         toAnalyze.delete();
                     }
                     // clear the window reference
                     RequestAnalyzerWebConsole.this.frame = null;
                 }
             });

             this.frame = frame;
         }

         // make sure the window is visible, try to bring to the front
         this.frame.setVisible(true);
         this.frame.toFront();
     }

     private boolean canOpenSwingGui(final ServletRequest request) {
         try {
             return !GraphicsEnvironment.isHeadless()
                     && InetAddress.getByName(request.getLocalAddr()).isLoopbackAddress();
         } catch (UnknownHostException e) {
             // unexpected, but still fall back to fail-safe false
             return false;
         }
     }

     private final File getLogFileCopy() throws ServletException {
         File result = null;
         InputStream input = null;
         OutputStream output = null;
         try {
             result = File.createTempFile(getServletName(), ".tmp");
             input = new FileInputStream(this.logFile);
             output = new FileOutputStream(result);
             IOUtils.copy(input, output);
             return result;
         } catch (IOException ioe) {
             throw new ServletException("Cannot create copy of log file", ioe);
         } finally {
             IOUtils.closeQuietly(input);
             IOUtils.closeQuietly(output);
         }
     }

     private String formatByteSize(final long value) {
         final String suffix;
         final String suffixedValue;

         if (value >= 0) {
             final BigDecimal KB = new BigDecimal(1000L);
             final BigDecimal MB = new BigDecimal(1000L * 1000);
             final BigDecimal GB = new BigDecimal(1000L * 1000 * 1000);

             BigDecimal bd = new BigDecimal(value);
             if (bd.compareTo(GB) > 0) {
                 bd = bd.divide(GB);
                 suffix = "GB";
             } else if (bd.compareTo(MB) > 0) {
                 bd = bd.divide(MB);
                 suffix = "MB";
             } else if (bd.compareTo(KB) > 0) {
                 bd = bd.divide(KB);
                 suffix = "kB";
             } else {
                 suffix = "B";
             }
             suffixedValue = bd.setScale(2, RoundingMode.UP).toString();
         } else {
             suffixedValue = "n/a";
             suffix = "";
         }

         return suffixedValue + suffix;
     }

 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one
	* or more contributor license agreements. See the NOTICE file
	* distributed with this work for additional information
	* regarding copyright ownership. The ASF licenses this file
	* to you under the Apache License, Version 2.0 (the
	* "License"); you may not use this file except in compliance
	* with the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing,
	* software distributed under the License is distributed on an
	* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
	* KIND, either express or implied. See the License for the
	* specific language governing permissions and limitations
	* under the License.
	*/
	package org.apache.sling.reqanalyzer.impl;

	import java.awt.AWTEvent;
	import java.awt.Dimension;
	import java.awt.GraphicsEnvironment;
	import java.awt.Toolkit;
	import java.awt.event.WindowAdapter;
	import java.awt.event.WindowEvent;
	import java.io.File;
	import java.io.FileInputStream;
	import java.io.FileOutputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.io.OutputStream;
	import java.io.PrintWriter;
	import java.math.BigDecimal;
	import java.math.RoundingMode;
	import java.net.InetAddress;
	import java.net.UnknownHostException;
	import java.util.Date;
	import java.util.zip.Deflater;
	import java.util.zip.ZipEntry;
	import java.util.zip.ZipOutputStream;

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

	import org.apache.commons.io.IOUtils;
	import org.apache.sling.reqanalyzer.impl.gui.MainFrame;
	import org.slf4j.Logger;
	import org.slf4j.LoggerFactory;

	@SuppressWarnings("serial")
	public class RequestAnalyzerWebConsole extends HttpServlet {

	private static final String WINDOW_MARKER = ".showWindow";

	private static final String RAW_FILE_MARKER = ".txt";

	private static final String ZIP_FILE_MARKER = ".txt.zip";

	/** default log */
	private final Logger log = LoggerFactory.getLogger(getClass());

	private final File logFile;

	private MainFrame frame;

	RequestAnalyzerWebConsole(final File logFile) {
	this.logFile = logFile;
	}

	void dispose() {
	if (this.frame != null) {
	MainFrame frame = this.frame;
	this.frame = null;
	AWTEvent e = new WindowEvent(frame, WindowEvent.WINDOW_CLOSING);
	Toolkit.getDefaultToolkit().getSystemEventQueue().postEvent(e);
	}
	}

	@Override
	protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
	if (req.getRequestURI().endsWith(RAW_FILE_MARKER) \|\| req.getRequestURI().endsWith(ZIP_FILE_MARKER)) {

	InputStream input = null;
	OutputStream output = null;
	try {
	input = new FileInputStream(this.logFile);
	output = resp.getOutputStream();

	if (req.getRequestURI().endsWith(ZIP_FILE_MARKER)) {
	ZipOutputStream zip = new ZipOutputStream(output);
	zip.setLevel(Deflater.BEST_SPEED);

	ZipEntry entry = new ZipEntry(this.logFile.getName());
	entry.setTime(this.logFile.lastModified());
	entry.setMethod(ZipEntry.DEFLATED);

	zip.putNextEntry(entry);

	output = zip;
	resp.setContentType("application/zip");
	} else {
	resp.setContentType("text/plain");
	resp.setCharacterEncoding("UTF-8");
	resp.setHeader("Content-Length", String.valueOf(this.logFile.length())); // might be bigger than
	}
	resp.setDateHeader("Last-Modified", this.logFile.lastModified());

	IOUtils.copy(input, output);
	} catch (IOException ioe) {
	throw new ServletException("Cannot create copy of log file", ioe);
	} finally {
	IOUtils.closeQuietly(input);

	if (output instanceof ZipOutputStream) {
	((ZipOutputStream) output).closeEntry();
	((ZipOutputStream) output).finish();
	}
	}

	resp.flushBuffer();

	} else if (req.getRequestURI().endsWith(WINDOW_MARKER)) {

	if (canOpenSwingGui(req)) {
	showWindow();
	}

	String target = req.getRequestURI();
	target = target.substring(0, target.length() - WINDOW_MARKER.length());
	resp.sendRedirect(target);
	resp.flushBuffer();

	} else {

	super.service(req, resp);

	}
	}

	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
	PrintWriter pw = resp.getWriter();

	final String fileSize = this.formatByteSize(this.logFile.length());
	pw.printf("<p class='statline ui-state-highlight'>Request Log Size %s</p>%n", fileSize);

	pw.println("<table class='nicetable ui-widget'>");
	pw.println(" <thead>");
	pw.println(" <tr>");
	pw.println(" <th class='ui-widget-header'>Name</th>");
	pw.println(" <th class='ui-widget-header'>Last Update</th>");
	pw.println(" <th class='ui-widget-header'>Size</th>");
	pw.println(" <th class='ui-widget-header'>Action</th>");
	pw.println(" </tr>");
	pw.println("</thead>");
	pw.println("<tbody>");

	pw.println("<tr>");

	pw.printf("<td><a href='${pluginRoot}%s'>%s</a> (<a href='${pluginRoot}%s'>zipped</a>)</td>%n", RAW_FILE_MARKER, this.logFile.getName(), ZIP_FILE_MARKER);
	pw.printf("<td>%s</td><td>%s</td>", new Date(this.logFile.lastModified()), fileSize);

	pw.println("<td><ul class='icons ui-widget'>");
	if (canOpenSwingGui(req)) {
	pw.printf(
	"<li title='Analyze Now' class='dynhover ui-state-default ui-corner-all'><a href='${pluginRoot}%s'><span class='ui-icon ui-icon-wrench'></span></a></li>%n",
	WINDOW_MARKER);
	// pw.printf(" (<a href='${pluginRoot}%s'>Analyze Now</a>)", WINDOW_MARKER);
	}
	// pw.println("<li title='Remove File' class='dynhover ui-state-default ui-corner-all'><span class='ui-icon ui-icon-trash'></span></li>");
	pw.println("</ul></td>");

	pw.println("</tr></tbody>");
	pw.println("</table>");

	}

	private synchronized void showWindow() throws ServletException, IOException {
	if (this.frame == null) {
	final File toAnalyze = this.getLogFileCopy();

	final Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
	MainFrame frame = new MainFrame(toAnalyze, Integer.MAX_VALUE, screenSize);

	// exit the application if the main frame is closed
	frame.addWindowListener(new WindowAdapter() {
	@Override
	public void windowClosing(WindowEvent e) {
	// remove the tmp file we showed
	if (toAnalyze.exists()) {
	toAnalyze.delete();
	}
	// clear the window reference
	RequestAnalyzerWebConsole.this.frame = null;
	}
	});

	this.frame = frame;
	}

	// make sure the window is visible, try to bring to the front
	this.frame.setVisible(true);
	this.frame.toFront();
	}

	private boolean canOpenSwingGui(final ServletRequest request) {
	try {
	return !GraphicsEnvironment.isHeadless()
	&& InetAddress.getByName(request.getLocalAddr()).isLoopbackAddress();
	} catch (UnknownHostException e) {
	// unexpected, but still fall back to fail-safe false
	return false;
	}
	}

	private final File getLogFileCopy() throws ServletException {
	File result = null;
	InputStream input = null;
	OutputStream output = null;
	try {
	result = File.createTempFile(getServletName(), ".tmp");
	input = new FileInputStream(this.logFile);
	output = new FileOutputStream(result);
	IOUtils.copy(input, output);
	return result;
	} catch (IOException ioe) {
	throw new ServletException("Cannot create copy of log file", ioe);
	} finally {
	IOUtils.closeQuietly(input);
	IOUtils.closeQuietly(output);
	}
	}

	private String formatByteSize(final long value) {
	final String suffix;
	final String suffixedValue;

	if (value >= 0) {
	final BigDecimal KB = new BigDecimal(1000L);
	final BigDecimal MB = new BigDecimal(1000L * 1000);
	final BigDecimal GB = new BigDecimal(1000L * 1000 * 1000);

	BigDecimal bd = new BigDecimal(value);
	if (bd.compareTo(GB) > 0) {
	bd = bd.divide(GB);
	suffix = "GB";
	} else if (bd.compareTo(MB) > 0) {
	bd = bd.divide(MB);
	suffix = "MB";
	} else if (bd.compareTo(KB) > 0) {
	bd = bd.divide(KB);
	suffix = "kB";
	} else {
	suffix = "B";
	}
	suffixedValue = bd.setScale(2, RoundingMode.UP).toString();
	} else {
	suffixedValue = "n/a";
	suffix = "";
	}

	return suffixedValue + suffix;
	}

	}