speedrun timer

src/dusk/livesplit.cpp

@@ -0,0 +1,183 @@
+#if _WIN32
+    #include <winsock2.h>
+    #include <ws2tcpip.h>
+    using socket_t = SOCKET;
+    static void closeSocket(socket_t s) { closesocket(s); }
+#else
+    #include <sys/socket.h>
+    #include <netinet/in.h>
+    #include <arpa/inet.h>
+    #include <unistd.h>
+    #include <fcntl.h>
+    #include <errno.h>
+    using socket_t = int;
+    static void closeSocket(socket_t s) { close(s); }
+    #ifndef INVALID_SOCKET
+        #define INVALID_SOCKET -1
+    #endif
+#endif
+
+#include <cstdio>
+#include "dusk/livesplit.h"
+#include "f_op/f_op_overlap_mng.h"
+
+namespace dusk::speedrun {
+
+static bool     running           = false;
+static uint64_t frameCount        = 0;
+static socket_t sock              = INVALID_SOCKET;
+static bool     wasLoading        = false;
+static bool     connected         = false;
+static bool     connectPending    = false;
+static bool     disconnectPending = false;
+
+static void sendCmd(const char* cmd) {
+    if (sock == INVALID_SOCKET) {
+        return;
+    }
+
+    char msg[64];
+    int len = snprintf(msg, sizeof(msg), "%s\r\n", cmd);
+
+    if (send(sock, msg, len, 0) >= 0) {
+        if (!connected) {
+            connected = connectPending = true;
+        }
+
+        return;
+    }
+
+#if _WIN32
+    int err = WSAGetLastError();
+    if (err == WSAEWOULDBLOCK || err == WSAENOTCONN) {
+        return;
+    }
+#else
+    if (errno == EAGAIN || errno == EWOULDBLOCK || errno == ENOTCONN) {
+        return;
+    }
+#endif
+
+    if (connected) disconnectPending = true;
+    closeSocket(sock);
+    sock = INVALID_SOCKET;
+    connected = connectPending = false;
+}
+
+uint64_t getFrameCount() {
+    return frameCount;
+}
+
+void onGameFrame() {
+    if (!running) {
+        return;
+    }
+
+    bool loading = fopOvlpM_IsDoingReq() != 0;
+
+    if (loading != wasLoading) {
+        sendCmd(loading ? "pausegametime" : "unpausegametime");
+        wasLoading = loading;
+    }
+
+    if (!loading) {
+        ++frameCount;
+    }
+}
+
+void start() {
+    if (running) {
+        return;
+    }
+    
+    running = true;
+    frameCount = 0;
+    wasLoading = false;
+    sendCmd("initgametime");
+    sendCmd("reset");
+    sendCmd("starttimer");
+}
+
+void reset() {
+    running = false;
+    frameCount = 0;
+    wasLoading = false;
+    sendCmd("reset");
+}
+
+void connectLiveSplit(const char* host, int port) {
+#if _WIN32
+    WSADATA wd{}; WSAStartup(MAKEWORD(2, 2), &wd);
+#endif
+
+    if (sock != INVALID_SOCKET) {
+        closeSocket(sock); sock = INVALID_SOCKET;
+    }
+
+    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+    if (sock == INVALID_SOCKET) {
+        return;
+    }
+
+#if _WIN32
+    u_long nb = 1;
+    ioctlsocket(sock, FIONBIO, &nb);
+#else
+    fcntl(sock, F_SETFL, fcntl(sock, F_GETFL, 0) | O_NONBLOCK);
+#endif
+
+    sockaddr_in addr{}; addr.sin_family = AF_INET;
+    addr.sin_port = htons((uint16_t)port);
+    inet_pton(AF_INET, host, &addr.sin_addr);
+    connect(sock, (sockaddr*)&addr, sizeof(addr));
+    sendCmd("initgametime");
+}
+
+void disconnectLiveSplit() {
+    if (sock != INVALID_SOCKET) {
+        closeSocket(sock);
+        sock = INVALID_SOCKET;
+        connected = false;
+    }
+}
+
+bool consumeConnectedEvent()    { bool v = connectPending;    connectPending    = false; return v; }
+bool consumeDisconnectedEvent() { bool v = disconnectPending; disconnectPending = false; return v; }
+
+void updateLiveSplit() {
+    if (sock == INVALID_SOCKET) {
+        return;
+    }
+
+    if (!connected) {
+        sendCmd("initgametime");
+        return;
+    }
+
+    if (!running) {
+        return;
+    }
+
+    const uint64_t totalMs  = frameCount * 1000 / 30;
+    const uint64_t totalSec = totalMs / 1000;
+    char cmd[32];
+    
+    snprintf(cmd, sizeof(cmd), "setgametime %u:%02u:%02u.%03u",
+        (uint32_t)(totalSec / 3600),
+        (uint32_t)((totalSec / 60) % 60),
+        (uint32_t)(totalSec % 60),
+        (uint32_t)(totalMs % 1000)
+    );
+
+    sendCmd(cmd);
+}
+
+void shutdown() {
+    disconnectLiveSplit();
+#if _WIN32
+    WSACleanup();
+#endif
+}
+
+}