refactor: standardize tool response format and update dependencies (#8)

* refactor: standardize tool response format and update dependencies

* refactor: simplify path handling and improve tool responses

Author: Wu-kung

README.md

@@ -30,7 +30,6 @@ MCP collection for the [umi](https://github.com/umijs/umi) framework.
 - `umi-version`, Show the version of the umi project.
 - `umi-route-list`, List all routes of the umi project. (Based on the appData under `.umi` directory)
 - `umi-appdata-list`, List detailed app data of the umi project. (Based on the appData under `.umi` directory)
-- `umi-changelog`, Show the changelog of the umi and @umijs/max.
 
 ## Usage
 

package.json

@@ -35,21 +35,21 @@
   "license": "MIT",
   "devDependencies": {
     "@modelcontextprotocol/inspector": "^0.8.2",
-    "@modelcontextprotocol/sdk": "^1.9.0",
+    "@modelcontextprotocol/sdk": "^1.10.2",
     "@trivago/prettier-plugin-sort-imports": "^5.2.2",
-    "@types/node": "^22.14.0",
+    "@types/node": "^22.15.2",
     "@types/resolve": "^1.20.6",
     "@types/yargs-parser": "^21.0.3",
-    "@umijs/tools": "^0.1.34",
-    "fastmcp": "^1.21.0",
+    "@umijs/tools": "^0.1.36",
+    "fastmcp": "^1.22.4",
     "prettier": "^3.5.3",
     "resolve": "^1.22.10",
-    "takumi": "^0.0.7",
+    "takumi": "^0.0.15",
     "tsx": "^4.19.3",
     "typescript": "^5.8.3",
-    "vitest": "^3.1.1",
+    "vitest": "^3.1.2",
     "yargs-parser": "^21.1.1",
-    "zod": "^3.24.2"
+    "zod": "^3.24.3"
   },
   "volta": {
     "node": "22.11.0",

pnpm-lock.yaml

@@ -24,5 +24,4 @@ export function registerTools(toolContext: ToolContext) {
   umiLint(toolContext);
   umiRouteList(toolContext);
   umiAppdataList(toolContext);
-  // umiChangelog(toolContext); // Temporarily commented out as not needed
 }

src/index.ts

@@ -22,7 +22,7 @@ export function parse(root: string) {
       `It is not a umi or @umijs/max or @alipay/bigfish project: ${root}`,
     );
   }
-  const binPath = (() => {
+  const getBinPath = () => {
     try {
       if (isUmi) {
         return resolve.sync('umi/bin/umi.js', { basedir: root });
@@ -32,9 +32,22 @@ export function parse(root: string) {
       }
       return resolve.sync('@alipay/bigfish/bin/bigfish.js', { basedir: root });
     } catch (e) {
+      const binDir = path.join(root, 'node_modules', '.bin');
+      if (fs.existsSync(binDir)) {
+        if (isUmi && fs.existsSync(path.join(binDir, 'umi'))) {
+          return path.join(binDir, 'umi');
+        }
+        if (isMax && fs.existsSync(path.join(binDir, 'max'))) {
+          return path.join(binDir, 'max');
+        }
+        if (fs.existsSync(path.join(binDir, 'bigfish'))) {
+          return path.join(binDir, 'bigfish');
+        }
+      }
       throw new UserError(`Failed to resolve bin path: ${e}`);
     }
-  })();
+  };
+  const binPath = getBinPath();
   const binName = (() => {
     if (isUmi) {
       return 'umi';

src/parse.ts

@@ -13,10 +13,10 @@ function winJoin(...args: string[]) {
   return winPath(join(...args));
 }
 
-export function getPaths(cwd: string, frameworkName: string) {
+export function getPaths(cwd: string) {
   const src = winJoin(cwd, 'src');
   const absSrcPath = existsSync(src) && statSync(src).isDirectory() ? src : cwd;
-  const absTmpPath = winJoin(absSrcPath, `.${frameworkName}`);
+  const absTmpPath = winJoin(absSrcPath, '.umi');
   return {
     absSrcPath,
     absTmpPath,

src/path.ts

@@ -17,9 +17,11 @@ export const umiAppdataList = async ({
     parameters: z.object({}),
     execute: async (): Promise<any> => {
       try {
-        const { absTmpPath } = getPaths(root, frameworkName);
-
-        if (!existsSync(absTmpPath)) {
+        const { absTmpPath } = getPaths(root);
+        if (
+          !existsSync(absTmpPath) ||
+          !existsSync(`${absTmpPath}/appData.json`)
+        ) {
           const { binPath } = parse(root);
           execSync(`${binPath} setup`, { cwd: root });
         }
@@ -30,18 +32,17 @@ export const umiAppdataList = async ({
           `${appDataPath} is not exist, please upgrade to the latest version of ${frameworkName}`,
         );
         const appDataJson = JSON.parse(readFileSync(appDataPath, 'utf-8'));
-        return {
-          success: true,
-          data: {
-            type: 'text',
-            text: JSON.stringify(appDataJson, null, 2),
-          },
+        const summary = {
+          structure: Object.keys(appDataJson),
+          totalSize: JSON.stringify(appDataJson).length,
+          moreInfo: 'see src/.umi/appData.json',
         };
-      } catch (error: any) {
         return {
-          success: false,
-          data: error.message || 'Failed to list app data',
+          type: 'text',
+          text: JSON.stringify(summary, null, 2),
         };
+      } catch (error: any) {
+        return error.message || 'Failed to list app data';
       }
     },
   });

src/tools/umi-appdata-list.ts

@@ -50,11 +50,14 @@ export const umiBuild = async ({
           cwd: root,
           timeout: 5 * 60000,
         });
-        return { success: true, data: result.toString() };
+        return {
+          type: 'text',
+          text: result.toString()
+        };
       } catch (error: any) {
         return {
-          success: false,
-          data: `Build failed. Please check the error message above.\n${error.message || error}`,
+          type: 'text',
+          text: `Build failed. Please check the error message above.\n${error.message || error}`,
         };
       }
     },

src/tools/umi-build.ts

@@ -16,9 +16,15 @@ export const umiConfig = async ({
       try {
         const { binPath } = parse(root);
         const result = execSync(`${binPath} config list`, { cwd: root });
-        return { success: true, data: result.toString() };
+        return {
+          type: 'text',
+          text: result.toString()
+        };
       } catch (error: any) {
-        return { success: false, data: error.message || 'Failed to list config' };
+        return {
+          type: 'text',
+          text: error.message || 'Failed to list config'
+        };
       }
     },
   });
@@ -33,9 +39,15 @@ export const umiConfig = async ({
       try {
         const { binPath } = parse(root);
         const result = execSync(`${binPath} config get ${key}`, { cwd: root });
-        return { success: true, data: result.toString() };
+        return {
+          type: 'text',
+          text: result.toString()
+        };
       } catch (error: any) {
-        return { success: false, data: error.message || `Failed to get config key ${key}` };
+        return {
+          type: 'text',
+          text: error.message || `Failed to get config key ${key}`
+        };
       }
     },
   });
@@ -53,9 +65,15 @@ export const umiConfig = async ({
         const result = execSync(`${binPath} config set ${key} ${value}`, {
           cwd: root,
         });
-        return { success: true, data: result.toString() };
+        return {
+          type: 'text',
+          text: result.toString()
+        };
       } catch (error: any) {
-        return { success: false, data: error.message || `Failed to set config key ${key}` };
+        return {
+          type: 'text',
+          text: error.message || `Failed to set config key ${key}`
+        };
       }
     },
   });
@@ -72,9 +90,15 @@ export const umiConfig = async ({
         const result = execSync(`${binPath} config remove ${key}`, {
           cwd: root,
         });
-        return { success: true, data: result.toString() };
+        return {
+          type: 'text',
+          text: result.toString()
+        };
       } catch (error: any) {
-        return { success: false, data: error.message || `Failed to remove config key ${key}` };
+        return {
+          type: 'text',
+          text: error.message || `Failed to remove config key ${key}`
+        };
       }
     },
   });