Support directives in nested functions/closure tracking

Author: pranaygp

packages/swc-plugin-workflow/transform/tests/fixture/ai-sdk-tool/input.js

@@ -0,0 +1,16 @@
+import { tool } from 'ai';
+import { z } from 'zod';
+
+export const weatherTool = tool({
+  description: 'Get the weather in a location',
+  parameters: z.object({
+    location: z.string().describe('The location to get the weather for'),
+  }),
+  execute: async ({ location }) => {
+    'use step';
+    const response = await fetch(
+      `https://api.weather.com/v1/current?location=${location}`
+    );
+    return response.json();
+  },
+});

packages/swc-plugin-workflow/transform/tests/fixture/ai-sdk-tool/output-client.js

@@ -0,0 +1,13 @@
+import { tool } from 'ai';
+import { z } from 'zod';
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    parameters: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: async ({ location })=>{
+        'use step';
+        const response = await fetch(`https://api.weather.com/v1/current?location=${location}`);
+        return response.json();
+    }
+});

packages/swc-plugin-workflow/transform/tests/fixture/ai-sdk-tool/output-step.js

@@ -0,0 +1,13 @@
+import { tool } from 'ai';
+import { z } from 'zod';
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    parameters: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: async ({ location })=>{
+        'use step';
+        const response = await fetch(`https://api.weather.com/v1/current?location=${location}`);
+        return response.json();
+    }
+});

packages/swc-plugin-workflow/transform/tests/fixture/ai-sdk-tool/output-workflow.js

@@ -0,0 +1,13 @@
+import { tool } from 'ai';
+import { z } from 'zod';
+export const weatherTool = tool({
+    description: 'Get the weather in a location',
+    parameters: z.object({
+        location: z.string().describe('The location to get the weather for')
+    }),
+    execute: async ({ location })=>{
+        'use step';
+        const response = await fetch(`https://api.weather.com/v1/current?location=${location}`);
+        return response.json();
+    }
+});