rivetkit 2.1.5 → 2.1.6-rc.1

package/dist/tsup/actor/errors.cjs

@@ -49,7 +49,7 @@
 
 
 
-var _chunkL47L3ZWJcjs = require('../chunk-L47L3ZWJ.cjs');
+var _chunk6G7ZNM27cjs = require('../chunk-6G7ZNM27.cjs');
 require('../chunk-HDQ2JUQT.cjs');
 
 
@@ -102,5 +102,5 @@ require('../chunk-HDQ2JUQT.cjs');
 
 
 
-exports.ActionNotFound = _chunkL47L3ZWJcjs.ActionNotFound; exports.ActionTimedOut = _chunkL47L3ZWJcjs.ActionTimedOut; exports.ActorAborted = _chunkL47L3ZWJcjs.ActorAborted; exports.ActorDuplicateKey = _chunkL47L3ZWJcjs.ActorDuplicateKey; exports.ActorError = _chunkL47L3ZWJcjs.ActorError; exports.ActorNotFound = _chunkL47L3ZWJcjs.ActorNotFound; exports.ActorStopping = _chunkL47L3ZWJcjs.ActorStopping; exports.ConnStateNotEnabled = _chunkL47L3ZWJcjs.ConnStateNotEnabled; exports.DatabaseNotEnabled = _chunkL47L3ZWJcjs.DatabaseNotEnabled; exports.EndpointMismatch = _chunkL47L3ZWJcjs.EndpointMismatch; exports.EventPayloadInvalid = _chunkL47L3ZWJcjs.EventPayloadInvalid; exports.FeatureNotImplemented = _chunkL47L3ZWJcjs.FeatureNotImplemented; exports.Forbidden = _chunkL47L3ZWJcjs.Forbidden; exports.INTERNAL_ERROR_CODE = _chunkL47L3ZWJcjs.INTERNAL_ERROR_CODE; exports.INTERNAL_ERROR_DESCRIPTION = _chunkL47L3ZWJcjs.INTERNAL_ERROR_DESCRIPTION; exports.IncomingMessageTooLong = _chunkL47L3ZWJcjs.IncomingMessageTooLong; exports.InternalError = _chunkL47L3ZWJcjs.InternalError; exports.InvalidActionRequest = _chunkL47L3ZWJcjs.InvalidActionRequest; exports.InvalidCanPublishResponse = _chunkL47L3ZWJcjs.InvalidCanPublishResponse; exports.InvalidCanSubscribeResponse = _chunkL47L3ZWJcjs.InvalidCanSubscribeResponse; exports.InvalidEncoding = _chunkL47L3ZWJcjs.InvalidEncoding; exports.InvalidParams = _chunkL47L3ZWJcjs.InvalidParams; exports.InvalidQueryJSON = _chunkL47L3ZWJcjs.InvalidQueryJSON; exports.InvalidRequest = _chunkL47L3ZWJcjs.InvalidRequest; exports.InvalidRequestHandlerResponse = _chunkL47L3ZWJcjs.InvalidRequestHandlerResponse; exports.InvalidStateType = _chunkL47L3ZWJcjs.InvalidStateType; exports.MalformedMessage = _chunkL47L3ZWJcjs.MalformedMessage; exports.MissingActorHeader = _chunkL47L3ZWJcjs.MissingActorHeader; exports.NamespaceMismatch = _chunkL47L3ZWJcjs.NamespaceMismatch; exports.OutgoingMessageTooLong = _chunkL47L3ZWJcjs.OutgoingMessageTooLong; exports.ProxyError = _chunkL47L3ZWJcjs.ProxyError; exports.QueueAlreadyCompleted = _chunkL47L3ZWJcjs.QueueAlreadyCompleted; exports.QueueCompleteNotConfigured = _chunkL47L3ZWJcjs.QueueCompleteNotConfigured; exports.QueueCompletionPayloadInvalid = _chunkL47L3ZWJcjs.QueueCompletionPayloadInvalid; exports.QueueFull = _chunkL47L3ZWJcjs.QueueFull; exports.QueueMessageInvalid = _chunkL47L3ZWJcjs.QueueMessageInvalid; exports.QueueMessageTooLarge = _chunkL47L3ZWJcjs.QueueMessageTooLarge; exports.QueuePayloadInvalid = _chunkL47L3ZWJcjs.QueuePayloadInvalid; exports.QueuePreviousMessageNotCompleted = _chunkL47L3ZWJcjs.QueuePreviousMessageNotCompleted; exports.RequestHandlerNotDefined = _chunkL47L3ZWJcjs.RequestHandlerNotDefined; exports.RestrictedFeature = _chunkL47L3ZWJcjs.RestrictedFeature; exports.RouteNotFound = _chunkL47L3ZWJcjs.RouteNotFound; exports.StateNotEnabled = _chunkL47L3ZWJcjs.StateNotEnabled; exports.USER_ERROR_CODE = _chunkL47L3ZWJcjs.USER_ERROR_CODE; exports.Unreachable = _chunkL47L3ZWJcjs.Unreachable; exports.Unsupported = _chunkL47L3ZWJcjs.Unsupported; exports.UserError = _chunkL47L3ZWJcjs.UserError; exports.VarsNotEnabled = _chunkL47L3ZWJcjs.VarsNotEnabled; exports.WebSocketHandlerNotDefined = _chunkL47L3ZWJcjs.WebSocketHandlerNotDefined; exports.WebSocketsNotEnabled = _chunkL47L3ZWJcjs.WebSocketsNotEnabled;
+exports.ActionNotFound = _chunk6G7ZNM27cjs.ActionNotFound; exports.ActionTimedOut = _chunk6G7ZNM27cjs.ActionTimedOut; exports.ActorAborted = _chunk6G7ZNM27cjs.ActorAborted; exports.ActorDuplicateKey = _chunk6G7ZNM27cjs.ActorDuplicateKey; exports.ActorError = _chunk6G7ZNM27cjs.ActorError; exports.ActorNotFound = _chunk6G7ZNM27cjs.ActorNotFound; exports.ActorStopping = _chunk6G7ZNM27cjs.ActorStopping; exports.ConnStateNotEnabled = _chunk6G7ZNM27cjs.ConnStateNotEnabled; exports.DatabaseNotEnabled = _chunk6G7ZNM27cjs.DatabaseNotEnabled; exports.EndpointMismatch = _chunk6G7ZNM27cjs.EndpointMismatch; exports.EventPayloadInvalid = _chunk6G7ZNM27cjs.EventPayloadInvalid; exports.FeatureNotImplemented = _chunk6G7ZNM27cjs.FeatureNotImplemented; exports.Forbidden = _chunk6G7ZNM27cjs.Forbidden; exports.INTERNAL_ERROR_CODE = _chunk6G7ZNM27cjs.INTERNAL_ERROR_CODE; exports.INTERNAL_ERROR_DESCRIPTION = _chunk6G7ZNM27cjs.INTERNAL_ERROR_DESCRIPTION; exports.IncomingMessageTooLong = _chunk6G7ZNM27cjs.IncomingMessageTooLong; exports.InternalError = _chunk6G7ZNM27cjs.InternalError; exports.InvalidActionRequest = _chunk6G7ZNM27cjs.InvalidActionRequest; exports.InvalidCanPublishResponse = _chunk6G7ZNM27cjs.InvalidCanPublishResponse; exports.InvalidCanSubscribeResponse = _chunk6G7ZNM27cjs.InvalidCanSubscribeResponse; exports.InvalidEncoding = _chunk6G7ZNM27cjs.InvalidEncoding; exports.InvalidParams = _chunk6G7ZNM27cjs.InvalidParams; exports.InvalidQueryJSON = _chunk6G7ZNM27cjs.InvalidQueryJSON; exports.InvalidRequest = _chunk6G7ZNM27cjs.InvalidRequest; exports.InvalidRequestHandlerResponse = _chunk6G7ZNM27cjs.InvalidRequestHandlerResponse; exports.InvalidStateType = _chunk6G7ZNM27cjs.InvalidStateType; exports.MalformedMessage = _chunk6G7ZNM27cjs.MalformedMessage; exports.MissingActorHeader = _chunk6G7ZNM27cjs.MissingActorHeader; exports.NamespaceMismatch = _chunk6G7ZNM27cjs.NamespaceMismatch; exports.OutgoingMessageTooLong = _chunk6G7ZNM27cjs.OutgoingMessageTooLong; exports.ProxyError = _chunk6G7ZNM27cjs.ProxyError; exports.QueueAlreadyCompleted = _chunk6G7ZNM27cjs.QueueAlreadyCompleted; exports.QueueCompleteNotConfigured = _chunk6G7ZNM27cjs.QueueCompleteNotConfigured; exports.QueueCompletionPayloadInvalid = _chunk6G7ZNM27cjs.QueueCompletionPayloadInvalid; exports.QueueFull = _chunk6G7ZNM27cjs.QueueFull; exports.QueueMessageInvalid = _chunk6G7ZNM27cjs.QueueMessageInvalid; exports.QueueMessageTooLarge = _chunk6G7ZNM27cjs.QueueMessageTooLarge; exports.QueuePayloadInvalid = _chunk6G7ZNM27cjs.QueuePayloadInvalid; exports.QueuePreviousMessageNotCompleted = _chunk6G7ZNM27cjs.QueuePreviousMessageNotCompleted; exports.RequestHandlerNotDefined = _chunk6G7ZNM27cjs.RequestHandlerNotDefined; exports.RestrictedFeature = _chunk6G7ZNM27cjs.RestrictedFeature; exports.RouteNotFound = _chunk6G7ZNM27cjs.RouteNotFound; exports.StateNotEnabled = _chunk6G7ZNM27cjs.StateNotEnabled; exports.USER_ERROR_CODE = _chunk6G7ZNM27cjs.USER_ERROR_CODE; exports.Unreachable = _chunk6G7ZNM27cjs.Unreachable; exports.Unsupported = _chunk6G7ZNM27cjs.Unsupported; exports.UserError = _chunk6G7ZNM27cjs.UserError; exports.VarsNotEnabled = _chunk6G7ZNM27cjs.VarsNotEnabled; exports.WebSocketHandlerNotDefined = _chunk6G7ZNM27cjs.WebSocketHandlerNotDefined; exports.WebSocketsNotEnabled = _chunk6G7ZNM27cjs.WebSocketsNotEnabled;
 //# sourceMappingURL=errors.cjs.map

package/dist/tsup/actor/errors.js

@@ -49,7 +49,7 @@ import {
   VarsNotEnabled,
   WebSocketHandlerNotDefined,
   WebSocketsNotEnabled
-} from "../chunk-7HTNH26M.js";
+} from "../chunk-AKUJ5OTO.js";
 import "../chunk-424PT5DM.js";
 export {
   ActionNotFound,

package/dist/tsup/{actor-router-consts-D29T1Z-K.d.cts → actor-router-consts-DU-1IdQj.d.cts}

@@ -21,4 +21,4 @@ declare const WS_TEST_PROTOCOL_PATH = "test_path.";
  **/
 declare const ALLOWED_PUBLIC_HEADERS: string[];
 
-export { ALLOWED_PUBLIC_HEADERS as A, HEADER_ACTOR_ID as H, PATH_CONNECT as P, WS_PROTOCOL_ACTOR as W, PATH_WEBSOCKET_PREFIX as a, HEADER_ACTOR_QUERY as b, HEADER_CONN_PARAMS as c, HEADER_ENCODING as d, HEADER_RIVET_ACTOR as e, HEADER_RIVET_TARGET as f, PATH_WEBSOCKET_BASE as g, WS_PROTOCOL_CONN_PARAMS as h, WS_PROTOCOL_ENCODING as i, WS_TEST_PROTOCOL_PATH as j, WS_PROTOCOL_STANDARD as k, WS_PROTOCOL_TARGET as l };
+export { ALLOWED_PUBLIC_HEADERS as A, HEADER_ACTOR_ID as H, PATH_CONNECT as P, WS_PROTOCOL_ACTOR as W, HEADER_ACTOR_QUERY as a, HEADER_CONN_PARAMS as b, HEADER_ENCODING as c, HEADER_RIVET_ACTOR as d, HEADER_RIVET_TARGET as e, PATH_WEBSOCKET_BASE as f, PATH_WEBSOCKET_PREFIX as g, WS_PROTOCOL_CONN_PARAMS as h, WS_PROTOCOL_ENCODING as i, WS_TEST_PROTOCOL_PATH as j, WS_PROTOCOL_STANDARD as k, WS_PROTOCOL_TARGET as l };

package/dist/tsup/{actor-router-consts-D29T1Z-K.d.ts → actor-router-consts-DU-1IdQj.d.ts}

@@ -21,4 +21,4 @@ declare const WS_TEST_PROTOCOL_PATH = "test_path.";
  **/
 declare const ALLOWED_PUBLIC_HEADERS: string[];
 
-export { ALLOWED_PUBLIC_HEADERS as A, HEADER_ACTOR_ID as H, PATH_CONNECT as P, WS_PROTOCOL_ACTOR as W, PATH_WEBSOCKET_PREFIX as a, HEADER_ACTOR_QUERY as b, HEADER_CONN_PARAMS as c, HEADER_ENCODING as d, HEADER_RIVET_ACTOR as e, HEADER_RIVET_TARGET as f, PATH_WEBSOCKET_BASE as g, WS_PROTOCOL_CONN_PARAMS as h, WS_PROTOCOL_ENCODING as i, WS_TEST_PROTOCOL_PATH as j, WS_PROTOCOL_STANDARD as k, WS_PROTOCOL_TARGET as l };
+export { ALLOWED_PUBLIC_HEADERS as A, HEADER_ACTOR_ID as H, PATH_CONNECT as P, WS_PROTOCOL_ACTOR as W, HEADER_ACTOR_QUERY as a, HEADER_CONN_PARAMS as b, HEADER_ENCODING as c, HEADER_RIVET_ACTOR as d, HEADER_RIVET_TARGET as e, PATH_WEBSOCKET_BASE as f, PATH_WEBSOCKET_PREFIX as g, WS_PROTOCOL_CONN_PARAMS as h, WS_PROTOCOL_ENCODING as i, WS_TEST_PROTOCOL_PATH as j, WS_PROTOCOL_STANDARD as k, WS_PROTOCOL_TARGET as l };

package/dist/tsup/chunk-2ELYUO6C.cjs

@@ -0,0 +1,112 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+
+
+var _chunkSQFCIDCGcjs = require('./chunk-SQFCIDCG.cjs');
+
+// src/db/mod.ts
+function db({
+  onMigrate
+} = {}) {
+  return {
+    createClient: async (ctx) => {
+      const override = ctx.overrideRawDatabaseClient ? await ctx.overrideRawDatabaseClient() : void 0;
+      if (override) {
+        return {
+          execute: async (query, ...args) => {
+            return await override.exec(query, ...args);
+          },
+          close: async () => {
+          }
+        };
+      }
+      if (!ctx.sqliteVfs) {
+        throw new Error(
+          "SqliteVfs instance not provided in context. The driver must provide a sqliteVfs instance."
+        );
+      }
+      const kvStore = _chunkSQFCIDCGcjs.createActorKvStore.call(void 0, ctx.kv, ctx.metrics);
+      const db2 = await ctx.sqliteVfs.open(ctx.actorId, kvStore);
+      let closed = false;
+      const mutex = new (0, _chunkSQFCIDCGcjs.AsyncMutex)();
+      const ensureOpen = () => {
+        if (closed) {
+          throw new Error("database is closed");
+        }
+      };
+      return {
+        execute: async (query, ...args) => {
+          return await mutex.run(async () => {
+            var _a;
+            ensureOpen();
+            const start = performance.now();
+            let result;
+            if (args.length > 0) {
+              const bindings = _chunkSQFCIDCGcjs.toSqliteBindings.call(void 0, args);
+              const token = query.trimStart().slice(0, 16).toUpperCase();
+              const returnsRows = token.startsWith("SELECT") || token.startsWith("PRAGMA") || token.startsWith("WITH") || /\bRETURNING\b/i.test(query);
+              if (returnsRows) {
+                const { rows, columns } = await db2.query(
+                  query,
+                  bindings
+                );
+                result = rows.map((row) => {
+                  const rowObj = {};
+                  for (let i = 0; i < columns.length; i++) {
+                    rowObj[columns[i]] = row[i];
+                  }
+                  return rowObj;
+                });
+              } else {
+                await db2.run(query, bindings);
+                result = [];
+              }
+            } else {
+              const results = [];
+              let columnNames = null;
+              await db2.exec(
+                query,
+                (row, columns) => {
+                  if (!columnNames) {
+                    columnNames = columns;
+                  }
+                  const rowObj = {};
+                  for (let i = 0; i < row.length; i++) {
+                    rowObj[columnNames[i]] = row[i];
+                  }
+                  results.push(rowObj);
+                }
+              );
+              result = results;
+            }
+            (_a = ctx.metrics) == null ? void 0 : _a.trackSql(query, performance.now() - start);
+            return result;
+          });
+        },
+        close: async () => {
+          const shouldClose = await mutex.run(async () => {
+            if (closed) return false;
+            closed = true;
+            return true;
+          });
+          if (shouldClose) {
+            await db2.close();
+          }
+        }
+      };
+    },
+    onMigrate: async (client) => {
+      if (onMigrate) {
+        await onMigrate(client);
+      }
+    },
+    onDestroy: async (client) => {
+      await client.close();
+    }
+  };
+}
+
+
+
+exports.db = db;
+//# sourceMappingURL=chunk-2ELYUO6C.cjs.map

package/dist/tsup/chunk-2ELYUO6C.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-2ELYUO6C.cjs","../../src/db/mod.ts"],"names":["db"],"mappings":"AAAA;AACE;AACA;AACA;AACF,wDAA6B;AAC7B;AACA;ACGO,SAAS,EAAA,CAAG;AAAA,EAClB;AACD,EAAA,EAA2B,CAAC,CAAA,EAAgC;AAC3D,EAAA,OAAO;AAAA,IACN,YAAA,EAAc,MAAA,CAAO,GAAA,EAAA,GAAQ;AAE5B,MAAA,MAAM,SAAA,EAAW,GAAA,CAAI,0BAAA,EAClB,MAAM,GAAA,CAAI,yBAAA,CAA0B,EAAA,EACpC,KAAA,CAAA;AAEH,MAAA,GAAA,CAAI,QAAA,EAAU;AAEb,QAAA,OAAO;AAAA,UACN,OAAA,EAAS,MAAA,CAMR,KAAA,EAAA,GACG,IAAA,EAAA,GACkB;AACrB,YAAA,OAAO,MAAM,QAAA,CAAS,IAAA,CAAW,KAAA,EAAO,GAAG,IAAI,CAAA;AAAA,UAChD,CAAA;AAAA,UACA,KAAA,EAAO,MAAA,CAAA,EAAA,GAAY;AAAA,UAEnB;AAAA,QACD,CAAA;AAAA,MACD;AAGA,MAAA,GAAA,CAAI,CAAC,GAAA,CAAI,SAAA,EAAW;AACnB,QAAA,MAAM,IAAI,KAAA;AAAA,UACT;AAAA,QACD,CAAA;AAAA,MACD;AAEA,MAAA,MAAM,QAAA,EAAU,kDAAA,GAAmB,CAAI,EAAA,EAAI,GAAA,CAAI,OAAO,CAAA;AACtD,MAAA,MAAMA,IAAAA,EAAK,MAAM,GAAA,CAAI,SAAA,CAAU,IAAA,CAAK,GAAA,CAAI,OAAA,EAAS,OAAO,CAAA;AACxD,MAAA,IAAI,OAAA,EAAS,KAAA;AACb,MAAA,MAAM,MAAA,EAAQ,IAAI,iCAAA,CAAW,CAAA;AAC7B,MAAA,MAAM,WAAA,EAAa,CAAA,EAAA,GAAM;AACxB,QAAA,GAAA,CAAI,MAAA,EAAQ;AACX,UAAA,MAAM,IAAI,KAAA,CAAM,oBAAoB,CAAA;AAAA,QACrC;AAAA,MACD,CAAA;AAEA,MAAA,OAAO;AAAA,QACN,OAAA,EAAS,MAAA,CAMR,KAAA,EAAA,GACG,IAAA,EAAA,GACkB;AACrB,UAAA,OAAO,MAAM,KAAA,CAAM,GAAA,CAAI,MAAA,CAAA,EAAA,GAAY;AAlExC,YAAA,IAAA,EAAA;AAmEM,YAAA,UAAA,CAAW,CAAA;AAEX,YAAA,MAAM,MAAA,EAAQ,WAAA,CAAY,GAAA,CAAI,CAAA;AAO9B,YAAA,IAAI,MAAA;AACJ,YAAA,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,CAAA,EAAG;AACpB,cAAA,MAAM,SAAA,EAAW,gDAAA,IAAqB,CAAA;AACtC,cAAA,MAAM,MAAA,EAAQ,KAAA,CACZ,SAAA,CAAU,CAAA,CACV,KAAA,CAAM,CAAA,EAAG,EAAE,CAAA,CACX,WAAA,CAAY,CAAA;AACd,cAAA,MAAM,YAAA,EACL,KAAA,CAAM,UAAA,CAAW,QAAQ,EAAA,GACzB,KAAA,CAAM,UAAA,CAAW,QAAQ,EAAA,GACzB,KAAA,CAAM,UAAA,CAAW,MAAM,EAAA,GACvB,gBAAA,CAAiB,IAAA,CAAK,KAAK,CAAA;AAE5B,cAAA,GAAA,CAAI,WAAA,EAAa;AAChB,gBAAA,MAAM,EAAE,IAAA,EAAM,QAAQ,EAAA,EAAI,MAAMA,GAAAA,CAAG,KAAA;AAAA,kBAClC,KAAA;AAAA,kBACA;AAAA,gBACD,CAAA;AACA,gBAAA,OAAA,EAAS,IAAA,CAAK,GAAA,CAAI,CAAC,GAAA,EAAA,GAAmB;AACrC,kBAAA,MAAM,OAAA,EAAkC,CAAC,CAAA;AACzC,kBAAA,IAAA,CAAA,IAAS,EAAA,EAAI,CAAA,EAAG,EAAA,EAAI,OAAA,CAAQ,MAAA,EAAQ,CAAA,EAAA,EAAK;AACxC,oBAAA,MAAA,CAAO,OAAA,CAAQ,CAAC,CAAC,EAAA,EAAI,GAAA,CAAI,CAAC,CAAA;AAAA,kBAC3B;AACA,kBAAA,OAAO,MAAA;AAAA,gBACR,CAAC,CAAA;AAAA,cACF,EAAA,KAAO;AACN,gBAAA,MAAMA,GAAAA,CAAG,GAAA,CAAI,KAAA,EAAO,QAAQ,CAAA;AAC5B,gBAAA,OAAA,EAAS,CAAC,CAAA;AAAA,cACX;AAAA,YACD,EAAA,KAAO;AACN,cAAA,MAAM,QAAA,EAAqC,CAAC,CAAA;AAC5C,cAAA,IAAI,YAAA,EAA+B,IAAA;AACnC,cAAA,MAAMA,GAAAA,CAAG,IAAA;AAAA,gBACR,KAAA;AAAA,gBACA,CAAC,GAAA,EAAgB,OAAA,EAAA,GAAsB;AACtC,kBAAA,GAAA,CAAI,CAAC,WAAA,EAAa;AACjB,oBAAA,YAAA,EAAc,OAAA;AAAA,kBACf;AACA,kBAAA,MAAM,OAAA,EAAkC,CAAC,CAAA;AACzC,kBAAA,IAAA,CAAA,IAAS,EAAA,EAAI,CAAA,EAAG,EAAA,EAAI,GAAA,CAAI,MAAA,EAAQ,CAAA,EAAA,EAAK;AACpC,oBAAA,MAAA,CAAO,WAAA,CAAY,CAAC,CAAC,EAAA,EAAI,GAAA,CAAI,CAAC,CAAA;AAAA,kBAC/B;AACA,kBAAA,OAAA,CAAQ,IAAA,CAAK,MAAM,CAAA;AAAA,gBACpB;AAAA,cACD,CAAA;AACA,cAAA,OAAA,EAAS,OAAA;AAAA,YACV;AAEA,YAAA,CAAA,GAAA,EAAA,GAAA,CAAI,OAAA,EAAA,GAAJ,KAAA,EAAA,KAAA,EAAA,EAAA,EAAA,CAAa,QAAA,CAAS,KAAA,EAAO,WAAA,CAAY,GAAA,CAAI,EAAA,EAAI,KAAA,CAAA;AACjD,YAAA,OAAO,MAAA;AAAA,UACR,CAAC,CAAA;AAAA,QACF,CAAA;AAAA,QACA,KAAA,EAAO,MAAA,CAAA,EAAA,GAAY;AAClB,UAAA,MAAM,YAAA,EAAc,MAAM,KAAA,CAAM,GAAA,CAAI,MAAA,CAAA,EAAA,GAAY;AAC/C,YAAA,GAAA,CAAI,MAAA,EAAQ,OAAO,KAAA;AACnB,YAAA,OAAA,EAAS,IAAA;AACT,YAAA,OAAO,IAAA;AAAA,UACR,CAAC,CAAA;AACD,UAAA,GAAA,CAAI,WAAA,EAAa;AAChB,YAAA,MAAMA,GAAAA,CAAG,KAAA,CAAM,CAAA;AAAA,UAChB;AAAA,QACD;AAAA,MACD,CAAA;AAAA,IACD,CAAA;AAAA,IACA,SAAA,EAAW,MAAA,CAAO,MAAA,EAAA,GAAW;AAC5B,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,MAAM,SAAA,CAAU,MAAM,CAAA;AAAA,MACvB;AAAA,IACD,CAAA;AAAA,IACA,SAAA,EAAW,MAAA,CAAO,MAAA,EAAA,GAAW;AAC5B,MAAA,MAAM,MAAA,CAAO,KAAA,CAAM,CAAA;AAAA,IACpB;AAAA,EACD,CAAA;AACD;AD1CA;AACA;AACE;AACF,gBAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-2ELYUO6C.cjs","sourcesContent":[null,"import type { DatabaseProvider, RawAccess } from \"./config\";\nimport { AsyncMutex, createActorKvStore, toSqliteBindings } from \"./shared\";\n\nexport type { RawAccess } from \"./config\";\n\ninterface DatabaseFactoryConfig {\n\tonMigrate?: (db: RawAccess) => Promise<void> | void;\n}\n\nexport function db({\n\tonMigrate,\n}: DatabaseFactoryConfig = {}): DatabaseProvider<RawAccess> {\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Check if override is provided\n\t\t\tconst override = ctx.overrideRawDatabaseClient\n\t\t\t\t? await ctx.overrideRawDatabaseClient()\n\t\t\t\t: undefined;\n\n\t\t\tif (override) {\n\t\t\t\t// Use the override\n\t\t\t\treturn {\n\t\t\t\t\texecute: async <\n\t\t\t\t\t\tTRow extends Record<string, unknown> = Record<\n\t\t\t\t\t\t\tstring,\n\t\t\t\t\t\t\tunknown\n\t\t\t\t\t\t>,\n\t\t\t\t\t>(\n\t\t\t\t\t\tquery: string,\n\t\t\t\t\t\t...args: unknown[]\n\t\t\t\t\t): Promise<TRow[]> => {\n\t\t\t\t\t\treturn await override.exec<TRow>(query, ...args);\n\t\t\t\t\t},\n\t\t\t\t\tclose: async () => {\n\t\t\t\t\t\t// Override clients don't need cleanup\n\t\t\t\t\t},\n\t\t\t\t} satisfies RawAccess;\n\t\t\t}\n\n\t\t\t// Construct KV-backed client using actor driver's KV operations\n\t\t\tif (!ctx.sqliteVfs) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"SqliteVfs instance not provided in context. The driver must provide a sqliteVfs instance.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tconst kvStore = createActorKvStore(ctx.kv, ctx.metrics);\n\t\t\tconst db = await ctx.sqliteVfs.open(ctx.actorId, kvStore);\n\t\t\tlet closed = false;\n\t\t\tconst mutex = new AsyncMutex();\n\t\t\tconst ensureOpen = () => {\n\t\t\t\tif (closed) {\n\t\t\t\t\tthrow new Error(\"database is closed\");\n\t\t\t\t}\n\t\t\t};\n\n\t\t\treturn {\n\t\t\t\texecute: async <\n\t\t\t\t\tTRow extends Record<string, unknown> = Record<\n\t\t\t\t\t\tstring,\n\t\t\t\t\t\tunknown\n\t\t\t\t\t>,\n\t\t\t\t>(\n\t\t\t\t\tquery: string,\n\t\t\t\t\t...args: unknown[]\n\t\t\t\t): Promise<TRow[]> => {\n\t\t\t\t\treturn await mutex.run(async () => {\n\t\t\t\t\t\tensureOpen();\n\n\t\t\t\t\t\tconst start = performance.now();\n\n\t\t\t\t\t\t// `db.exec` does not support binding `?` placeholders.\n\t\t\t\t\t\t// Use `db.query` for statements that return rows and `db.run` for\n\t\t\t\t\t\t// statements that mutate data when parameters are provided.\n\t\t\t\t\t\t// Keep using `db.exec` for non-parameterized SQL because it\n\t\t\t\t\t\t// supports multi-statement migrations.\n\t\t\t\t\t\tlet result: TRow[];\n\t\t\t\t\t\tif (args.length > 0) {\n\t\t\t\t\t\t\tconst bindings = toSqliteBindings(args);\n\t\t\t\t\t\t\tconst token = query\n\t\t\t\t\t\t\t\t.trimStart()\n\t\t\t\t\t\t\t\t.slice(0, 16)\n\t\t\t\t\t\t\t\t.toUpperCase();\n\t\t\t\t\t\t\tconst returnsRows =\n\t\t\t\t\t\t\t\ttoken.startsWith(\"SELECT\") ||\n\t\t\t\t\t\t\t\ttoken.startsWith(\"PRAGMA\") ||\n\t\t\t\t\t\t\t\ttoken.startsWith(\"WITH\") ||\n\t\t\t\t\t\t\t\t/\\bRETURNING\\b/i.test(query);\n\n\t\t\t\t\t\t\tif (returnsRows) {\n\t\t\t\t\t\t\t\tconst { rows, columns } = await db.query(\n\t\t\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\t\t\tbindings,\n\t\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\t\tresult = rows.map((row: unknown[]) => {\n\t\t\t\t\t\t\t\t\tconst rowObj: Record<string, unknown> = {};\n\t\t\t\t\t\t\t\t\tfor (let i = 0; i < columns.length; i++) {\n\t\t\t\t\t\t\t\t\t\trowObj[columns[i]] = row[i];\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t\treturn rowObj;\n\t\t\t\t\t\t\t\t}) as TRow[];\n\t\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t\tawait db.run(query, bindings);\n\t\t\t\t\t\t\t\tresult = [] as TRow[];\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tconst results: Record<string, unknown>[] = [];\n\t\t\t\t\t\t\tlet columnNames: string[] | null = null;\n\t\t\t\t\t\t\tawait db.exec(\n\t\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\t\t(row: unknown[], columns: string[]) => {\n\t\t\t\t\t\t\t\t\tif (!columnNames) {\n\t\t\t\t\t\t\t\t\t\tcolumnNames = columns;\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t\tconst rowObj: Record<string, unknown> = {};\n\t\t\t\t\t\t\t\t\tfor (let i = 0; i < row.length; i++) {\n\t\t\t\t\t\t\t\t\t\trowObj[columnNames[i]] = row[i];\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t\tresults.push(rowObj);\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tresult = results as TRow[];\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tctx.metrics?.trackSql(query, performance.now() - start);\n\t\t\t\t\t\treturn result;\n\t\t\t\t\t});\n\t\t\t\t},\n\t\t\t\tclose: async () => {\n\t\t\t\t\tconst shouldClose = await mutex.run(async () => {\n\t\t\t\t\t\tif (closed) return false;\n\t\t\t\t\t\tclosed = true;\n\t\t\t\t\t\treturn true;\n\t\t\t\t\t});\n\t\t\t\t\tif (shouldClose) {\n\t\t\t\t\t\tawait db.close();\n\t\t\t\t\t}\n\t\t\t\t},\n\t\t\t} satisfies RawAccess;\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\tif (onMigrate) {\n\t\t\t\tawait onMigrate(client);\n\t\t\t}\n\t\t},\n\t\tonDestroy: async (client) => {\n\t\t\tawait client.close();\n\t\t},\n\t};\n}\n"]}