statusCode paramaterized.

Author: subwiz

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

@@ -18,12 +18,13 @@ public class AnyUrlHttpServerMain {
 
     private static void printHelp(PrintStream out) {
         out.println("Parameters: ");
-        out.println("\t-p    [Mandatory] port number.");
-        out.println("\t-f    File to serve. When not given, prints <p>Hello World!</p>");
-        out.println("\t-c    Response Content-Type. Default is text/html.");
-        out.println("\t-r    Response character encoding. Default is utf-8.");
-        out.println("\t-H    * Response header in the format: `header:value'.");
-        out.println("\t-h    Print this help.");
+        out.println("  -p    [Mandatory] port number.");
+        out.println("  -f    File to serve. When not given, prints <p>Hello World!</p>");
+        out.println("  -c    Response Content-Type. Default is text/html.");
+        out.println("  -r    Response character encoding. Default is utf-8.");
+        out.println("  -H    * Response header in the format: `header:value'.");
+        out.println("  -s    Response status code.");
+        out.println("  -h    Print this help.");
         out.println("Parameters with * can be used more than once.");
     }
 
@@ -82,6 +83,16 @@ public static void main(String[] args) throws Exception {
             servlet.setHeaders(headers);
         }
 
+        if(options.has("s")) {
+            try {
+                int statusCode = Integer.parseInt(options.valueOf("s").toString());
+                servlet.setStatusCode(statusCode);
+            }
+            catch(NumberFormatException ex) {
+                throw new IllegalArgumentException("Param -s must be a valid status code.");
+            }
+        }
+        
         Server server = new Server(port);
         server.setStopAtShutdown(true);
 

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

@@ -22,6 +22,7 @@ public class AnyUrlServlet extends HttpServlet {
     private String charset = "utf-8";
     private File file;
     private MultiValueMap<String, String> headers;
+    private int statusCode = HttpServletResponse.SC_OK;
 
     public void setContentType(String contentType) {
         this.contentType = contentType;
@@ -39,6 +40,10 @@ public void setHeaders(MultiValueMap headers) {
         this.headers = headers;
     }
 
+    public void setStatusCode(int statusCode) {
+        this.statusCode = statusCode;
+    }
+
     @Override
     protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
         process(req, resp);
@@ -72,6 +77,8 @@ private void process(HttpServletRequest req, HttpServletResponse resp) throws Se
             }
         }
 
+        resp.setStatus(statusCode);
+        
         try(OutputStream os = resp.getOutputStream();) {
             if(file != null && file.exists() && file.canRead()) {
                 try(InputStream is = new FileInputStream(file)) {