@tanstack/ai-code-mode/src/code-wrapper.ts

Code Mode for TanStack AI - LLM-driven code execution in secure sandboxes

/**
 * Wrap user code in an async IIFE to support top-level await
 *
 * The code is wrapped so that:
 * 1. Top-level await works
 * 2. Return values are captured and serialized (for isolate boundary transfer)
 * 3. Errors propagate correctly
 */
export function wrapCode(code: string): string {
  // Results must be serialized to JSON because objects can't be transferred
  // directly across the isolate boundary in isolated-vm
  return `
(async function() {
  try {
    const __userResult = await (async function() {
${code}
    })();
    return JSON.stringify(__userResult);
  } catch (__error) {
    throw __error;
  }
})()
`
}