rivetkit 2.3.11-rc.8 → 2.3.11-rc.9

package/dist/tsup/mod.js

@@ -1,6 +1,6 @@
 import {
   decodeWorkflowHistoryTransport
-} from "./chunk-4EDGOGCB.js";
+} from "./chunk-C5TORAVP.js";
 import {
   ACTOR_CONTEXT_INTERNAL_SYMBOL,
   ActorConfigSchema,
@@ -23,13 +23,13 @@ import {
   isResponseLike,
   tryParseEndpoint,
   updateRunnerConfig
-} from "./chunk-JBQ63WSM.js";
+} from "./chunk-HOZ3ZLZ2.js";
 import {
   makePrefixedKey,
   removePrefixFromKey
 } from "./chunk-JI6GZ2C2.js";
+import "./chunk-XOOESJY7.js";
 import "./chunk-7AFZMFPQ.js";
-import "./chunk-V6ANE6JM.js";
 import {
   LogLevelSchema,
   VERSION,
@@ -62,7 +62,7 @@ import {
   parsePortEnv,
   stringifyError,
   toUint8Array
-} from "./chunk-AGNBLCNJ.js";
+} from "./chunk-LZ4TSBIP.js";
 import {
   INTERNAL_ERROR_CODE,
   RivetError,
@@ -2795,6 +2795,83 @@ function createWriteThroughProxy(value, commit, beforeChange) {
     }
   );
 }
+function unwrapProxy(value) {
+  let current = value;
+  while (current !== null && typeof current === "object") {
+    const target = onChange.target(current);
+    if (target === current) {
+      break;
+    }
+    current = target;
+  }
+  return current;
+}
+function isPlainObject(value) {
+  const proto = Object.getPrototypeOf(value);
+  return proto === Object.prototype || proto === null;
+}
+function unwrapDeep(value, seen) {
+  const unwrapped = unwrapProxy(value);
+  if (!unwrapped || typeof unwrapped !== "object") {
+    return unwrapped;
+  }
+  if (seen.has(unwrapped)) {
+    return unwrapped;
+  }
+  seen.add(unwrapped);
+  if (Array.isArray(unwrapped)) {
+    for (let i = 0; i < unwrapped.length; i++) {
+      const child = unwrapDeep(unwrapped[i], seen);
+      if (child !== unwrapped[i]) {
+        unwrapped[i] = child;
+      }
+    }
+    return unwrapped;
+  }
+  if (unwrapped instanceof Map) {
+    const replacements = [];
+    for (const [key, child] of unwrapped.entries()) {
+      const nextKey = unwrapDeep(key, seen);
+      const nextChild = unwrapDeep(child, seen);
+      if (nextKey !== key || nextChild !== child) {
+        replacements.push([key, nextKey, nextChild]);
+      }
+    }
+    for (const [key, nextKey, nextChild] of replacements) {
+      if (nextKey !== key) {
+        unwrapped.delete(key);
+      }
+      unwrapped.set(nextKey, nextChild);
+    }
+    return unwrapped;
+  }
+  if (unwrapped instanceof Set) {
+    const replacements = [];
+    for (const child of unwrapped.values()) {
+      const next = unwrapDeep(child, seen);
+      if (next !== child) {
+        replacements.push([child, next]);
+      }
+    }
+    for (const [child, next] of replacements) {
+      unwrapped.delete(child);
+      unwrapped.add(next);
+    }
+    return unwrapped;
+  }
+  if (isPlainObject(unwrapped)) {
+    for (const key of Object.keys(unwrapped)) {
+      const child = unwrapDeep(unwrapped[key], seen);
+      if (child !== unwrapped[key]) {
+        unwrapped[key] = child;
+      }
+    }
+  }
+  return unwrapped;
+}
+function unwrapWriteThroughProxy(value) {
+  return unwrapDeep(value, /* @__PURE__ */ new Set());
+}
 
 // src/registry/native.ts
 var textEncoder = new TextEncoder();
@@ -3483,8 +3560,23 @@ var NativeConnAdapter = class {
   }
   get state() {
     const nextState = this.#readState();
+    if (!this.#ctx) {
+      return this.#createStateProxy(nextState);
+    }
+    const connState = getNativeConnPersistState(
+      this.#runtime,
+      this.#ctx,
+      this.#conn
+    );
+    if (connState.stateProxy === void 0 || connState.stateProxyTarget !== nextState) {
+      connState.stateProxyTarget = nextState;
+      connState.stateProxy = this.#createStateProxy(nextState);
+    }
+    return connState.stateProxy;
+  }
+  #createStateProxy(state) {
     return createWriteThroughProxy(
-      nextState,
+      state,
       (nextValue) => {
         this.#writeState(nextValue, { writeNative: true });
       },
@@ -3494,11 +3586,14 @@ var NativeConnAdapter = class {
     );
   }
   set state(value) {
-    assertJsonCompatValue(value);
-    this.#writeState(value, { writeNative: true });
+    const nextValue = unwrapWriteThroughProxy(value);
+    assertJsonCompatValue(nextValue);
+    this.#writeState(nextValue, { writeNative: true });
   }
   initializeState(value) {
-    this.#writeState(value, { writeNative: false });
+    this.#writeState(unwrapWriteThroughProxy(value), {
+      writeNative: false
+    });
   }
   get isHibernatable() {
     return callNativeSync(
@@ -4551,14 +4646,17 @@ var ActorContextHandleAdapter = class {
       throw stateNotEnabledError();
     }
     this.#assertCanMutateState();
-    assertJsonCompatValue(value);
-    this.#writeState(value, { scheduleSave: true });
+    const nextValue = unwrapWriteThroughProxy(value);
+    assertJsonCompatValue(nextValue);
+    this.#writeState(nextValue, { scheduleSave: true });
   }
   initializeState(value) {
     if (!this.#stateEnabled) {
       return;
     }
-    this.#writeState(value, { scheduleSave: false });
+    this.#writeState(unwrapWriteThroughProxy(value), {
+      scheduleSave: false
+    });
   }
   get vars() {
     const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);