feat(ai): Implement ConnectionService Lifecycle Tools (#8181)

- Added startServer and stopServer methods to ConnectionService
- Exposed start_ws_server and stop_ws_server tools in OpenAPI
- Enables agents to manage the WebSocket bridge lifecycle

Author: tobiu

ai/mcp/server/neural-link/openapi.yaml

@@ -241,6 +241,58 @@ paths:
               schema:
                 $ref: '#/components/schemas/ErrorResponse'
 
+  /server/start:
+    post:
+      summary: Start WebSocket Server
+      operationId: start_ws_server
+      description: |
+        Starts the WebSocket server to accept connections from the Neo.mjs app.
+
+        **When to Use:**
+        Use this to recover the bridge if the server was previously stopped or crashed.
+      tags: [Lifecycle]
+      requestBody:
+        content:
+          application/json:
+            schema:
+              type: object
+              properties: {}
+      responses:
+        '200':
+          description: Server started successfully
+        '500':
+          description: Internal server error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+
+  /server/stop:
+    post:
+      summary: Stop WebSocket Server
+      operationId: stop_ws_server
+      description: |
+        Stops the WebSocket server and closes all active connections.
+
+        **When to Use:**
+        Use this to simulate a network disconnect or to perform a clean shutdown of the bridge.
+      tags: [Lifecycle]
+      requestBody:
+        content:
+          application/json:
+            schema:
+              type: object
+              properties: {}
+      responses:
+        '200':
+          description: Server stopped successfully
+        '500':
+          description: Internal server error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+
   /worker/topology:
     post:
       summary: Get Worker Topology

ai/mcp/server/neural-link/services/ConnectionService.mjs

@@ -71,6 +71,19 @@ class ConnectionService extends Base {
      * @returns {Promise<void>}
      */
     async initAsync() {
+        await this.startServer();
+    }
+
+    /**
+     * Starts the WebSocket server.
+     * @returns {Promise<void>}
+     */
+    async startServer() {
+        if (this.wss) {
+            logger.warn('WebSocket Server is already running.');
+            return;
+        }
+
         this.wss = new WebSocketServer({port: this.port});
 
         this.wss.on('connection', (ws)  => this.#handleConnection(ws));
@@ -79,6 +92,38 @@ class ConnectionService extends Base {
         logger.info(`WebSocket Server listening on port ${this.port}`);
     }
 
+    /**
+     * Stops the WebSocket server and closes all active connections.
+     * @returns {Promise<void>}
+     */
+    async stopServer() {
+        if (!this.wss) {
+            logger.warn('WebSocket Server is not running.');
+            return;
+        }
+
+        logger.info('Stopping WebSocket Server...');
+
+        // Close all active sessions
+        for (const [sessionId, ws] of this.sessions) {
+            ws.close();
+            this.#handleDisconnect(sessionId);
+        }
+
+        return new Promise((resolve, reject) => {
+            this.wss.close((err) => {
+                if (err) {
+                    logger.error('Error closing WebSocket Server:', err);
+                    reject(err);
+                } else {
+                    logger.info('WebSocket Server stopped.');
+                    this.wss = null;
+                    resolve();
+                }
+            });
+        });
+    }
+
     /**
      * Handles a new WebSocket connection.
      * @param {WebSocket} ws

ai/mcp/server/neural-link/services/toolService.mjs

@@ -21,7 +21,9 @@ const serviceMapping = {
     get_worker_topology   : RuntimeService.getWorkerTopology.bind(RuntimeService),
     healthcheck           : HealthService.healthcheck.bind(HealthService),
     reload_page           : RuntimeService.reloadPage.bind(RuntimeService),
-    set_component_property: ComponentService.setComponentProperty.bind(ComponentService)
+    set_component_property: ComponentService.setComponentProperty.bind(ComponentService),
+    start_ws_server       : ConnectionService.startServer.bind(ConnectionService),
+    stop_ws_server        : ConnectionService.stopServer.bind(ConnectionService)
 };
 
 const toolService = Neo.create(ToolService, {