Compare commits
10 Commits
Author | SHA1 | Date | |
---|---|---|---|
7ba5bfbf9e | |||
e96b02357d | |||
c2e8927c42 | |||
a7ad8770c9 | |||
5c55999771 | |||
6e1d843f79 | |||
57b20f69cd | |||
bdf7e6bbae | |||
7a116c6789 | |||
6dace63cb0 |
24
build.gradle
24
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.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
1
gradle.properties
Normal file
@ -0,0 +1 @@
|
||||
projectVersion=1.0.5
|
@ -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");
|
||||
}
|
||||
|
||||
}
|
||||
|
4
src/main/java/net/pixtaded/crab/common/Logs.java
Normal file
4
src/main/java/net/pixtaded/crab/common/Logs.java
Normal file
@ -0,0 +1,4 @@
|
||||
package net.pixtaded.crab.common;
|
||||
|
||||
public record Logs(int sizeInBytes, String content) {
|
||||
}
|
@ -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);
|
||||
}
|
||||
}
|
||||
|
@ -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());
|
||||
|
@ -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
|
||||
|
@ -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);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user