omg connection and user auth on socket is working

frontend/js/chat.js

@@ -2,38 +2,51 @@ const form = document.getElementById('form');
 const input = document.getElementById('input');
 const messages = document.getElementById('messages');
 
-form.addEventListener('submit', (e) => {
-    e.preventDefault();
-    if (input.value) {
-        socket.emit('chat message', input.value);
-        input.value = '';
-    }
-});
-fetch('/auth/token')
-    .then(response => {
+// Function to get the token
+async function getToken() {
+    try {
+        const response = await fetch('/auth/token');
         if (!response.ok) {
-            throw new Error('Network response was not ok');
+            console.log('Network response was not ok');
         }
-        return response.text();
-    })
-    .then(token => {
-        localStorage.setItem('token', token); // Save to localStorage
-    })
-    .catch(error => {
+        return await response.text();
+    } catch (error) {
         console.error('There was a problem with token fetching', error);
+        return null;
+    }
+}
+// Initialize socket connection
+async function initializeSocket() {
+    const token = await getToken();
+    if (!token) {
+        console.error('Not logged in');
+        return;
+    }
+
+    const socket = io({
+        auth: {
+            token: token
+        }
     });
 
-const token = localStorage.getItem('token');
-const socket = io({
-    auth: {
-        token: token
-    }
-});
+    socket.on('connect', () => {
+        console.log('Connected to server');
+    });
 
-socket.on('chat message', ({msg}) => {
-    const item = document.createElement('li');
-    item.textContent = msg;
-    messages.appendChild(item);
-    window.scrollTo(0, document.body.scrollHeight);
-});
+    socket.on('chat message', ({msg}) => {
+        const item = document.createElement('li');
+        item.textContent = msg;
+        messages.appendChild(item);
+        window.scrollTo(0, document.body.scrollHeight);
+    });
 
+    form.addEventListener('submit', (e) => {
+        e.preventDefault();
+        if (input.value) {
+            socket.emit('chat message', input.value);
+            input.value = '';
+        }
+    });
+}
+
+initializeSocket();

index.js

@@ -123,25 +123,22 @@ io.use((socket, next) => {
   } else {
       next(new Error('Not logged in'));
   }
-  const username = socket.handshake.username;
 })
 
 // socket io connection
 io.on('connection', (socket) => {
-        const username = socket.user.username;
-        console.log(username + ' connected');
-
-        // disconnect event
-        socket.on('disconnect', () => {
-            console.log(username + ' have  disconnected');
-        });
-
-        // chat message event
-        socket.on('chat message', (msg) => {
-            const message = username + ': ' + msg;
-            console.log(message);
-            io.emit('chat message', message);
-        });
+    const username = socket.user.username;
+    console.log(username + ' connected');
+    // disconnect event
+    socket.on('disconnect', () => {
+        console.log(username + ' have disconnected');
+    });
+    // chat message event
+    socket.on('chat message', (msg) => {
+        const message = username + ': ' + msg;
+        console.log(message);
+        io.emit('chat message', message);
+    });
 })
 // run server
 server.listen(port, () => {