rivetkit 0.0.0-pr.4727.e649173 → 0.0.0-pr.4738.bcfa930

package/dist/browser/client.d.ts

@@ -933,15 +933,6 @@ interface ActorKv {
     list<T extends ActorKvValueType = "text", K extends ActorKvKeyType = "text">(prefix: Uint8Array | string, options?: ActorKvListOptions<T, K>): Promise<Array<[ActorKvKeyTypeMap[K], ActorKvValueTypeMap[T]]>>;
     [key: string]: any;
 }
-interface ActorSql {
-    exec(sql: string, callback?: (row: unknown[], columns: string[]) => void): Promise<void>;
-    run(sql: string, params?: unknown[] | Record<string, unknown>): Promise<void>;
-    query(sql: string, params?: unknown[] | Record<string, unknown>): Promise<{
-        columns: string[];
-        rows: unknown[][];
-    }>;
-    [key: string]: any;
-}
 interface ActorSchedule {
     after(duration: number, action: string, ...args: unknown[]): Promise<void>;
     at(timestamp: number, action: string, ...args: unknown[]): Promise<void>;
@@ -1019,7 +1010,6 @@ interface ActorContext<TState, TConnParams, TConnState, TVars, TInput, TDatabase
     state: TState;
     vars: TVars;
     readonly kv: ActorKv;
-    readonly sql: ActorSql;
     readonly db: InferDatabaseClient<TDatabase>;
     readonly schedule: ActorSchedule;
     readonly queue: ActorQueue<TQueues>;
@@ -1365,12 +1355,12 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     logging: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodObject<{
         baseLogger: z$2.ZodOptional<z$2.ZodCustom<Logger, Logger>>;
         level: z$2.ZodOptional<z$2.ZodEnum<{
-            error: "error";
-            trace: "trace";
             fatal: "fatal";
+            error: "error";
             warn: "warn";
             info: "info";
             debug: "debug";
+            trace: "trace";
             silent: "silent";
         }>>;
     }, z$2.core.$strip>>>;
@@ -1405,6 +1395,10 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
         totalSlots: z$2.ZodDefault<z$2.ZodNumber>;
         envoyKey: z$2.ZodOptional<z$2.ZodString>;
     }, z$2.core.$strip>>>;
+    shutdown: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodObject<{
+        gracePeriodMs: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodNumber>>;
+        disableSignalHandlers: z$2.ZodDefault<z$2.ZodOptional<z$2.ZodBoolean>>;
+    }, z$2.core.$strip>>>;
 }, z$2.core.$strip>, z$2.ZodTransform<{
     endpoint: string | undefined;
     namespace: string;
@@ -1428,7 +1422,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     noWelcome: boolean;
     logging: {
         baseLogger?: Logger | undefined;
-        level?: "error" | "trace" | "fatal" | "warn" | "info" | "debug" | "silent" | undefined;
+        level?: "fatal" | "error" | "warn" | "info" | "debug" | "trace" | "silent" | undefined;
     };
     headers: Record<string, string>;
     httpBasePath: string;
@@ -1441,6 +1435,10 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
         totalSlots: number;
         envoyKey?: string | undefined;
     };
+    shutdown: {
+        gracePeriodMs: number;
+        disableSignalHandlers: boolean;
+    };
     staticDir?: string | undefined;
     httpHost?: string | undefined;
     configurePool?: {
@@ -1462,7 +1460,7 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
     noWelcome: boolean;
     logging: {
         baseLogger?: Logger | undefined;
-        level?: "error" | "trace" | "fatal" | "warn" | "info" | "debug" | "silent" | undefined;
+        level?: "fatal" | "error" | "warn" | "info" | "debug" | "trace" | "silent" | undefined;
     };
     endpoint: string | undefined;
     token: string | undefined;
@@ -1484,6 +1482,10 @@ declare const RegistryConfigSchema: z$2.ZodPipe<z$2.ZodObject<{
         totalSlots: number;
         envoyKey?: string | undefined;
     };
+    shutdown: {
+        gracePeriodMs: number;
+        disableSignalHandlers: boolean;
+    };
     staticDir?: string | undefined;
     httpHost?: string | undefined;
     configurePool?: {

package/dist/browser/client.js

@@ -225,7 +225,7 @@ function noopNext() {
 // package.json
 var package_default = {
   name: "rivetkit",
-  version: "0.0.0-pr.4727.e649173",
+  version: "0.0.0-pr.4738.bcfa930",
   description: "Lightweight libraries for building stateful actors on edge platforms",
   license: "Apache-2.0",
   keywords: [
@@ -455,107 +455,6 @@ import {
   stdTimeFunctions
 } from "pino";
 import { z } from "zod/v4";
-
-// src/common/log-levels.ts
-var LogLevels = {
-  TRACE: 0,
-  DEBUG: 1,
-  INFO: 2,
-  WARN: 3,
-  ERROR: 4,
-  CRITICAL: 5
-};
-
-// src/common/logfmt.ts
-var LOG_LEVEL_COLORS = {
-  [LogLevels.CRITICAL]: "\x1B[31m",
-  // Red
-  [LogLevels.ERROR]: "\x1B[31m",
-  // Red
-  [LogLevels.WARN]: "\x1B[33m",
-  // Yellow
-  [LogLevels.INFO]: "\x1B[32m",
-  // Green
-  [LogLevels.DEBUG]: "\x1B[36m",
-  // Cyan
-  [LogLevels.TRACE]: "\x1B[36m"
-  // Cyan
-};
-var RESET_COLOR = "\x1B[0m";
-function stringify(data) {
-  let line = "";
-  const entries = Object.entries(data);
-  for (let i = 0; i < entries.length; i++) {
-    const [key, valueRaw] = entries[i];
-    let isNull = false;
-    let valueString;
-    if (valueRaw == null) {
-      isNull = true;
-      valueString = "";
-    } else {
-      valueString = valueRaw.toString();
-    }
-    if (valueString.length > 512 && key !== "msg" && key !== "error")
-      valueString = `${valueString.slice(0, 512)}...`;
-    const needsQuoting = valueString.indexOf(" ") > -1 || valueString.indexOf("=") > -1;
-    const needsEscaping = valueString.indexOf('"') > -1 || valueString.indexOf("\\") > -1;
-    valueString = valueString.replace(/\n/g, "\\n");
-    if (needsEscaping) valueString = valueString.replace(/["\\]/g, "\\$&");
-    if (needsQuoting || needsEscaping) valueString = `"${valueString}"`;
-    if (valueString === "" && !isNull) valueString = '""';
-    if (LOGGER_CONFIG.enableColor) {
-      let color = "\x1B[2m";
-      if (key === "level") {
-        const level = LogLevels[valueString];
-        const levelColor = LOG_LEVEL_COLORS[level];
-        if (levelColor) {
-          color = levelColor;
-        }
-      } else if (key === "msg") {
-        color = "\x1B[32m";
-      } else if (key === "trace") {
-        color = "\x1B[34m";
-      }
-      line += `\x1B[0m\x1B[1m${key}\x1B[0m\x1B[2m=\x1B[0m${color}${valueString}${RESET_COLOR}`;
-    } else {
-      line += `${key}=${valueString}`;
-    }
-    if (i !== entries.length - 1) {
-      line += " ";
-    }
-  }
-  return line;
-}
-function formatTimestamp(date) {
-  const year = date.getUTCFullYear();
-  const month = String(date.getUTCMonth() + 1).padStart(2, "0");
-  const day = String(date.getUTCDate()).padStart(2, "0");
-  const hours = String(date.getUTCHours()).padStart(2, "0");
-  const minutes = String(date.getUTCMinutes()).padStart(2, "0");
-  const seconds = String(date.getUTCSeconds()).padStart(2, "0");
-  const milliseconds = String(date.getUTCMilliseconds()).padStart(3, "0");
-  return `${year}-${month}-${day}T${hours}:${minutes}:${seconds}.${milliseconds}Z`;
-}
-function castToLogValue(v) {
-  if (typeof v === "string" || typeof v === "number" || typeof v === "bigint" || typeof v === "boolean" || v === null || v === void 0) {
-    return v;
-  }
-  if (v instanceof Error) {
-    return String(v);
-  }
-  try {
-    return JSON.stringify(v);
-  } catch {
-    return "[cannot stringify]";
-  }
-}
-var LOGGER_CONFIG = {
-  enableColor: false,
-  enableSpreadObject: false,
-  enableErrorStack: false
-};
-
-// src/common/log.ts
 var baseLogger;
 var configuredLogLevel;
 var loggerCache = /* @__PURE__ */ new Map();
@@ -585,27 +484,6 @@ function getPinoLevel(logLevel) {
 function getIncludeTarget() {
   return getLogTarget();
 }
-function customWrite(level, o) {
-  const entries = {};
-  if (getLogTimestamp() && o.time) {
-    const date = typeof o.time === "number" ? new Date(o.time) : /* @__PURE__ */ new Date();
-    entries.ts = formatTimestamp(date);
-  }
-  entries.level = level.toUpperCase();
-  if (o.target) {
-    entries.target = o.target;
-  }
-  if (o.msg) {
-    entries.msg = o.msg;
-  }
-  for (const [key, value] of Object.entries(o)) {
-    if (key !== "time" && key !== "level" && key !== "target" && key !== "msg" && key !== "pid" && key !== "hostname") {
-      entries[key] = castToLogValue(value);
-    }
-  }
-  const output = stringify(entries);
-  console.log(output);
-}
 function configureDefaultLogger(logLevel) {
   if (logLevel) {
     configuredLogLevel = logLevel;
@@ -621,64 +499,7 @@ function configureDefaultLogger(logLevel) {
         return { level: number };
       }
     },
-    timestamp: getLogTimestamp() ? stdTimeFunctions.epochTime : false,
-    browser: {
-      write: {
-        fatal: customWrite.bind(null, "fatal"),
-        error: customWrite.bind(null, "error"),
-        warn: customWrite.bind(null, "warn"),
-        info: customWrite.bind(null, "info"),
-        debug: customWrite.bind(null, "debug"),
-        trace: customWrite.bind(null, "trace")
-      }
-    },
-    hooks: {
-      logMethod(inputArgs, method, level) {
-        const levelMap = {
-          10: "trace",
-          20: "debug",
-          30: "info",
-          40: "warn",
-          50: "error",
-          60: "fatal"
-        };
-        const levelName = levelMap[level] || "info";
-        const time = getLogTimestamp() ? Date.now() : void 0;
-        const bindings = this.bindings?.() || {};
-        if (inputArgs.length >= 2) {
-          const [objOrMsg, msg] = inputArgs;
-          if (typeof objOrMsg === "object" && objOrMsg !== null) {
-            customWrite(levelName, {
-              ...bindings,
-              ...objOrMsg,
-              msg,
-              time
-            });
-          } else {
-            customWrite(levelName, {
-              ...bindings,
-              msg: String(objOrMsg),
-              time
-            });
-          }
-        } else if (inputArgs.length === 1) {
-          const [objOrMsg] = inputArgs;
-          if (typeof objOrMsg === "object" && objOrMsg !== null) {
-            customWrite(levelName, {
-              ...bindings,
-              ...objOrMsg,
-              time
-            });
-          } else {
-            customWrite(levelName, {
-              ...bindings,
-              msg: String(objOrMsg),
-              time
-            });
-          }
-        }
-      }
-    }
+    timestamp: getLogTimestamp() ? stdTimeFunctions.epochTime : false
   });
   loggerCache.clear();
 }
@@ -3878,8 +3699,8 @@ function getGatewayTarget(state) {
   }
   return state;
 }
-function isStaleResolvedActorError(group, code) {
-  return group === "actor" && (code === "not_found" || code === "stopping" || code.startsWith("destroyed_"));
+function isStaleResolvedActorError(group, code, message) {
+  return group === "actor" && (code === "not_found" || code === "stopping" || code === "not_configured" && message === "Actor capability 'actor event inbox' is not configured." || code.startsWith("destroyed_"));
 }
 async function checkForSchedulingError(group, code, actorId, query, driver) {
   const name = getActorNameFromQuery(query);
@@ -4041,15 +3862,15 @@ var ActorConnRaw = class {
    * re-resolves to a fresh actor. Returns true if the identity was
    * invalidated.
    */
-  #invalidateActorIfStale(group, code) {
-    if (!isDynamicActorQuery(this.#actorResolutionState) || !isStaleResolvedActorError(group, code)) {
+  #invalidateActorIfStale(group, code, message) {
+    if (!isDynamicActorQuery(this.#actorResolutionState) || !isStaleResolvedActorError(group, code, message)) {
       return false;
     }
     this.#clearResolvedActorIdentity();
     return true;
   }
-  #shouldReconnectForStaleActor(group, code) {
-    return isDynamicActorQuery(this.#actorResolutionState) && isStaleResolvedActorError(group, code) && this.#onOpenPromise !== void 0 && this.#connStatus !== "connected";
+  #shouldReconnectForStaleActor(group, code, message) {
+    return isDynamicActorQuery(this.#actorResolutionState) && isStaleResolvedActorError(group, code, message) && this.#onOpenPromise !== void 0 && this.#connStatus !== "connected";
   }
   send(name, body, options) {
     return this.#sendQueueMessage(name, body, options);
@@ -4363,7 +4184,7 @@ var ActorConnRaw = class {
       const { group, code, message, metadata, actionId } = response.body.val;
       if (actionId !== null) {
         const inFlight = this.#takeActionInFlight(Number(actionId));
-        this.#invalidateActorIfStale(group, code);
+        this.#invalidateActorIfStale(group, code, message);
         logger2().warn({
           msg: "action error",
           actionId,
@@ -4384,7 +4205,7 @@ var ActorConnRaw = class {
           message,
           metadata
         });
-        if (this.#shouldReconnectForStaleActor(group, code)) {
+        if (this.#shouldReconnectForStaleActor(group, code, message)) {
           this.#clearResolvedActorIdentity();
           this.#onOpenPromise?.reject(
             new RivetError(group, code, message, metadata)
@@ -4412,7 +4233,7 @@ var ActorConnRaw = class {
         if (this.#onOpenPromise) {
           this.#onOpenPromise.reject(errorToThrow);
         }
-        this.#invalidateActorIfStale(group, code);
+        this.#invalidateActorIfStale(group, code, message);
         for (const [id, inFlight] of this.#actionsInFlight.entries()) {
           inFlight.reject(errorToThrow);
           this.#actionsInFlight.delete(id);
@@ -5115,6 +4936,9 @@ async function resolveGatewayTarget(driver, target) {
 }
 
 // src/client/actor-handle.ts
+var DYNAMIC_LIFECYCLE_RETRY_WINDOW_MS = 100;
+var GET_FOR_KEY_DYNAMIC_QUERY_MAX_ATTEMPTS = 24;
+var GET_OR_CREATE_DYNAMIC_QUERY_MAX_ATTEMPTS = 180;
 var ActorHandleRaw = class {
   #client;
   #driver;
@@ -5177,7 +5001,7 @@ var ActorHandleRaw = class {
           attempt,
           maxAttempts
         )) {
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
@@ -5185,16 +5009,21 @@ var ActorHandleRaw = class {
           group,
           code,
           attempt,
-          maxAttempts
+          maxAttempts,
+          message
         )) {
           this.#clearResolvedActorId();
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code,
+          message
+        );
         if (invalidated && attempt < maxAttempts - 1) {
-          useQueryTarget = code === "stopping" || code.startsWith("destroyed_");
+          useQueryTarget = this.#shouldRetryViaQueryTarget() && (code === "stopping" || code.startsWith("destroyed_"));
           if (useQueryTarget) {
             await this.#waitForRetryWindow();
           }
@@ -5290,7 +5119,7 @@ var ActorHandleRaw = class {
           attempt,
           maxAttempts
         )) {
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
@@ -5298,10 +5127,11 @@ var ActorHandleRaw = class {
           group,
           code,
           attempt,
-          maxAttempts
+          maxAttempts,
+          message
         )) {
           this.#clearResolvedActorId();
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
@@ -5313,10 +5143,14 @@ var ActorHandleRaw = class {
             metadata
           );
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code,
+          message
+        );
         if (invalidated && attempt < maxAttempts - 1) {
           if (group === "actor" && code === "stopping") {
-            useQueryTarget = true;
+            useQueryTarget = this.#shouldRetryViaQueryTarget();
             await new Promise((resolve) => setTimeout(resolve, 100));
           }
           continue;
@@ -5338,19 +5172,24 @@ var ActorHandleRaw = class {
     }
   }
   async #waitForRetryWindow() {
-    await new Promise((resolve) => setTimeout(resolve, 100));
+    await new Promise(
+      (resolve) => setTimeout(resolve, DYNAMIC_LIFECYCLE_RETRY_WINDOW_MS)
+    );
   }
   #getDynamicQueryMaxAttempts() {
     if (!isDynamicActorQuery(this.#actorResolutionState)) {
       return 1;
     }
-    return "getOrCreateForKey" in this.#actorResolutionState ? 60 : 24;
+    return "getOrCreateForKey" in this.#actorResolutionState ? GET_OR_CREATE_DYNAMIC_QUERY_MAX_ATTEMPTS : GET_FOR_KEY_DYNAMIC_QUERY_MAX_ATTEMPTS;
   }
-  #shouldRetryDynamicLifecycleError(group, code, attempt, maxAttempts) {
+  #shouldRetryViaQueryTarget() {
+    return isDynamicActorQuery(this.#actorResolutionState);
+  }
+  #shouldRetryDynamicLifecycleError(group, code, attempt, maxAttempts, message) {
     if (!isDynamicActorQuery(this.#actorResolutionState) || attempt >= maxAttempts - 1 || group !== "actor") {
       return false;
     }
-    return code === "not_found" || code === "stopping" || code === "destroying" || code.startsWith("destroyed_");
+    return code === "not_found" || code === "stopping" || code === "not_configured" && message === "Actor capability 'actor event inbox' is not configured." || code === "destroying" || code.startsWith("destroyed_");
   }
   #clearResolvedActorId() {
     this.#resolvedActorId = void 0;
@@ -5375,8 +5214,8 @@ var ActorHandleRaw = class {
     this.#clearResolvedActorId();
     return true;
   }
-  #invalidateResolvedActorId(group, code) {
-    if (!isDynamicActorQuery(this.#actorResolutionState) || !isStaleResolvedActorError(group, code)) {
+  #invalidateResolvedActorId(group, code, message) {
+    if (!isDynamicActorQuery(this.#actorResolutionState) || !isStaleResolvedActorError(group, code, message)) {
       return false;
     }
     this.#clearResolvedActorId();
@@ -5482,7 +5321,7 @@ var ActorHandleRaw = class {
           attempt,
           maxAttempts
         )) {
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
@@ -5490,16 +5329,21 @@ var ActorHandleRaw = class {
           group,
           code,
           attempt,
-          maxAttempts
+          maxAttempts,
+          message
         )) {
           this.#clearResolvedActorId();
-          useQueryTarget = true;
+          useQueryTarget = this.#shouldRetryViaQueryTarget();
           await this.#waitForRetryWindow();
           continue;
         }
-        const invalidated = this.#invalidateResolvedActorId(group, code);
+        const invalidated = this.#invalidateResolvedActorId(
+          group,
+          code,
+          message
+        );
         if (invalidated && attempt < maxAttempts - 1) {
-          useQueryTarget = code === "stopping" || code.startsWith("destroyed_");
+          useQueryTarget = this.#shouldRetryViaQueryTarget() && (code === "stopping" || code.startsWith("destroyed_"));
           if (useQueryTarget) {
             await this.#waitForRetryWindow();
           }
@@ -5518,7 +5362,7 @@ var ActorHandleRaw = class {
     if (!error) {
       return null;
     }
-    const { group, code } = error;
+    const { group, code, message } = error;
     if (await this.#shouldRetrySchedulingError(
       group,
       code,
@@ -5527,7 +5371,7 @@ var ActorHandleRaw = class {
       maxAttempts
     )) {
       return {
-        useQueryTarget: true,
+        useQueryTarget: this.#shouldRetryViaQueryTarget(),
         waitForRetryWindow: true
       };
     }
@@ -5535,17 +5379,22 @@ var ActorHandleRaw = class {
       group,
       code,
       attempt,
-      maxAttempts
+      maxAttempts,
+      message
     )) {
       this.#clearResolvedActorId();
       return {
-        useQueryTarget: true,
+        useQueryTarget: this.#shouldRetryViaQueryTarget(),
         waitForRetryWindow: true
       };
     }
-    const invalidated = this.#invalidateResolvedActorId(group, code);
+    const invalidated = this.#invalidateResolvedActorId(
+      group,
+      code,
+      message
+    );
     if (invalidated && attempt < maxAttempts - 1) {
-      const useQueryTarget = code === "stopping" || code.startsWith("destroyed_");
+      const useQueryTarget = this.#shouldRetryViaQueryTarget() && (code === "stopping" || code.startsWith("destroyed_"));
       return {
         useQueryTarget,
         waitForRetryWindow: useQueryTarget