dsh-desktop-kit 2.7.6 → 2.7.8

package/plugins/dsh-mcp-manager/lib/api.js

@@ -32,8 +32,17 @@ async function readEnvelope(request) {
   return { rpcId: body.rpcId, method: body.method, payload: body.payload ?? {} };
 }
 
-function publicServer(server) {
-  return {
+function promptText(result) {
+  const messages = Array.isArray(result?.messages) ? result.messages : [];
+  return messages.map((message) => {
+    const content = message?.content;
+    if (typeof content === "string") return content;
+    if (Array.isArray(content)) return content.map((block) => block?.text ?? "").filter(Boolean).join("\n");
+    return content?.text ?? "";
+  }).filter(Boolean).join("\n\n");
+}
+
+function publicServer(server) {  return {
     id: server.id,
     scope: server.scope,
     transport: server.url ? "http" : "stdio",
@@ -239,6 +248,28 @@ export function registerApi(ctx, manager) {
       error: record.error,
     };
   });
+  operation("prompts", async (payload) => {
+    const config = loadConfig({ cwd: process.cwd() });
+    const server = config.servers.find((item) => item.id === String(payload.id ?? ""));
+    if (!server) throw new Error("server not found");
+    const record = await manager.connect(server);
+    if (!record.client?.connected) return { prompts: [], error: record.error };
+    try {
+      const prompts = await manager.prompts(server.id);
+      record.prompts = prompts;
+      return { prompts, error: "" };
+    } catch (error) {
+      return { prompts: [], error: error instanceof Error ? error.message : String(error) };
+    }
+  });
+  operation("prompt", async (payload) => {
+    const config = loadConfig({ cwd: process.cwd() });
+    const server = config.servers.find((item) => item.id === String(payload.id ?? ""));
+    if (!server) throw new Error("server not found");
+    await manager.connect(server);
+    const result = await manager.prompt(server.id, String(payload.name ?? ""), payload.arguments ?? {});
+    return { text: promptText(result), description: result?.description ?? "" };
+  });
   operation("call", async (payload) => {
     const config = loadConfig({ cwd: process.cwd() });
     const server = config.servers.find((item) => item.id === String(payload.id ?? ""));

package/plugins/dsh-mcp-manager/lib/client.js

@@ -165,6 +165,7 @@ export class McpClient {
     this.capabilities = {};
     this.tools = [];
     this.resources = [];
+    this.prompts = [];
   }
 
   async connect() {
@@ -212,6 +213,23 @@ export class McpClient {
     return this.transport.request("resources/read", { uri }, this.config.timeoutMs);
   }
 
+  async listPrompts() {
+    if (!this.capabilities.prompts) return [];
+    const prompts = [];
+    let cursor;
+    do {
+      const result = await this.transport.request("prompts/list", cursor ? { cursor } : {}, this.config.timeoutMs);
+      if (Array.isArray(result?.prompts)) prompts.push(...result.prompts);
+      cursor = result?.nextCursor;
+    } while (cursor);
+    this.prompts = prompts;
+    return prompts;
+  }
+
+  async getPrompt(name, args = {}) {
+    return this.transport.request("prompts/get", { name, arguments: args }, this.config.timeoutMs);
+  }
+
   async close() {
     await this.transport.close?.();
     this.connected = false;

package/plugins/dsh-mcp-manager/lib/manager.js

@@ -53,7 +53,7 @@ export class McpManager {
         throw new Error("MCP manager is closed");
       }
       const tools = await client.listTools();
-      const record = { server, client, tools, resources: [], error: "" };
+      const record = { server, client, tools, resources: [], prompts: [], error: "" };
       this.clients.set(server.id, record);
       for (const tool of tools) {
         const name = this.toolName(server.id, tool.name);
@@ -62,7 +62,7 @@ export class McpManager {
       return record;
     } catch (error) {
       await client.close().catch(() => {});
-      const record = { server, client: null, tools: [], resources: [], error: error instanceof Error ? error.message : String(error) };
+      const record = { server, client: null, tools: [], resources: [], prompts: [], error: error instanceof Error ? error.message : String(error) };
       this.clients.set(server.id, record);
       return record;
     }
@@ -122,6 +122,18 @@ export class McpManager {
     return record.client.readResource(uri);
   }
 
+  async prompts(serverId) {
+    const record = this.clients.get(serverId);
+    if (!record?.client?.connected) throw new Error(`MCP server not connected: ${serverId}`);
+    return record.client.listPrompts();
+  }
+
+  async prompt(serverId, name, args) {
+    const record = this.clients.get(serverId);
+    if (!record?.client?.connected) throw new Error(`MCP server not connected: ${serverId}`);
+    return record.client.getPrompt(name, args);
+  }
+
   status() {
     return [...this.clients.values()].map((record) => ({
       id: record.server.id,