@pikku/inspector 0.12.28 → 0.12.29

package/src/utils/workflow/dsl/extract-dsl-workflow.ts

@@ -7,6 +7,7 @@ import type {
   ParallelGroupStepMeta,
   FanoutStepMeta,
   CancelStepMeta,
+  SuspendStepMeta,
   SetStepMeta,
   SwitchStepMeta,
   SwitchCaseMeta,
@@ -21,6 +22,7 @@ import type {
 import {
   isWorkflowDoCall,
   isWorkflowSleepCall,
+  isWorkflowSuspendCall,
   isThrowCancelException,
   extractCancelReason,
   isParallelFanout,
@@ -512,6 +514,10 @@ function extractExpressionStatement(
       return extractSleepStep(call, context)
     }
 
+    if (isWorkflowSuspendCall(call, context.checker)) {
+      return extractSuspendStep(call, context)
+    }
+
     // Check for parallel group or fanout
     if (isParallelFanout(call)) {
       return extractParallelFanout(call, context)
@@ -698,6 +704,31 @@ function extractSleepStep(
   }
 }
 
+/**
+ * Extract suspend step from workflow.suspend() call
+ */
+function extractSuspendStep(
+  call: ts.CallExpression,
+  context: ExtractionContext
+): SuspendStepMeta | null {
+  const args = call.arguments
+  if (args.length < 1) return null
+
+  try {
+    const reason = extractStringLiteral(args[0], context.checker)
+    return {
+      type: 'suspend',
+      reason,
+    }
+  } catch (error) {
+    context.errors.push({
+      message: `Failed to extract suspend step: ${error instanceof Error ? error.message : String(error)}`,
+      node: call,
+    })
+    return null
+  }
+}
+
 /**
  * Extract cancel step from throw WorkflowCancelledException statement
  */

package/src/utils/workflow/dsl/patterns.ts

@@ -42,6 +42,25 @@ export function isWorkflowSleepCall(
   )
 }
 
+/**
+ * Check if a call expression is workflow.suspend()
+ */
+export function isWorkflowSuspendCall(
+  node: ts.CallExpression,
+  _checker: ts.TypeChecker
+): boolean {
+  if (!ts.isPropertyAccessExpression(node.expression)) {
+    return false
+  }
+
+  const propAccess = node.expression
+  return (
+    propAccess.name.text === 'suspend' &&
+    ts.isIdentifier(propAccess.expression) &&
+    propAccess.expression.text === 'workflow'
+  )
+}
+
 /**
  * Check if a throw statement throws WorkflowCancelledException
  * Matches: throw new WorkflowCancelledException(...) or throw WorkflowCancelledException(...)

package/src/visit.ts

@@ -87,6 +87,12 @@ export const visitSetup = (
   )
 
   addFileWithFactory(node, checker, state.configFactories, 'CreateConfig')
+  addFileWithFactory(
+    node,
+    checker,
+    state.serverLifecycleFactories,
+    'ServerLifecycle'
+  )
 
   addRPCInvocations(node, state, logger)
   addWireAddon(node, state, logger)