@rivetkit/rivetkit-napi 0.0.0-main.337772d → 0.0.0-main.4994268

package/index.d.ts

@@ -23,10 +23,16 @@ export interface StateDeltaPayload {
   connHibernation: Array<StateDeltaConnHibernationEntry>
   connHibernationRemoved: Array<string>
 }
+export interface WorkflowKvWritePayload {
+  key: Buffer
+  value: Buffer
+}
 export interface JsRequestSaveOpts {
   immediate?: boolean
   maxWaitMs?: number
 }
+export declare function decodeInspectorRequest(bytes: Buffer, advertisedVersion: number): Buffer
+export declare function encodeInspectorResponse(bytes: Buffer, targetVersion: number): Buffer
 export interface JsInspectorSnapshot {
   stateRevision: number
   connectionsRevision: number
@@ -35,10 +41,8 @@ export interface JsInspectorSnapshot {
   queueSize: number
   connectedClients: number
 }
-export interface JsHttpResponse {
-  status?: number
-  headers?: Record<string, string>
-  body?: Buffer
+export interface JsActorRuntimeSocketEndpointInfo {
+  path: string
 }
 export interface JsQueueSendResult {
   status: string
@@ -47,11 +51,32 @@ export interface JsQueueSendResult {
 export interface JsActionDefinition {
   name: string
 }
+/**
+ * One entry in the actor's `inspector.tabs[]` declaration. Either a
+ * custom-tab descriptor (id + label + source dir) or a built-in modifier
+ * (id + hidden=true). Validation already happened on the TS side; the
+ * runtime just splits the discriminator.
+ */
+export interface JsInspectorTabEntry {
+  id: string
+  /** Required for custom entries; omitted for HideBuiltin. */
+  label?: string
+  /** Required for custom entries — absolute path to the source directory. */
+  source?: string
+  /**
+   * Optional icon id for custom entries. Dashboard maps strings to
+   * glyphs; unknown ids fall back to a generic icon.
+   */
+  icon?: string
+  /** Set to true for HideBuiltin entries. */
+  hidden?: boolean
+}
 export interface JsActorConfig {
   name?: string
   icon?: string
   hasDatabase?: boolean
   remoteSqlite?: boolean
+  enableActorRuntimeSocket?: boolean
   hasState?: boolean
   canHibernateWebsocket?: boolean
   stateSaveIntervalMs?: number
@@ -72,12 +97,12 @@ export interface JsActorConfig {
   connectionLivenessTimeoutMs?: number
   connectionLivenessIntervalMs?: number
   maxQueueSize?: number
+  maxSchedules?: number
   maxQueueMessageSize?: number
   maxIncomingMessageSize?: number
   maxOutgoingMessageSize?: number
-  preloadMaxWorkflowBytes?: number
-  preloadMaxConnectionsBytes?: number
   actions?: Array<JsActionDefinition>
+  inspectorTabs?: Array<JsInspectorTabEntry>
 }
 export interface JsBindParam {
   kind: string
@@ -99,6 +124,10 @@ export interface NativeExecuteResult {
   changes: number
   lastInsertRowId?: number
 }
+export interface JsSqliteBatchStatement {
+  sql: string
+  params?: Array<JsBindParam>
+}
 export interface JsSqliteVfsMetrics {
   requestBuildNs: number
   serializeNs: number
@@ -112,6 +141,12 @@ export interface JsSqliteVfsMetrics {
   writeBufferDirtyPages: number
   dbSizePages: number
 }
+export interface JsHttpResponse {
+  status?: number
+  headers?: Record<string, string>
+  body?: Buffer
+  stream?: boolean
+}
 export interface JsQueueNextOptions {
   names?: Array<string>
   timeoutMs?: number
@@ -155,6 +190,8 @@ export interface JsServeConfig {
   namespace: string
   poolName: string
   engineBinaryPath?: string
+  engineHost?: string
+  enginePort?: number
   handleInspectorHttpInRuntime?: boolean
   serverlessBasePath?: string
   serverlessPackageVersion: string
@@ -164,6 +201,16 @@ export interface JsServeConfig {
   serverlessValidateEndpoint: boolean
   serverlessMaxStartPayloadBytes: number
 }
+export interface JsListenerConfig {
+  /** Host to bind. Defaults to `0.0.0.0` when not provided. */
+  host?: string
+  port: number
+  /**
+   * Optional static file root mounted as a fallback below the framework
+   * routes.
+   */
+  publicDir?: string
+}
 export interface JsServerlessRequest {
   method: string
   url: string
@@ -174,15 +221,54 @@ export interface JsServerlessResponseHead {
   status: number
   headers: Record<string, string>
 }
-export interface JsRegistryDiagnostics {
-  mode: string
-  envoyActiveActorCount?: number
+export interface JsApplicationResponse {
+  status: number
+  headers: Record<string, string>
+  body?: Buffer
+  stream?: boolean
+}
+export interface JsRegistryRouteResponse {
+  status: number
+  headers: Record<string, string>
+  body: Buffer
 }
 export interface JsServerlessStreamError {
   group: string
   code: string
   message: string
 }
+export interface JsScheduledEventInfo {
+  id: string
+  action: string
+  args: Buffer
+  runAt: number
+}
+export interface JsCronJobInfo {
+  name: string
+  kind: string
+  action: string
+  args: Buffer
+  nextRunAt: number
+  lastRunAt?: number
+  expression?: string
+  timezone?: string
+  intervalMs?: number
+  maxHistory: number
+}
+export interface JsScheduleErrorInfo {
+  group: string
+  code: string
+  message: string
+  metadata?: any
+}
+export interface JsCronFire {
+  action: string
+  scheduledAt: number
+  firedAt: number
+  finishedAt?: number
+  result: string
+  error?: JsScheduleErrorInfo
+}
 /** Options for KV list operations. */
 export interface JsKvListOptions {
   reverse?: boolean
@@ -195,12 +281,12 @@ export interface JsKvEntry {
 }
 /** N-API wrapper around `rivetkit-core::ActorContext`. */
 export declare class ActorContext {
-  constructor(actorId: string, name: string, region: string)
   state(): Buffer
   beginOnStateChange(): void
   endOnStateChange(): void
   kv(): Kv
   sql(): JsNativeDatabase
+  provisionActorRuntimeSocket(): Promise<JsActorRuntimeSocketEndpointInfo>
   schedule(): Schedule
   queue(): Queue
   setAlarm(timestampMs?: number | undefined | null): void
@@ -215,6 +301,7 @@ export declare class ActorContext {
   takePendingHibernationChanges(): Array<string>
   dirtyHibernatableConns(): Array<ConnHandle>
   saveState(payload: StateDeltaPayload): Promise<void>
+  saveStateAndWorkflowBatch(writes: Array<WorkflowKvWritePayload>): Promise<void>
   actorId(): string
   name(): string
   key(): Array<JsActorKeySegment>
@@ -240,6 +327,8 @@ export declare class ActorContext {
   disconnectConns(predicate: (...args: any[]) => any): Promise<void>
   broadcast(name: string, args: Buffer): void
   waitUntil(promise: Promise<any>): void
+  waitForTrackedShutdownWork(): Promise<boolean>
+  waitForTrackedShutdownWorkUnbounded(): Promise<void>
   registerTask(promise: Promise<any>): void
   runtimeState(): object
   clearRuntimeState(): void
@@ -268,8 +357,28 @@ export declare class JsNativeDatabase {
   run(sql: string, params?: Array<JsBindParam> | undefined | null): Promise<ExecuteResult>
   query(sql: string, params?: Array<JsBindParam> | undefined | null): Promise<QueryResult>
   execute(sql: string, params?: Array<JsBindParam> | undefined | null): Promise<NativeExecuteResult>
+  executeBatch(statements: Array<JsSqliteBatchStatement>): Promise<Array<NativeExecuteResult>>
   exec(sql: string): Promise<QueryResult>
   close(): Promise<void>
+  beginTransaction(timeoutMs?: number | undefined | null): Promise<JsSqliteTransaction>
+}
+export declare class JsSqliteTransaction {
+  execute(sql: string, params?: Array<JsBindParam> | undefined | null): Promise<NativeExecuteResult>
+  exec(sql: string): Promise<QueryResult>
+  commit(): Promise<void>
+  rollback(): Promise<void>
+}
+/** JavaScript-facing writer for a streaming actor HTTP response. */
+export declare class HttpResponseBodyStream {
+  cancelled(): Promise<void>
+  write(chunk: Buffer): Promise<void>
+  end(): Promise<void>
+  error(message: string): Promise<void>
+}
+/** JavaScript-facing reader for a streaming actor HTTP request. */
+export declare class HttpRequestBodyStream {
+  read(): Promise<Buffer | null>
+  cancel(): Promise<void>
 }
 export declare class Kv {
   get(key: Buffer): Promise<Buffer | null>
@@ -287,11 +396,12 @@ export declare class Queue {
   next(options?: JsQueueNextOptions | undefined | null, signal?: CancellationToken | undefined | null): Promise<QueueMessage | null>
   nextBatch(options?: JsQueueNextBatchOptions | undefined | null, signal?: CancellationToken | undefined | null): Promise<Array<QueueMessage>>
   waitForNames(names: Array<string>, options?: JsQueueWaitOptions | undefined | null, signal?: CancellationToken | undefined | null): Promise<QueueMessage>
-  waitForNamesAvailable(names: Array<string>, options?: JsQueueWaitOptions | undefined | null): Promise<void>
+  waitForNamesAvailable(names: Array<string>, options?: JsQueueWaitOptions | undefined | null, signal?: CancellationToken | undefined | null): Promise<void>
   enqueueAndWait(name: string, body: Buffer, options?: JsQueueEnqueueAndWaitOptions | undefined | null, signal?: CancellationToken | undefined | null): Promise<Buffer | null>
   tryNext(options?: JsQueueTryNextOptions | undefined | null): QueueMessage | null
   tryNextBatch(options?: JsQueueTryNextBatchOptions | undefined | null): Array<QueueMessage>
   maxSize(): number
+  reset(): Promise<void>
   inspectMessages(): Promise<Array<JsQueueInspectMessage>>
 }
 export declare class QueueMessage {
@@ -306,6 +416,13 @@ export declare class CoreRegistry {
   constructor()
   register(name: string, factory: NapiActorFactory): void
   serve(config: JsServeConfig): Promise<void>
+  /**
+   * Wait until the serverful envoy has completed its Engine registration.
+   *
+   * Readiness is the envoy client's `ToEnvoyInit` signal, not HTTP health or
+   * merely constructing the envoy handle.
+   */
+  waitReady(): Promise<void>
   /**
    * Trip the shutdown token and tear down any live serverless runtime.
    *
@@ -314,12 +431,26 @@ export declare class CoreRegistry {
    * separately to avoid re-entrancy.
    */
   shutdown(): Promise<void>
-  diagnostics(): Promise<JsRegistryDiagnostics>
+  actorStopThresholdMs(): Promise<number | null>
+  health(): Promise<JsRegistryRouteResponse>
+  metadata(): JsRegistryRouteResponse
+  metrics(): JsRegistryRouteResponse
+  serveApplicationListener(listener: JsListenerConfig, application: (...args: any[]) => any, config: JsServeConfig): Promise<void>
+  serveListener(listener: JsListenerConfig, config: JsServeConfig, application?: (...args: any[]) => any | undefined | null): Promise<void>
   handleServerlessRequest(req: JsServerlessRequest, onStreamEvent: (...args: any[]) => any, cancelToken: CancellationToken, config: JsServeConfig): Promise<JsServerlessResponseHead>
 }
 export declare class Schedule {
-  after(durationMs: number, actionName: string, args: Buffer): void
-  at(timestampMs: number, actionName: string, args: Buffer): void
+  after(durationMs: number, actionName: string, args: Buffer): Promise<string>
+  at(timestampMs: number, actionName: string, args: Buffer): Promise<string>
+  cancel(id: string): Promise<boolean>
+  get(id: string): Promise<JsScheduledEventInfo | null>
+  list(): Promise<Array<JsScheduledEventInfo>>
+  cronSet(name: string, expression: string, timezone: string | undefined | null, actionName: string, args: Buffer, maxHistory?: number | undefined | null): Promise<void>
+  cronEvery(name: string, intervalMs: number, actionName: string, args: Buffer, maxHistory?: number | undefined | null): Promise<void>
+  cronGet(name: string): Promise<JsCronJobInfo | null>
+  cronList(): Promise<Array<JsCronJobInfo>>
+  cronDelete(name: string): Promise<boolean>
+  cronHistory(name: string, limit?: number | undefined | null): Promise<Array<JsCronFire>>
 }
 export declare class WebSocket {
   send(data: Buffer, binary: boolean): void

package/index.js

@@ -310,13 +310,18 @@ if (!nativeBinding) {
   throw new Error(`Failed to load native binding`)
 }
 
-const { ActorContext, NapiActorFactory, CancellationToken, ConnHandle, JsNativeDatabase, Kv, Queue, QueueMessage, CoreRegistry, Schedule, WebSocket } = nativeBinding
+const { ActorContext, decodeInspectorRequest, encodeInspectorResponse, NapiActorFactory, CancellationToken, ConnHandle, JsNativeDatabase, JsSqliteTransaction, HttpResponseBodyStream, HttpRequestBodyStream, Kv, Queue, QueueMessage, CoreRegistry, Schedule, WebSocket } = nativeBinding
 
 module.exports.ActorContext = ActorContext
+module.exports.decodeInspectorRequest = decodeInspectorRequest
+module.exports.encodeInspectorResponse = encodeInspectorResponse
 module.exports.NapiActorFactory = NapiActorFactory
 module.exports.CancellationToken = CancellationToken
 module.exports.ConnHandle = ConnHandle
 module.exports.JsNativeDatabase = JsNativeDatabase
+module.exports.JsSqliteTransaction = JsSqliteTransaction
+module.exports.HttpResponseBodyStream = HttpResponseBodyStream
+module.exports.HttpRequestBodyStream = HttpRequestBodyStream
 module.exports.Kv = Kv
 module.exports.Queue = Queue
 module.exports.QueueMessage = QueueMessage

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/rivetkit-napi",
-	"version": "0.0.0-main.337772d",
+	"version": "0.0.0-main.4994268",
 	"description": "Native N-API addon for RivetKit providing envoy client and SQLite access",
 	"license": "Apache-2.0",
 	"main": "index.js",
@@ -39,19 +39,19 @@
 		"build": "node scripts/build.mjs",
 		"build:release": "node scripts/build.mjs --release",
 		"build:force": "node scripts/build.mjs --force",
-		"build:force:release": "node scripts/build.mjs --force --release"
+		"build:force:release": "node scripts/build.mjs --force --release",
+		"build:embed": "node scripts/build.mjs"
 	},
 	"dependencies": {
 		"@napi-rs/cli": "^2.18.4",
-		"@rivetkit/engine-envoy-protocol": "0.0.0-main.337772d"
+		"@rivetkit/engine-envoy-protocol": "0.0.0-main.4994268"
 	},
 	"optionalDependencies": {
-		"@rivetkit/rivetkit-napi-darwin-arm64": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-darwin-x64": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-linux-arm64-gnu": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-linux-arm64-musl": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-linux-x64-gnu": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-linux-x64-musl": "0.0.0-main.337772d",
-		"@rivetkit/rivetkit-napi-win32-x64-msvc": "0.0.0-main.337772d"
+		"@rivetkit/rivetkit-napi-darwin-arm64": "0.0.0-main.4994268",
+		"@rivetkit/rivetkit-napi-darwin-x64": "0.0.0-main.4994268",
+		"@rivetkit/rivetkit-napi-linux-arm64-gnu": "0.0.0-main.4994268",
+		"@rivetkit/rivetkit-napi-linux-arm64-musl": "0.0.0-main.4994268",
+		"@rivetkit/rivetkit-napi-linux-x64-gnu": "0.0.0-main.4994268",
+		"@rivetkit/rivetkit-napi-linux-x64-musl": "0.0.0-main.4994268"
 	}
 }

package/scripts/build.mjs

@@ -16,6 +16,14 @@ if (process.env.SKIP_NAPI_BUILD === "1") {
 	process.exit(0);
 }
 
+// The per-actor inspector UI (frontend/dist/inspector-ui, embedded into
+// rivetkit-core by its build.rs) must be built before this napi build runs.
+// It is NOT built here: rivetkit-core's embed needs rivetkit/inspector-tab,
+// which is downstream of this package in the build graph, so building it from
+// the napi build would invert the dependency order. CI builds it via
+// `turbo build:inspector-ui` in docker/build/*.Dockerfile before `napi build`;
+// for local builds run `pnpm -F @rivetkit/engine-frontend build:inspector-ui`
+// (or `turbo build:inspector-ui`) first.
 const cmd = ["build", "--platform", ...extraFlags];
 console.log(`[rivetkit-napi/build] running: napi ${cmd.join(" ")}`);
 execFileSync("napi", cmd, { stdio: "inherit" });