@insanitybit/bun-prewarm 0.0.1

package/README.md

@@ -0,0 +1,34 @@
+# bun-prewarm
+
+Prewarm Bun runtime state inside a Firecracker VM before snapshotting.
+
+## Purpose
+
+This package warms Bun-related filesystem, binary, and runtime state so that after snapshot restore, Bun startup and first-execution latency are lower.
+
+## Installation
+
+```sh
+npm install -g @insanitybit/bun-prewarm
+```
+
+## Usage
+
+Run explicitly during VM snapshot build:
+
+```sh
+bun-prewarm
+# then snapshot the VM
+```
+
+## What it does
+
+1. Runs `bun --version` to fault the Bun binary into memory
+2. Executes a small JS workload under Bun to warm parsing and runtime
+3. Runs `bun install --no-save` to warm package manager code paths
+
+## Requirements
+
+- Bun must be installed and available on `$PATH`
+- Linux environment (Firecracker VM)
+- Node.js 18+ (for running the prewarm script itself)

package/bin/bun-prewarm.js

@@ -0,0 +1,93 @@
+#!/usr/bin/env node
+
+const { execSync } = require("child_process");
+const { mkdtempSync, writeFileSync } = require("fs");
+const { tmpdir } = require("os");
+const { join } = require("path");
+
+/**
+ * Prewarm Bun runtime state for Firecracker VM snapshotting.
+ *
+ * Steps:
+ * 1. Run `bun --version` to fault the binary into memory
+ * 2. Execute a small JS workload under Bun
+ * 3. Run `bun install --no-save` to warm install code paths
+ */
+
+function log(message) {
+  console.log(`[bun-prewarm] ${message}`);
+}
+
+function run(command, options = {}) {
+  log(`Running: ${command}`);
+  try {
+    const output = execSync(command, {
+      encoding: "utf8",
+      stdio: ["pipe", "pipe", "pipe"],
+      ...options,
+    });
+    return output.trim();
+  } catch (error) {
+    console.error(`[bun-prewarm] Command failed: ${command}`);
+    console.error(`[bun-prewarm] Exit code: ${error.status}`);
+    if (error.stderr) {
+      console.error(`[bun-prewarm] stderr: ${error.stderr}`);
+    }
+    process.exit(1);
+  }
+}
+
+// Step 1: Fault Bun binary and runtime
+log("Step 1: Faulting Bun binary into memory");
+const bunVersion = run("bun --version");
+log(`Bun version: ${bunVersion}`);
+
+// Step 2: Execute a small JS program under Bun
+log("Step 2: Executing JS workload under Bun");
+
+const tempDir = mkdtempSync(join(tmpdir(), "bun-prewarm-"));
+log(`Created temp directory: ${tempDir}`);
+
+// A simple workload that exercises parsing, runtime, and execution
+const workloadScript = `
+// Non-trivial loop to prevent total optimization removal
+let sum = 0;
+for (let i = 0; i < 100000; i++) {
+  sum += Math.sqrt(i) * Math.sin(i);
+}
+
+// Touch some built-in modules
+const crypto = require("crypto");
+const hash = crypto.createHash("sha256").update("prewarm").digest("hex");
+
+// Print results to prove execution
+console.log("Workload result:", sum.toFixed(4));
+console.log("Hash:", hash.slice(0, 16));
+console.log("Bun prewarm workload completed");
+`;
+
+const workloadPath = join(tempDir, "workload.js");
+writeFileSync(workloadPath, workloadScript);
+log(`Wrote workload script to: ${workloadPath}`);
+
+const workloadOutput = run(`bun ${workloadPath}`);
+log(`Workload output:\n${workloadOutput}`);
+
+// Step 3: Warm Bun install code paths
+log("Step 3: Warming Bun install code paths");
+
+// Create a minimal package.json in the temp directory
+const minimalPackageJson = JSON.stringify({ name: "prewarm-dummy", private: true }, null, 2);
+writeFileSync(join(tempDir, "package.json"), minimalPackageJson);
+
+try {
+  run("bun install --no-save", { cwd: tempDir });
+  log("Bun install completed");
+} catch {
+  // Failure in install step is non-fatal as per design doc
+  // (network may not be available, etc.)
+  log("Bun install step skipped or failed (non-fatal)");
+}
+
+// Done
+log("Prewarm complete");

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "@insanitybit/bun-prewarm",
+  "version": "0.0.1",
+  "description": "Prewarm Bun runtime state inside a Firecracker VM before snapshotting",
+  "license": "MIT",
+  "private": false,
+  "bin": {
+    "bun-prewarm": "bin/bun-prewarm.js"
+  },
+  "files": [
+    "bin"
+  ],
+  "engines": {
+    "node": ">=18"
+  }
+}