pizzakv/main.zig

const std = @import("std");

const net = std.net;
const posix = std.posix;
const fmt = std.fmt;

const socket = @import("socket.zig");
const command = @import("command.zig");
const persistence = @import("persistence.zig");

const PORT = 8085;

pub fn main() !void {
    const listener = try socket.init(PORT);
    defer posix.close(listener);

    std.debug.print("2025 pizzakv! TCP Listening on port {any}\n<danilo@fragoso.dev>\n---------\n", .{PORT});
    std.debug.print("Commands:\n\nread key\nwrite key|value\ndelete key\nkeys\nreads prefix\nstatus\n", .{});
    std.debug.print("---------\n", .{});

    try persistence.init();

    while (true) {
        var client_address: net.Address = undefined;
        var client_address_len: posix.socklen_t = @sizeOf(net.Address);

        const conn = posix.accept(listener, &client_address.any, &client_address_len, 0) catch |err| {
            std.debug.print("error accept: {any}", .{err});
            continue;
        };

        const thread = try std.Thread.spawn(.{}, handleConnection, .{conn});
        thread.detach();
    }
}

pub fn handleConnection(conn: posix.socket_t) !void {
    defer posix.close(conn);

    var requestBuffer: [1024 * 1024]u8 = undefined;
    var responseBuffer: [1024 * 1024 * 4]u8 = undefined;

    while (true) {
        const n = try socket.readUntilCR(conn, &requestBuffer);
        if (n == 0) {
            break;
        }

        const cmdResponse = command.parse(requestBuffer[0..n]) orelse {
            socket.write(conn, "error\r") catch |err| {
                std.debug.print("error writing: {any}", .{err});
            };
            continue;
        };

        @memcpy(responseBuffer[0..cmdResponse.len], cmdResponse);
        responseBuffer[cmdResponse.len] = '\r';
        socket.write(conn, responseBuffer[0 .. cmdResponse.len + 1]) catch |err| {
            std.debug.print("error writing: {any}", .{err});
        };
    }
}