@gethmy/mcp 2.2.0 → 2.2.1

package/src/server.ts

@@ -59,13 +59,7 @@ import {
 } from "./context-assembly.js";
 import { autoExpandGraph } from "./graph-expansion.js";
 import { runLifecycleMaintenance } from "./lifecycle-maintenance.js";
-import {
-  type CardData,
-  type ColumnData,
-  generatePrompt,
-  type MemoryData,
-  type PromptVariant,
-} from "./prompt-builder.js";
+import type { PromptVariant } from "./prompt-builder.js";
 
 /**
  * Dependencies injected into tool handlers.
@@ -2699,14 +2693,11 @@ async function handleToolCall(
 
     // Prompt generation
     case "harmony_generate_prompt": {
-      // Get card - either by UUID or short ID
-      let cardData: CardData;
-      let columnData: ColumnData | null = null;
+      // Resolve card ID — either directly or via short ID
+      let cardId: string;
 
       if (args.cardId) {
-        const cardId = z.string().uuid().parse(args.cardId);
-        const cardResult = await client.getCard(cardId);
-        cardData = cardResult.card as CardData;
+        cardId = z.string().uuid().parse(args.cardId);
       } else if (args.shortId !== undefined) {
         const shortId = z.number().int().positive().parse(args.shortId);
         const projectId =
@@ -2717,35 +2708,12 @@ async function handleToolCall(
           );
         }
         const cardResult = await client.getCardByShortId(projectId, shortId);
-        cardData = cardResult.card as CardData;
+        cardId = (cardResult.card as { id: string }).id;
       } else {
         throw new Error("Either cardId or shortId must be provided");
       }
 
-      // Try to get column info from board
-      const projectIdForBoard =
-        (args.projectId as string) ||
-        getActiveProjectId() ||
-        (cardData as unknown as { project_id: string }).project_id;
-      if (projectIdForBoard) {
-        try {
-          const board = await client.getBoard(projectIdForBoard, {
-            summary: true,
-          });
-          const columnId = (cardData as unknown as { column_id: string })
-            .column_id;
-          const column = (
-            board.columns as Array<{ id: string; name: string }>
-          ).find((col) => col.id === columnId);
-          if (column) {
-            columnData = { name: column.name };
-          }
-        } catch {
-          // Column info not available, continue without it
-        }
-      }
-
-      const variant = (args.variant as PromptVariant) || "execute";
+      // Parse MCP-specific context options
       const contextOptions: Record<string, boolean> = {};
       if (args.includeSubtasks !== undefined) {
         contextOptions.includeSubtasks =
@@ -2761,89 +2729,24 @@ async function handleToolCall(
           args.includeDescription === "true";
       }
 
-      // Assemble context using the context assembly engine
-      let assembledContextStr: string | undefined;
-      let assemblyId: string | undefined;
-      let memories: MemoryData[] | undefined;
-
-      try {
-        const workspaceId = deps.getActiveWorkspaceId();
-        if (workspaceId && cardData.title) {
-          const cardLabels = (cardData.labels || []).map((l) => l.name);
-          const taskContext = [cardData.title, cardData.description || ""]
-            .filter(Boolean)
-            .join(" ");
-
-          const assembled = await assembleContext({
-            workspaceId,
-            projectId: getActiveProjectId() || undefined,
-            taskContext,
-            cardLabels,
-            cardId: cardData.id,
-            client,
-          });
-
-          if (assembled.context) {
-            assembledContextStr = assembled.context;
-            assemblyId = assembled.manifest.assemblyId;
-            cacheManifest(assembled.manifest);
-          }
-        }
-      } catch {
-        // Context assembly failed, try legacy fallback
-        try {
-          const workspaceId = deps.getActiveWorkspaceId();
-          if (workspaceId && cardData.title) {
-            const memoryResult = await client.searchMemoryEntities(
-              workspaceId,
-              cardData.title,
-              {
-                project_id: getActiveProjectId() || undefined,
-                limit: 5,
-              },
-            );
-            if (memoryResult.entities?.length > 0) {
-              memories = memoryResult.entities.map((e: unknown) => {
-                const entity = e as {
-                  id: string;
-                  type: string;
-                  title: string;
-                  content: string;
-                  confidence: number;
-                  tags: string[];
-                };
-                return {
-                  id: entity.id,
-                  type: entity.type,
-                  title: entity.title,
-                  content: entity.content,
-                  confidence: entity.confidence,
-                  tags: entity.tags || [],
-                };
-              });
-            }
-          }
-        } catch {
-          // Memory fetch also failed, continue without memories
-        }
-      }
-
-      const result = generatePrompt({
-        card: cardData,
-        column: columnData,
-        variant,
-        contextOptions,
+      // Delegate to the shared prompt generation pipeline
+      const result = await client.generateCardPrompt({
+        cardId,
+        workspaceId: deps.getActiveWorkspaceId() || "",
+        projectId:
+          (args.projectId as string) || getActiveProjectId() || undefined,
+        variant: (args.variant as PromptVariant) || "execute",
         customConstraints: args.customConstraints as string | undefined,
-        memories,
-        assembledContext: assembledContextStr,
-        assemblyId,
+        contextOptions,
       });
 
+      // MCP-specific: cache the assembly manifest for the feedback loop
+      if (result.assemblyId) {
+        trackSessionAssembly(cardId, result.assemblyId);
+      }
+
       return {
         success: true,
-        cardId: cardData.id,
-        shortId: cardData.short_id,
-        title: cardData.title,
         ...result,
       };
     }
@@ -3037,8 +2940,7 @@ async function handleToolCall(
       // Track memory write action and flush (fire-and-forget)
       const updateMemSession = getActiveMemorySession();
       if (updateMemSession) {
-        const updateTitle =
-          (updates.title as string) || entityId.slice(0, 8);
+        const updateTitle = (updates.title as string) || entityId.slice(0, 8);
         appendMemoryAction(
           updateMemSession.cardId,
           `Updated memory: ${updateTitle}`,
@@ -4053,13 +3955,17 @@ export class HarmonyMCPServer {
     await this.server.connect(transport);
     console.error("Harmony MCP server running on stdio");
 
-    // Initialize auto-session tracking
+    // Initialize auto-session tracking with MCP client identity detection
     const configDeps = createConfigDeps();
     initAutoSession(
       async (client, cardId, status) => {
         await runEndSessionPipeline(client, configDeps, cardId, status);
       },
       () => getClient(),
+      () => {
+        const cv = this.server.getClientVersion();
+        return cv ? { name: cv.name, version: cv.version } : null;
+      },
     );
 
     // Graceful shutdown: end all auto-sessions