by.stub.utils.HandlerUtils Maven / Gradle / Ivy

Go to download
/*
A Java-based HTTP stub server

Copyright (C) 2012 Alexander Zagniotov, Isa Goksu and Eric Mrak

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see .
 */

package by.stub.utils;

import by.stub.exception.Stubby4JException;
import org.eclipse.jetty.http.HttpHeaders;
import org.eclipse.jetty.http.MimeTypes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Alexander Zagniotov
 * @since 6/24/12, 1:00 AM
 */
public final class HandlerUtils {

   private HandlerUtils() {

   }

   public static void configureErrorResponse(final HttpServletResponse response, final int httpStatus, final String message) throws IOException {
      response.setStatus(httpStatus);
      response.sendError(httpStatus, message);
      response.flushBuffer();
   }

   public static String getHtmlResourceByName(final String templateSuffix) {
      final String htmlTemplatePath = String.format("/html/%s.html", templateSuffix);
      final InputStream inputStream = HandlerUtils.class.getResourceAsStream(htmlTemplatePath);
      if (inputStream == null) {
         throw new Stubby4JException(String.format("Could not find resource %s", htmlTemplatePath));
      }
      return StringUtils.inputStreamToString(inputStream);
   }

   public static String[] getHighlightableHtmlAttributes(final String templateSuffix) {
      final String templatePath = String.format("/html/%s", templateSuffix);
      final InputStream inputStream = HandlerUtils.class.getResourceAsStream(templatePath);
      if (inputStream == null) {
         throw new Stubby4JException(String.format("Could not find resource %s", templatePath));
      }
      return StringUtils.inputStreamToString(inputStream).split("\\n");
   }

   public static String constructHeaderServerName() {
      final Package pkg = HandlerUtils.class.getPackage();
      final String implementationVersion = pkg.getImplementationVersion() == null ?
            "x.x.x" : pkg.getImplementationVersion();
      final String implementationTitle = pkg.getImplementationTitle() == null ?
            "Java-based HTTP stub server" : pkg.getImplementationTitle();
      return String.format("stubby4j/%s (%s)", implementationVersion, implementationTitle);
   }

   public static  Map constructParamMap(final String queryString) {

      if (!StringUtils.isSet(queryString))
         return new HashMap();

      final Map paramMap = new HashMap();
      final String[] pairs = queryString.split("&");
      for (final String pair : pairs) {
         final String[] splittedPair = pair.split("=");
         paramMap.put(splittedPair[0], splittedPair[1]);
      }

      return paramMap;
   }

   public static void setResponseMainHeaders(final HttpServletResponse response) {
      response.setHeader(HttpHeaders.SERVER, HandlerUtils.constructHeaderServerName());
      response.setHeader(HttpHeaders.DATE, new Date().toString());
      response.setHeader(HttpHeaders.CONTENT_TYPE, MimeTypes.TEXT_HTML_UTF_8);
      response.setHeader(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate"); // HTTP 1.1.
      response.setHeader(HttpHeaders.PRAGMA, "no-cache"); // HTTP 1.0.
      response.setDateHeader(HttpHeaders.EXPIRES, 0);
   }

   public static String linkifyRequestUrl(final String scheme, final Object uri, final String host, final int port) {
      final String fullUrl = String.format("%s://%s:%s%s", scheme, host, port, uri);
      return String.format("%s", fullUrl, fullUrl);
   }

   public static String populateHtmlTemplate(final String templateName, final Object... params) {
      final StringBuilder builder = new StringBuilder();
      builder.append(String.format(getHtmlResourceByName(templateName), params));
      return builder.toString();
   }

   public static String extractPostRequestBody(final HttpServletRequest request, final String source) throws IOException {
      if (!request.getMethod().equalsIgnoreCase("post")) return null;

      try {
         return StringUtils.inputStreamToString(request.getInputStream());
      } catch (final Exception ex) {
         final String err = String.format("Error when extracting POST body: %s, returning null..", ex.toString());
         ConsoleUtils.logIncomingRequestError(request, source, err);
         return null;
      }
   }

   public static Object highlightResponseMarkup(final Object value, final String[] attributesToHighlight) {
      String valueString = value.toString();
      valueString = valueString.replaceAll("\"(.*?)\"", "\"$1\"");
      valueString = valueString.replaceAll("=('|\")(.*?)('|\")", "=$1$2$3");
      valueString = valueString.replaceAll(">(.*?)<", ">$1<");
      //valueString = valueString.replaceAll("\"(.*?)\"", "\"$1\"");

      valueString = valueString.replaceAll("</(.*?)>", "apmerlt;/$1apmergt;");
      valueString = valueString.replaceAll("<(.*?)>", "apmerlt;$1apmergt;");

      final String[] firstSet = {"apmergt;", "apmerlt;", "\\{", "\\}", "\\[", "\\]", "null", "true", "false"};
      for (final String attrib : firstSet) {
         valueString = valueString.replaceAll(attrib, String.format("%s", attrib));
      }

      valueString = valueString.replaceAll(">([0-9\\.\\$]+)<", ">$1<");
      //valueString = valueString.replaceAll("\"", "\"");

      for (final String attrib : attributesToHighlight) {
         final String trimmedAttrib = attrib.trim();
         valueString = valueString.replaceAll(trimmedAttrib, String.format("%s", trimmedAttrib));
      }
      valueString = valueString.replaceAll("=", "=");
      valueString = valueString.replaceAll("clazzor", "class");
      valueString = valueString.replaceAll("ekuals", "=");
      valueString = valueString.replaceAll("apmer", "&");

      return valueString;
   }
}