Fix trigger script

Normalize trigger scripts across workbenches

Author: pranaygp

workbench/example/workflows/1_simple.ts

@@ -1,8 +1,18 @@
 import { FatalError } from 'workflow';
 
+function deepStepCallSync() {
+  throw new FatalError('Deep step call error');
+}
+
+async function deepStepCall() {
+  deepStepCallSync();
+}
+
 async function add(a: number, b: number): Promise<number> {
   'use step';
 
+  await deepStepCall();
+
   // Mimic a retryable error 50% of the time
   if (Math.random() < 0.5) {
     throw new Error('Retryable error');
@@ -16,10 +26,19 @@ async function add(a: number, b: number): Promise<number> {
   return a + b;
 }
 
+async function deepWorkflowCall(i: number) {
+  throw new Error('Error in workflow itself');
+
+  const a = await add(i, 7);
+  return a;
+}
+
 export async function simple(i: number) {
   'use workflow';
   console.log('Simple workflow started');
 
+  await deepWorkflowCall(i);
+
   const a = await add(i, 7);
   console.log('Workflow step 1 completed - Result:', a);
 

workbench/example/workflows/7_full.ts

@@ -0,0 +1,43 @@
+import { createWebhook, sleep } from 'workflow';
+
+export async function handleUserSignup(email: string) {
+  'use workflow';
+
+  const user = await createUser(email);
+  await sendWelcomeEmail(user);
+
+  await sleep('5s');
+
+  const webhook = createWebhook();
+  await sendOnboardingEmail(user, webhook.url);
+
+  await webhook;
+  console.log('Webhook Resolved');
+
+  return { userId: user.id, status: 'onboarded' };
+}
+
+async function createUser(email: string) {
+  'use step';
+
+  console.log(`Creating a new user with email: ${email}`);
+
+  return { id: crypto.randomUUID(), email };
+}
+
+async function sendWelcomeEmail(user: { id: string; email: string }) {
+  'use step';
+
+  console.log(`Sending welcome email to user: ${user.id}`);
+}
+
+async function sendOnboardingEmail(
+  user: { id: string; email: string },
+  callback: string
+) {
+  'use step';
+
+  console.log(`Sending onboarding email to user: ${user.id}`);
+
+  console.log(`Click this link to resolve the webhook: ${callback}`);
+}

workbench/hono/.gitignore

@@ -4,3 +4,4 @@ manifest.js
 .output
 .data
 .vercel
+_workflows.ts

workbench/hono/_workflows.ts

@@ -0,0 +1,26 @@
+// Auto-generated by workbench/scripts/generate-workflows-registry.js
+// Do not edit this file manually - it will be regenerated on build
+
+import * as workflow_0_demo from './workflows/0_demo.js';
+import * as workflow_1_simple from './workflows/1_simple.js';
+import * as workflow_2_control_flow from './workflows/2_control_flow.js';
+import * as workflow_3_streams from './workflows/3_streams.js';
+import * as workflow_4_ai from './workflows/4_ai.js';
+import * as workflow_5_hooks from './workflows/5_hooks.js';
+import * as workflow_6_batching from './workflows/6_batching.js';
+import * as workflow_7_full from './workflows/7_full.js';
+import * as workflow_98_duplicate_case from './workflows/98_duplicate_case.js';
+import * as workflow_99_e2e from './workflows/99_e2e.js';
+
+export const allWorkflows = {
+  'workflows/0_demo.ts': workflow_0_demo,
+  'workflows/1_simple.ts': workflow_1_simple,
+  'workflows/2_control_flow.ts': workflow_2_control_flow,
+  'workflows/3_streams.ts': workflow_3_streams,
+  'workflows/4_ai.ts': workflow_4_ai,
+  'workflows/5_hooks.ts': workflow_5_hooks,
+  'workflows/6_batching.ts': workflow_6_batching,
+  'workflows/7_full.ts': workflow_7_full,
+  'workflows/98_duplicate_case.ts': workflow_98_duplicate_case,
+  'workflows/99_e2e.ts': workflow_99_e2e,
+} as const;

workbench/hono/_workflows.ts

@@ -5,6 +5,9 @@
   "version": "0.0.0",
   "license": "Apache-2.0",
   "scripts": {
+    "generate:workflows": "node ../scripts/generate-workflows-registry.js",
+    "predev": "pnpm generate:workflows",
+    "prebuild": "pnpm generate:workflows",
     "dev": "nitro dev",
     "build": "nitro build"
   },

workbench/hono/package.json

@@ -40,3 +40,6 @@ yarn-error.log*
 *.tsbuildinfo
 next-env.d.ts
 .env*.local
+
+# workflow
+_workflows.ts

workbench/nextjs-turbopack/.gitignore

@@ -1,8 +1,6 @@
 import { getRun, start } from 'workflow/api';
 import { hydrateWorkflowArguments } from 'workflow/internal/serialization';
-import * as batchingWorkflow from '@/workflows/6_batching';
-import * as duplicateE2e from '@/workflows/98_duplicate_case';
-import * as e2eWorkflows from '@/workflows/99_e2e';
+import { allWorkflows } from '@/_workflows';
 
 export async function POST(req: Request) {
   const url = new URL(req.url);
@@ -35,16 +33,23 @@ export async function POST(req: Request) {
   );
 
   try {
-    let workflows;
-    if (workflowFile === 'workflows/99_e2e.ts') {
-      workflows = e2eWorkflows;
-    } else if (workflowFile === 'workflows/6_batching.ts') {
-      workflows = batchingWorkflow;
-    } else {
-      workflows = duplicateE2e;
+    const workflows = allWorkflows[workflowFile as keyof typeof allWorkflows];
+    if (!workflows) {
+      return Response.json(
+        { error: `Workflow file "${workflowFile}" not found` },
+        { status: 404 }
+      );
+    }
+
+    const workflow = workflows[workflowFn as keyof typeof workflows];
+    if (!workflow) {
+      return Response.json(
+        { error: `Function "${workflowFn}" not found in ${workflowFile}` },
+        { status: 400 }
+      );
     }
 
-    const run = await start((workflows as any)[workflowFn], args);
+    const run = await start(workflow as any, args);
     console.log('Run:', run);
     return Response.json(run);
   } catch (err) {

workbench/nextjs-turbopack/app/api/trigger/route.ts

@@ -4,9 +4,12 @@
   "private": true,
   "license": "Apache-2.0",
   "scripts": {
+    "generate:workflows": "node ../scripts/generate-workflows-registry.js",
+    "predev": "pnpm generate:workflows",
+    "prebuild": "pnpm generate:workflows",
     "dev": "next dev --turbopack",
     "build": "next build --turbopack",
-    "clean": "rm -rf .next .swc app/.well-known/workflow",
+    "clean": "rm -rf .next .swc app/.well-known/workflow _workflows.ts",
     "start": "next start",
     "lint": "next lint"
   },

workbench/nextjs-turbopack/package.json

@@ -0,0 +1 @@
+../../example/workflows/1_simple.ts