404 page handler

Author: 16BitS3G4l

lib/src/main/java/com/devsegal/jserve/BaseHTTPServer.java

@@ -4,6 +4,8 @@
 import java.net.ServerSocket;
 import java.net.Socket;
 import java.util.HashMap;
+import java.util.stream.Stream;
+import java.nio.file.Files;
 import java.nio.file.Path;
 
 public class BaseHTTPServer implements Runnable {
@@ -29,11 +31,47 @@ public void setupOriginalServerPath(String path) {
         setupOriginalServerPath(Path.of(path));
     }
 
+    // TODO: replace this method with more readable one and better structured
+    // idea: put lambda in one function 
+    // TODO: allow nested folders in public asset folder
+    public void setupPublicAssetFolder(Path assetFolder, String assetFolderPrefix) {
+        Stream<Path> publicAssets = null;
+
+        try {
+            publicAssets = Files.list(assetFolder);
+            publicAssets.forEach(path -> {
+                System.out.println(assetFolderPrefix + "/" + path.getFileName());
+
+                routesToHandlers.put(assetFolderPrefix + "/" + path.getFileName().toString() + "GET", (request, response) -> {
+                    response.setResponseHeaders(new ResponseHeaders("text/html", "close"));
+                    response.readContentFromFile(path, false);
+                    response.send();
+                });
+            });
+
+        } catch(IOException e) {
+            e.printStackTrace();
+        } finally {
+            publicAssets.close();
+        }
+    }
+
+    public void setupPublicAssetFolder(Path assetFolder) {
+        setupPublicAssetFolder(assetFolder, "");
+    }
+
+    public void setupPublicAssetFolder(String path) {
+        setupPublicAssetFolder(Path.of(path));
+    }
+
+    public void setupPublicAssetFolder(String assetFolder, String assetFolderPrefix) {
+        setupPublicAssetFolder(Path.of(assetFolder), assetFolderPrefix);
+    }
+
     public void setupNotFoundPageHandler(NotFoundPageHandler notFoundPageHandler) {
         this.notFoundPageHandler = notFoundPageHandler;
     }
 
-
     public void route(String path, String method, WebRouteHandler handler) {
         routesToHandlers.put(path + method, handler);
     }

lib/src/main/java/com/devsegal/jserve/ConnectionHandler.java

@@ -13,7 +13,6 @@ class ConnectionHandler implements Runnable {
 	private BufferedReader requestReader;
 	private ResponseWriter responseWriter;
 	private Path originalServerPath;
-	private Path publicAssetPath;
 	private Socket connection;
 	private NotFoundPageHandler notFoundPageHandler;
 	private HashMap<String, WebRouteHandler> routesToHandlers;