webmcp-nexus-core 0.1.12 → 0.1.13

package/dist/index.cjs

@@ -32,6 +32,7 @@ var index_exports = {};
 __export(index_exports, {
   extractProperties: () => extractProperties,
   extractToolsFromFile: () => extractToolsFromFile,
+  generateFieldSchemasInjectionCode: () => generateFieldSchemasInjectionCode,
   generateSchema: () => generateSchema,
   generateSchemaInjectionCode: () => generateSchemaInjectionCode,
   mapType: () => mapType,
@@ -44,7 +45,7 @@ module.exports = __toCommonJS(index_exports);
 // src/ts-extractor.ts
 var import_node_path = __toESM(require("path"), 1);
 var import_ts_morph = require("ts-morph");
-var REGISTRATION_FUNCTIONS = ["registerGlobalTools", "useWebMcpTools"];
+var REGISTRATION_FUNCTIONS = ["registerGlobalTools", "useWebMcpTools", "withWebMcpTools"];
 var isDebug = () => (process.env.DEBUG ?? "").toLowerCase().includes("webmcp");
 function resolveWithAlias(spec, alias) {
   if (!alias) return null;
@@ -158,6 +159,21 @@ function extractFunctionMetadata(node, _sourceFile) {
     }
     return { description, readOnly, properties };
   }
+  if (node.getKind() === import_ts_morph.SyntaxKind.MethodDeclaration) {
+    const methodDecl = node;
+    const jsDocs = methodDecl.getJsDocs();
+    if (jsDocs.length > 0) {
+      description = jsDocs[0].getDescription()?.trim() ?? "";
+      const tags = jsDocs[0].getTags();
+      readOnly = tags.some((tag) => tag.getTagName() === "readonly");
+    }
+    const params = methodDecl.getParameters();
+    if (params.length > 0) {
+      const paramType = params[0].getType();
+      properties = extractProperties(paramType);
+    }
+    return { description, readOnly, properties };
+  }
   if (node.getKind() === import_ts_morph.SyntaxKind.ArrowFunction || node.getKind() === import_ts_morph.SyntaxKind.FunctionExpression) {
     const funcNode = node;
     const parent = funcNode.getParent();
@@ -424,6 +440,119 @@ function resolveNamespaceImportArg(arg, sourceFile, project, projectRoot, alias)
   }
   return tools;
 }
+var REACT_LIFECYCLE_METHODS = /* @__PURE__ */ new Set([
+  "constructor",
+  "render",
+  "componentDidMount",
+  "componentDidUpdate",
+  "componentWillUnmount",
+  "shouldComponentUpdate",
+  "getSnapshotBeforeUpdate",
+  "componentDidCatch",
+  "getDerivedStateFromProps",
+  "getDerivedStateFromError",
+  "UNSAFE_componentWillMount",
+  "UNSAFE_componentWillReceiveProps",
+  "UNSAFE_componentWillUpdate"
+]);
+function resolveClassComponentArg(args, sourceFile, _project, projectRoot) {
+  const tools = [];
+  if (args.length === 0) return tools;
+  const firstArg = args[0];
+  if (firstArg.getKind() !== import_ts_morph.SyntaxKind.Identifier) return tools;
+  const symbol = firstArg.getSymbol();
+  if (!symbol) return tools;
+  const declarations = symbol.getDeclarations();
+  if (declarations.length === 0) return tools;
+  let classDecl;
+  for (const decl of declarations) {
+    if (decl.getKind() === import_ts_morph.SyntaxKind.ClassDeclaration) {
+      classDecl = decl;
+      break;
+    }
+    if (decl.getKind() === import_ts_morph.SyntaxKind.VariableDeclaration) {
+      const init = decl.getInitializer?.();
+      if (init && init.getKind() === import_ts_morph.SyntaxKind.ClassExpression) {
+        classDecl = init;
+        break;
+      }
+    }
+  }
+  if (!classDecl) return tools;
+  const className = firstArg.getText();
+  if (!className) {
+    console.warn(`[webmcp] withWebMcpTools: anonymous class is not supported, skipping.`);
+    return tools;
+  }
+  let explicitMethods = null;
+  if (args.length > 1 && args[1].getKind() === import_ts_morph.SyntaxKind.ArrayLiteralExpression) {
+    explicitMethods = /* @__PURE__ */ new Set();
+    const arrayLiteral = args[1];
+    for (const element of arrayLiteral.getElements()) {
+      if (element.getKind() === import_ts_morph.SyntaxKind.StringLiteral) {
+        explicitMethods.add(element.getLiteralValue());
+      }
+    }
+  }
+  const filePath = sourceFile.getFilePath();
+  const baseDir = projectRoot ?? process.cwd();
+  const relPath = "./" + import_node_path.default.relative(baseDir, filePath).replace(/\.tsx?$/, "");
+  const members = classDecl.getMembers?.() ?? [];
+  for (const member of members) {
+    const memberKind = member.getKind();
+    if (memberKind === import_ts_morph.SyntaxKind.MethodDeclaration) {
+      const name = member.getName?.();
+      if (!name || REACT_LIFECYCLE_METHODS.has(name)) continue;
+      if (explicitMethods && !explicitMethods.has(name)) continue;
+      const metadata = extractFunctionMetadata(member, sourceFile);
+      if (!explicitMethods && !metadata.description) continue;
+      tools.push({
+        name,
+        description: metadata.description,
+        properties: metadata.properties,
+        readOnly: metadata.readOnly,
+        sourceFile: relPath,
+        injectionTarget: `${className}.prototype.${name}`,
+        memberType: "prototype"
+      });
+    }
+    if (memberKind === import_ts_morph.SyntaxKind.PropertyDeclaration) {
+      const name = member.getName?.();
+      if (!name || REACT_LIFECYCLE_METHODS.has(name)) continue;
+      if (explicitMethods && !explicitMethods.has(name)) continue;
+      const initializer = member.getInitializer?.();
+      if (!initializer) continue;
+      const initKind = initializer.getKind();
+      if (initKind !== import_ts_morph.SyntaxKind.ArrowFunction && initKind !== import_ts_morph.SyntaxKind.FunctionExpression) continue;
+      let description = "";
+      let readOnly = false;
+      const jsDocs = member.getJsDocs?.();
+      if (jsDocs && jsDocs.length > 0) {
+        description = jsDocs[0].getDescription?.()?.trim() ?? "";
+        const tags = jsDocs[0].getTags?.() ?? [];
+        readOnly = tags.some((tag) => tag.getTagName() === "readonly");
+      }
+      if (!explicitMethods && !description) continue;
+      let properties = [];
+      const funcNode = initializer;
+      const params = funcNode.getParameters?.();
+      if (params && params.length > 0) {
+        const paramType = params[0].getType();
+        properties = extractProperties(paramType);
+      }
+      tools.push({
+        name,
+        description,
+        properties,
+        readOnly,
+        sourceFile: relPath,
+        injectionTarget: `${className}.__webmcpFieldSchemas`,
+        memberType: "field"
+      });
+    }
+  }
+  return tools;
+}
 function extractToolsFromFile(fileContent, filePath, projectRoot, alias) {
   const hasRegistration = REGISTRATION_FUNCTIONS.some((fn) => fileContent.includes(fn));
   if (!hasRegistration) return null;
@@ -467,6 +596,11 @@ function extractToolsFromFile(fileContent, filePath, projectRoot, alias) {
         start: callExpr.getStart()
       });
       const args = callExpr.getArguments();
+      if (exprText === "withWebMcpTools") {
+        const tools = resolveClassComponentArg(args, sourceFile, project, projectRoot);
+        result.tools.push(...tools);
+        return;
+      }
       for (const arg of args) {
         if (arg.getKind() === import_ts_morph.SyntaxKind.ObjectLiteralExpression) {
           const tools = resolveObjectLiteralArg(arg, sourceFile, project, projectRoot);
@@ -528,6 +662,19 @@ function generateSchemaInjectionCode(injectionTarget, description, properties, r
   };
   return `${injectionTarget}.__webmcpSchema = ${JSON.stringify(schemaObj, null, 2)};`;
 }
+function generateFieldSchemasInjectionCode(className, fieldTools) {
+  const schemas = {};
+  for (const tool of fieldTools) {
+    const inputSchema = generateSchema(tool.properties);
+    delete inputSchema.description;
+    schemas[tool.name] = {
+      description: tool.description,
+      inputSchema,
+      readOnly: tool.readOnly
+    };
+  }
+  return `${className}.__webmcpFieldSchemas = ${JSON.stringify(schemas, null, 2)};`;
+}
 function mapTypeToSchema(prop) {
   const schema = { type: "string" };
   if (prop.description) {
@@ -575,7 +722,7 @@ function mapTypeToSchema(prop) {
 // src/transform.ts
 var isDebug2 = () => (process.env.DEBUG ?? "").toLowerCase().includes("webmcp");
 function transformCode(code, filePath, options = {}) {
-  if (!code.includes("registerGlobalTools") && !code.includes("useWebMcpTools")) {
+  if (!code.includes("registerGlobalTools") && !code.includes("useWebMcpTools") && !code.includes("withWebMcpTools")) {
     return { code, transformed: false };
   }
   if (isDebug2()) {
@@ -589,6 +736,7 @@ function transformCode(code, filePath, options = {}) {
   }
   const injections = [];
   for (const tool of result.tools) {
+    if (tool.memberType === "field") continue;
     const injectionCode = generateSchemaInjectionCode(
       tool.injectionTarget,
       tool.description,
@@ -597,18 +745,42 @@ function transformCode(code, filePath, options = {}) {
     );
     injections.push(injectionCode);
   }
+  const fieldToolsByClass = /* @__PURE__ */ new Map();
+  for (const tool of result.tools) {
+    if (tool.memberType !== "field") continue;
+    const className = tool.injectionTarget.replace(".__webmcpFieldSchemas", "");
+    if (!fieldToolsByClass.has(className)) {
+      fieldToolsByClass.set(className, []);
+    }
+    fieldToolsByClass.get(className).push({
+      name: tool.name,
+      description: tool.description,
+      properties: tool.properties,
+      readOnly: tool.readOnly
+    });
+  }
+  for (const [className, fieldTools] of fieldToolsByClass) {
+    injections.push(generateFieldSchemasInjectionCode(className, fieldTools));
+  }
   if (injections.length === 0) {
     return { code, transformed: false };
   }
   const firstCall = result.registrationCalls.sort((a, b) => a.start - b.start)[0];
+  let insertPos = firstCall.start;
+  const textBefore = code.slice(0, firstCall.start);
+  const exportDefaultMatch = textBefore.match(/(export\s+default\s+)$/);
+  if (exportDefaultMatch) {
+    insertPos = firstCall.start - exportDefaultMatch[1].length;
+  }
   const injectionBlock = "\n// [webmcp-nexus-core] \u6784\u5EFA\u65F6\u6CE8\u5165\u7684 schema \u5143\u6570\u636E\n" + injections.join("\n") + "\n";
-  const newCode = code.slice(0, firstCall.start) + injectionBlock + code.slice(firstCall.start);
+  const newCode = code.slice(0, insertPos) + injectionBlock + code.slice(insertPos);
   return { code: newCode, transformed: true };
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   extractProperties,
   extractToolsFromFile,
+  generateFieldSchemasInjectionCode,
   generateSchema,
   generateSchemaInjectionCode,
   mapType,

package/dist/index.d.cts

@@ -40,6 +40,24 @@ declare function generateSchema(properties: PropertyInfo[], description?: string
  * @returns 注入代码字符串
  */
 declare function generateSchemaInjectionCode(injectionTarget: string, description: string, properties: PropertyInfo[], readOnly: boolean): string;
+/**
+ * 为 class field 箭头函数生成聚合的静态属性注入代码。
+ *
+ * 输出形如：
+ * ```
+ * ClassName.__webmcpFieldSchemas = { "method1": {...}, "method2": {...} };
+ * ```
+ *
+ * @param className - class 名称
+ * @param fieldTools - class field 工具元数据列表
+ * @returns 注入代码字符串
+ */
+declare function generateFieldSchemasInjectionCode(className: string, fieldTools: Array<{
+    name: string;
+    description: string;
+    properties: PropertyInfo[];
+    readOnly: boolean;
+}>): string;
 declare function mapTypeToSchema(prop: PropertyInfo): JsonSchema;
 
 /**
@@ -69,8 +87,15 @@ interface ExtractedTool {
      * 注入目标表达式。
      * 对于对象字面量参数中的本地变量：变量名，如 "searchInPanel"
      * 对于 namespace import：namespace.exportName，如 "userApi.getUser"
+     * 对于 class 原型方法："ClassName.prototype.methodName"
      */
     injectionTarget: string;
+    /**
+     * class 成员类型。仅在 withWebMcpTools 场景下有值。
+     * - 'prototype'：原型方法（MethodDeclaration）
+     * - 'field'：class field 箭头函数（PropertyDeclaration）
+     */
+    memberType?: 'prototype' | 'field';
 }
 /** 注册调用的提取结果 */
 interface ExtractionResult {
@@ -79,7 +104,7 @@ interface ExtractionResult {
     /** 注册调用的位置信息（用于在调用前注入代码） */
     registrationCalls: {
         /** 调用类型 */
-        type: 'registerGlobalTools' | 'useWebMcpTools';
+        type: 'registerGlobalTools' | 'useWebMcpTools' | 'withWebMcpTools';
         /** 调用在源码中的起始位置（字符偏移量） */
         start: number;
     }[];
@@ -104,7 +129,7 @@ declare function mapType(type: Type): string;
 /** 从一个 Type 提取所有属性信息 */
 declare function extractProperties(type: Type, depth?: number): PropertyInfo[];
 /**
- * 从源文件中提取所有 registerGlobalTools / useWebMcpTools 调用引用的工具
+ * 从源文件中提取所有 registerGlobalTools / useWebMcpTools / withWebMcpTools 调用引用的工具
  *
  * @param fileContent - 源文件内容
  * @param filePath - 源文件绝对路径
@@ -126,4 +151,4 @@ interface TransformResult {
 }
 declare function transformCode(code: string, filePath: string, options?: TransformOptions): TransformResult;
 
-export { type AliasMap, type ExtractedTool, type ExtractionResult, type JsonSchema, type PropertyInfo, type TransformOptions, type TransformResult, extractProperties, extractToolsFromFile, generateSchema, generateSchemaInjectionCode, mapType, mapTypeToSchema, resolveWithAlias, transformCode };
+export { type AliasMap, type ExtractedTool, type ExtractionResult, type JsonSchema, type PropertyInfo, type TransformOptions, type TransformResult, extractProperties, extractToolsFromFile, generateFieldSchemasInjectionCode, generateSchema, generateSchemaInjectionCode, mapType, mapTypeToSchema, resolveWithAlias, transformCode };

package/dist/index.d.ts

@@ -40,6 +40,24 @@ declare function generateSchema(properties: PropertyInfo[], description?: string
  * @returns 注入代码字符串
  */
 declare function generateSchemaInjectionCode(injectionTarget: string, description: string, properties: PropertyInfo[], readOnly: boolean): string;
+/**
+ * 为 class field 箭头函数生成聚合的静态属性注入代码。
+ *
+ * 输出形如：
+ * ```
+ * ClassName.__webmcpFieldSchemas = { "method1": {...}, "method2": {...} };
+ * ```
+ *
+ * @param className - class 名称
+ * @param fieldTools - class field 工具元数据列表
+ * @returns 注入代码字符串
+ */
+declare function generateFieldSchemasInjectionCode(className: string, fieldTools: Array<{
+    name: string;
+    description: string;
+    properties: PropertyInfo[];
+    readOnly: boolean;
+}>): string;
 declare function mapTypeToSchema(prop: PropertyInfo): JsonSchema;
 
 /**
@@ -69,8 +87,15 @@ interface ExtractedTool {
      * 注入目标表达式。
      * 对于对象字面量参数中的本地变量：变量名，如 "searchInPanel"
      * 对于 namespace import：namespace.exportName，如 "userApi.getUser"
+     * 对于 class 原型方法："ClassName.prototype.methodName"
      */
     injectionTarget: string;
+    /**
+     * class 成员类型。仅在 withWebMcpTools 场景下有值。
+     * - 'prototype'：原型方法（MethodDeclaration）
+     * - 'field'：class field 箭头函数（PropertyDeclaration）
+     */
+    memberType?: 'prototype' | 'field';
 }
 /** 注册调用的提取结果 */
 interface ExtractionResult {
@@ -79,7 +104,7 @@ interface ExtractionResult {
     /** 注册调用的位置信息（用于在调用前注入代码） */
     registrationCalls: {
         /** 调用类型 */
-        type: 'registerGlobalTools' | 'useWebMcpTools';
+        type: 'registerGlobalTools' | 'useWebMcpTools' | 'withWebMcpTools';
         /** 调用在源码中的起始位置（字符偏移量） */
         start: number;
     }[];
@@ -104,7 +129,7 @@ declare function mapType(type: Type): string;
 /** 从一个 Type 提取所有属性信息 */
 declare function extractProperties(type: Type, depth?: number): PropertyInfo[];
 /**
- * 从源文件中提取所有 registerGlobalTools / useWebMcpTools 调用引用的工具
+ * 从源文件中提取所有 registerGlobalTools / useWebMcpTools / withWebMcpTools 调用引用的工具
  *
  * @param fileContent - 源文件内容
  * @param filePath - 源文件绝对路径
@@ -126,4 +151,4 @@ interface TransformResult {
 }
 declare function transformCode(code: string, filePath: string, options?: TransformOptions): TransformResult;
 
-export { type AliasMap, type ExtractedTool, type ExtractionResult, type JsonSchema, type PropertyInfo, type TransformOptions, type TransformResult, extractProperties, extractToolsFromFile, generateSchema, generateSchemaInjectionCode, mapType, mapTypeToSchema, resolveWithAlias, transformCode };
+export { type AliasMap, type ExtractedTool, type ExtractionResult, type JsonSchema, type PropertyInfo, type TransformOptions, type TransformResult, extractProperties, extractToolsFromFile, generateFieldSchemasInjectionCode, generateSchema, generateSchemaInjectionCode, mapType, mapTypeToSchema, resolveWithAlias, transformCode };

package/dist/index.js

@@ -4,7 +4,7 @@ import {
   Project,
   SyntaxKind
 } from "ts-morph";
-var REGISTRATION_FUNCTIONS = ["registerGlobalTools", "useWebMcpTools"];
+var REGISTRATION_FUNCTIONS = ["registerGlobalTools", "useWebMcpTools", "withWebMcpTools"];
 var isDebug = () => (process.env.DEBUG ?? "").toLowerCase().includes("webmcp");
 function resolveWithAlias(spec, alias) {
   if (!alias) return null;
@@ -118,6 +118,21 @@ function extractFunctionMetadata(node, _sourceFile) {
     }
     return { description, readOnly, properties };
   }
+  if (node.getKind() === SyntaxKind.MethodDeclaration) {
+    const methodDecl = node;
+    const jsDocs = methodDecl.getJsDocs();
+    if (jsDocs.length > 0) {
+      description = jsDocs[0].getDescription()?.trim() ?? "";
+      const tags = jsDocs[0].getTags();
+      readOnly = tags.some((tag) => tag.getTagName() === "readonly");
+    }
+    const params = methodDecl.getParameters();
+    if (params.length > 0) {
+      const paramType = params[0].getType();
+      properties = extractProperties(paramType);
+    }
+    return { description, readOnly, properties };
+  }
   if (node.getKind() === SyntaxKind.ArrowFunction || node.getKind() === SyntaxKind.FunctionExpression) {
     const funcNode = node;
     const parent = funcNode.getParent();
@@ -384,6 +399,119 @@ function resolveNamespaceImportArg(arg, sourceFile, project, projectRoot, alias)
   }
   return tools;
 }
+var REACT_LIFECYCLE_METHODS = /* @__PURE__ */ new Set([
+  "constructor",
+  "render",
+  "componentDidMount",
+  "componentDidUpdate",
+  "componentWillUnmount",
+  "shouldComponentUpdate",
+  "getSnapshotBeforeUpdate",
+  "componentDidCatch",
+  "getDerivedStateFromProps",
+  "getDerivedStateFromError",
+  "UNSAFE_componentWillMount",
+  "UNSAFE_componentWillReceiveProps",
+  "UNSAFE_componentWillUpdate"
+]);
+function resolveClassComponentArg(args, sourceFile, _project, projectRoot) {
+  const tools = [];
+  if (args.length === 0) return tools;
+  const firstArg = args[0];
+  if (firstArg.getKind() !== SyntaxKind.Identifier) return tools;
+  const symbol = firstArg.getSymbol();
+  if (!symbol) return tools;
+  const declarations = symbol.getDeclarations();
+  if (declarations.length === 0) return tools;
+  let classDecl;
+  for (const decl of declarations) {
+    if (decl.getKind() === SyntaxKind.ClassDeclaration) {
+      classDecl = decl;
+      break;
+    }
+    if (decl.getKind() === SyntaxKind.VariableDeclaration) {
+      const init = decl.getInitializer?.();
+      if (init && init.getKind() === SyntaxKind.ClassExpression) {
+        classDecl = init;
+        break;
+      }
+    }
+  }
+  if (!classDecl) return tools;
+  const className = firstArg.getText();
+  if (!className) {
+    console.warn(`[webmcp] withWebMcpTools: anonymous class is not supported, skipping.`);
+    return tools;
+  }
+  let explicitMethods = null;
+  if (args.length > 1 && args[1].getKind() === SyntaxKind.ArrayLiteralExpression) {
+    explicitMethods = /* @__PURE__ */ new Set();
+    const arrayLiteral = args[1];
+    for (const element of arrayLiteral.getElements()) {
+      if (element.getKind() === SyntaxKind.StringLiteral) {
+        explicitMethods.add(element.getLiteralValue());
+      }
+    }
+  }
+  const filePath = sourceFile.getFilePath();
+  const baseDir = projectRoot ?? process.cwd();
+  const relPath = "./" + nodePath.relative(baseDir, filePath).replace(/\.tsx?$/, "");
+  const members = classDecl.getMembers?.() ?? [];
+  for (const member of members) {
+    const memberKind = member.getKind();
+    if (memberKind === SyntaxKind.MethodDeclaration) {
+      const name = member.getName?.();
+      if (!name || REACT_LIFECYCLE_METHODS.has(name)) continue;
+      if (explicitMethods && !explicitMethods.has(name)) continue;
+      const metadata = extractFunctionMetadata(member, sourceFile);
+      if (!explicitMethods && !metadata.description) continue;
+      tools.push({
+        name,
+        description: metadata.description,
+        properties: metadata.properties,
+        readOnly: metadata.readOnly,
+        sourceFile: relPath,
+        injectionTarget: `${className}.prototype.${name}`,
+        memberType: "prototype"
+      });
+    }
+    if (memberKind === SyntaxKind.PropertyDeclaration) {
+      const name = member.getName?.();
+      if (!name || REACT_LIFECYCLE_METHODS.has(name)) continue;
+      if (explicitMethods && !explicitMethods.has(name)) continue;
+      const initializer = member.getInitializer?.();
+      if (!initializer) continue;
+      const initKind = initializer.getKind();
+      if (initKind !== SyntaxKind.ArrowFunction && initKind !== SyntaxKind.FunctionExpression) continue;
+      let description = "";
+      let readOnly = false;
+      const jsDocs = member.getJsDocs?.();
+      if (jsDocs && jsDocs.length > 0) {
+        description = jsDocs[0].getDescription?.()?.trim() ?? "";
+        const tags = jsDocs[0].getTags?.() ?? [];
+        readOnly = tags.some((tag) => tag.getTagName() === "readonly");
+      }
+      if (!explicitMethods && !description) continue;
+      let properties = [];
+      const funcNode = initializer;
+      const params = funcNode.getParameters?.();
+      if (params && params.length > 0) {
+        const paramType = params[0].getType();
+        properties = extractProperties(paramType);
+      }
+      tools.push({
+        name,
+        description,
+        properties,
+        readOnly,
+        sourceFile: relPath,
+        injectionTarget: `${className}.__webmcpFieldSchemas`,
+        memberType: "field"
+      });
+    }
+  }
+  return tools;
+}
 function extractToolsFromFile(fileContent, filePath, projectRoot, alias) {
   const hasRegistration = REGISTRATION_FUNCTIONS.some((fn) => fileContent.includes(fn));
   if (!hasRegistration) return null;
@@ -427,6 +555,11 @@ function extractToolsFromFile(fileContent, filePath, projectRoot, alias) {
         start: callExpr.getStart()
       });
       const args = callExpr.getArguments();
+      if (exprText === "withWebMcpTools") {
+        const tools = resolveClassComponentArg(args, sourceFile, project, projectRoot);
+        result.tools.push(...tools);
+        return;
+      }
       for (const arg of args) {
         if (arg.getKind() === SyntaxKind.ObjectLiteralExpression) {
           const tools = resolveObjectLiteralArg(arg, sourceFile, project, projectRoot);
@@ -488,6 +621,19 @@ function generateSchemaInjectionCode(injectionTarget, description, properties, r
   };
   return `${injectionTarget}.__webmcpSchema = ${JSON.stringify(schemaObj, null, 2)};`;
 }
+function generateFieldSchemasInjectionCode(className, fieldTools) {
+  const schemas = {};
+  for (const tool of fieldTools) {
+    const inputSchema = generateSchema(tool.properties);
+    delete inputSchema.description;
+    schemas[tool.name] = {
+      description: tool.description,
+      inputSchema,
+      readOnly: tool.readOnly
+    };
+  }
+  return `${className}.__webmcpFieldSchemas = ${JSON.stringify(schemas, null, 2)};`;
+}
 function mapTypeToSchema(prop) {
   const schema = { type: "string" };
   if (prop.description) {
@@ -535,7 +681,7 @@ function mapTypeToSchema(prop) {
 // src/transform.ts
 var isDebug2 = () => (process.env.DEBUG ?? "").toLowerCase().includes("webmcp");
 function transformCode(code, filePath, options = {}) {
-  if (!code.includes("registerGlobalTools") && !code.includes("useWebMcpTools")) {
+  if (!code.includes("registerGlobalTools") && !code.includes("useWebMcpTools") && !code.includes("withWebMcpTools")) {
     return { code, transformed: false };
   }
   if (isDebug2()) {
@@ -549,6 +695,7 @@ function transformCode(code, filePath, options = {}) {
   }
   const injections = [];
   for (const tool of result.tools) {
+    if (tool.memberType === "field") continue;
     const injectionCode = generateSchemaInjectionCode(
       tool.injectionTarget,
       tool.description,
@@ -557,17 +704,41 @@ function transformCode(code, filePath, options = {}) {
     );
     injections.push(injectionCode);
   }
+  const fieldToolsByClass = /* @__PURE__ */ new Map();
+  for (const tool of result.tools) {
+    if (tool.memberType !== "field") continue;
+    const className = tool.injectionTarget.replace(".__webmcpFieldSchemas", "");
+    if (!fieldToolsByClass.has(className)) {
+      fieldToolsByClass.set(className, []);
+    }
+    fieldToolsByClass.get(className).push({
+      name: tool.name,
+      description: tool.description,
+      properties: tool.properties,
+      readOnly: tool.readOnly
+    });
+  }
+  for (const [className, fieldTools] of fieldToolsByClass) {
+    injections.push(generateFieldSchemasInjectionCode(className, fieldTools));
+  }
   if (injections.length === 0) {
     return { code, transformed: false };
   }
   const firstCall = result.registrationCalls.sort((a, b) => a.start - b.start)[0];
+  let insertPos = firstCall.start;
+  const textBefore = code.slice(0, firstCall.start);
+  const exportDefaultMatch = textBefore.match(/(export\s+default\s+)$/);
+  if (exportDefaultMatch) {
+    insertPos = firstCall.start - exportDefaultMatch[1].length;
+  }
   const injectionBlock = "\n// [webmcp-nexus-core] \u6784\u5EFA\u65F6\u6CE8\u5165\u7684 schema \u5143\u6570\u636E\n" + injections.join("\n") + "\n";
-  const newCode = code.slice(0, firstCall.start) + injectionBlock + code.slice(firstCall.start);
+  const newCode = code.slice(0, insertPos) + injectionBlock + code.slice(insertPos);
   return { code: newCode, transformed: true };
 }
 export {
   extractProperties,
   extractToolsFromFile,
+  generateFieldSchemasInjectionCode,
   generateSchema,
   generateSchemaInjectionCode,
   mapType,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "webmcp-nexus-core",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "description": "Core logic for WebMCP build plugins - TypeScript type extraction and JSON Schema generation",
   "license": "MIT",
   "keywords": [