server and handlers

Author: Katsute

README.md

@@ -23,7 +23,6 @@
 <br>
 
 > **Warning**
->
 > simplehttpserver5 is not compatible with any previous version of [simplehttpserver](https://github.com/Ktt-Development/simplehttpserver)
 
 🚧 WIP

src/main/java/dev/katsute/simplehttpserver/ContextUtility.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+import java.util.regex.Pattern;
+
+abstract class ContextUtility {
+
+    // replace consecutive slashes and back slashes with a single forward slash
+    @SuppressWarnings("RegExpRedundantEscape")
+    private static final Pattern forwardSlash = Pattern.compile("\\/{2,}|\\\\+");
+    // trim start and end slashes as well as whitespace
+    @SuppressWarnings("RegExpRedundantEscape")
+    private static final Pattern trimSlash = Pattern.compile("^\\s*\\/*|\\/*\\s*$");
+
+    static String getContext(final String context, final boolean leading, final boolean trailing){
+        final String linSlash = forwardSlash.matcher(context).replaceAll("/");
+        final String strippedSlash = trimSlash.matcher(linSlash).replaceAll("");
+        return strippedSlash.length() == 0
+            ? leading || trailing ? "/" : ""
+            : (leading ? "/" : "") + strippedSlash + (trailing ? "/" : "");
+    }
+
+}

src/main/java/dev/katsute/simplehttpserver/FileRecord.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+import java.nio.charset.StandardCharsets;
+import java.util.*;
+
+public class FileRecord extends Record {
+
+    final String fileName, contentType;
+    private final byte[] bytes;
+
+    FileRecord(final Map.Entry<String,Map<String,?>> entry){
+        super(entry);
+
+        fileName    = Objects.requireNonNull(Objects.requireNonNull(getHeader("Content-Disposition").getParameter("filename")));
+        contentType = Objects.requireNonNull(Objects.requireNonNull(getHeader("Content-Type")).getValue());
+        bytes       = getValue().getBytes(StandardCharsets.UTF_8);
+    }
+
+    public final String getFileName(){
+        return fileName;
+    }
+
+    public final String getContentType(){
+        return contentType;
+    }
+
+    public final byte[] getBytes(){
+        return Arrays.copyOf(bytes, bytes.length);
+    }
+
+}

src/main/java/dev/katsute/simplehttpserver/HttpServerExtensions.java

@@ -0,0 +1,57 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+import com.sun.net.httpserver.*;
+
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.util.Map;
+
+interface HttpServerExtensions {
+
+    InetSocketAddress bind(final int port) throws IOException;
+
+    InetSocketAddress bind(final int port, final int backlog) throws IOException;
+
+    //
+
+    void setSessionHandler(final HttpSessionHandler sessionHandler);
+
+    HttpSessionHandler getSessionHandler();
+
+    HttpSession getSession(final HttpExchange exchange);
+
+    //
+
+    HttpHandler getContextHandler(final String context);
+
+    HttpHandler getContextHandler(final HttpContext context);
+
+    Map<HttpContext,HttpHandler> getContexts();
+
+    String getRandomContext();
+
+    String getRandomContext(final String context);
+
+    //
+
+    void stop();
+
+}

src/main/java/dev/katsute/simplehttpserver/HttpSession.java

@@ -0,0 +1,33 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+public abstract class HttpSession {
+
+    HttpSession(){ }
+
+    public abstract String getSessionID();
+
+    public abstract long getCreationTime();
+
+    public abstract long getLastAccessed();
+
+    public abstract void update();
+
+}

src/main/java/dev/katsute/simplehttpserver/HttpSessionHandler.java

@@ -0,0 +1,126 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+import com.sun.net.httpserver.Headers;
+import com.sun.net.httpserver.HttpExchange;
+
+import java.net.HttpCookie;
+import java.util.*;
+
+public class HttpSessionHandler {
+
+    private final Map<String,HttpSession> sessions = Collections.synchronizedMap(new HashMap<>());
+
+    private final String cookie;
+
+    public HttpSessionHandler(){
+        this("__session-id");
+    }
+
+    public HttpSessionHandler(final String cookie){
+        this.cookie = cookie;
+    }
+
+    public synchronized String assignSessionID(final HttpExchange exchange){
+        String id;
+        do id = UUID.randomUUID().toString();
+        while(sessions.containsKey(id));
+        return id;
+    }
+
+    private String getSetSession(final Headers headers){
+        if(headers.containsKey("Set-Cookie"))
+           for(final String value : headers.get("Set-Cookie"))
+               if(value.startsWith(cookie + "="))
+                   return value.substring(cookie.length() + 1, value.indexOf(";"));
+       return null;
+    }
+
+    public final HttpSession getSession(final HttpExchange exchange){
+        final String sessionId;
+        final HttpSession session;
+
+        @SuppressWarnings("SpellCheckingInspection")
+        final String rcookies = exchange.getRequestHeaders().getFirst("Cookie");
+        final Map<String,String> cookies = new HashMap<>();
+
+        if(rcookies != null && !rcookies.isEmpty()){
+            final String[] pairs = rcookies.split("; ");
+            for(final String pair : pairs){
+                final String[] value = pair.split("=");
+                cookies.put(value[0], value[1]);
+            }
+        }
+
+        final String setSession = getSetSession(exchange.getResponseHeaders());
+        sessionId = setSession != null ? setSession : cookies.get(cookie);
+
+        synchronized(this){
+            if(!sessions.containsKey(sessionId)){
+                session = new HttpSession() {
+                    private final String sessionID;
+                    private final long creationTime;
+                    private long lastAccessTime;
+
+                    {
+                        sessionID = assignSessionID(exchange);
+                        creationTime = System.currentTimeMillis();
+                        lastAccessTime = creationTime;
+                        sessions.put(sessionID, this);
+                    }
+
+                    @Override
+                    public final String getSessionID(){
+                        return sessionID;
+                    }
+
+                    //
+
+                    @Override
+                    public final long getCreationTime(){
+                        return creationTime;
+                    }
+
+                    @Override
+                    public final long getLastAccessed(){
+                        return lastAccessTime;
+                    }
+
+                    @Override
+                    public synchronized final void update(){
+                        lastAccessTime = System.currentTimeMillis();
+                    }
+
+                };
+
+                final HttpCookie OUT = new HttpCookie(cookie, session.getSessionID());
+                OUT.setPath("/");
+                OUT.setHttpOnly(true);
+
+                exchange.getResponseHeaders().add("Set-Cookie", OUT.toString());
+                sessions.put(session.getSessionID(), session);
+            }else{
+                session = sessions.get(sessionId);
+            }
+        }
+        return session;
+    }
+
+}

src/main/java/dev/katsute/simplehttpserver/MultipartFormData.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright (C) 2022 Katsute <https://github.com/Katsute>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package dev.katsute.simplehttpserver;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class MultipartFormData {
+
+    private final Map<String,Record> data;
+
+    MultipartFormData(final Map<String,Record> data){
+        this.data = new HashMap<>(data);
+    }
+
+    public final Record getEntry(final String name){
+        return data.get(name);
+    }
+
+    public final Map<String,Record> getEntries(){
+        return new HashMap<>(data);
+    }
+
+}