Compare commits

...

10 Commits

8 changed files with 90 additions and 32 deletions

View File

@ -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.1'
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
}

1
gradle.properties Normal file
View File

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

View File

@ -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.*;
@ -18,6 +19,7 @@ public class CrabClient implements Crab {
private BufferedReader in;
private int lastBufferLength = 0;
private String nickname;
private Logs cache = new Logs(0, "");
public CrabClient() {
this.nickname = "";
@ -37,7 +39,6 @@ public class CrabClient implements Crab {
try {
if (this.serverAddress == null)
setup();
connect();
communicate();
} catch (IOException e) {
System.err.println("Error connecting to the server: " + e.getMessage());
@ -68,7 +69,7 @@ public class CrabClient implements Crab {
System.out.print("Enter your nickname (leave empty for no nickname): ");
nickname = scanner.nextLine();
if (nickname != "")
if (!nickname.isEmpty())
nickname = "<" + nickname + "> ";
}
@ -82,9 +83,7 @@ public class CrabClient implements Crab {
private void communicate() throws IOException {
Scanner scanner = new Scanner(System.in);
String message;
while (true) {
System.out.print("\033[H\033[2J");
getLogs();
System.out.print("Enter a message (or type '/exit' to exit): ");
message = scanner.nextLine();
@ -98,12 +97,14 @@ public class CrabClient implements Crab {
}
private void sendPacket(byte PID, String argument) throws IOException {
connect();
String formattedMessage = String.valueOf((char) PID) + argument + "\n";
out.print(formattedMessage);
out.flush();
receiveResponse(PID);
closeConnection();
}
private void receiveResponse(byte PID) throws IOException {
@ -113,13 +114,15 @@ public class CrabClient implements Crab {
int response = in.read(buffer);
lastBufferLength = Integer.parseInt(new String(buffer).trim());
} case LOGS -> {
if (cache.sizeInBytes() != lastBufferLength) {
byte[] bytes = socket.getInputStream().readNBytes(lastBufferLength);
System.out.print(Sanitizer.sanitizeString(new String(bytes, StandardCharsets.UTF_8), false));
cache = new Logs(lastBufferLength, new String(bytes, StandardCharsets.UTF_8));
}
clearScreen();
System.out.print(Sanitizer.sanitizeString(cache.content(), false));
} default -> {
}
}
closeConnection();
connect();
}
private void closeConnection() {
@ -136,4 +139,9 @@ public class CrabClient implements Crab {
sendPacket(LOGS_SIZE, "");
sendPacket(LOGS, "");
}
private void clearScreen() {
System.out.print("\033[999999S\033[H\033[2J");
}
}

View File

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

View File

@ -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);
}
}

View File

@ -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;
@ -12,7 +13,8 @@ public class CrabServer implements Crab {
private Socket socket;
private boolean isStopped = false;
private int port;
private Database db;
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());

View File

@ -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

View File

@ -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);
}
}
}