septum 0.1.4 → 0.1.6

package/src/mcp/tools/trace-vertical-slice.ts

@@ -1,48 +1,74 @@
-import type { ValidatedSeptumConfig } from "../../core/config/schema.ts";
-import type { SeptumRepository } from "../../core/database/repository.ts";
-import { VerticalSliceTracer } from "../../core/resolver/vertical-slice-tracer.ts";
-import type { VerticalSliceTraceResponse } from "../../types/index.ts";
-
-export interface TraceVerticalSliceArgs {
-  query: string;
-}
-
-export function handleTraceVerticalSlice(
-  repo: SeptumRepository,
-  config: ValidatedSeptumConfig,
-  args: TraceVerticalSliceArgs
-) {
-  if (!args.query) {
-    throw new Error(
-      "Missing required argument: 'query' (e.g. 'POST /orders/{id}/status', 'OrderController@updateStatus', or task intent 'Ubah status order di dashboard')"
-    );
-  }
-
-  const workspaceRoot = (config.settings as Record<string, unknown>)?.workspace_root as string || process.cwd();
-  const tracer = new VerticalSliceTracer(repo, workspaceRoot);
-  const result: VerticalSliceTraceResponse = tracer.trace(args.query);
-
-  let formattedText = result.message;
-
-  if (result.found && result.alternatives && result.alternatives.length > 0) {
-    formattedText += `\n\n• Other Matching Slices:\n`;
-    for (const alt of result.alternatives) {
-      formattedText += `    - ${alt.method} ${alt.uri} -> ${alt.controller}@${alt.action}\n`;
-    }
-  } else if (!result.found && result.alternatives && result.alternatives.length > 0) {
-    formattedText += `\n\nAvailable Slices:\n`;
-    for (const alt of result.alternatives) {
-      formattedText += `    - ${alt.method} ${alt.uri} -> ${alt.controller}@${alt.action}\n`;
-    }
-  }
-
-  return {
-    content: [
-      {
-        type: "text",
-        text: formattedText.trim(),
-      },
-    ],
-    metadata: result,
-  };
-}
+import type { ValidatedSeptumConfig } from "../../core/config/schema.ts";
+import type { SeptumRepository } from "../../core/database/repository.ts";
+import { VerticalSliceTracer } from "../../core/resolver/vertical-slice-tracer.ts";
+import type { VerticalSliceTraceResponse } from "../../types/index.ts";
+import type { TraceVerticalSliceArgs } from "../schemas.ts";
+
+export type { TraceVerticalSliceArgs } from "../schemas.ts";
+
+export function handleTraceVerticalSlice(
+  repo: SeptumRepository,
+  config: ValidatedSeptumConfig,
+  args: Partial<TraceVerticalSliceArgs> & { route?: string },
+  workspaceRoot?: string
+) {
+  const query = (
+    args.query ||
+    args.symbol ||
+    args.route ||
+    ""
+  ).toString().trim();
+
+  if (!query) {
+    throw new Error(
+      "Missing required argument: 'query' (e.g. 'POST /orders/{id}/status', 'OrderController@updateStatus', or task intent 'Ubah status order di dashboard')"
+    );
+  }
+
+  const root =
+    workspaceRoot ||
+    args.workspace_path ||
+    ((config.settings as Record<string, unknown>)?.workspace_root as string) ||
+    process.cwd();
+  const maxDepth = Math.min(Math.max(Math.trunc(Number(args.max_depth ?? 5)), 1), 8);
+  const tracer = new VerticalSliceTracer(repo, root);
+  const result: VerticalSliceTraceResponse = (tracer.trace as any)(query, maxDepth);
+  const chain = result.chain || result.slice || [];
+
+  let summary = "";
+  if (result.found && chain.length > 0) {
+    summary = chain
+      .map((node) => `[${(node.stage || "STAGE").toUpperCase()}] ${node.symbol}`)
+      .join(" ➔ ");
+  } else if (result.found && result.entrypoint) {
+    summary = `[ENTRYPOINT] ${result.entrypoint.method} ${result.entrypoint.uri} ➔ ${result.entrypoint.controller}@${result.entrypoint.action}`;
+  } else {
+    summary = `[NOT FOUND] No vertical slice matching '${query}'`;
+  }
+
+  const payload = {
+    status: result.found ? "success" : "not_found",
+    query,
+    found: result.found,
+    confidence: result.confidence || (result.found ? "exact" : undefined),
+    is_exact_match: result.is_exact_match,
+    architecture_style: result.architecture_style,
+    max_depth_applied: result.max_depth_applied ?? maxDepth,
+    entrypoint: result.entrypoint || null,
+    chain,
+    alternatives: result.alternatives || [],
+    summary,
+    message: result.message || summary,
+  };
+
+  return {
+    content: [
+      {
+        type: "text" as const,
+        text: JSON.stringify(payload, null, 2),
+      },
+    ],
+    metadata: payload,
+  };
+}
+