#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>

#define SOCKET_FILE "/tmp/mysocket"

// WARNING: UNTESTED

int main() {
    int server_fd, client_fd;
    struct sockaddr_un addr;
    char buf[1024];
    ssize_t num_read;

    // Remove any previous socket file
    unlink(SOCKET_FILE);

    server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
    if (server_fd == -1) {
        perror("socket");
        exit(EXIT_FAILURE);
    }

    memset(&addr, 0, sizeof(struct sockaddr_un));
    addr.sun_family = AF_UNIX;
    strncpy(addr.sun_path, SOCKET_FILE, sizeof(addr.sun_path) - 1);

    if (bind(server_fd, (struct sockaddr *)&addr, sizeof(struct sockaddr_un)) == -1) {
        perror("bind");
        exit(EXIT_FAILURE);
    }

    if (listen(server_fd, 5) == -1) {
        perror("listen");
        exit(EXIT_FAILURE);
    }

    std::cout << "Listening on " << SOCKET_FILE << "..." << std::endl;

    while (true) {
        client_fd = accept(server_fd, NULL, NULL);
        if (client_fd == -1) {
            perror("accept");
            continue;
        }

        num_read = read(client_fd, buf, sizeof(buf) - 1);
        if (num_read > 0) {
            buf[num_read] = '\0';
            std::cout << "Received: " << buf << std::endl;
            write(client_fd, buf, num_read); // Echo back
        }
        close(client_fd);
    }

    close(server_fd);
    unlink(SOCKET_FILE);
    return 0;
}