From 82e1d078f588ded5205875646e1597efc0b4dba8 Mon Sep 17 00:00:00 2001
From: fullgream <fullgream@i2pmail.org>
Date: Sun, 23 Mar 2025 03:34:41 +0300
Subject: [PATCH] edit frontend api code

---
 server/frontend/public/js/connect/api.js  | 6 +++++-
 server/frontend/public/js/connect/main.js | 4 ++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/server/frontend/public/js/connect/api.js b/server/frontend/public/js/connect/api.js
index 58bc457..7469e68 100644
--- a/server/frontend/public/js/connect/api.js
+++ b/server/frontend/public/js/connect/api.js
@@ -11,6 +11,10 @@ function getTraceId() {
 		.join("");
 }
 
+const curUrl = new URL(location.href);
+const [address, port] = curUrl.searchParams.get("server").split(/:/);
+const isTLSmode = curUrl.searchParams.get("encrypted") === "true";
+
 class ProtoApiMethods {
 	constructor (api) {
 		this.api = api;
@@ -138,7 +142,7 @@ class ApiHTML {
 			)
 			.then(user => {
 				const { token } = user;
-				localStorage.setItem("my-token", token);
+				localStorage.setItem(`my-token>${!isTLSmode ? "ws" : "wss"}://${address}:${port}`, token);
 				this.renderMainMenu(user, bgUrl);
 			});
 		}
diff --git a/server/frontend/public/js/connect/main.js b/server/frontend/public/js/connect/main.js
index b7db2c7..09048b7 100644
--- a/server/frontend/public/js/connect/main.js
+++ b/server/frontend/public/js/connect/main.js
@@ -30,7 +30,7 @@ document.body.innerHTML = `<nav id="navbar-main" class="navbar navbar-connect-da
 
 	<div class="vr" style="margin-right:5px"></div>
 	<div class="d-flex" style="margin-right:10px" id="navbar-user-menu">
-		<button id="toggle-theme" class="btn btn-outline-light" type="button" data-toggle="collapse" data-target="#navbarNav" aria-controls="navbarNav" aria-expanded="false" aria-label="Toggle navigation"><i class="fas fa-sun"></i></button>
+		<button id="toggle-theme" class="btn btn-outline-light" type="button" data-toggle="collapse" data-target="#navbarNav" aria-controls="navbarNav" aria-expanded="false"><i class="fas fa-sun"></i></button>
 	</div>
 </nav>
 <div id="server.area">
@@ -56,7 +56,7 @@ socket.run()
 		console.log("socket sends:", data);
 		document.title = data.name;
 		document.getElementById("server-name").innerText = data.name;
-		const token = localStorage.getItem("my-token");
+		const token = localStorage.getItem(`my-token>${!isTLSmode ? "ws" : "wss"}://${address}:${port}`);
 		let user;
 		if (token) {
 			try {