From 2ed915a08266f0f4472b13279a3957c32a75667c Mon Sep 17 00:00:00 2001
From: Leo dev <klesti.selimay@icloud.com>
Date: Sat, 4 Oct 2025 13:58:46 +0200
Subject: [PATCH] Better addresses

---
 src/app/home/page.tsx        | 4 ++--
 src/app/server/[id]/page.tsx | 4 ++--
 src/lib/auth.ts              | 6 +++++-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/app/home/page.tsx b/src/app/home/page.tsx
index 9568bcf..9f4e133 100644
--- a/src/app/home/page.tsx
+++ b/src/app/home/page.tsx
@@ -6,7 +6,7 @@ import AppSidebar from "@/components/Sidebar";
 import Message from "@/types/message";
 import useUser, { User } from "@/hooks/get-user";
 import { NumberMap } from "@/types/typeUtils";
-import auth from "@/lib/auth";
+import auth, { makeAddress } from "@/lib/auth";
 
 // In this testing instance we are assuming that Alice's node is localhost
 // To make this functional we need to add a settings feature to the api
@@ -22,7 +22,7 @@ export default function AppHome() {
 
   useEffect(() => {
     if (!ip || !user || loading) return;
-    auth(ip, wsRef, () => {}, setMessages);
+    auth(makeAddress(ip, 7090), wsRef, () => {}, setMessages);
     setUserList((prev) => {
       prev[user.id] = user;
       return prev;
diff --git a/src/app/server/[id]/page.tsx b/src/app/server/[id]/page.tsx
index 84c22b5..6cc5bb0 100644
--- a/src/app/server/[id]/page.tsx
+++ b/src/app/server/[id]/page.tsx
@@ -8,7 +8,7 @@ import Message from "@/types/message";
 import useUser, { User } from "@/hooks/get-user";
 import { NumberMap } from "@/types/typeUtils";
 import ServerType from "@/types/server";
-import auth from "@/lib/auth";
+import auth, { makeAddress } from "@/lib/auth";
 
 export default function Server() {
   const { id: ip } = useParams<{ id: string }>();
@@ -20,7 +20,7 @@ export default function Server() {
 
   useEffect(() => {
     if (!ip || !user || loading) return;
-    auth(ip, wsRef, setServer, setMessages);
+    auth(makeAddress(ip), wsRef, setServer, setMessages);
     setUserList((prev) => {
       prev[user.id] = user;
       return prev;
diff --git a/src/lib/auth.ts b/src/lib/auth.ts
index 99da046..f044f97 100644
--- a/src/lib/auth.ts
+++ b/src/lib/auth.ts
@@ -3,6 +3,10 @@ import Server from "@/types/server";
 import axios from "axios";
 import { Dispatch, RefObject, SetStateAction } from "react";
 
+export function makeAddress(ip: string, defaultPort = 7080) {
+  return ip.includes(":") ? `${ip}` : `${ip}:${defaultPort}`;
+}
+
 export default async function auth(
   ip: string,
   wsRef: RefObject<WebSocket | null>,
@@ -14,7 +18,7 @@ export default async function auth(
   ).token;
 
   // Open the WebSocket connection
-  const ws = new WebSocket(`ws://${ip}:7080`);
+  const ws = new WebSocket(`ws://${ip}`);
   wsRef.current = ws;
 
   ws.onopen = () => {