WIP

backend/agent-manager.ts

@@ -12,7 +12,7 @@ import { R } from "redbean-node";
 export class AgentManager {
 
     protected socket : DockgeSocket;
-    protected instanceSocketList : Record<string, SocketClient> = {};
+    protected agentSocketList : Record<string, SocketClient> = {};
 
     constructor(socket: DockgeSocket) {
         this.socket = socket;
@@ -27,7 +27,7 @@ export class AgentManager {
                 reject(new Error("Invalid Dockge URL"));
             }
 
-            if (this.instanceSocketList[endpoint]) {
+            if (this.agentSocketList[endpoint]) {
                 reject(new Error("The Dockge URL already exists"));
             }
 
@@ -80,10 +80,20 @@ export class AgentManager {
 
     /**
      *
-     * @param endpoint
+     * @param url
      */
-    remove(endpoint : string) {
+    async remove(url : string) {
+        let bean = await R.findOne("agent", " url = ? ", [
+            url,
+        ]);
 
+        if (bean) {
+            await R.trash(bean);
+            let endpoint = bean.endpoint;
+            delete this.agentSocketList[endpoint];
+        } else {
+            throw new Error("Agent not found");
+        }
     }
 
     connect(url : string, username : string, password : string) {
@@ -100,7 +110,7 @@ export class AgentManager {
             return;
         }
 
-        if (this.instanceSocketList[endpoint]) {
+        if (this.agentSocketList[endpoint]) {
             log.debug("agent-manager", "Already connected to the socket server: " + endpoint);
             return;
         }
@@ -171,11 +181,11 @@ export class AgentManager {
             }
         });
 
-        this.instanceSocketList[endpoint] = client;
+        this.agentSocketList[endpoint] = client;
     }
 
     disconnect(endpoint : string) {
-        let client = this.instanceSocketList[endpoint];
+        let client = this.agentSocketList[endpoint];
         client?.disconnect();
     }
 
@@ -198,14 +208,14 @@ export class AgentManager {
     }
 
     disconnectAll() {
-        for (let endpoint in this.instanceSocketList) {
+        for (let endpoint in this.agentSocketList) {
             this.disconnect(endpoint);
         }
     }
 
     emitToEndpoint(endpoint: string, eventName: string, ...args : unknown[]) {
         log.debug("agent-manager", "Emitting event to endpoint: " + endpoint);
-        let client = this.instanceSocketList[endpoint];
+        let client = this.agentSocketList[endpoint];
         if (!client) {
             log.error("agent-manager", "Socket client not found for endpoint: " + endpoint);
             throw new Error("Socket client not found for endpoint: " + endpoint);
@@ -215,7 +225,7 @@ export class AgentManager {
 
     emitToAllEndpoints(eventName: string, ...args : unknown[]) {
         log.debug("agent-manager", "Emitting event to all endpoints");
-        for (let endpoint in this.instanceSocketList) {
+        for (let endpoint in this.agentSocketList) {
             this.emitToEndpoint(endpoint, eventName, ...args);
         }
     }

backend/dockge-server.ts

@@ -663,10 +663,10 @@ export class DockgeServer {
      * @param {string} userID
      * @param {string?} currentSocketID
      */
-    disconnectAllSocketClients(userID: number, currentSocketID? : string) {
+    disconnectAllSocketClients(userID: number | undefined, currentSocketID? : string) {
         for (const rawSocket of this.io.sockets.sockets.values()) {
             let socket = rawSocket as DockgeSocket;
-            if (socket.userID === userID && socket.id !== currentSocketID) {
+            if ((!userID || socket.userID === userID) && socket.id !== currentSocketID) {
                 try {
                     socket.emit("refresh");
                     socket.disconnect();

backend/socket-handlers/manage-agent-socket-handler.ts

@@ -11,7 +11,6 @@ export class ManageAgentSocketHandler extends SocketHandler {
             try {
                 log.debug("manage-agent-socket-handler", "addAgent");
                 checkLogin(socket);
-
                 let manager = socket.instanceManager;
                 await manager.test(data.url, data.username, data.password);
                 await manager.add(data.url, data.username, data.password);
@@ -19,6 +18,9 @@ export class ManageAgentSocketHandler extends SocketHandler {
                 // connect to the agent
                 manager.connect(data.url, data.username, data.password);
 
+                // Refresh another sockets
+                // It is a bit difficult to control another browser sessions to connect/disconnect agents, so force them to refresh the page will be easier.
+                server.disconnectAllSocketClients(undefined, socket.id);
                 manager.sendAgentList();
 
                 callback({
@@ -33,11 +35,21 @@ export class ManageAgentSocketHandler extends SocketHandler {
         });
 
         // removeAgent
-        socket.on("removeAgent", async (data : unknown, callback : unknown) => {
+        socket.on("removeAgent", async (url : unknown, callback : unknown) => {
             try {
                 log.debug("manage-agent-socket-handler", "removeAgent");
                 checkLogin(socket);
-                await socket.instanceManager.remove(data.endpoint);
+                let manager = socket.instanceManager;
+                await manager.remove(url);
+
+                server.disconnectAllSocketClients(undefined, socket.id);
+                manager.sendAgentList();
+
+                callback({
+                    ok: true,
+                    msg: "agentRemovedSuccessfully",
+                    msgi18n: true,
+                });
             } catch (e) {
                 callbackError(e, callback);
             }