From 58a7cbe7b17dccded72e16dc11ad3d981535a163 Mon Sep 17 00:00:00 2001
From: JovannMC <jovannmc@femboyfurry.net>
Date: Wed, 18 Dec 2024 13:04:27 +0300
Subject: [PATCH] Allow overriding slimevr server/port to search on

idk why i didn't have this before, lol

thanks @ButterscotchV
---
 src/languages/en.json      |  4 +++-
 src/main.ts                | 24 ++++++++++++++++++++++--
 src/static/html/index.html | 38 ++++++++++++++++++++++++++++++++++++++
 src/static/js/index.ts     | 38 +++++++++++++++++++++++++++++++++++++-
 4 files changed, 100 insertions(+), 4 deletions(-)

diff --git a/src/languages/en.json b/src/languages/en.json
index aaea101..0b7dd4a 100644
--- a/src/languages/en.json
+++ b/src/languages/en.json
@@ -100,7 +100,9 @@
             "autoStart": "Start connection on open",
             "autoOff": "Turn off trackers on exit",
             "trackerVisualizationFPS": "Tracker visualization FPS",
-            "heartbeat": "Tracker heartbeat interval"
+            "heartbeat": "Tracker heartbeat interval",
+            "serverAddress": "SlimeVR server address",
+            "serverPort": "SlimeVR server port"
         },
         "updates": {
             "title": "Update checking",
diff --git a/src/main.ts b/src/main.ts
index b502002..8d52b9b 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -40,6 +40,8 @@ let firstLaunch = false;
 let isClosing = false;
 let hasInitializedLogDir = false;
 let foundSlimeVR = false;
+let serverAddress = "255.255.255.255";
+let serverPort = 6969;
 
 let canLogToFile = true;
 let loggingMode = 1;
@@ -86,6 +88,9 @@ try {
     updateChannel = config.global?.updates?.updateChannel ?? "stable";
     heartbeatInterval = config.global?.trackers?.heartbeatInterval ?? 2000;
     loggingMode = config.global?.debug?.loggingMode ?? 1;
+
+    serverAddress = config.global?.serverAddress ?? "255.255.255.255";
+    serverPort = config.global?.serverPort ?? 6969;
 } catch (err) {
     // If the config file doesn't exist or is empty, create it
     log("First launch, creating config file and showing onboarding screen (after load)");
@@ -637,6 +642,16 @@ ipcMain.on("set-auto-off", (_event, arg) => {
     log(`Auto-off set to: ${arg}`, "settings");
 });
 
+ipcMain.on("set-server-address", (_event, arg) => {
+    serverAddress = arg;
+    log(`Server address set to: ${arg}`, "settings");
+});
+
+ipcMain.on("set-server-port", (_event, arg) => {
+    serverPort = arg;
+    log(`Server port set to: ${arg}`, "settings");
+});
+
 ipcMain.on("open-support-page", async () => {
     await shell.openExternal("https://github.com/OCSYT/SlimeTora/wiki/FAQ#i-found-an-issuebug");
 });
@@ -1194,7 +1209,9 @@ async function processQueue() {
             `SlimeTora v${app.getVersion()}`,
             new FirmwareFeatureFlags(new Map([])),
             BoardType.HARITORA,
-            MCUType.HARITORA
+            MCUType.HARITORA,
+            serverAddress,
+            serverPort
         );
 
         await newTracker.init();
@@ -1650,11 +1667,14 @@ const heartbeatTracker = new EmulatedTracker(
     `SlimeTora v${app.getVersion()} heartbeat`,
     new FirmwareFeatureFlags(new Map([])),
     BoardType.HARITORA,
-    MCUType.HARITORA
+    MCUType.HARITORA,
+    serverAddress,
+    serverPort
 );
 
 setupTrackerEvents(heartbeatTracker, true);
 await heartbeatTracker.init();
+log(`Looking for SlimeVR server on: ${serverAddress}:${serverPort}`, "connection");
 
 /*
  * Logging
diff --git a/src/static/html/index.html b/src/static/html/index.html
index 4e16228..f119b8f 100644
--- a/src/static/html/index.html
+++ b/src/static/html/index.html
@@ -432,6 +432,43 @@ <h3 class="title is-4 is-spaced" data-i18n="programSettings.other.title">Other s
                                 </div>
                             </div>
                         </div>
+                        <br />
+                        <div class="columns is-centered">
+                            <div class="column is-two-fifths is-centered">
+                                <div class="input-container">
+                                    <label
+                                        for="server-address"
+                                        class="label has-text-warning"
+                                        data-i18n="programSettings.other.serverAddress"
+                                        >SlimeVR server address</label
+                                    >
+                                    <input
+                                        type="text"
+                                        id="server-address"
+                                        placeholder="255.255.255.255"
+                                        class="input"
+                                        style="max-width: 100%"
+                                    />
+                                </div>
+                            </div>
+                            <div class="column is-two-fifths is-centered">
+                                <div class="input-container">
+                                    <label
+                                        for="server-port"
+                                        class="label has-text-warning"
+                                        data-i18n="programSettings.other.serverPort"
+                                        >SlimeVR server port</label
+                                    >
+                                    <input
+                                        type="text"
+                                        id="server-port"
+                                        placeholder="6969"
+                                        class="input"
+                                        style="max-width: 100%"
+                                    />
+                                </div>
+                            </div>
+                        </div>
                     </div>
                 </div>
             </div>
@@ -728,6 +765,7 @@ <h2 class="subtitle has-text-centered" data-i18n="debugging.text">
                         </div>
                     </div>
                     <br />
+                    <br />
                     <div class="buttons is-centered">
                         <button class="button is-primary" onclick="openSupport()" data-i18n="debugging.openSupportPage">
                             Open support page
diff --git a/src/static/js/index.ts b/src/static/js/index.ts
index e2e5f47..d32ee7c 100644
--- a/src/static/js/index.ts
+++ b/src/static/js/index.ts
@@ -32,6 +32,9 @@ let autoOff = false;
 let trackerVisualizationFPS = 10;
 let trackerHeartbeatInterval = 2000;
 
+let serverAddress = "255.255.255.255";
+let serverPort = 6969;
+
 let appUpdatesEnabled = true;
 let translationsUpdatesEnabled = true;
 let updateChannel = "stable";
@@ -145,6 +148,8 @@ document.addEventListener("DOMContentLoaded", async function () {
     canLogToFile = settings.global?.debug?.canLogToFile ?? true;
     loggingMode = settings.global?.debug?.loggingMode ?? 1;
     bypassCOMPortLimit = settings.global?.debug?.bypassCOMPortLimit ?? false;
+    serverAddress = settings.global?.serverAddress ?? "255.255.255.255";
+    serverPort = settings.global?.serverPort ?? 6969;
 
     // Set switch states based on settings
     setSwitchState("compact-view-switch", compactView);
@@ -178,6 +183,12 @@ document.addEventListener("DOMContentLoaded", async function () {
     const trackerHeartbeatIntervalInput = document.getElementById("tracker-heartbeat-interval") as HTMLInputElement;
     trackerHeartbeatIntervalInput.value = trackerHeartbeatInterval.toString();
 
+    const serverAddressInput = document.getElementById("server-address") as HTMLInputElement;
+    serverAddressInput.value = serverAddress;
+
+    const serverPortInput = document.getElementById("server-port") as HTMLInputElement;
+    serverPortInput.value = serverPort.toString();
+
     // Set the selected COM ports
     const comPortsSwitches = Array.from(document.getElementById("com-ports").querySelectorAll("input"));
     const selectedPorts: string[] = settings.global?.connectionMode?.comPorts ?? [];
@@ -1234,6 +1245,30 @@ function addEventListeners() {
         window.ipc.send("set-tracker-heartbeat-interval", heartbeatInterval);
     });
 
+    document.getElementById("server-address").addEventListener("change", async function () {
+        serverAddress = (document.getElementById("server-address") as HTMLInputElement).value;
+        window.log(`Changed server address to: ${serverAddress}`);
+        window.ipc.send("save-setting", {
+            global: {
+                serverAddress: serverAddress,
+            },
+        });
+
+        window.ipc.send("set-server-address", serverAddress);
+    });
+
+    document.getElementById("server-port").addEventListener("change", async function () {
+        serverPort = parseInt((document.getElementById("server-port") as HTMLInputElement).value);
+        window.log(`Changed server port to: ${serverPort}`);
+        window.ipc.send("save-setting", {
+            global: {
+                serverPort: serverPort,
+            },
+        });
+
+        window.ipc.send("set-server-port", serverPort);
+    });
+
     document.getElementById("language-select").addEventListener("change", async function () {
         const language: string = (document.getElementById("language-select") as HTMLSelectElement).value;
         selectLanguage(language);
@@ -1544,4 +1579,5 @@ window.openSupport = () => {
     window.ipc.send("open-support-page", null);
 };
 
-export {};
+export { };
+