added -H parameter for headers.

Author: subwiz

src/main/java/org/wiztools/anyurlhttpserver/AnyUrlHttpServerMain.java

@@ -2,11 +2,15 @@
 
 import java.io.File;
 import java.io.PrintStream;
+import java.util.List;
 import joptsimple.OptionParser;
 import joptsimple.OptionSet;
 import org.eclipse.jetty.server.Server;
 import org.eclipse.jetty.servlet.ServletContextHandler;
 import org.eclipse.jetty.servlet.ServletHolder;
+import org.wiztools.commons.MultiValueMap;
+import org.wiztools.commons.MultiValueMapArrayList;
+import org.wiztools.commons.MultiValueMapLinkedHashSet;
 
 /**
  *
@@ -20,10 +24,12 @@ private static void printHelp(PrintStream out) {
         out.println("\t-f\t: File to serve. When not given, prints <p>Hello World!</p>");
         out.println("\t-c\t: Response Content-Type. Default is text/html.");
         out.println("\t-r\t: Response character encoding. Default is utf-8.");
+        out.println("\t-H\t: Header in the format: `header:value'.");
+        out.println("\t-h\t: Print this help.");
     }
 
     public static void main(String[] args) throws Exception {
-        OptionParser parser = new OptionParser("p:f:c:r:h");
+        OptionParser parser = new OptionParser("p:f:c:r:H:h");
         OptionSet options = parser.parse(args);
         if(options.has("h")) {
             printHelp(System.out);
@@ -47,29 +53,35 @@ public static void main(String[] args) throws Exception {
             System.exit(1);
         }
 
-        final File file;
+        AnyUrlServlet servlet = new AnyUrlServlet();
+        
         if(options.has("f")) {
             String fileStr = options.valueOf("f").toString();
-            file = new File(fileStr);
-        }
-        else {
-            file = null;
+            servlet.setFile(new File(fileStr));
         }
 
-        String contentType = null;
         if(options.has("c")) {
-            contentType = options.valueOf("c").toString();
+            servlet.setContentType(options.valueOf("c").toString());
         }
 
-        String charset = null;
         if(options.has("r")) {
-            charset = options.valueOf("r").toString();
+            servlet.setCharset(options.valueOf("r").toString());
         }
 
-        AnyUrlServlet servlet = new AnyUrlServlet();
-        servlet.setFile(file);
-        servlet.setContentType(contentType);
-        servlet.setCharset(charset);
+        if(options.has("H")) {
+            MultiValueMap headers = new MultiValueMapLinkedHashSet();
+            for(Object t: options.valuesOf("H")) {
+                final String headerLine = t.toString().trim();
+                final int sepIdx = headerLine.indexOf(':');
+                if((sepIdx > 0) && (sepIdx < headerLine.length())) {
+                    final String header = headerLine.substring(0, sepIdx);
+                    final String value = headerLine.substring(sepIdx + 1, headerLine.length());
+                    System.out.printf("\nHeader: Value => %s:%s\n", header, value);
+                    headers.put(header, value);
+                }
+            }
+            servlet.setHeaders(headers);
+        }
 
         Server server = new Server(port);
         server.setStopAtShutdown(true);

src/main/java/org/wiztools/anyurlhttpserver/AnyUrlServlet.java

@@ -10,6 +10,7 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import org.wiztools.commons.Charsets;
+import org.wiztools.commons.MultiValueMap;
 import org.wiztools.commons.StreamUtil;
 
 /**
@@ -20,6 +21,7 @@ public class AnyUrlServlet extends HttpServlet {
     private String contentType = "text/html";
     private String charset = "utf-8";
     private File file;
+    private MultiValueMap<String, String> headers;
 
     public void setContentType(String contentType) {
         this.contentType = contentType;
@@ -32,6 +34,10 @@ public void setCharset(String charset) {
     public void setFile(File file) {
         this.file = file;
     }
+    
+    public void setHeaders(MultiValueMap headers) {
+        this.headers = headers;
+    }
 
     @Override
     protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
@@ -57,6 +63,15 @@ protected void doDelete(HttpServletRequest req, HttpServletResponse resp) throws
     private void process(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
         resp.setContentType(contentType);
         resp.setCharacterEncoding(charset);
+        
+        if(headers != null) {
+            for(String key: headers.keySet()) {
+                for(String value: headers.get(key)) {
+                    resp.addHeader(key, value);
+                }
+            }
+        }
+        
         try(OutputStream os = resp.getOutputStream();) {
             if(file != null && file.exists() && file.canRead()) {
                 try(InputStream is = new FileInputStream(file)) {