Add body-parser

Author: simonwep

src/express/http/Request.java

@@ -3,6 +3,7 @@
 import com.sun.net.httpserver.Headers;
 import com.sun.net.httpserver.HttpExchange;
 import express.http.cookie.Cookie;
+import express.multipart.MultiPartData;
 
 import java.io.IOException;
 import java.io.InputStream;
@@ -21,22 +22,27 @@
  */
 public class Request {
 
-  private final HttpExchange HTTP_EXCHANGE;
   private final URI URI;
+  private final HttpExchange HTTP_EXCHANGE;
   private final InputStream BODY;
   private final Headers HEADER;
+  private final String CONTENT_TYPE;
 
   private final HashMap<String, Cookie> COOKIES;
   private final HashMap<String, String> QUERYS;
+  private HashMap<String, MultiPartData> formData;
   private HashMap<String, String> params;
 
   public Request(HttpExchange exchange) {
     this.HTTP_EXCHANGE = exchange;
     this.URI = exchange.getRequestURI();
     this.HEADER = exchange.getRequestHeaders();
     this.BODY = exchange.getRequestBody();
+    this.CONTENT_TYPE = HEADER.get("Content-Type").get(0);
 
+    this.formData = new HashMap<>();
     this.params = new HashMap<>();
+
     this.QUERYS = parseRawQuery(exchange.getRequestURI());
     this.COOKIES = parseCookies(exchange.getRequestHeaders());
   }
@@ -100,7 +106,7 @@ public String getHost() {
    * @return The request content-type.
    */
   public String getContentType() {
-    return HEADER.get("Content-Type").get(0);
+    return CONTENT_TYPE;
   }
 
   /**
@@ -124,6 +130,14 @@ public String getMethod() {
     return HTTP_EXCHANGE.getRequestMethod();
   }
 
+  public HashMap<String, MultiPartData> getFormData() {
+    return this.formData;
+  }
+
+  public void setFormData(HashMap<String, MultiPartData> formData) {
+    this.formData = formData;
+  }
+
   /**
    * Returns an param from a dynamic url.
    * see README

src/express/middleware/BodyParser.java

@@ -0,0 +1,43 @@
+package express.middleware;
+
+import express.events.HttpRequest;
+import express.http.Request;
+import express.http.Response;
+import express.multipart.MultiPartData;
+import express.multipart.MultiPartStream;
+
+import java.io.IOException;
+import java.util.HashMap;
+
+public class BodyParser implements HttpRequest {
+
+  private final long MAX_SIZE;
+
+  public BodyParser() {
+    this.MAX_SIZE = -1;
+  }
+
+  public BodyParser(long maxFieldSize) {
+    this.MAX_SIZE = maxFieldSize;
+  }
+
+  @Override
+  public void handle(Request req, Response res) {
+
+    HashMap<String, MultiPartData> dataMap = new HashMap<>();
+    try {
+      String boundary = "--" + req.getContentType().replaceAll("multipart/form-data; boundary=", "") + "\r\n";
+      MultiPartStream stream = new MultiPartStream(req.getBody(), boundary.getBytes(), MAX_SIZE);
+      MultiPartData data;
+
+      while ((data = stream.read()) != null) {
+        dataMap.put(data.getHead(), data);
+      }
+
+      req.setFormData(dataMap);
+    } catch (IOException e) {
+      // TODO: Handle error
+      e.printStackTrace();
+    }
+  }
+}

src/express/multipart/MultiPartData.java

@@ -0,0 +1,26 @@
+package express.multipart;
+
+public class MultiPartData {
+
+  private final MultiPartStatus MPS;
+  private final String HEAD;
+  private final byte[] DATA;
+
+  public MultiPartData(MultiPartStatus mps, String head, byte[] data) {
+    this.MPS = mps;
+    this.HEAD = head;
+    this.DATA = data;
+  }
+
+  public MultiPartStatus getStatus() {
+    return MPS;
+  }
+
+  public String getHead() {
+    return HEAD;
+  }
+
+  public byte[] getBytes() {
+    return DATA;
+  }
+}

src/express/multipart/MultiPartStatus.java

@@ -0,0 +1,5 @@
+package express.multipart;
+
+public enum MultiPartStatus {
+  OUT_OF_SIZE, OK
+}

src/express/multipart/MultiPartStream.java

@@ -0,0 +1,78 @@
+package express.multipart;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Arrays;
+
+public class MultiPartStream {
+
+  private final ByteArrayOutputStream BUFFER = new ByteArrayOutputStream();
+  private final InputStream IS;
+  private final byte[] BOUNDARY;
+  private final long MAX_SIZE;
+
+  public MultiPartStream(InputStream body, byte[] boundary, long maxSize) {
+    this.IS = body;
+    this.BOUNDARY = boundary;
+    this.MAX_SIZE = maxSize;
+  }
+
+  public MultiPartData read() throws IOException {
+    byte[] buffer = new byte[BOUNDARY.length];
+    ByteArrayOutputStream b = new ByteArrayOutputStream();
+    String head = null;
+
+    int i;      // Read index
+    int il = 0; // Last byte
+    int boundaryIndex = 0; // Boundary index to detect boundary
+
+    // Read up to next boundary
+    while ((i = IS.read()) != -1) {
+
+      // Check if the byte is a boundary start
+      if ((byte) i == BOUNDARY[boundaryIndex]) {
+        buffer[boundaryIndex] = (byte) i;
+        boundaryIndex++;
+
+        // Check if the boundary is fully matched
+        if (boundaryIndex == BOUNDARY.length)
+          break;
+
+      } else {
+
+        // Check if the buffer contains data
+        if (boundaryIndex > 0)
+          b.write(Arrays.copyOf(buffer, boundaryIndex - 1));
+        b.write(i);
+
+        // Check if the end of the head is reached
+        if (head == null && i == '\n' && il == '\r') {
+          head = new String(b.toByteArray());
+          b.reset();
+        } else if (head != null) {
+
+          // Check if there is an limit
+          if (MAX_SIZE > 0 && b.size() > MAX_SIZE) {
+            return new MultiPartData(MultiPartStatus.OUT_OF_SIZE, head, new byte[0]);
+          }
+        }
+
+        // Reset index, save last byte
+        boundaryIndex = 0;
+        il = i;
+      }
+    }
+
+    // Check if the end is reached
+    if (i == -1)
+      return null;
+
+    if (b.size() == 0)
+      return read();
+
+    return new MultiPartData(MultiPartStatus.OK, head, b.toByteArray());
+  }
+
+
+}

src/test/Use.java

@@ -2,7 +2,6 @@
 
 import express.Express;
 import express.middleware.Static;
-
 import java.io.IOException;
 
 public class Use {