@intentius/chant-lexicon-temporal 0.1.6 → 0.1.8

package/src/op/serializer.ts

@@ -75,7 +75,7 @@ function generateWorkflow(config: OpConfig): string {
   const lines: string[] = [
     "// Generated by chant — do not edit directly.",
     `// Source: ${config.name}.op.ts`,
-    "import { proxyActivities, condition, defineSignal, setHandler } from '@temporalio/workflow';",
+    "import { proxyActivities, condition, defineSignal, setHandler, upsertSearchAttributes } from '@temporalio/workflow';",
     "import { TEMPORAL_ACTIVITY_PROFILES } from '@intentius/chant-lexicon-temporal';",
     "import type * as activities from './activities';",
     "",
@@ -107,29 +107,91 @@ function generateWorkflow(config: OpConfig): string {
   // Workflow function
   lines.push(`export async function ${fnName}(): Promise<void> {`);
 
+  // Initial search attributes — OpName plus any user-provided attrs.
+  // Each value is wrapped in a single-element array (classic
+  // upsertSearchAttributes API takes arrays).
+  const initialAttrs: Record<string, string[]> = {
+    OpName: [config.name],
+  };
+  for (const [k, v] of Object.entries(config.searchAttributes ?? {})) {
+    initialAttrs[k] = [v];
+  }
+  lines.push(`  upsertSearchAttributes(${JSON.stringify(initialAttrs)});`);
+  lines.push("");
+
+  // Counter for outcome-attribute capture variables (workflow-scoped).
+  let resultCounter = 0;
+  const nextResultVar = (): string => `__r${resultCounter++}`;
+
+  // Build a `String(<var>?.<from-path>)` fragment from a dot-path.
+  const stringifyFromPath = (varName: string, from?: string): string => {
+    if (!from) return `String(${varName})`;
+    const parts = from.split(".");
+    return `String(${varName}?.${parts.join("?.")})`;
+  };
+
+  // Emit `upsertSearchAttributes({ <name>: [<expr>] })` for an outcome attr.
+  const emitOutcomeUpsert = (
+    step: ActivityStep,
+    varName: string,
+    indent = "  ",
+  ): string | null => {
+    if (!step.outcomeAttribute) return null;
+    const { name, from } = step.outcomeAttribute;
+    return `${indent}upsertSearchAttributes({ ${JSON.stringify(name)}: [${stringifyFromPath(varName, from)}] });`;
+  };
+
   const renderPhases = (phases: PhaseDefinition[]) => {
     for (const phase of phases) {
       const phaseLines: string[] = [];
       phaseLines.push(`  // Phase: ${phase.name}`);
+      phaseLines.push(`  upsertSearchAttributes({ Phase: ${JSON.stringify([phase.name])} });`);
 
       const activitySteps = phase.steps.filter(isActivityStep);
       const gateSteps = phase.steps.filter(isGateStep);
 
       if (phase.parallel && activitySteps.length > 1) {
-        phaseLines.push("  await Promise.all([");
-        for (const step of activitySteps) {
-          const argsStr = step.args && Object.keys(step.args).length > 0
-            ? JSON.stringify(step.args)
-            : "{}";
-          phaseLines.push(`    ${step.fn}(${argsStr}),`);
+        // Capture results into an array if any step has an outcome attribute,
+        // otherwise just await Promise.all without the destructure.
+        const anyOutcome = activitySteps.some((s) => s.outcomeAttribute);
+        if (anyOutcome) {
+          const vars = activitySteps.map(() => nextResultVar());
+          phaseLines.push(`  const [${vars.join(", ")}] = await Promise.all([`);
+          for (let i = 0; i < activitySteps.length; i++) {
+            const step = activitySteps[i];
+            const argsStr = step.args && Object.keys(step.args).length > 0
+              ? JSON.stringify(step.args)
+              : "{}";
+            phaseLines.push(`    ${step.fn}(${argsStr}),`);
+          }
+          phaseLines.push("  ]);");
+          for (let i = 0; i < activitySteps.length; i++) {
+            const upsert = emitOutcomeUpsert(activitySteps[i], vars[i]);
+            if (upsert) phaseLines.push(upsert);
+          }
+        } else {
+          phaseLines.push("  await Promise.all([");
+          for (const step of activitySteps) {
+            const argsStr = step.args && Object.keys(step.args).length > 0
+              ? JSON.stringify(step.args)
+              : "{}";
+            phaseLines.push(`    ${step.fn}(${argsStr}),`);
+          }
+          phaseLines.push("  ]);");
         }
-        phaseLines.push("  ]);");
       } else {
         for (const step of activitySteps) {
           const argsStr = step.args && Object.keys(step.args).length > 0
             ? JSON.stringify(step.args)
             : "{}";
-          phaseLines.push(`  await ${step.fn}(${argsStr});`);
+          if (step.outcomeAttribute) {
+            const v = nextResultVar();
+            phaseLines.push(`  const ${v} = await ${step.fn}(${argsStr});`);
+            const upsert = emitOutcomeUpsert(step, v);
+            if (upsert) phaseLines.push(upsert);
+          } else {
+            phaseLines.push(`  await ${step.fn}(${argsStr});`);
+          }
         }
       }
 

package/src/plugin.test.ts

@@ -32,7 +32,7 @@ describe("temporal plugin", () => {
     const tools = temporalPlugin.mcpTools?.();
     expect(Array.isArray(tools)).toBe(true);
     expect(tools?.length).toBe(1);
-    expect(tools?.[0].name).toBe("diff");
+    expect(tools?.[0].name).toBe("temporal:diff");
   });
 
   it("mcpResources() returns at least 2 resources including resource-catalog", () => {
@@ -40,7 +40,7 @@ describe("temporal plugin", () => {
     expect(Array.isArray(resources)).toBe(true);
     expect((resources?.length ?? 0)).toBeGreaterThanOrEqual(2);
     const uris = resources?.map((r) => r.uri);
-    expect(uris).toContain("resource-catalog");
+    expect(uris).toContain("temporal:resource-catalog");
   });
 
   it("skills() returns 2 skill entries", () => {

package/src/plugin.ts

@@ -103,6 +103,7 @@ export const temporalPlugin: LexiconPlugin = {
       createDiffTool(
         temporalSerializer,
         "Compare current Temporal build output (docker-compose.yml, temporal-setup.sh, schedules/) against previous version",
+        "temporal",
       ),
     ];
   },
@@ -114,6 +115,7 @@ export const temporalPlugin: LexiconPlugin = {
         "Temporal Resource Types",
         "All supported Temporal resource types: TemporalServer, TemporalNamespace, SearchAttribute, TemporalSchedule",
         "lexicon-temporal.json",
+        "temporal",
       ),
       {
         uri: "examples/dev-server",
@@ -283,4 +285,9 @@ export const temporalPlugin: LexiconPlugin = {
     const { generateDocs } = await import("./codegen/docs");
     return generateDocs(options);
   },
+
+  async describeResources(options) {
+    const { describeResources } = await import("./describe-resources");
+    return describeResources(options);
+  },
 };