@freestyle-sh/with-sandbox-agent 0.0.1

package/dist/index.d.ts

@@ -0,0 +1,42 @@
+import * as freestyle_sandboxes from 'freestyle-sandboxes';
+import { VmWith, VmWithInstance, VmSpec } from 'freestyle-sandboxes';
+import { SandboxAgent } from 'sandbox-agent';
+
+type KnownEnvKeys = "ANTHROPIC_API_KEY" | "OPENAI_API_KEY" | "AMP_API_KEY";
+type EnvRecord = Partial<Record<KnownEnvKeys, string>> & Record<string, string>;
+type VmSandboxAgentOptions = {
+    host?: string;
+    port?: number;
+    token?: string;
+    env?: EnvRecord;
+};
+type VmSandboxAgentResolvedOptions = {
+    host: string;
+    port: number;
+    token?: string;
+    env: Record<string, string>;
+};
+declare class VmSandboxAgent extends VmWith<VmSandboxAgentInstance> {
+    options: VmSandboxAgentResolvedOptions;
+    constructor(options?: VmSandboxAgentOptions);
+    configureSnapshotSpec(spec: VmSpec): VmSpec;
+    createInstance(): VmSandboxAgentInstance;
+    installServiceName(): string;
+}
+declare class VmSandboxAgentInstance extends VmWithInstance {
+    builder: VmSandboxAgent;
+    constructor(builder: VmSandboxAgent);
+    port(): number;
+    host(): string;
+    client({ domain }?: {
+        domain?: string;
+    }): Promise<{
+        client: SandboxAgent;
+        domain: string;
+    }>;
+    /** @internal */
+    get _vm(): freestyle_sandboxes.Vm;
+}
+
+export { VmSandboxAgent };
+export type { VmSandboxAgentOptions, VmSandboxAgentResolvedOptions };

package/dist/index.js

@@ -0,0 +1,119 @@
+import { VmWith, VmSpec, VmWithInstance } from 'freestyle-sandboxes';
+import { SandboxAgent } from 'sandbox-agent';
+
+class VmSandboxAgent extends VmWith {
+  options;
+  constructor(options) {
+    super();
+    this.options = {
+      host: options?.host ?? "0.0.0.0",
+      port: options?.port ?? 2468,
+      token: options?.token,
+      env: options?.env ?? {}
+    };
+  }
+  configureSnapshotSpec(spec) {
+    const envExports = Object.entries(this.options.env).map(([k, v]) => `export ${k}="${v}"`).join("\n");
+    const tokenExport = this.options.token ? `export SANDBOX_TOKEN="${this.options.token}"` : "";
+    const tokenArg = this.options.token ? `--token "$SANDBOX_TOKEN"` : "--no-token";
+    return this.composeSpecs(
+      spec,
+      new VmSpec({
+        additionalFiles: {
+          "/opt/install-sandbox-agent.sh": {
+            content: `#!/bin/bash
+set -e
+export PATH="/root/.sandbox-agent/bin:$PATH"
+curl -fsSL https://releases.rivet.dev/sandbox-agent/latest/install.sh | sh
+`
+          },
+          "/opt/preinstall-agents.sh": {
+            content: `#!/bin/bash
+set -e
+export PATH="/root/.sandbox-agent/bin:$PATH"
+sandbox-agent install-agent claude
+sandbox-agent install-agent codex
+sandbox-agent install-agent opencode
+sandbox-agent install-agent amp
+`
+          },
+          "/opt/run-sandbox-agent.sh": {
+            content: `#!/bin/bash
+export PATH="/root/.sandbox-agent/bin:$PATH"
+${tokenExport}
+${envExports}
+sandbox-agent server ${tokenArg} --host ${this.options.host} --port ${this.options.port}
+`
+          }
+        },
+        systemd: {
+          services: [
+            {
+              name: "install-sandbox-agent",
+              mode: "oneshot",
+              exec: ["bash /opt/install-sandbox-agent.sh"],
+              timeoutSec: 300
+            },
+            {
+              name: "preinstall-agents",
+              mode: "oneshot",
+              after: ["install-sandbox-agent.service"],
+              requires: ["install-sandbox-agent.service"],
+              exec: ["bash /opt/preinstall-agents.sh"],
+              timeoutSec: 600
+            },
+            {
+              name: "sandbox-agent",
+              mode: "service",
+              after: ["preinstall-agents.service"],
+              requires: ["preinstall-agents.service"],
+              restartPolicy: {
+                policy: "always"
+              },
+              exec: ["bash /opt/run-sandbox-agent.sh"]
+            }
+          ]
+        }
+      })
+    );
+  }
+  createInstance() {
+    return new VmSandboxAgentInstance(this);
+  }
+  installServiceName() {
+    return "install-sandbox-agent.service";
+  }
+}
+class VmSandboxAgentInstance extends VmWithInstance {
+  builder;
+  constructor(builder) {
+    super();
+    this.builder = builder;
+  }
+  port() {
+    return this.builder.options.port;
+  }
+  host() {
+    return this.builder.options.host;
+  }
+  async client({ domain } = {}) {
+    const resolvedDomain = domain ?? `${crypto.randomUUID()}-sandbox-agent.style.dev`;
+    const freestyle = this._vm._freestyle;
+    await freestyle.domains.mappings.create({
+      domain: resolvedDomain,
+      vmId: this.vm.vmId,
+      vmPort: this.port()
+    });
+    const client = await SandboxAgent.connect({
+      baseUrl: `https://${resolvedDomain}`,
+      token: this.builder.options.token
+    });
+    return { client, domain: resolvedDomain };
+  }
+  /** @internal */
+  get _vm() {
+    return this.vm;
+  }
+}
+
+export { VmSandboxAgent };

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "@freestyle-sh/with-sandbox-agent",
+  "version": "0.0.1",
+  "private": false,
+  "dependencies": {
+    "freestyle-sandboxes": "^0.1.14",
+    "sandbox-agent": "^0.1.3"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "pkgroll": "^2.11.2",
+    "typescript": "^5.8.3"
+  },
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "source": "./src/index.ts",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "pkgroll"
+  }
+}