Sandbox Execution

import { Module } from '@nestjs/common';
import { LoopCoreModule } from '@loopstack/core';
import { SandboxFilesystemModule } from '@loopstack/sandbox-filesystem';
import { SandboxToolModule } from '@loopstack/sandbox-tool';
 
@Module({
  imports: [LoopCoreModule, SandboxToolModule, SandboxFilesystemModule],
  providers: [SandboxWorkflow],
  exports: [SandboxWorkflow],
})
export class SandboxModule {}

import { z } from 'zod';
import { BaseWorkflow, Final, Initial, InjectTool, ToolResult, Transition, Workflow } from '@loopstack/common';
import {
  SandboxCreateDirectory,
  SandboxDelete,
  SandboxReadFile,
  SandboxWriteFile,
} from '@loopstack/sandbox-filesystem';
import { SandboxDestroy, SandboxInit } from '@loopstack/sandbox-tool';
 
@Workflow({
  uiConfig: __dirname + '/sandbox.ui.yaml',
  schema: z.object({ outputDir: z.string().default(process.cwd() + '/out') }),
})
export class SandboxWorkflow extends BaseWorkflow {
  containerId?: string;
 
  @InjectTool() sandboxInit: SandboxInit;
  @InjectTool() sandboxDestroy: SandboxDestroy;
  @InjectTool() sandboxWriteFile: SandboxWriteFile;
  @InjectTool() sandboxReadFile: SandboxReadFile;
  @InjectTool() sandboxCreateDirectory: SandboxCreateDirectory;
 
  @Initial({ to: 'sandbox_ready' })
  async initSandbox() {
    const args = this.ctx.args as { outputDir: string };
    const result: ToolResult = await this.sandboxInit.call({
      containerId: 'my-sandbox',
      imageName: 'node:18',
      containerName: 'sandbox-container',
      projectOutPath: args.outputDir,
      rootPath: 'workspace',
    });
    this.containerId = result.data.containerId;
  }
 
  @Transition({ from: 'sandbox_ready', to: 'file_written' })
  async writeFile() {
    await this.sandboxCreateDirectory.call({
      containerId: this.containerId!,
      path: '/workspace/src',
      recursive: true,
    });
 
    await this.sandboxWriteFile.call({
      containerId: this.containerId!,
      path: '/workspace/src/hello.js',
      content: "console.log('Hello from sandbox!');",
      encoding: 'utf8',
      createParentDirs: true,
    });
  }
 
  @Transition({ from: 'file_written', to: 'file_read' })
  async readFile() {
    const result: ToolResult = await this.sandboxReadFile.call({
      containerId: this.containerId!,
      path: '/workspace/src/hello.js',
      encoding: 'utf8',
    });
    // result.data.content contains the file content
  }
 
  @Final({ from: 'file_read' })
  async destroySandbox() {
    await this.sandboxDestroy.call({
      containerId: this.containerId!,
      removeContainer: true,
    });
  }
}