Implement Neural Link Client Core (Tree & Props) #8170

tobiu · tobiu · commit a598a765cab6 · 2025-12-28T14:59:44.000+01:00
diff --git a/ai/mcp/server/neural-link/openapi.yaml b/ai/mcp/server/neural-link/openapi.yaml
@@ -67,6 +67,68 @@ paths:
               schema:
                 $ref: '#/components/schemas/ErrorResponse'
 
+  /component/vdom/get:
+    post:
+      summary: Get Component VDOM Tree
+      operationId: get_vdom_tree
+      x-pass-as-object: true
+      description: Retrieves the VDOM tree of a component.
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/GetComponentTreeRequest'
+      responses:
+        '200':
+          description: The VDOM tree structure
+          content:
+            application/json:
+              schema:
+                type: object
+        '400':
+          description: Invalid request body
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+        '500':
+          description: Internal server error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+
+  /component/vnode/get:
+    post:
+      summary: Get Component VNode Tree
+      operationId: get_vnode_tree
+      x-pass-as-object: true
+      description: Retrieves the VNode tree of a component.
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/GetComponentTreeRequest'
+      responses:
+        '200':
+          description: The VNode tree structure
+          content:
+            application/json:
+              schema:
+                type: object
+        '400':
+          description: Invalid request body
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+        '500':
+          description: Internal server error
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+
   /page/reload:
     post:
       summary: Reload Page
@@ -143,6 +205,12 @@ components:
     GetComponentTreeRequest:
       type: object
       properties:
+        rootId:
+          type: string
+          description: Optional root component ID. Defaults to the App's MainView if available.
+        depth:
+          type: integer
+          description: Depth limit. 1=Self, 2=Self+Children, -1=Infinite.
         windowId:
           type: string
           description: The target window ID
diff --git a/ai/mcp/server/neural-link/services/ConnectionService.mjs b/ai/mcp/server/neural-link/services/ConnectionService.mjs
@@ -231,11 +231,37 @@ class ConnectionService extends Base {
     /**
      * Retrieves the full component tree of the application.
      * @param {Object} opts            The options object.
+     * @param {Number} [opts.depth]    The depth limit.
+     * @param {String} [opts.rootId]   Optional root component ID.
      * @param {String} [opts.windowId] The target window ID.
      * @returns {Promise<Object>} The component tree structure.
      */
-    async getComponentTree({windowId}) {
-        return await this.#call(windowId, 'get_component_tree', {})
+    async getComponentTree({depth, rootId, windowId}) {
+        return await this.#call(windowId, 'get_component_tree', {depth, rootId})
+    }
+
+    /**
+     * Retrieves the VDOM tree of a component.
+     * @param {Object} opts            The options object.
+     * @param {Number} [opts.depth]    The depth limit.
+     * @param {String} [opts.rootId]   Optional root component ID.
+     * @param {String} [opts.windowId] The target window ID.
+     * @returns {Promise<Object>} The VDOM tree structure.
+     */
+    async getVdomTree({depth, rootId, windowId}) {
+        return await this.#call(windowId, 'get_vdom_tree', {depth, rootId})
+    }
+
+    /**
+     * Retrieves the VNode tree of a component.
+     * @param {Object} opts            The options object.
+     * @param {Number} [opts.depth]    The depth limit.
+     * @param {String} [opts.rootId]   Optional root component ID.
+     * @param {String} [opts.windowId] The target window ID.
+     * @returns {Promise<Object>} The VNode tree structure.
+     */
+    async getVnodeTree({depth, rootId, windowId}) {
+        return await this.#call(windowId, 'get_vnode_tree', {depth, rootId})
     }
 
     /**
diff --git a/ai/mcp/server/neural-link/services/toolService.mjs b/ai/mcp/server/neural-link/services/toolService.mjs
@@ -10,6 +10,8 @@ const openApiFilePath = path.join(__dirname, '../openapi.yaml');
 const serviceMapping = {
     get_component_property: ConnectionService.getComponentProperty.bind(ConnectionService),
     get_component_tree    : ConnectionService.getComponentTree.bind(ConnectionService),
+    get_vdom_tree         : ConnectionService.getVdomTree.bind(ConnectionService),
+    get_vnode_tree        : ConnectionService.getVnodeTree.bind(ConnectionService),
     reload_page           : ConnectionService.reloadPage.bind(ConnectionService),
     set_component_property: ConnectionService.setComponentProperty.bind(ConnectionService)
 };
diff --git a/src/ai/Client.mjs b/src/ai/Client.mjs
@@ -119,28 +119,91 @@ class Client extends Base {
      * @returns {Promise<*>} The result of the operation
      */
     async handleRequest(method, params) {
+        let me = this,
+            component;
+
         switch (method) {
-            case 'reload_page':
-                Neo.Main.reloadWindow();
-                return {status: 'reloading'};
+            case 'get_component_property':
+                component = Neo.getComponent(params.id);
+                if (!component) throw new Error(`Component not found: ${params.id}`);
+                return {value: me.safeSerialize(component[params.property])};
 
             case 'get_component_tree':
-                // TODO: Implement actual tree retrieval
-                return {root: 'viewport'};
+                return {tree: me.serializeComponent(me.getComponentRoot(params.rootId), params.depth || -1)};
 
-            case 'get_component_property':
-                // TODO: Implement property retrieval
-                return {value: null};
+            case 'get_vdom_tree':
+                component = me.getComponentRoot(params.rootId);
+                if (!component) throw new Error('Root component not found');
+                return {vdom: component.vdom};
+
+            case 'get_vnode_tree':
+                component = me.getComponentRoot(params.rootId);
+                if (!component) throw new Error('Root component not found');
+                return {vnode: component.vnode};
+
+            case 'reload_page':
+                Neo.Main.reloadWindow();
+                return {status: 'reloading'};
 
             case 'set_component_property':
-                // TODO: Implement property setting
+                component = Neo.getComponent(params.id);
+                if (!component) throw new Error(`Component not found: ${params.id}`);
+                component[params.property] = params.value;
                 return {success: true};
 
             default:
                 throw new Error(`Unknown method: ${method}`);
         }
     }
 
+    /**
+     * @param {String} [rootId]
+     * @returns {Neo.component.Base|null}
+     */
+    getComponentRoot(rootId) {
+        if (rootId) {
+            return Neo.getComponent(rootId)
+        }
+
+        const apps = Object.values(Neo.apps || {});
+
+        if (apps.length > 0) {
+            return apps[0].mainView
+        }
+
+        return null
+    }
+
+    /**
+     * @param {*} value
+     * @returns {*}
+     */
+    safeSerialize(value) {
+        const type = Neo.typeOf(value);
+
+        if (type === 'NeoInstance') {
+            return {
+                neoInstance: true,
+                id         : value.id,
+                className  : value.className
+            }
+        }
+
+        if (type === 'Object') {
+            const result = {};
+            Object.entries(value).forEach(([k, v]) => {
+                result[k] = this.safeSerialize(v)
+            });
+            return result
+        }
+
+        if (type === 'Array') {
+            return value.map(v => this.safeSerialize(v))
+        }
+
+        return value
+    }
+
     /**
      * Sends a JSON-RPC response
      * @param {Number|String} id
@@ -155,6 +218,32 @@ class Client extends Base {
             }))
         }
     }
+
+    /**
+     * @param {Neo.component.Base} component
+     * @param {Number} maxDepth
+     * @param {Number} currentDepth
+     * @returns {Object}
+     */
+    serializeComponent(component, maxDepth, currentDepth=1) {
+        if (!component) return null;
+
+        const result = {
+            id       : component.id,
+            className: component.className,
+            ntype    : component.ntype
+        };
+
+        if (maxDepth === -1 || currentDepth < maxDepth) {
+            const children = Neo.manager.Component.getChildren(component);
+
+            if (children && children.length > 0) {
+                result.items = children.map(child => this.serializeComponent(child, maxDepth, currentDepth + 1))
+            }
+        }
+
+        return result
+    }
 }
 
 export default Neo.setupClass(Client);
