feat: update

Author: Groupguanfang

.cursor/mcp.json

@@ -2,7 +2,7 @@
   "mcpServers": {
     "arkts": {
       "command": "node",
-      "args": ["dist/bin.mjs"]
+      "args": [".cache/bin.cjs"]
     }
   }
 }

package.json

@@ -50,6 +50,7 @@
     "axios": "^1.15.0",
     "minisearch": "^7.2.0",
     "nodejieba": "^3.5.8",
+    "source-map-support": "^0.5.21",
     "zod": "^4.3.6"
   },
   "devDependencies": {

pnpm-lock.yaml

@@ -2,6 +2,7 @@
 import process from 'node:process'
 import { StdioServerTransport } from '@modelcontextprotocol/server'
 import { createArktsMcpServer } from './index'
+import 'source-map-support/register'
 
 async function main(): Promise<void> {
   const server = await createArktsMcpServer()

src/bin.ts

@@ -22,28 +22,27 @@ export async function createArktsMcpServer(options: ArktsMcpServerOptions = {}):
   // Lazy or eager load the tools
   if (import.meta.env.BUILD_TYPE === 'BIN') {
     const mods = Object.entries(import.meta.glob<ToolModule>('./tools/**/*.tool.{ts,md}', { eager: true }))
-    for (const [path, mod] of mods) {
-      installTools(path, mod, mods)
-    }
+    await Promise.all(mods.map(async ([path, mod]) => /* @__PURE__ */ await installTools(path, mod, mods)))
   }
   else {
     const mods = Object.entries(import.meta.glob<ToolModule>('./tools/**/*.tool.{ts,md}'))
-    for (const [path, mod] of mods) {
-      if (path.endsWith('.md')) continue
-      installTools(path, await mod(), mods)
-    }
+    await Promise.all(mods.map(async ([path, mod]) => /* @__PURE__ */ await installTools(path, await mod(), mods)))
   }
 
   return server
 
   async function installTools(path: string, module: ToolModule, mods: [string, (() => Promise<ToolModule>) | ToolModule][]): Promise<void> {
     if (typeof module.default === 'object' && module.default !== null) {
+      const description = await findDescription(path, mods)
+      const title = await findTitle(description ?? '')
+
       server.registerTool(module.default.name, {
-        description: await findDescription(path, mods),
+        title,
+        description,
         ...module.default,
       }, module.default.execute)
     }
-    module.install?.(server)
+    await module.install?.(server)
   }
 
   async function findDescription(path: string, mods: [string, (() => Promise<ToolModule | MarkdownModule>) | ToolModule | MarkdownModule][]): Promise<string | undefined> {
@@ -54,4 +53,10 @@ export async function createArktsMcpServer(options: ArktsMcpServerOptions = {}):
       return loadedMod.default as string
     }
   }
+
+  async function findTitle(description: string): Promise<string | undefined> {
+    const maybeTitle = description.trim().split('\n')?.[0]
+    if (maybeTitle?.startsWith('# ')) return maybeTitle.slice(2)
+    return undefined
+  }
 }

src/index.ts

@@ -51,10 +51,7 @@ export async function requestCatalogTree(language: 'cn' | 'en'): Promise<Catalog
   return response.data
 }
 
-export async function searchCatalogTree(
-  text: string,
-  flattenTree: FlattenCatalogTreeItem[],
-): Promise<SearchResult[]> {
+export async function searchCatalogTree(text: string, flattenTree: FlattenCatalogTreeItem[]): Promise<SearchResult[]> {
   const miniSearch = new MiniSearch({
     fields: ['nodeName'],
     idField: 'nodeId',

src/tools/docs/getCatalogTree.tool.ts

@@ -1,13 +1,16 @@
-import type { McpServer, StandardSchemaWithJSON, ToolCallback } from '@modelcontextprotocol/server'
+import type { McpServer, StandardSchemaWithJSON, ToolAnnotations, ToolCallback } from '@modelcontextprotocol/server'
 import type { Awaitable } from '@vstils/core'
 
 export type InstallHook = (server: McpServer) => Awaitable<unknown>
 
 export interface McpTool<OutputArgs extends StandardSchemaWithJSON = StandardSchemaWithJSON, InputArgs extends StandardSchemaWithJSON | undefined = undefined> {
   readonly name: string
+  readonly title?: string
   readonly description?: string
   readonly inputSchema?: InputArgs
   readonly outputSchema?: OutputArgs
+  readonly annotations?: ToolAnnotations
+  readonly _meta?: Record<string, unknown>
   readonly execute: ToolCallback<InputArgs>
 }
 
@@ -16,6 +19,8 @@ export interface ToolModule {
   readonly default?: McpTool
 }
 
-export function defineMcpTool<OutputArgs extends StandardSchemaWithJSON = StandardSchemaWithJSON, InputArgs extends StandardSchemaWithJSON | undefined = undefined>(mcpTool: McpTool<OutputArgs, InputArgs>): McpTool<OutputArgs, InputArgs> {
-  return mcpTool
-}
+// eslint-disable-next-line antfu/top-level-function
+export const defineMcpTool = <
+  OutputArgs extends StandardSchemaWithJSON = StandardSchemaWithJSON,
+  InputArgs extends StandardSchemaWithJSON | undefined = undefined,
+>(mcpTool: McpTool<OutputArgs, InputArgs>): McpTool<OutputArgs, InputArgs> => mcpTool

src/tools/types.ts

@@ -20,22 +20,24 @@ export default defineConfig({
       format: 'esm',
       minify: true,
       dts: false,
+      sourcemap: 'inline',
       env: { BUILD_TYPE: 'BIN' },
       plugins: [markdownPlugin],
       deps: {
         onlyBundle: false,
       },
-      copy: './src/assets',
     } satisfies PackUserConfig,
     {
       entry: 'src/index.ts',
       format: ['esm', 'cjs'],
       dts: true,
+      sourcemap: 'inline',
       env: { BUILD_TYPE: 'LIB' },
       plugins: [markdownPlugin],
       deps: {
         onlyBundle: false,
       },
+      copy: 'src/assets',
     } satisfies PackUserConfig,
     process.argv.includes('--build-exe')
       ? ({
@@ -44,6 +46,7 @@ export default defineConfig({
           outDir: '.cache',
           // Use BIN here so the Rolldown graph matches dist/bin.cjs. EXE previously produced a
           // lazy-init Zod layout that broke @modelcontextprotocol/server's module-scope z.url().
+          sourcemap: 'inline',
           env: { BUILD_TYPE: 'BIN' },
           deps: {
             alwaysBundle: Object.keys(packageJson.dependencies).map(dep => new RegExp(`^${dep}`)),
@@ -54,6 +57,15 @@ export default defineConfig({
             outDir: 'target',
             fileName: 'arkts-mcp',
           },
+          minify: {
+            compress: {
+              keepNames: {
+                class: true,
+                function: true,
+              },
+            },
+            mangle: false,
+          },
           plugins: [markdownPlugin, nodejiebaPlugin],
           copy: [
             ...nodejiebaPlugin.api?.buildCopyConfig(baseCopyDir) ?? [],