@computesdk/workbench 2.0.0 → 2.0.2

package/dist/bin/workbench.js

@@ -32,10 +32,26 @@ function setSandbox(state, sandbox, provider) {
   state.currentSandbox = sandbox;
   state.currentProvider = provider;
   state.sandboxCreatedAt = /* @__PURE__ */ new Date();
+  updatePromptIfNeeded(state);
 }
 function clearSandbox(state) {
   state.currentSandbox = null;
   state.sandboxCreatedAt = null;
+  updatePromptIfNeeded(state);
+}
+function updatePromptIfNeeded(state) {
+  if (state._replServer) {
+    const prompt = getPrompt(state);
+    state._replServer.setPrompt(prompt);
+  }
+}
+function getPrompt(state) {
+  if (!state.currentSandbox) {
+    return "> ";
+  }
+  const provider = state.currentProvider || "unknown";
+  const sandboxId = state.currentSandbox.sandboxId || "";
+  return `${provider}:${sandboxId}> `;
 }
 function hasSandbox(state) {
   return state.currentSandbox !== null;
@@ -91,6 +107,7 @@ function showWelcome(availableProviders, currentProvider, useDirectMode) {
   console.log(c.bold(c.cyan("\n\u2554\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2557")));
   console.log(c.bold(c.cyan("\u2551   ComputeSDK Workbench                               \u2551")));
   console.log(c.bold(c.cyan("\u255A\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u255D\n")));
+  console.log(c.dim("Prompt shows connection status: > (disconnected) or provider:sandbox> (connected)\n"));
   if (availableProviders.length > 0) {
     const backendProviders = availableProviders.filter((p) => p !== "gateway");
     console.log(`Providers available: ${backendProviders.join(", ")}`);
@@ -124,7 +141,12 @@ function showInfo(state) {
     return;
   }
   console.log("\n" + c.bold("Current Sandbox:"));
+  if (state.currentSandbox.sandboxId) {
+    console.log(`  Sandbox ID: ${c.cyan(state.currentSandbox.sandboxId)}`);
+  }
   console.log(`  Provider: ${c.green(state.currentProvider || "unknown")}`);
+  const modeLabel = state.useDirectMode ? "direct \u{1F517}" : "gateway \u{1F310}";
+  console.log(`  Mode: ${c.blue(modeLabel)}`);
   console.log(`  Created: ${state.sandboxCreatedAt?.toLocaleString() || "unknown"}`);
   console.log(`  Uptime: ${formatUptime(state)}`);
   console.log("");
@@ -221,12 +243,28 @@ ${c.bold("Running Commands:")}
     ${c.cyan('git.clone("https://github.com/user/repo")')}
     ${c.cyan("git.status()")}
   
-  ${c.dim("Filesystem:")}
+  ${c.dim("Filesystem (via commands):")}
     ${c.cyan('ls("/home")')}
     ${c.cyan('cat("/etc/hosts")')}
     ${c.cyan('rm.rf("/tmp")')}          ${c.dim("// Force remove")}
     ${c.cyan('rm.auto("/path")')}       ${c.dim("// Smart remove")}
   
+  ${c.dim("Filesystem (direct API):")}
+    ${c.cyan('filesystem.readFile("/etc/hosts")')}
+    ${c.cyan('filesystem.writeFile("/app/server.js", code)')}
+    ${c.cyan('filesystem.mkdir("/app")')}
+    ${c.cyan('filesystem.readdir("/home")')}
+    ${c.cyan('filesystem.exists("/path")')}
+    ${c.cyan('filesystem.remove("/file")')}
+  
+  ${c.dim("Sandbox Methods:")}
+    ${c.cyan("getUrl({ port: 3000 })")}   ${c.dim("// Get public URL")}
+    ${c.cyan(`runCode("console.log('hi')", "node")`)}
+    ${c.cyan("sandboxInfo()")}            ${c.dim("// Get sandbox details")}
+    ${c.cyan("getInstance()")}            ${c.dim("// Get native instance")}
+  
+  ${c.dim('Note: No need to use "await" - promises are auto-awaited!')}
+  
   ${c.dim("Compute CLI:")}
     ${c.cyan("compute.isSetup()")}      ${c.dim("// Check if daemon is running")}
     ${c.cyan("compute.setup()")}        ${c.dim("// Install + start daemon")}
@@ -256,47 +294,62 @@ function logWarning(message) {
 }
 
 // src/cli/providers.ts
+import {
+  PROVIDER_AUTH as SHARED_PROVIDER_AUTH,
+  PROVIDER_NAMES as SHARED_PROVIDER_NAMES,
+  getProviderConfigFromEnv
+} from "computesdk";
 var PROVIDER_NAMES = [
   "gateway",
-  "e2b",
-  "railway",
-  "daytona",
-  "modal",
-  "runloop",
-  "vercel",
-  "cloudflare",
-  "codesandbox",
-  "blaxel"
+  ...SHARED_PROVIDER_NAMES
 ];
-var PROVIDER_ENV_VARS = {
-  gateway: ["COMPUTESDK_API_KEY"],
-  e2b: ["E2B_API_KEY"],
-  railway: ["RAILWAY_API_KEY", "RAILWAY_PROJECT_ID", "RAILWAY_ENVIRONMENT_ID"],
-  daytona: ["DAYTONA_API_KEY"],
-  modal: ["MODAL_TOKEN_ID", "MODAL_TOKEN_SECRET"],
-  runloop: ["RUNLOOP_API_KEY"],
-  vercel: ["VERCEL_TOKEN", "VERCEL_TEAM_ID", "VERCEL_PROJECT_ID"],
-  cloudflare: ["CLOUDFLARE_API_TOKEN", "CLOUDFLARE_ACCOUNT_ID"],
-  codesandbox: ["CSB_API_KEY"],
-  blaxel: ["BL_API_KEY", "BL_WORKSPACE"]
+var PROVIDER_AUTH = {
+  gateway: [["COMPUTESDK_API_KEY"]],
+  ...SHARED_PROVIDER_AUTH
 };
 function getProviderStatus(provider) {
+  const authOptions = PROVIDER_AUTH[provider];
   if (typeof process === "undefined") {
     return {
       name: provider,
       isComplete: false,
       present: [],
-      missing: [...PROVIDER_ENV_VARS[provider]]
+      missing: [...authOptions[0]]
     };
   }
-  const requiredVars = PROVIDER_ENV_VARS[provider];
-  const present = requiredVars.filter((varName) => !!process.env?.[varName]);
-  const missing = requiredVars.filter((varName) => !process.env?.[varName]);
+  const allVars = new Set(authOptions.flat());
+  const presentSet = /* @__PURE__ */ new Set();
+  for (const v of allVars) {
+    if (process.env?.[v]) presentSet.add(v);
+  }
+  let bestOption = null;
+  for (const option of authOptions) {
+    const missing = [];
+    let presentCount = 0;
+    for (const v of option) {
+      if (presentSet.has(v)) {
+        presentCount++;
+      } else {
+        missing.push(v);
+      }
+    }
+    if (missing.length === 0) {
+      return {
+        name: provider,
+        isComplete: true,
+        present: [...presentSet],
+        missing: []
+      };
+    }
+    if (!bestOption || presentCount > bestOption.presentCount) {
+      bestOption = { presentCount, missing };
+    }
+  }
   return {
     name: provider,
-    isComplete: missing.length === 0,
-    present: [...present],
-    missing: [...missing]
+    isComplete: false,
+    present: [...presentSet],
+    missing: bestOption?.missing ?? []
   };
 }
 function getAvailableProviders() {
@@ -428,47 +481,12 @@ async function loadProvider(providerName) {
   }
 }
 function getProviderConfig(providerName) {
-  const config2 = {};
-  switch (providerName) {
-    case "e2b":
-      if (process.env.E2B_API_KEY) config2.apiKey = process.env.E2B_API_KEY;
-      break;
-    case "railway":
-      if (process.env.RAILWAY_API_KEY) config2.apiKey = process.env.RAILWAY_API_KEY;
-      if (process.env.RAILWAY_PROJECT_ID) config2.projectId = process.env.RAILWAY_PROJECT_ID;
-      if (process.env.RAILWAY_ENVIRONMENT_ID) config2.environmentId = process.env.RAILWAY_ENVIRONMENT_ID;
-      break;
-    case "daytona":
-      if (process.env.DAYTONA_API_KEY) config2.apiKey = process.env.DAYTONA_API_KEY;
-      break;
-    case "modal":
-      if (process.env.MODAL_TOKEN_ID) config2.tokenId = process.env.MODAL_TOKEN_ID;
-      if (process.env.MODAL_TOKEN_SECRET) config2.tokenSecret = process.env.MODAL_TOKEN_SECRET;
-      break;
-    case "runloop":
-      if (process.env.RUNLOOP_API_KEY) config2.apiKey = process.env.RUNLOOP_API_KEY;
-      break;
-    case "vercel":
-      if (process.env.VERCEL_TOKEN) config2.token = process.env.VERCEL_TOKEN;
-      if (process.env.VERCEL_TEAM_ID) config2.teamId = process.env.VERCEL_TEAM_ID;
-      if (process.env.VERCEL_PROJECT_ID) config2.projectId = process.env.VERCEL_PROJECT_ID;
-      break;
-    case "cloudflare":
-      if (process.env.CLOUDFLARE_API_TOKEN) config2.apiToken = process.env.CLOUDFLARE_API_TOKEN;
-      if (process.env.CLOUDFLARE_ACCOUNT_ID) config2.accountId = process.env.CLOUDFLARE_ACCOUNT_ID;
-      break;
-    case "codesandbox":
-      if (process.env.CSB_API_KEY) config2.apiKey = process.env.CSB_API_KEY;
-      break;
-    case "blaxel":
-      if (process.env.BL_API_KEY) config2.apiKey = process.env.BL_API_KEY;
-      if (process.env.BL_WORKSPACE) config2.workspace = process.env.BL_WORKSPACE;
-      break;
-    case "gateway":
-      if (process.env.COMPUTESDK_API_KEY) config2.apiKey = process.env.COMPUTESDK_API_KEY;
-      break;
+  if (providerName === "gateway") {
+    const config2 = {};
+    if (process.env.COMPUTESDK_API_KEY) config2.apiKey = process.env.COMPUTESDK_API_KEY;
+    return config2;
   }
-  return config2;
+  return getProviderConfigFromEnv(providerName);
 }
 
 // src/cli/commands.ts
@@ -642,9 +660,34 @@ async function runCommand(state, command) {
   } catch (error) {
     const duration = Date.now() - startTime;
     logError(`Failed ${c.dim(`(${formatDuration(duration)})`)} - ${error instanceof Error ? error.message : String(error)}`);
+    if (isStaleConnectionError(error)) {
+      clearSandbox(state);
+      logWarning("Sandbox connection lost. Next command will create a new sandbox.");
+    }
     throw error;
   }
 }
+function isStaleConnectionError(error) {
+  if (!(error instanceof Error)) return false;
+  const message = error.message.toLowerCase();
+  const stalePhrases = [
+    "websocket",
+    "connection refused",
+    "connection reset",
+    "connection closed",
+    "socket hang up",
+    "econnrefused",
+    "econnreset",
+    "etimedout",
+    "not found",
+    "sandbox not found",
+    "unauthorized",
+    "401",
+    "403",
+    "404"
+  ];
+  return stalePhrases.some((phrase) => message.includes(phrase));
+}
 async function switchProvider(state, mode, providerName) {
   let useDirect = false;
   let actualProvider = mode;
@@ -800,7 +843,8 @@ import * as path2 from "path";
 import * as os from "os";
 function createREPL(state) {
   const replServer = repl.start({
-    prompt: "workbench> ",
+    prompt: "> ",
+    // Initial prompt, will be updated by state management
     useColors: true,
     terminal: true,
     useGlobal: false,
@@ -811,6 +855,7 @@ function createREPL(state) {
   setupSmartEvaluator(replServer, state);
   setupAutocomplete(replServer, state);
   setupHistory(replServer);
+  state._replServer = replServer;
   return replServer;
 }
 function injectCmdContext(replServer) {
@@ -918,10 +963,94 @@ function injectWorkbenchCommands(replServer, state) {
   };
   replServer.context.env = () => showEnv();
   replServer.context.help = showHelp;
+  replServer.context.getUrl = async (options) => {
+    const sandbox = state.currentSandbox;
+    if (!sandbox) {
+      throw new Error("No active sandbox. Run a command to auto-create one.");
+    }
+    return sandbox.getUrl(options);
+  };
+  replServer.context.sandboxInfo = async () => {
+    const sandbox = state.currentSandbox;
+    if (!sandbox) {
+      throw new Error("No active sandbox. Run a command to auto-create one.");
+    }
+    return sandbox.getInfo();
+  };
+  replServer.context.runCode = async (code, runtime) => {
+    const sandbox = state.currentSandbox;
+    if (!sandbox) {
+      throw new Error("No active sandbox. Run a command to auto-create one.");
+    }
+    return sandbox.runCode(code, runtime);
+  };
+  replServer.context.filesystem = {
+    get readFile() {
+      return async (path4) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.readFile(path4);
+      };
+    },
+    get writeFile() {
+      return async (path4, content) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.writeFile(path4, content);
+      };
+    },
+    get mkdir() {
+      return async (path4) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.mkdir(path4);
+      };
+    },
+    get readdir() {
+      return async (path4) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.readdir(path4);
+      };
+    },
+    get exists() {
+      return async (path4) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.exists(path4);
+      };
+    },
+    get remove() {
+      return async (path4) => {
+        const sandbox = state.currentSandbox;
+        if (!sandbox) {
+          throw new Error("No active sandbox. Run a command to auto-create one.");
+        }
+        return sandbox.filesystem.remove(path4);
+      };
+    }
+  };
+  replServer.context.getInstance = () => {
+    const sandbox = state.currentSandbox;
+    if (!sandbox) {
+      throw new Error("No active sandbox. Run a command to auto-create one.");
+    }
+    return sandbox.getInstance();
+  };
 }
 function setupSmartEvaluator(replServer, state) {
   const originalEval = replServer.eval;
-  const workbenchCommands = /* @__PURE__ */ new Set(["help", "providers", "info", "env", "restart", "destroy", "mode", "verbose"]);
+  const workbenchCommands = /* @__PURE__ */ new Set(["help", "providers", "info", "env", "restart", "destroy", "mode", "verbose", "sandboxInfo"]);
   replServer.eval = function(cmd3, context, filename, callback) {
     const trimmedCmd = cmd3.trim();
     const providerMatch = trimmedCmd.match(/^provider(?:\s+(direct|gateway))?\s+(\w+)$/);
@@ -980,6 +1109,15 @@ function setupSmartEvaluator(replServer, state) {
         }
         return;
       }
+      if (result && typeof result.then === "function") {
+        try {
+          const output = await result;
+          callback(null, output);
+        } catch (error) {
+          callback(error, void 0);
+        }
+        return;
+      }
       callback(null, result);
     });
   };
@@ -998,52 +1136,81 @@ function setupAutocomplete(replServer, state) {
     "help": [],
     "verbose": [],
     "exit": [],
-    ".exit": []
+    ".exit": [],
+    // Sandbox methods
+    "getUrl": [],
+    "runCode": [],
+    "sandboxInfo": [],
+    "getInstance": []
+    // Filesystem is an object, so it gets dot notation autocomplete automatically
   };
   replServer.completer = function(line, callback) {
-    const trimmed = line.trim();
-    if (!line.includes(" ") && !line.includes(".")) {
-      const commands = Object.keys(workbenchCommands);
-      const hits = commands.filter((cmd3) => cmd3.startsWith(trimmed));
+    try {
+      const trimmed = line.trim();
+      if (!line.includes(" ") && !line.includes(".")) {
+        const commands = Object.keys(workbenchCommands);
+        const hits = commands.filter((cmd3) => cmd3.startsWith(trimmed));
+        if (originalCompleter) {
+          originalCompleter.call(replServer, line, (err, result) => {
+            if (err || !result) {
+              callback(null, [hits, trimmed]);
+              return;
+            }
+            if (!Array.isArray(result) || result.length !== 2) {
+              callback(null, [hits, trimmed]);
+              return;
+            }
+            const [contextHits, partial] = result;
+            if (!Array.isArray(contextHits)) {
+              callback(null, [hits, trimmed]);
+              return;
+            }
+            const allHits = [.../* @__PURE__ */ new Set([...hits, ...contextHits])].sort();
+            const completionPrefix = typeof partial === "string" ? partial : trimmed;
+            callback(null, [allHits, completionPrefix]);
+          });
+          return;
+        }
+        callback(null, [hits.length ? hits : commands, trimmed]);
+        return;
+      }
+      if (line.includes(" ") && !line.includes(".")) {
+        const parts = line.split(" ");
+        const command = parts[0].trim();
+        const partial = parts.slice(1).join(" ").trim();
+        const suggestions = workbenchCommands[command];
+        if (suggestions !== void 0) {
+          if (suggestions.length > 0) {
+            const hits = suggestions.filter((s) => s.startsWith(partial)).map((s) => `${command} ${s}`);
+            callback(null, [hits.length ? hits : suggestions.map((s) => `${command} ${s}`), line]);
+          } else {
+            callback(null, [[], line]);
+          }
+          return;
+        }
+      }
       if (originalCompleter) {
         originalCompleter.call(replServer, line, (err, result) => {
           if (err || !result) {
-            callback(null, [hits, trimmed]);
+            callback(null, [[], line]);
             return;
           }
-          const [contextHits, partial] = result;
-          if (!Array.isArray(contextHits)) {
-            callback(null, [hits, trimmed]);
+          if (!Array.isArray(result) || result.length !== 2) {
+            callback(null, [[], line]);
             return;
           }
-          const allHits = [.../* @__PURE__ */ new Set([...hits, ...contextHits])].sort();
-          callback(null, [allHits, partial]);
+          const [completions, partial] = result;
+          if (!Array.isArray(completions) || typeof partial !== "string") {
+            callback(null, [[], line]);
+            return;
+          }
+          callback(null, [completions, partial]);
         });
-        return;
-      }
-      callback(null, [hits.length ? hits : commands, trimmed]);
-      return;
-    }
-    if (line.includes(" ") && !line.includes(".")) {
-      const parts = line.split(" ");
-      const command = parts[0].trim();
-      const partial = parts.slice(1).join(" ").trim();
-      const suggestions = workbenchCommands[command];
-      if (suggestions && suggestions.length > 0) {
-        const hits = suggestions.filter((s) => s.startsWith(partial)).map((s) => `${command} ${s}`);
-        callback(null, [hits.length ? hits : suggestions.map((s) => `${command} ${s}`), line]);
-        return;
+      } else {
+        callback(null, [[], line]);
       }
-    }
-    if (originalCompleter) {
-      originalCompleter.call(replServer, line, (err, result) => {
-        if (err || !result) {
-          callback(null, [[], line]);
-          return;
-        }
-        callback(null, result);
-      });
-    } else {
+    } catch (error) {
+      console.error("Autocomplete error:", error);
       callback(null, [[], line]);
     }
   };