@memrov2/pentcloud-tool-runtime 0.1.0 → 0.1.1-dev.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memrov2/pentcloud-tool-runtime",
-  "version": "0.1.0",
+  "version": "0.1.1-dev.0",
   "type": "module",
   "license": "MIT",
   "description": "Cloudflare-native tool-runtime tool pack",

package/src/worker.d.mts

@@ -1,7 +1,9 @@
 import { DurableObject } from "cloudflare:workers";
 import type { ToolCommand, ToolResult, PackLock } from "./index.mjs";
 declare class ToolRunner<I> extends DurableObject {
+  ready(): Promise<{ image: string; instance: "lite" | "standard-1"; readyMs: number }>;
   run(input: I, files?: Record<string, Uint8Array>): Promise<ToolResult>;
+  retire(): Promise<{ stopped: boolean; retired: true }>;
   stop(): Promise<{ stopped: boolean }>;
   capabilities(): Promise<{
     platform: string;

package/src/worker.mjs

@@ -34,6 +34,7 @@ export function createRunner(lock, buildCommand) {
       return result;
     }
     async #ready() {
+      if (await this.ctx.storage.get("retired")) throw new Error("Owned tool runtime is retired");
       const container = this.ctx.container,
         image = container.images[lock.packId],
         start = Date.now();
@@ -47,13 +48,7 @@ export function createRunner(lock, buildCommand) {
       } else
         container.start({
           image,
-          instance: {
-            vcpu: lock.runtime?.vcpu ?? 1,
-            memoryMib:
-              lock.runtime?.memoryMib ??
-              (lock.packId === "nuclei" ? 1024 : 512),
-            diskMb: lock.runtime?.diskMb ?? 2000,
-          },
+          instance: lock.packId === "password" ? "standard-1" : "lite",
           enableInternet: lock.runtime?.enableInternet ?? true,
           labels: { pack: lock.packId, version: lock.packVersion },
         });
@@ -92,7 +87,10 @@ export function createRunner(lock, buildCommand) {
         throw new Error("Pack image and SDK pins do not match");
       if ((await container.inspect())?.image !== image)
         throw new Error("Unable to verify pack image identity");
-      return { image, readyMs: Date.now() - start };
+      return { image, instance: lock.packId === "password" ? "standard-1" : "lite", readyMs: Date.now() - start };
+    }
+    ready() {
+      return this.#serialize(() => this.#ready());
     }
     run(input, files) {
       const { requiredFiles = [], ...command } = buildCommand(input),
@@ -140,6 +138,13 @@ export function createRunner(lock, buildCommand) {
         };
       });
     }
+    retire() {
+      return this.#serialize(async () => {
+        await this.ctx.storage.put("retired", true);
+        if (this.ctx.container.running) await this.ctx.container.destroy();
+        return { stopped: !this.ctx.container.running, retired: true };
+      });
+    }
     stop() {
       return this.#serialize(async () => {
         if (this.ctx.container.running) await this.ctx.container.destroy();