@mcp-b/do-runtime 0.2.2 → 0.3.0

package/dist/src/api/http.d.ts

@@ -35,6 +35,7 @@
  */
 type IoAwaiter = {
     awaitIo<T>(promise: Promise<T>): Promise<T>;
+    awaitIo<T, R>(promise: Promise<T>, func: (value: T) => R | PromiseLike<R>): Promise<R>;
 };
 /** Wrap a host-provided request so consuming or streaming its body resumes gated. */
 export declare function gateRequestBody(ctx: IoAwaiter, request: Request): Request;

package/dist/src/fixtures/await-transform.actor.d.ts

@@ -0,0 +1 @@
+export declare function readAfterForeignAwait(storage: DurableObjectStorage): Promise<unknown>;

package/dist/src/gate.d.ts

@@ -0,0 +1,4 @@
+/** Re-enter the actor that owns this transformed await; fail open outside actors. */
+export declare function __gate<T>(value: T): T | Promise<Awaited<T>>;
+/** Gate every operation used by `for await`, including early return and throw. */
+export declare function __gateAsyncIterable<T, IterableType extends AsyncIterable<T> | Iterable<T>>(iterable: IterableType): IterableType;

package/dist/src/io/io-context.d.ts

@@ -466,6 +466,12 @@ export declare class IoContext {
      */
     awaitIo<T>(promise: Promise<T>): Promise<T>;
     awaitIo<T, R>(promise: Promise<T>, func: (value: T) => R | PromiseLike<R>): Promise<R>;
+    /**
+     * The await transform's hot path. It preserves provenance without capturing
+     * an Error stack for every await: the first call and every 64th call refresh
+     * the sampled site, while intervening calls keep that stack and say so.
+     */
+    awaitIoFromTransform<T>(promise: Promise<T>): Promise<T>;
     /**
      * Waits for the given I/O while holding the input lock, so that all other I/O is blocked
      * from completing in the meantime (unless it is also holding the same input lock).

package/dist/src/vite.d.ts

@@ -0,0 +1,7 @@
+import { type FilterPattern, type Plugin } from "vite";
+export interface DoRuntimeAwaitTransformOptions {
+    include?: FilterPattern;
+    exclude?: FilterPattern;
+}
+/** Rewrite syntactic awaits in selected actor-bundled modules to re-enter their input gate. */
+export declare function doRuntimeAwaitTransform(options?: DoRuntimeAwaitTransformOptions): Plugin;

package/dist/vite.js

@@ -0,0 +1,55 @@
+import MagicString from "magic-string";
+import { Visitor, createFilter } from "vite";
+//#region src/vite.ts
+var MARKER = "/* @do-runtime-gated */";
+var IMPORT = "import { __gate, __gateAsyncIterable } from \"@mcp-b/do-runtime/gate\";";
+function patterns(pattern) {
+	if (pattern === void 0 || pattern === null) return [];
+	return typeof pattern === "string" || pattern instanceof RegExp ? [pattern] : pattern;
+}
+/** Rewrite syntactic awaits in selected actor-bundled modules to re-enter their input gate. */
+function doRuntimeAwaitTransform(options) {
+	const filter = createFilter(options?.include, [
+		"**/node_modules/@mcp-b/do-runtime/**",
+		"**/@mcp-b/do-runtime/gate",
+		...patterns(options?.exclude)
+	]);
+	return {
+		name: "do-runtime-await-transform",
+		enforce: "post",
+		transform(code, id) {
+			if (!code.includes("await") || code.includes(MARKER) || !filter(id)) return null;
+			const source = new MagicString(code);
+			let transformed = false;
+			const program = this.parse(code);
+			new Visitor({
+				AwaitExpression(node) {
+					source.prependLeft(node.argument.start, "__gate((");
+					source.appendRight(node.argument.end, "))");
+					transformed = true;
+				},
+				ForOfStatement(node) {
+					if (!node.await) return;
+					source.prependLeft(node.right.start, "__gateAsyncIterable((");
+					source.appendRight(node.right.end, "))");
+					transformed = true;
+				}
+			}).visit(program);
+			if (!transformed) return null;
+			const insertionPoint = code.startsWith("#!") ? code.indexOf("\n") + 1 : 0;
+			source.appendLeft(insertionPoint, `${MARKER}\n${IMPORT}\n`);
+			return {
+				code: source.toString(),
+				map: source.generateMap({
+					hires: "boundary",
+					includeContent: true,
+					source: id
+				})
+			};
+		}
+	};
+}
+//#endregion
+export { doRuntimeAwaitTransform };
+
+//# sourceMappingURL=vite.js.map

package/dist/vite.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vite.js","names":[],"sources":["../src/vite.ts"],"sourcesContent":["import MagicString from \"magic-string\";\nimport {\n  createFilter,\n  Visitor,\n  type FilterPattern,\n  type Plugin,\n} from \"vite\";\n\nconst MARKER = \"/* @do-runtime-gated */\";\nconst IMPORT =\n  'import { __gate, __gateAsyncIterable } from \"@mcp-b/do-runtime/gate\";';\n\nexport interface DoRuntimeAwaitTransformOptions {\n  include?: FilterPattern;\n  exclude?: FilterPattern;\n}\n\nfunction patterns(pattern: FilterPattern | undefined): readonly (string | RegExp)[] {\n  if (pattern === undefined || pattern === null) return [];\n  return typeof pattern === \"string\" || pattern instanceof RegExp ? [pattern] : pattern;\n}\n\n/** Rewrite syntactic awaits in selected actor-bundled modules to re-enter their input gate. */\nexport function doRuntimeAwaitTransform(options?: DoRuntimeAwaitTransformOptions): Plugin {\n  const filter = createFilter(options?.include, [\n    \"**/node_modules/@mcp-b/do-runtime/**\",\n    \"**/@mcp-b/do-runtime/gate\",\n    ...patterns(options?.exclude),\n  ]);\n\n  return {\n    name: \"do-runtime-await-transform\",\n    enforce: \"post\",\n    transform(code, id) {\n      if (!code.includes(\"await\") || code.includes(MARKER) || !filter(id)) return null;\n\n      const source = new MagicString(code);\n      let transformed = false;\n      const program = this.parse(code);\n      new Visitor({\n        AwaitExpression(node) {\n          source.prependLeft(node.argument.start, \"__gate((\");\n          source.appendRight(node.argument.end, \"))\");\n          transformed = true;\n        },\n        ForOfStatement(node) {\n          if (!node.await) return;\n          source.prependLeft(node.right.start, \"__gateAsyncIterable((\");\n          source.appendRight(node.right.end, \"))\");\n          transformed = true;\n        },\n      }).visit(program);\n      if (!transformed) return null;\n\n      const insertionPoint = code.startsWith(\"#!\") ? code.indexOf(\"\\n\") + 1 : 0;\n      source.appendLeft(insertionPoint, `${MARKER}\\n${IMPORT}\\n`);\n      return {\n        code: source.toString(),\n        map: source.generateMap({ hires: \"boundary\", includeContent: true, source: id }),\n      };\n    },\n  };\n}\n"],"mappings":";;;AAQA,IAAM,SAAS;AACf,IAAM,SACJ;AAOF,SAAS,SAAS,SAAkE;CAClF,IAAI,YAAY,KAAA,KAAa,YAAY,MAAM,OAAO,CAAC;CACvD,OAAO,OAAO,YAAY,YAAY,mBAAmB,SAAS,CAAC,OAAO,IAAI;AAChF;;AAGA,SAAgB,wBAAwB,SAAkD;CACxF,MAAM,SAAS,aAAa,SAAS,SAAS;EAC5C;EACA;EACA,GAAG,SAAS,SAAS,OAAO;CAC9B,CAAC;CAED,OAAO;EACL,MAAM;EACN,SAAS;EACT,UAAU,MAAM,IAAI;GAClB,IAAI,CAAC,KAAK,SAAS,OAAO,KAAK,KAAK,SAAS,MAAM,KAAK,CAAC,OAAO,EAAE,GAAG,OAAO;GAE5E,MAAM,SAAS,IAAI,YAAY,IAAI;GACnC,IAAI,cAAc;GAClB,MAAM,UAAU,KAAK,MAAM,IAAI;GAC/B,IAAI,QAAQ;IACV,gBAAgB,MAAM;KACpB,OAAO,YAAY,KAAK,SAAS,OAAO,UAAU;KAClD,OAAO,YAAY,KAAK,SAAS,KAAK,IAAI;KAC1C,cAAc;IAChB;IACA,eAAe,MAAM;KACnB,IAAI,CAAC,KAAK,OAAO;KACjB,OAAO,YAAY,KAAK,MAAM,OAAO,uBAAuB;KAC5D,OAAO,YAAY,KAAK,MAAM,KAAK,IAAI;KACvC,cAAc;IAChB;GACF,CAAC,CAAC,CAAC,MAAM,OAAO;GAChB,IAAI,CAAC,aAAa,OAAO;GAEzB,MAAM,iBAAiB,KAAK,WAAW,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,IAAI;GACxE,OAAO,WAAW,gBAAgB,GAAG,OAAO,IAAI,OAAO,GAAG;GAC1D,OAAO;IACL,MAAM,OAAO,SAAS;IACtB,KAAK,OAAO,YAAY;KAAE,OAAO;KAAY,gBAAgB;KAAM,QAAQ;IAAG,CAAC;GACjF;EACF;CACF;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mcp-b/do-runtime",
-  "version": "0.2.2",
+  "version": "0.3.0",
   "description": "Cloudflare's Durable Object runtime (workerd), ported to TypeScript: actors with input/output gates, SQLite storage, facets and alarms, running in the browser and in Node.",
   "keywords": [
     "actors",
@@ -57,6 +57,14 @@
       "types": "./dist/src/api/cloudflare-workers.d.ts",
       "import": "./dist/cloudflare-workers.js"
     },
+    "./gate": {
+      "types": "./dist/src/gate.d.ts",
+      "import": "./dist/gate.js"
+    },
+    "./vite": {
+      "types": "./dist/src/vite.d.ts",
+      "import": "./dist/vite.js"
+    },
     "./conformance": {
       "types": "./dist/conformance/host.d.ts",
       "import": "./dist/conformance.js"
@@ -65,7 +73,16 @@
   },
   "dependencies": {
     "@ungap/structured-clone": "1.3.3",
-    "capnweb": "0.10.0"
+    "capnweb": "0.10.0",
+    "magic-string": "0.30.21"
+  },
+  "peerDependencies": {
+    "vite": "^8.1.5"
+  },
+  "peerDependenciesMeta": {
+    "vite": {
+      "optional": true
+    }
   },
   "devDependencies": {
     "@arethetypeswrong/cli": "0.18.5",
@@ -79,7 +96,6 @@
     "playwright": "1.61.1",
     "publint": "0.3.23",
     "typescript": "^5.9.3",
-    "vite": "^8.1.5",
     "vitest": "4.1.10",
     "workerd": "1.20260820.1",
     "wrangler": "^4.114.0"