Apply workflow function transformation in "step" mode

Author: TooTallNate

.changeset/cruel-crabs-mate.md

@@ -0,0 +1,5 @@
+---
+"@workflow/swc-plugin": patch
+---
+
+Apply workflow function transformation in "step" mode

packages/swc-plugin-workflow/transform/src/lib.rs

@@ -22,7 +22,7 @@ export async function validStep() {
     return 42;
 }
 export const validWorkflow = async ()=>{
-    'use workflow';
-    return 'test';
+    throw new Error("You attempted to execute workflow validWorkflow function directly. To start a workflow, use start(validWorkflow) from workflow/api");
 };
+validWorkflow.workflowId = "workflow//input.js//validWorkflow";
 registerStepFunction("step//input.js//validStep", validStep);

packages/swc-plugin-workflow/transform/tests/errors/non-async-functions/output-step.js

@@ -7,11 +7,9 @@ async function stepFunctionWithoutExport(a, b) {
     return a - b;
 }
 export async function workflowFunction(a, b) {
-    'use workflow';
-    const result = await stepFunction(a, b);
-    const result2 = await stepFunctionWithoutExport(a, b);
-    return result + result2;
+    throw new Error("You attempted to execute workflow workflowFunction function directly. To start a workflow, use start(workflowFunction) from workflow/api");
 }
+workflowFunction.workflowId = "workflow//input.js//workflowFunction";
 export async function normalFunction(a, b) {
     return a * b;
 }

packages/swc-plugin-workflow/transform/tests/fixture/mixed-functions/output-step.js

@@ -7,8 +7,10 @@ const localArrow = async (input)=>{
     return input.bar;
 };
 export async function workflow(input) {
-    return input.foo;
+    throw new Error("You attempted to execute workflow workflow function directly. To start a workflow, use start(workflow) from workflow/api");
 }
+workflow.workflowId = "workflow//input.js//workflow";
 export const arrowWorkflow = async (input)=>{
     return input.bar;
 };
+arrowWorkflow.workflowId = "workflow//input.js//arrowWorkflow";

packages/swc-plugin-workflow/transform/tests/fixture/module-level-workflow/output-step.js

@@ -1,37 +1,21 @@
 import { registerStepFunction } from "workflow/internal/private";
-/**__internal_workflows{"workflows":{"input.js":{"example":{"workflowId":"workflow//input.js//example"}}},"steps":{"input.js":{"arrowStep":{"stepId":"step//input.js//arrowStep"},"helpers/objectStep":{"stepId":"step//input.js//helpers/objectStep"},"letArrowStep":{"stepId":"step//input.js//letArrowStep"},"step":{"stepId":"step//input.js//step"},"varArrowStep":{"stepId":"step//input.js//varArrowStep"}}}}*/;
+/**__internal_workflows{"workflows":{"input.js":{"example":{"workflowId":"workflow//input.js//example"}}},"steps":{"input.js":{"arrowStep":{"stepId":"step//input.js//arrowStep"},"helpers/objectStep":{"stepId":"step//input.js//example/helpers/objectStep"},"letArrowStep":{"stepId":"step//input.js//letArrowStep"},"step":{"stepId":"step//input.js//step"},"varArrowStep":{"stepId":"step//input.js//varArrowStep"}}}}*/;
 // Function declaration step
 async function example$step(a, b) {
     return a + b;
 }
 var example$arrowStep = async (x, y)=>x * y;
 var example$letArrowStep = async (x, y)=>x - y;
 var example$varArrowStep = async (x, y)=>x / y;
-var helpers$objectStep = async (x, y)=>{
+var example$helpers$objectStep = async (x, y)=>{
     return x + y + 10;
 };
 export async function example(a, b) {
-    "use workflow";
-    const step = example$step;
-    // Arrow function with const
-    const arrowStep = example$arrowStep;
-    // Arrow function with let
-    let letArrowStep = example$letArrowStep;
-    // Arrow function with var
-    var varArrowStep = example$varArrowStep;
-    // Object with step method
-    const helpers = {
-        objectStep: helpers$objectStep
-    };
-    const val = await step(a, b);
-    const val2 = await arrowStep(a, b);
-    const val3 = await letArrowStep(a, b);
-    const val4 = await varArrowStep(a, b);
-    const val5 = await helpers.objectStep(a, b);
-    return val + val2 + val3 + val4 + val5;
+    throw new Error("You attempted to execute workflow example function directly. To start a workflow, use start(example) from workflow/api");
 }
+example.workflowId = "workflow//input.js//example";
 registerStepFunction("step//input.js//example/step", example$step);
 registerStepFunction("step//input.js//example/arrowStep", example$arrowStep);
 registerStepFunction("step//input.js//example/letArrowStep", example$letArrowStep);
 registerStepFunction("step//input.js//example/varArrowStep", example$varArrowStep);
-registerStepFunction("step//input.js//helpers/objectStep", helpers$objectStep);
+registerStepFunction("step//input.js//example/helpers/objectStep", example$helpers$objectStep);

packages/swc-plugin-workflow/transform/tests/fixture/nested-step-in-workflow/output-step.js

@@ -1,4 +1,4 @@
-/**__internal_workflows{"workflows":{"input.js":{"example":{"workflowId":"workflow//input.js//example"}}},"steps":{"input.js":{"arrowStep":{"stepId":"step//input.js//arrowStep"},"helpers/objectStep":{"stepId":"step//input.js//helpers/objectStep"},"letArrowStep":{"stepId":"step//input.js//letArrowStep"},"step":{"stepId":"step//input.js//step"},"varArrowStep":{"stepId":"step//input.js//varArrowStep"}}}}*/;
+/**__internal_workflows{"workflows":{"input.js":{"example":{"workflowId":"workflow//input.js//example"}}},"steps":{"input.js":{"arrowStep":{"stepId":"step//input.js//arrowStep"},"helpers/objectStep":{"stepId":"step//input.js//example/helpers/objectStep"},"letArrowStep":{"stepId":"step//input.js//letArrowStep"},"step":{"stepId":"step//input.js//step"},"varArrowStep":{"stepId":"step//input.js//varArrowStep"}}}}*/;
 export async function example(a, b) {
     var step = globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//example/step");
     // Arrow function with const
@@ -9,7 +9,7 @@ export async function example(a, b) {
     var varArrowStep = globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//example/varArrowStep");
     // Object with step method
     const helpers = {
-        objectStep: globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//helpers/objectStep")
+        objectStep: globalThis[Symbol.for("WORKFLOW_USE_STEP")]("step//input.js//example/helpers/objectStep")
     };
     const val = await step(a, b);
     const val2 = await arrowStep(a, b);

packages/swc-plugin-workflow/transform/tests/fixture/nested-step-in-workflow/output-workflow.js

@@ -80,23 +80,9 @@ const arrowWrapperReturnNamedFunctionVar = (a, b, c)=>{
     return fn;
 };
 export async function wflow() {
-    'use workflow';
-    let count = 42;
-    const namedStepWithClosureVars = wflow$namedStepWithClosureVars;
-    const agent = new DurableAgent({
-        arrowFunctionWithClosureVars: _anonymousStep2,
-        namedFunctionWithClosureVars: _anonymousStep3,
-        methodWithClosureVars: _anonymousStep4
-    });
-    await stepWrapperReturnArrowFunctionVar(1, 2, 3)();
-    await stepWrapperReturnNamedFunction(1, 2, 3)();
-    await stepWrapperReturnArrowFunction(1, 2, 3)();
-    await stepWrapperReturnNamedFunctionVar(1, 2, 3)();
-    await arrowWrapperReturnArrowFunctionVar(1, 2, 3)();
-    await arrowWrapperReturnNamedFunction(1, 2, 3)();
-    await arrowWrapperReturnArrowFunction(1, 2, 3)();
-    await arrowWrapperReturnNamedFunctionVar(1, 2, 3)();
+    throw new Error("You attempted to execute workflow wflow function directly. To start a workflow, use start(wflow) from workflow/api");
 }
+wflow.workflowId = "workflow//input.js//wflow";
 registerStepFunction("step//input.js//stepWrapperReturnArrowFunctionVar/fn", stepWrapperReturnArrowFunctionVar$fn);
 registerStepFunction("step//input.js//stepWrapperReturnNamedFunction/f", stepWrapperReturnNamedFunction$f);
 registerStepFunction("step//input.js//stepWrapperReturnArrowFunction/_anonymousStep0", stepWrapperReturnArrowFunction$_anonymousStep0);

packages/swc-plugin-workflow/transform/tests/fixture/nested-step-with-closure/output-step.js

@@ -6,27 +6,8 @@ import * as z from 'zod';
 var test$_anonymousStep0 = async ()=>gateway('openai/gpt-5');
 var test$_anonymousStep1 = async ({ location })=>`Weather in ${location}: Sunny, 72°F`;
 export async function test() {
-    'use workflow';
-    const agent = new DurableAgent({
-        model: _anonymousStep0,
-        tools: {
-            getWeather: tool({
-                description: 'Get weather for a location',
-                inputSchema: z.object({
-                    location: z.string()
-                }),
-                execute: _anonymousStep1
-            })
-        }
-    });
-    await agent.stream({
-        messages: [
-            {
-                role: 'user',
-                content: 'What is the weather in San Francisco?'
-            }
-        ]
-    });
+    throw new Error("You attempted to execute workflow test function directly. To start a workflow, use start(test) from workflow/api");
 }
+test.workflowId = "workflow//input.js//test";
 registerStepFunction("step//input.js//test/_anonymousStep0", test$_anonymousStep0);
 registerStepFunction("step//input.js//test/_anonymousStep1", test$_anonymousStep1);

packages/swc-plugin-workflow/transform/tests/fixture/nested-steps-in-object-constructor/output-step.js

@@ -1,5 +1,5 @@
 /**__internal_workflows{"workflows":{"input.js":{"workflow":{"workflowId":"workflow//input.js//workflow"}}}}*/;
 export async function workflow(a, b) {
-    'use workflow';
-    return add(a, b);
+    throw new Error("You attempted to execute workflow workflow function directly. To start a workflow, use start(workflow) from workflow/api");
 }
+workflow.workflowId = "workflow//input.js//workflow";