@rivetkit/rivetkit-wasm 0.0.0-main.17f6330 → 0.0.0-main.3118df2

package/index.d.ts

@@ -8,6 +8,9 @@ export class ActorContext {
 	free(): void;
 	keepAwake(promise: Promise<any>): void;
 	saveState(payload: any): Promise<void>;
+	saveStateAndWorkflowBatch(
+		writes: Array<{ key: Uint8Array; value: Uint8Array }>,
+	): Promise<void>;
 	waitUntil(promise: Promise<any>): void;
 	abortSignal(): any;
 	connectConn(params: Uint8Array, request: any): Promise<ConnHandle>;
@@ -120,6 +123,7 @@ export class Queue {
 	waitForNamesAvailable(names: any, options: any): Promise<void>;
 	send(name: string, body: Uint8Array): Promise<QueueMessage>;
 	maxSize(): number;
+	reset(): Promise<void>;
 }
 
 export class QueueMessage {
@@ -148,6 +152,17 @@ export class SqliteDb {
 	close(): Promise<void>;
 	query(sql: string, params: any): Promise<any>;
 	execute(sql: string, params: any): Promise<any>;
+	executeBatch(statements: Array<any>): Promise<Array<any>>;
+	beginTransaction(timeout_ms?: number | null): Promise<SqliteTransaction>;
+}
+
+export class SqliteTransaction {
+	private constructor();
+	free(): void;
+	exec(sql: string): Promise<any>;
+	execute(sql: string, params: any): Promise<any>;
+	commit(): Promise<void>;
+	rollback(): Promise<void>;
 }
 
 export class WebSocketHandle {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/rivetkit-wasm",
-	"version": "0.0.0-main.17f6330",
+	"version": "0.0.0-main.3118df2",
 	"description": "WebAssembly bindings for RivetKit core on edge JavaScript runtimes",
 	"license": "Apache-2.0",
 	"type": "module",

package/pkg/rivetkit_wasm.d.ts

@@ -1,9 +1,9 @@
 /* tslint:disable */
 /* eslint-disable */
-export function bridgeRivetErrorPrefix(): string;
 export function uint8ArrayFromBytes(bytes: Uint8Array): Uint8Array;
 export function start(): void;
 export function awaitPromise(promise: Promise<any>): Promise<any>;
+export function bridgeRivetErrorPrefix(): string;
 export function roundTripBytes(bytes: Uint8Array): Uint8Array;
 export class ActorContext {
   free(): void;
@@ -26,6 +26,7 @@ export class ActorContext {
   endWebsocketCallback(region_id: number): void;
   beginWebsocketCallback(): number;
   dirtyHibernatableConns(): Array<any>;
+  saveStateAndWorkflowBatch(writes: any): Promise<void>;
   kv(): Kv;
   waitForTrackedShutdownWork(): Promise<boolean>;
   takePendingHibernationChanges(): Array<any>;
@@ -98,6 +99,7 @@ export class Queue {
   inspectMessages(): Promise<Array<any>>;
   waitForNamesAvailable(names: any, options: any): Promise<void>;
   send(name: string, body: Uint8Array): Promise<QueueMessage>;
+  reset(): Promise<void>;
   maxSize(): number;
 }
 export class QueueMessage {
@@ -119,12 +121,22 @@ export class Schedule {
 export class SqliteDb {
   private constructor();
   free(): void;
+  executeBatch(statements: any): Promise<any>;
+  beginTransaction(timeout_ms?: number | null): Promise<SqliteTransaction>;
   run(sql: string, params: any): Promise<any>;
   exec(sql: string): Promise<any>;
   close(): Promise<void>;
   query(sql: string, params: any): Promise<any>;
   execute(sql: string, params: any): Promise<any>;
 }
+export class SqliteTransaction {
+  private constructor();
+  free(): void;
+  exec(sql: string): Promise<any>;
+  commit(): Promise<void>;
+  execute(sql: string, params: any): Promise<any>;
+  rollback(): Promise<void>;
+}
 export class WebSocketHandle {
   private constructor();
   free(): void;
@@ -145,6 +157,7 @@ export interface InitOutput {
   readonly __wbg_kv_free: (a: number, b: number) => void;
   readonly __wbg_queuemessage_free: (a: number, b: number) => void;
   readonly __wbg_sqlitedb_free: (a: number, b: number) => void;
+  readonly __wbg_sqlitetransaction_free: (a: number, b: number) => void;
   readonly __wbg_websockethandle_free: (a: number, b: number) => void;
   readonly actorcontext_abortSignal: (a: number) => [number, number, number];
   readonly actorcontext_actorId: (a: number) => [number, number];
@@ -172,6 +185,7 @@ export interface InitOutput {
   readonly actorcontext_restartRunHandler: (a: number) => void;
   readonly actorcontext_runtimeState: (a: number) => any;
   readonly actorcontext_saveState: (a: number, b: any) => any;
+  readonly actorcontext_saveStateAndWorkflowBatch: (a: number, b: any) => any;
   readonly actorcontext_setAlarm: (a: number, b: number, c: number) => [number, number];
   readonly actorcontext_sleep: (a: number) => [number, number];
   readonly actorcontext_sql: (a: number) => number;
@@ -213,6 +227,7 @@ export interface InitOutput {
   readonly queue_inspectMessages: (a: number) => any;
   readonly queue_maxSize: (a: number) => number;
   readonly queue_nextBatch: (a: number, b: any, c: number) => any;
+  readonly queue_reset: (a: number) => any;
   readonly queue_send: (a: number, b: number, c: number, d: number, e: number) => any;
   readonly queue_tryNextBatch: (a: number, b: any) => [number, number, number];
   readonly queue_waitForNames: (a: number, b: any, c: any, d: number) => any;
@@ -226,11 +241,17 @@ export interface InitOutput {
   readonly roundTripBytes: (a: number, b: number) => [number, number];
   readonly schedule_after: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
   readonly schedule_at: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
+  readonly sqlitedb_beginTransaction: (a: number, b: number, c: number) => any;
   readonly sqlitedb_close: (a: number) => any;
   readonly sqlitedb_exec: (a: number, b: number, c: number) => any;
   readonly sqlitedb_execute: (a: number, b: number, c: number, d: any) => any;
+  readonly sqlitedb_executeBatch: (a: number, b: any) => any;
   readonly sqlitedb_query: (a: number, b: number, c: number, d: any) => any;
   readonly sqlitedb_run: (a: number, b: number, c: number, d: any) => any;
+  readonly sqlitetransaction_commit: (a: number) => any;
+  readonly sqlitetransaction_exec: (a: number, b: number, c: number) => any;
+  readonly sqlitetransaction_execute: (a: number, b: number, c: number, d: any) => any;
+  readonly sqlitetransaction_rollback: (a: number) => any;
   readonly uint8ArrayFromBytes: (a: number, b: number) => any;
   readonly websockethandle_close: (a: number, b: number, c: number, d: number) => any;
   readonly websockethandle_send: (a: number, b: number, c: number, d: number) => [number, number];
@@ -248,9 +269,9 @@ export interface InitOutput {
   readonly __wbindgen_free: (a: number, b: number, c: number) => void;
   readonly __wbindgen_export_6: WebAssembly.Table;
   readonly __externref_table_dealloc: (a: number) => void;
-  readonly closure1598_externref_shim: (a: number, b: number, c: any) => void;
-  readonly closure1950_externref_shim: (a: number, b: number, c: any) => void;
-  readonly closure2181_externref_shim: (a: number, b: number, c: any, d: any) => void;
+  readonly closure1858_externref_shim: (a: number, b: number, c: any) => void;
+  readonly closure2231_externref_shim: (a: number, b: number, c: any) => void;
+  readonly closure2462_externref_shim: (a: number, b: number, c: any, d: any) => void;
   readonly __wbindgen_start: () => void;
 }
 

package/pkg/rivetkit_wasm.js

@@ -221,22 +221,6 @@ function _assertClass(instance, klass) {
         throw new Error(`expected instance of ${klass.name}`);
     }
 }
-/**
- * @returns {string}
- */
-export function bridgeRivetErrorPrefix() {
-    let deferred1_0;
-    let deferred1_1;
-    try {
-        const ret = wasm.bridgeRivetErrorPrefix();
-        deferred1_0 = ret[0];
-        deferred1_1 = ret[1];
-        return getStringFromWasm0(ret[0], ret[1]);
-    } finally {
-        wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
-    }
-}
-
 /**
  * @param {Uint8Array} bytes
  * @returns {Uint8Array}
@@ -261,6 +245,22 @@ export function awaitPromise(promise) {
     return ret;
 }
 
+/**
+ * @returns {string}
+ */
+export function bridgeRivetErrorPrefix() {
+    let deferred1_0;
+    let deferred1_1;
+    try {
+        const ret = wasm.bridgeRivetErrorPrefix();
+        deferred1_0 = ret[0];
+        deferred1_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+    }
+}
+
 /**
  * @param {Uint8Array} bytes
  * @returns {Uint8Array}
@@ -275,15 +275,15 @@ export function roundTripBytes(bytes) {
 }
 
 function __wbg_adapter_46(arg0, arg1, arg2) {
-    wasm.closure1598_externref_shim(arg0, arg1, arg2);
+    wasm.closure1858_externref_shim(arg0, arg1, arg2);
 }
 
 function __wbg_adapter_55(arg0, arg1, arg2) {
-    wasm.closure1950_externref_shim(arg0, arg1, arg2);
+    wasm.closure2231_externref_shim(arg0, arg1, arg2);
 }
 
-function __wbg_adapter_268(arg0, arg1, arg2, arg3) {
-    wasm.closure2181_externref_shim(arg0, arg1, arg2, arg3);
+function __wbg_adapter_248(arg0, arg1, arg2, arg3) {
+    wasm.closure2462_externref_shim(arg0, arg1, arg2, arg3);
 }
 
 const __wbindgen_enum_BinaryType = ["blob", "arraybuffer"];
@@ -443,6 +443,14 @@ export class ActorContext {
         const ret = wasm.actorcontext_dirtyHibernatableConns(this.__wbg_ptr);
         return ret;
     }
+    /**
+     * @param {any} writes
+     * @returns {Promise<void>}
+     */
+    saveStateAndWorkflowBatch(writes) {
+        const ret = wasm.actorcontext_saveStateAndWorkflowBatch(this.__wbg_ptr, writes);
+        return ret;
+    }
     /**
      * @returns {Kv}
      */
@@ -1076,6 +1084,13 @@ export class Queue {
         const ret = wasm.queue_send(this.__wbg_ptr, ptr0, len0, ptr1, len1);
         return ret;
     }
+    /**
+     * @returns {Promise<void>}
+     */
+    reset() {
+        const ret = wasm.queue_reset(this.__wbg_ptr);
+        return ret;
+    }
     /**
      * @returns {number}
      */
@@ -1241,6 +1256,22 @@ export class SqliteDb {
         const ptr = this.__destroy_into_raw();
         wasm.__wbg_sqlitedb_free(ptr, 0);
     }
+    /**
+     * @param {any} statements
+     * @returns {Promise<any>}
+     */
+    executeBatch(statements) {
+        const ret = wasm.sqlitedb_executeBatch(this.__wbg_ptr, statements);
+        return ret;
+    }
+    /**
+     * @param {number | null} [timeout_ms]
+     * @returns {Promise<SqliteTransaction>}
+     */
+    beginTransaction(timeout_ms) {
+        const ret = wasm.sqlitedb_beginTransaction(this.__wbg_ptr, !isLikeNone(timeout_ms), isLikeNone(timeout_ms) ? 0 : timeout_ms);
+        return ret;
+    }
     /**
      * @param {string} sql
      * @param {any} params
@@ -1293,6 +1324,68 @@ export class SqliteDb {
     }
 }
 
+const SqliteTransactionFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_sqlitetransaction_free(ptr >>> 0, 1));
+
+export class SqliteTransaction {
+
+    static __wrap(ptr) {
+        ptr = ptr >>> 0;
+        const obj = Object.create(SqliteTransaction.prototype);
+        obj.__wbg_ptr = ptr;
+        SqliteTransactionFinalization.register(obj, obj.__wbg_ptr, obj);
+        return obj;
+    }
+
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        SqliteTransactionFinalization.unregister(this);
+        return ptr;
+    }
+
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_sqlitetransaction_free(ptr, 0);
+    }
+    /**
+     * @param {string} sql
+     * @returns {Promise<any>}
+     */
+    exec(sql) {
+        const ptr0 = passStringToWasm0(sql, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.sqlitetransaction_exec(this.__wbg_ptr, ptr0, len0);
+        return ret;
+    }
+    /**
+     * @returns {Promise<void>}
+     */
+    commit() {
+        const ret = wasm.sqlitetransaction_commit(this.__wbg_ptr);
+        return ret;
+    }
+    /**
+     * @param {string} sql
+     * @param {any} params
+     * @returns {Promise<any>}
+     */
+    execute(sql, params) {
+        const ptr0 = passStringToWasm0(sql, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.sqlitetransaction_execute(this.__wbg_ptr, ptr0, len0, params);
+        return ret;
+    }
+    /**
+     * @returns {Promise<void>}
+     */
+    rollback() {
+        const ret = wasm.sqlitetransaction_rollback(this.__wbg_ptr);
+        return ret;
+    }
+}
+
 const WebSocketHandleFinalization = (typeof FinalizationRegistry === 'undefined')
     ? { register: () => {}, unregister: () => {} }
     : new FinalizationRegistry(ptr => wasm.__wbg_websockethandle_free(ptr >>> 0, 1));
@@ -1569,7 +1662,7 @@ function __wbg_get_imports() {
                 const a = state0.a;
                 state0.a = 0;
                 try {
-                    return __wbg_adapter_268(a, state0.b, arg0, arg1);
+                    return __wbg_adapter_248(a, state0.b, arg0, arg1);
                 } finally {
                     state0.a = a;
                 }
@@ -1712,6 +1805,10 @@ function __wbg_get_imports() {
     imports.wbg.__wbg_setonopen_2da654e1f39745d5 = function(arg0, arg1) {
         arg0.onopen = arg1;
     };
+    imports.wbg.__wbg_sqlitetransaction_new = function(arg0) {
+        const ret = SqliteTransaction.__wrap(arg0);
+        return ret;
+    };
     imports.wbg.__wbg_stack_0ed75d68575b0f3c = function(arg0, arg1) {
         const ret = arg1.stack;
         const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
@@ -1781,24 +1878,24 @@ function __wbg_get_imports() {
         const ret = false;
         return ret;
     };
-    imports.wbg.__wbindgen_closure_wrapper3733 = function(arg0, arg1, arg2) {
-        const ret = makeMutClosure(arg0, arg1, 1599, __wbg_adapter_46);
+    imports.wbg.__wbindgen_closure_wrapper4188 = function(arg0, arg1, arg2) {
+        const ret = makeMutClosure(arg0, arg1, 1859, __wbg_adapter_46);
         return ret;
     };
-    imports.wbg.__wbindgen_closure_wrapper3735 = function(arg0, arg1, arg2) {
-        const ret = makeMutClosure(arg0, arg1, 1599, __wbg_adapter_46);
+    imports.wbg.__wbindgen_closure_wrapper4190 = function(arg0, arg1, arg2) {
+        const ret = makeMutClosure(arg0, arg1, 1859, __wbg_adapter_46);
         return ret;
     };
-    imports.wbg.__wbindgen_closure_wrapper3737 = function(arg0, arg1, arg2) {
-        const ret = makeMutClosure(arg0, arg1, 1599, __wbg_adapter_46);
+    imports.wbg.__wbindgen_closure_wrapper4192 = function(arg0, arg1, arg2) {
+        const ret = makeMutClosure(arg0, arg1, 1859, __wbg_adapter_46);
         return ret;
     };
-    imports.wbg.__wbindgen_closure_wrapper3739 = function(arg0, arg1, arg2) {
-        const ret = makeMutClosure(arg0, arg1, 1599, __wbg_adapter_46);
+    imports.wbg.__wbindgen_closure_wrapper4194 = function(arg0, arg1, arg2) {
+        const ret = makeMutClosure(arg0, arg1, 1859, __wbg_adapter_46);
         return ret;
     };
-    imports.wbg.__wbindgen_closure_wrapper4580 = function(arg0, arg1, arg2) {
-        const ret = makeMutClosure(arg0, arg1, 1951, __wbg_adapter_55);
+    imports.wbg.__wbindgen_closure_wrapper5060 = function(arg0, arg1, arg2) {
+        const ret = makeMutClosure(arg0, arg1, 2232, __wbg_adapter_55);
         return ret;
     };
     imports.wbg.__wbindgen_debug_string = function(arg0, arg1) {

package/pkg/rivetkit_wasm_bg.wasm.d.ts

@@ -9,6 +9,7 @@ export const __wbg_coreregistry_free: (a: number, b: number) => void;
 export const __wbg_kv_free: (a: number, b: number) => void;
 export const __wbg_queuemessage_free: (a: number, b: number) => void;
 export const __wbg_sqlitedb_free: (a: number, b: number) => void;
+export const __wbg_sqlitetransaction_free: (a: number, b: number) => void;
 export const __wbg_websockethandle_free: (a: number, b: number) => void;
 export const actorcontext_abortSignal: (a: number) => [number, number, number];
 export const actorcontext_actorId: (a: number) => [number, number];
@@ -36,6 +37,7 @@ export const actorcontext_requestSaveAndWait: (a: number, b: any) => any;
 export const actorcontext_restartRunHandler: (a: number) => void;
 export const actorcontext_runtimeState: (a: number) => any;
 export const actorcontext_saveState: (a: number, b: any) => any;
+export const actorcontext_saveStateAndWorkflowBatch: (a: number, b: any) => any;
 export const actorcontext_setAlarm: (a: number, b: number, c: number) => [number, number];
 export const actorcontext_sleep: (a: number) => [number, number];
 export const actorcontext_sql: (a: number) => number;
@@ -77,6 +79,7 @@ export const queue_enqueueAndWait: (a: number, b: number, c: number, d: number,
 export const queue_inspectMessages: (a: number) => any;
 export const queue_maxSize: (a: number) => number;
 export const queue_nextBatch: (a: number, b: any, c: number) => any;
+export const queue_reset: (a: number) => any;
 export const queue_send: (a: number, b: number, c: number, d: number, e: number) => any;
 export const queue_tryNextBatch: (a: number, b: any) => [number, number, number];
 export const queue_waitForNames: (a: number, b: any, c: any, d: number) => any;
@@ -90,11 +93,17 @@ export const queuemessage_name: (a: number) => [number, number];
 export const roundTripBytes: (a: number, b: number) => [number, number];
 export const schedule_after: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
 export const schedule_at: (a: number, b: number, c: number, d: number, e: number, f: number) => void;
+export const sqlitedb_beginTransaction: (a: number, b: number, c: number) => any;
 export const sqlitedb_close: (a: number) => any;
 export const sqlitedb_exec: (a: number, b: number, c: number) => any;
 export const sqlitedb_execute: (a: number, b: number, c: number, d: any) => any;
+export const sqlitedb_executeBatch: (a: number, b: any) => any;
 export const sqlitedb_query: (a: number, b: number, c: number, d: any) => any;
 export const sqlitedb_run: (a: number, b: number, c: number, d: any) => any;
+export const sqlitetransaction_commit: (a: number) => any;
+export const sqlitetransaction_exec: (a: number, b: number, c: number) => any;
+export const sqlitetransaction_execute: (a: number, b: number, c: number, d: any) => any;
+export const sqlitetransaction_rollback: (a: number) => any;
 export const uint8ArrayFromBytes: (a: number, b: number) => any;
 export const websockethandle_close: (a: number, b: number, c: number, d: number) => any;
 export const websockethandle_send: (a: number, b: number, c: number, d: number) => [number, number];
@@ -112,7 +121,7 @@ export const __wbindgen_export_4: WebAssembly.Table;
 export const __wbindgen_free: (a: number, b: number, c: number) => void;
 export const __wbindgen_export_6: WebAssembly.Table;
 export const __externref_table_dealloc: (a: number) => void;
-export const closure1598_externref_shim: (a: number, b: number, c: any) => void;
-export const closure1950_externref_shim: (a: number, b: number, c: any) => void;
-export const closure2181_externref_shim: (a: number, b: number, c: any, d: any) => void;
+export const closure1858_externref_shim: (a: number, b: number, c: any) => void;
+export const closure2231_externref_shim: (a: number, b: number, c: any) => void;
+export const closure2462_externref_shim: (a: number, b: number, c: any, d: any) => void;
 export const __wbindgen_start: () => void;