Add shadowJar instead of application, bump the project's version to 1.0.5

build.gradle

@@ -1,19 +1,31 @@
+import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar
+
 plugins {
     id 'java'
-    id 'application'
+    id 'com.gradleup.shadow' version '9.0.0-beta4'
 }
 
 group = 'net.pixtaded'
-version = '1.0.4'
+version = projectVersion
 
 repositories {
     mavenCentral()
 }
 
-application {
-    mainClass = 'net.pixtaded.crab.Main'
-}
-
 dependencies {
     implementation('org.xerial:sqlite-jdbc:3.47.1.0')
 }
+
+tasks.build.dependsOn tasks.shadowJar
+
+tasks.named('jar', Jar) {
+    manifest {
+        attributes 'Main-Class': 'net.pixtaded.crab.Main'
+    }
+}
+
+tasks.named('shadowJar', ShadowJar) {
+    archiveBaseName = 'crab'
+    archiveClassifier = ''
+    archiveVersion = projectVersion
+}

gradle.properties

@@ -0,0 +1 @@
+projectVersion=1.0.5

src/main/java/net/pixtaded/crab/client/CrabClient.java

@@ -1,6 +1,7 @@
 package net.pixtaded.crab.client;
 
 import net.pixtaded.crab.common.Crab;
+import net.pixtaded.crab.common.Logs;
 import net.pixtaded.crab.common.Sanitizer;
 
 import java.io.*;

src/main/java/net/pixtaded/crab/common/Logs.java

@@ -1,4 +1,4 @@
-package net.pixtaded.crab.client;
+package net.pixtaded.crab.common;
 
 public record Logs(int sizeInBytes, String content) {
 }

src/main/java/net/pixtaded/crab/common/Sanitizer.java

@@ -2,11 +2,15 @@ package net.pixtaded.crab.common;
 
 public class Sanitizer {
     public static String sanitizeString(String s, boolean sanitizeNewlines) {
-        String sanitized = s.replaceAll("\033", "");
+        String sanitized = s.replaceAll("[\010\015\033]", "");
         if (sanitizeNewlines) {
             sanitized = sanitized.replaceAll("\n", "\\\\n");
             if (!s.endsWith("\n")) sanitized += '\n';
         }
         return sanitized;
     }
+
+    public static String formatMessage(long timeMillis, String address, String content) {
+        return String.format("[%td.%1$tm.%1$tY %1$tR] {%s} %s", timeMillis, address, content);
+    }
 }

src/main/java/net/pixtaded/crab/server/CrabServer.java

@@ -1,5 +1,6 @@
 package net.pixtaded.crab.server;
 import net.pixtaded.crab.common.Crab;
+import net.pixtaded.crab.common.Logs;
 
 import java.io.IOException;
 import java.net.ServerSocket;
@@ -13,6 +14,7 @@ public class CrabServer implements Crab {
     private boolean isStopped = false;
     private int port;
     private final Database db;
+    public Logs cache = new Logs(0, "");
 
     public CrabServer() {
         this.db = new Database("data.db");
@@ -28,6 +30,7 @@ public class CrabServer implements Crab {
         try {
             if (this.port == 0)
                 setup();
+            this.cache = getDb().getLogs();
             listen();
         } catch (IOException e) {
             if (!isStopped) System.err.println("Error starting server: " + e.getMessage());

src/main/java/net/pixtaded/crab/server/Database.java

@@ -1,15 +1,14 @@
 package net.pixtaded.crab.server;
 
+import net.pixtaded.crab.common.Logs;
 import net.pixtaded.crab.common.Sanitizer;
 
 import java.sql.*;
 import java.util.Date;
-import java.util.Locale;
 
 public class Database implements AutoCloseable {
 
     private Connection connection;
-    private String logs = "";
 
     public Database(String dbName) {
         try {
@@ -50,22 +49,17 @@ public class Database implements AutoCloseable {
         }
     }
 
-    public String getLogs() {
+    public Logs getLogs() {
         StringBuilder s = new StringBuilder();
         try (ResultSet rs = query("SELECT time, address, msg FROM messages")) {
             while (rs.next()) {
-                s.append(String.format("[%td.%1$tm.%1$tY %1$tR] {%s} %s", rs.getLong(1), rs.getString(2), rs.getString(3)));
+                s.append(Sanitizer.formatMessage(rs.getLong(1), rs.getString(2), rs.getString(3)));
             }
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
-        logs = s.toString();
-        return logs;
-    }
-
-    public int getLogSize() {
-        if (logs.isEmpty()) return 0;
-        else return logs.getBytes().length;
+        String logsString = s.toString();
+        return new Logs(logsString.isEmpty() ? 0 : logsString.getBytes().length, logsString);
     }
 
     @Override

src/main/java/net/pixtaded/crab/server/ServerThread.java

@@ -1,5 +1,8 @@
 package net.pixtaded.crab.server;
 
+import net.pixtaded.crab.common.Logs;
+import net.pixtaded.crab.common.Sanitizer;
+
 import java.io.*;
 import java.net.Socket;
 import java.nio.charset.StandardCharsets;
@@ -30,18 +33,30 @@ public class ServerThread implements Runnable {
     public void run() {
         try {
             byte[] PID = input.readNBytes(1);
+            if (PID.length == 0) {
+                socket.close();
+                return;
+            }
             switch (PID[0]) {
                 case MESSAGE -> {
-                    server.getDb().logMessage(new Date(), socket.getInetAddress().getHostAddress(), new String(input.readNBytes(4096), StandardCharsets.UTF_8).trim());
-                    socket.close();
+                    Date date = new Date();
+                    String msg = new String(input.readNBytes(4096), StandardCharsets.UTF_8).trim();
+                    String address = socket.getInetAddress().getHostAddress();
+
+                    String s = Sanitizer.sanitizeString(msg, true);
+                    String newContent = server.cache.content() + Sanitizer.formatMessage(date.getTime(), address, s);
+                    server.cache = new Logs(newContent.getBytes().length, newContent);
+
+                    new Thread(new LogDBThread(date, address, msg)).start();
                 } case LOGS -> {
-                    respond(server.getDb().getLogs());
+                    respond(server.cache.content());
                 } case LOGS_SIZE -> {
-                    respond(String.valueOf(server.getDb().getLogSize()));
+                    respond(String.valueOf(server.cache.sizeInBytes()));
                 } default -> {
                     System.out.println("PID not implemented: " + PID[0]);
                 }
             }
+            socket.close();
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
@@ -50,10 +65,27 @@ public class ServerThread implements Runnable {
     private void respond(byte[] data) throws IOException {
         socket.getOutputStream().write(data);
         socket.getOutputStream().flush();
-        socket.close();
     }
 
     private void respond(String utf8) throws IOException {
         respond(utf8.getBytes(StandardCharsets.UTF_8));
     }
+
+    private class LogDBThread implements Runnable {
+
+        Date date;
+        String msg;
+        String address;
+
+        public LogDBThread(Date date, String address, String msg) {
+            this.date = date;
+            this.msg = msg;
+            this.address = address;
+        }
+
+        @Override
+        public void run() {
+            server.getDb().logMessage(date, address, msg);
+        }
+    }
 }