Resolve TcpServer merge conflict (keep main version)

Author: viktorlindell12

Dockerfile

@@ -0,0 +1,11 @@
+FROM maven:3-eclipse-temurin-25-alpine AS build
+WORKDIR /build
+COPY src/ src/
+COPY pom.xml pom.xml
+RUN mvn compile
+
+FROM eclipse-temurin:25-jre-alpine
+RUN addgroup -S appgroup && adduser -S appuser -G appgroup
+COPY --from=build /build/target/classes/ /app/
+ENTRYPOINT ["java", "-classpath", "/app", "org.example.App"]
+USER appuser

pom.xml

@@ -14,9 +14,16 @@
         <junit.jupiter.version>6.0.2</junit.jupiter.version>
         <assertj.core.version>3.27.7</assertj.core.version>
         <mockito.version>5.21.0</mockito.version>
+        <bucket4j.version>8.14.0</bucket4j.version>
+
     </properties>
 
     <dependencies>
+        <dependency>
+            <groupId>com.bucket4j</groupId>
+            <artifactId>bucket4j_jdk17-core</artifactId>
+            <version>${bucket4j.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.junit.jupiter</groupId>
             <artifactId>junit-jupiter</artifactId>
@@ -47,6 +54,18 @@
             <version>4.3.0</version>
             <scope>test</scope>
         </dependency>
+
+        <dependency>
+            <groupId>tools.jackson.core</groupId>
+            <artifactId>jackson-databind</artifactId>
+            <version>3.0.3</version>
+        </dependency>
+        <dependency>
+            <groupId>tools.jackson.dataformat</groupId>
+            <artifactId>jackson-dataformat-yaml</artifactId>
+            <version>3.0.3</version>
+        </dependency>
+
     </dependencies>
     <build>
         <plugins>

src/main/java/org/example/ConnectionHandler.java

@@ -0,0 +1,42 @@
+package org.example;
+
+import org.example.httpparser.HttpParser;
+
+import java.io.IOException;
+import java.net.Socket;
+
+public class ConnectionHandler implements AutoCloseable {
+
+    Socket client;
+    String uri;
+
+    public ConnectionHandler(Socket client) {
+        this.client = client;
+    }
+
+    public void runConnectionHandler() throws IOException {
+        StaticFileHandler sfh = new StaticFileHandler();
+        HttpParser parser = new HttpParser();
+        parser.setReader(client.getInputStream());
+        parser.parseRequest();
+        parser.parseHttp();
+        resolveTargetFile(parser.getUri());
+        sfh.sendGetRequest(client.getOutputStream(), uri);
+    }
+
+    private void resolveTargetFile(String uri) {
+        if (uri.matches("/$")) { //matches(/)
+            this.uri = "index.html";
+        } else if (uri.matches("^(?!.*\\.html$).*$")) {
+            this.uri = uri.concat(".html");
+        } else {
+            this.uri = uri;
+        }
+
+    }
+
+    @Override
+    public void close() throws Exception {
+        client.close();
+    }
+}

src/main/java/org/example/StaticFileHandler.java

@@ -0,0 +1,56 @@
+package org.example;
+
+import org.example.http.HttpResponseBuilder;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.io.PrintWriter;
+import java.nio.file.Files;
+import java.util.Map;
+
+public class StaticFileHandler {
+    private final String WEB_ROOT;
+    private byte[] fileBytes;
+    private int statusCode;
+
+    //Constructor for production
+    public StaticFileHandler() {
+        WEB_ROOT = "www";
+    }
+
+    //Constructor for tests, otherwise the www folder won't be seen
+    public StaticFileHandler(String webRoot){
+        WEB_ROOT = webRoot;
+    }
+
+    private void handleGetRequest(String uri) throws IOException {
+
+        File file = new File(WEB_ROOT, uri);
+        if(file.exists()) {
+            fileBytes = Files.readAllBytes(file.toPath());
+            statusCode = 200;
+        } else {
+            File errorFile = new File(WEB_ROOT, "pageNotFound.html");
+            if(errorFile.exists()) {
+                fileBytes = Files.readAllBytes(errorFile.toPath());
+            } else {
+                fileBytes = "404 Not Found".getBytes();
+            }
+            statusCode = 404;
+        }
+    }
+
+    public void sendGetRequest(OutputStream outputStream, String uri) throws IOException{
+        handleGetRequest(uri);
+
+        HttpResponseBuilder response = new HttpResponseBuilder();
+        response.setStatusCode(statusCode);
+        response.setHeaders(Map.of("Content-Type", "text/html; charset=utf-8"));
+        response.setBody(fileBytes);
+        PrintWriter writer = new PrintWriter(outputStream, true);
+        writer.println(response.build());
+
+    }
+
+}

src/main/java/org/example/TcpServer.java

@@ -0,0 +1,36 @@
+package org.example;
+
+import java.io.IOException;
+import java.net.ServerSocket;
+import java.net.Socket;
+
+public class TcpServer {
+
+    private final int port;
+
+    public TcpServer(int port) {
+        this.port = port;
+    }
+
+    public void start() {
+        System.out.println("Starting TCP server on port " + port);
+
+        try (ServerSocket serverSocket = new ServerSocket(port)) {
+            while (true) {
+                Socket clientSocket = serverSocket.accept(); // block
+                System.out.println("Client connected: " + clientSocket.getRemoteSocketAddress());
+                Thread.ofVirtual().start(() -> handleClient(clientSocket));
+            }
+        } catch (IOException e) {
+            throw new RuntimeException("Failed to start TCP server", e);
+        }
+    }
+
+    private void handleClient(Socket client) {
+        try (ConnectionHandler connectionHandler = new ConnectionHandler(client)) {
+            connectionHandler.runConnectionHandler();
+        } catch (Exception e) {
+            throw new RuntimeException("Error handling client connection " + e);
+        }
+    }
+}

src/main/java/org/example/config/AppConfig.java

@@ -0,0 +1,53 @@
+package org.example.config;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+@JsonIgnoreProperties(ignoreUnknown = true)
+public record AppConfig(
+        @JsonProperty("server") ServerConfig server,
+        @JsonProperty("logging") LoggingConfig logging
+) {
+    public static AppConfig defaults() {
+        return new AppConfig(ServerConfig.defaults(), LoggingConfig.defaults());
+    }
+
+    public AppConfig withDefaultsApplied() {
+        ServerConfig serverConfig = (server == null ? ServerConfig.defaults() : server.withDefaultsApplied());
+        LoggingConfig loggingConfig = (logging == null ? LoggingConfig.defaults() : logging.withDefaultsApplied());
+        return new AppConfig(serverConfig, loggingConfig);
+    }
+
+    @JsonIgnoreProperties(ignoreUnknown = true)
+    public record ServerConfig(
+            @JsonProperty("port") Integer port,
+            @JsonProperty("rootDir") String rootDir
+    ) {
+        public static ServerConfig defaults() {
+            return new ServerConfig(8080, "./www");
+        }
+
+        public ServerConfig withDefaultsApplied() {
+            int p = (port == null ? 8080 : port);
+            if (p < 1 || p > 65535) {
+                throw new IllegalArgumentException("Invalid port number: " + p + ". Port must be between 1 and 65535");
+            }
+            String rd = (rootDir == null || rootDir.isBlank()) ? "./www" : rootDir;
+            return new ServerConfig(p, rd);
+        }
+    }
+
+    @JsonIgnoreProperties(ignoreUnknown = true)
+    public record LoggingConfig(
+            @JsonProperty("level") String level
+    ) {
+        public static LoggingConfig defaults() {
+            return new LoggingConfig("INFO");
+        }
+
+        public LoggingConfig withDefaultsApplied() {
+            String lvl = (level == null || level.isBlank()) ? "INFO" : level;
+            return new LoggingConfig(lvl);
+        }
+    }
+}

src/main/java/org/example/config/ConfigLoader.java

@@ -0,0 +1,71 @@
+package org.example.config;
+
+import tools.jackson.databind.ObjectMapper;
+import tools.jackson.databind.json.JsonMapper;
+import tools.jackson.dataformat.yaml.YAMLFactory;
+import tools.jackson.dataformat.yaml.YAMLMapper;
+
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Objects;
+
+public final class ConfigLoader {
+
+    private static volatile AppConfig cached;
+
+    private ConfigLoader() {}
+
+    public static AppConfig loadOnce(Path configPath) {
+        if (cached != null) return cached;
+
+        synchronized (ConfigLoader.class) {
+            if (cached == null){
+                cached = load(configPath).withDefaultsApplied();
+            }
+            return cached;
+        }
+    }
+
+    public static AppConfig get(){
+        if (cached == null){
+            throw new IllegalStateException("Config not loaded. call ConfigLoader.loadOnce(...) at startup.");
+        }
+        return cached;
+
+    }
+
+    public static AppConfig load(Path configPath) {
+        Objects.requireNonNull(configPath, "configPath");
+
+        if (!Files.exists(configPath)) {
+            return AppConfig.defaults();
+        }
+
+        ObjectMapper objectMapper = createMapperFor(configPath);
+
+        try (InputStream stream = Files.newInputStream(configPath)){
+            AppConfig config = objectMapper.readValue(stream, AppConfig.class);
+            return config == null ? AppConfig.defaults() : config;
+        } catch (Exception e){
+            throw new IllegalStateException("failed to read config file " + configPath.toAbsolutePath(), e);
+        }
+    }
+
+    private static ObjectMapper createMapperFor(Path configPath) {
+        String name = configPath.getFileName().toString().toLowerCase();
+
+        if (name.endsWith(".yml") || name.endsWith(".yaml")) {
+            return YAMLMapper.builder(new YAMLFactory()).build();
+
+        } else if (name.endsWith(".json")) {
+            return JsonMapper.builder().build();
+        } else  {
+            return YAMLMapper.builder(new YAMLFactory()).build();
+        }
+    }
+
+    static void resetForTests() {
+        cached = null;
+    }
+}

src/main/java/org/example/http/HttpResponseBuilder.java

@@ -0,0 +1,62 @@
+package org.example.http;
+//
+import java.nio.charset.StandardCharsets;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+public class HttpResponseBuilder {
+
+    private static final String PROTOCOL = "HTTP/1.1";
+    private int statusCode = 200;
+    private String body = "";
+    private byte[] bytebody;
+    private Map<String, String> headers = new LinkedHashMap<>();
+
+    private static final String CRLF = "\r\n";
+
+
+    public void setStatusCode(int statusCode) {
+        this.statusCode = statusCode;
+    }
+    public void setBody(String body) {
+        this.body = body != null ? body : "";
+    }
+
+    public void setBody(byte[] body) {
+        this.bytebody = body;
+    }
+    public void setHeaders(Map<String, String> headers) {
+        this.headers = new LinkedHashMap<>(headers);
+    }
+
+    private static final Map<Integer, String> REASON_PHRASES = Map.of(
+            200, "OK",
+            201, "Created",
+            400, "Bad Request",
+            404, "Not Found",
+            500, "Internal Server Error");
+    public String build(){
+        StringBuilder sb = new StringBuilder();
+        int contentLength;
+        if(body.isEmpty() && bytebody != null){
+            contentLength = bytebody.length;
+            setBody(new String(bytebody, StandardCharsets.UTF_8));
+        }else{
+            contentLength = body.getBytes(StandardCharsets.UTF_8).length;
+        }
+
+
+        String reason = REASON_PHRASES.getOrDefault(statusCode, "OK");
+        sb.append(PROTOCOL).append(" ").append(statusCode).append(" ").append(reason).append(CRLF);
+        headers.forEach((k,v) -> sb.append(k).append(": ").append(v).append(CRLF));
+        sb.append("Content-Length: ")
+                .append(contentLength);
+        sb.append(CRLF);
+        sb.append("Connection: close").append(CRLF);
+        sb.append(CRLF);
+        sb.append(body);
+        return sb.toString();
+
+    }
+
+}