src/controllers/api/loginController.ts

@@ -88,8 +88,7 @@ export const loginController: RequestHandler = async (request, response) => {
     account.LastLogin = new Date();
     await account.save();
 
-    // Tell WebUI its nonce has been invalidated
+    sendWsBroadcastTo(account._id.toString(), { nonce_updated: true });
-    sendWsBroadcastTo(account._id.toString(), { logged_out: true });
 
     response.json(createLoginResponse(myAddress, myUrlBase, account.toJSON(), buildLabel));
 };

src/controllers/api/logoutController.ts

@@ -21,8 +21,7 @@ export const logoutController: RequestHandler = async (req, res) => {
         }
     );
     if (stat.modifiedCount) {
-        // Tell WebUI its nonce has been invalidated
+        sendWsBroadcastTo(req.query.accountId as string, { nonce_updated: true });
-        sendWsBroadcastTo(req.query.accountId as string, { logged_out: true });
     }
 
     res.writeHead(200, {

src/services/wsService.ts

@@ -73,8 +73,9 @@ interface IWsMsgToClient {
     auth_fail?: {
         isRegister: boolean;
     };
-    logged_out?: boolean;
+    nonce_updated?: boolean;
     update_inventory?: boolean;
+    logged_out?: boolean;
 }
 
 const wsOnConnect = (ws: ws, req: http.IncomingMessage): void => {

static/webui/script.js

@@ -81,12 +81,15 @@ function openWebSocket() {
                 single.loadRoute("/webui/");
             }
         }
-        if ("logged_out" in msg) {
+        if ("nonce_updated" in msg) {
             sendAuth();
         }
         if ("update_inventory" in msg) {
             updateInventory();
         }
+        if ("logged_out" in msg) {
+            logout();
+        }
     };
     window.ws.onclose = function () {
         ws_is_open = false;