@rivetkit/engine-runner 2.0.33 → 2.0.34-rc.2

package/dist/mod.d.cts

@@ -2,6 +2,7 @@ import * as protocol from '@rivetkit/engine-runner-protocol';
 import { GatewayId, RequestId } from '@rivetkit/engine-runner-protocol';
 import { Logger } from 'pino';
 import WebSocket from 'ws';
+import { UniversalWebSocket } from '@rivetkit/virtual-websocket';
 
 interface PendingRequest {
     resolve: (response: Response) => void;
@@ -51,49 +52,15 @@ declare function idToStr(id: ArrayBuffer): string;
 declare const HIBERNATABLE_SYMBOL: unique symbol;
 declare class WebSocketTunnelAdapter {
     #private;
-    /** @experimental */
     readonly request: Request;
     get [HIBERNATABLE_SYMBOL](): boolean;
-    constructor(tunnel: Tunnel, actorId: string, requestId: string, serverMessageIndex: number, hibernatable: boolean, isRestoringHibernatable: boolean, 
-    /** @experimental */
-    request: Request, sendCallback: (data: ArrayBuffer | string, isBinary: boolean) => void, closeCallback: (code?: number, reason?: string) => void);
-    get bufferedAmount(): number;
+    constructor(tunnel: Tunnel, actorId: string, requestId: string, serverMessageIndex: number, hibernatable: boolean, isRestoringHibernatable: boolean, request: Request, sendCallback: (data: ArrayBuffer | string, isBinary: boolean) => void, closeCallback: (code?: number, reason?: string) => void);
+    get websocket(): UniversalWebSocket;
     _handleOpen(requestId: ArrayBuffer): void;
     _handleMessage(requestId: ArrayBuffer, data: string | Uint8Array, serverMessageIndex: number, isBinary: boolean): boolean;
     _handleClose(_requestId: ArrayBuffer, code?: number, reason?: string): void;
-    _handleError(error: Error): void;
     _closeWithoutCallback(code?: number, reason?: string): void;
-    get readyState(): number;
-    get binaryType(): string;
-    set binaryType(value: string);
-    get extensions(): string;
-    get protocol(): string;
-    get url(): string;
-    get onopen(): ((this: any, ev: any) => any) | null;
-    set onopen(value: ((this: any, ev: any) => any) | null);
-    get onclose(): ((this: any, ev: any) => any) | null;
-    set onclose(value: ((this: any, ev: any) => any) | null);
-    get onerror(): ((this: any, ev: any) => any) | null;
-    set onerror(value: ((this: any, ev: any) => any) | null);
-    get onmessage(): ((this: any, ev: any) => any) | null;
-    set onmessage(value: ((this: any, ev: any) => any) | null);
-    send(data: string | ArrayBuffer | ArrayBufferView | Blob | Buffer): void;
     close(code?: number, reason?: string): void;
-    addEventListener(type: string, listener: (event: any) => void, options?: boolean | any): void;
-    removeEventListener(type: string, listener: (event: any) => void, options?: boolean | any): void;
-    dispatchEvent(event: any): boolean;
-    static readonly CONNECTING = 0;
-    static readonly OPEN = 1;
-    static readonly CLOSING = 2;
-    static readonly CLOSED = 3;
-    readonly CONNECTING = 0;
-    readonly OPEN = 1;
-    readonly CLOSING = 2;
-    readonly CLOSED = 3;
-    ping(data?: any, mask?: boolean, cb?: (err: Error) => void): void;
-    pong(data?: any, mask?: boolean, cb?: (err: Error) => void): void;
-    /** @experimental */
-    terminate(): void;
 }
 
 interface ActorConfig {
@@ -274,7 +241,6 @@ interface KvListOptions {
 declare class Runner {
     #private;
     get config(): RunnerConfig;
-    __pegboardWebSocket?: WebSocket;
     runnerId?: string;
     get log(): Logger | undefined;
     constructor(config: RunnerConfig);
@@ -299,9 +265,7 @@ declare class Runner {
     setAlarm(actorId: string, alarmTs: number | null, generation?: number): void;
     clearAlarm(actorId: string, generation?: number): void;
     /** Asserts WebSocket exists and is ready. */
-    __webSocketReady(): this is this & {
-        __pegboardWebSocket: NonNullable<Runner["__pegboardWebSocket"]>;
-    };
+    getPegboardWebSocketIfReady(): WebSocket | undefined;
     __sendToServer(message: protocol.ToServer): void;
     sendHibernatableWebSocketMessageAck(gatewayId: ArrayBuffer, requestId: ArrayBuffer, index: number): void;
     /**

package/dist/mod.d.ts

@@ -2,6 +2,7 @@ import * as protocol from '@rivetkit/engine-runner-protocol';
 import { GatewayId, RequestId } from '@rivetkit/engine-runner-protocol';
 import { Logger } from 'pino';
 import WebSocket from 'ws';
+import { UniversalWebSocket } from '@rivetkit/virtual-websocket';
 
 interface PendingRequest {
     resolve: (response: Response) => void;
@@ -51,49 +52,15 @@ declare function idToStr(id: ArrayBuffer): string;
 declare const HIBERNATABLE_SYMBOL: unique symbol;
 declare class WebSocketTunnelAdapter {
     #private;
-    /** @experimental */
     readonly request: Request;
     get [HIBERNATABLE_SYMBOL](): boolean;
-    constructor(tunnel: Tunnel, actorId: string, requestId: string, serverMessageIndex: number, hibernatable: boolean, isRestoringHibernatable: boolean, 
-    /** @experimental */
-    request: Request, sendCallback: (data: ArrayBuffer | string, isBinary: boolean) => void, closeCallback: (code?: number, reason?: string) => void);
-    get bufferedAmount(): number;
+    constructor(tunnel: Tunnel, actorId: string, requestId: string, serverMessageIndex: number, hibernatable: boolean, isRestoringHibernatable: boolean, request: Request, sendCallback: (data: ArrayBuffer | string, isBinary: boolean) => void, closeCallback: (code?: number, reason?: string) => void);
+    get websocket(): UniversalWebSocket;
     _handleOpen(requestId: ArrayBuffer): void;
     _handleMessage(requestId: ArrayBuffer, data: string | Uint8Array, serverMessageIndex: number, isBinary: boolean): boolean;
     _handleClose(_requestId: ArrayBuffer, code?: number, reason?: string): void;
-    _handleError(error: Error): void;
     _closeWithoutCallback(code?: number, reason?: string): void;
-    get readyState(): number;
-    get binaryType(): string;
-    set binaryType(value: string);
-    get extensions(): string;
-    get protocol(): string;
-    get url(): string;
-    get onopen(): ((this: any, ev: any) => any) | null;
-    set onopen(value: ((this: any, ev: any) => any) | null);
-    get onclose(): ((this: any, ev: any) => any) | null;
-    set onclose(value: ((this: any, ev: any) => any) | null);
-    get onerror(): ((this: any, ev: any) => any) | null;
-    set onerror(value: ((this: any, ev: any) => any) | null);
-    get onmessage(): ((this: any, ev: any) => any) | null;
-    set onmessage(value: ((this: any, ev: any) => any) | null);
-    send(data: string | ArrayBuffer | ArrayBufferView | Blob | Buffer): void;
     close(code?: number, reason?: string): void;
-    addEventListener(type: string, listener: (event: any) => void, options?: boolean | any): void;
-    removeEventListener(type: string, listener: (event: any) => void, options?: boolean | any): void;
-    dispatchEvent(event: any): boolean;
-    static readonly CONNECTING = 0;
-    static readonly OPEN = 1;
-    static readonly CLOSING = 2;
-    static readonly CLOSED = 3;
-    readonly CONNECTING = 0;
-    readonly OPEN = 1;
-    readonly CLOSING = 2;
-    readonly CLOSED = 3;
-    ping(data?: any, mask?: boolean, cb?: (err: Error) => void): void;
-    pong(data?: any, mask?: boolean, cb?: (err: Error) => void): void;
-    /** @experimental */
-    terminate(): void;
 }
 
 interface ActorConfig {
@@ -274,7 +241,6 @@ interface KvListOptions {
 declare class Runner {
     #private;
     get config(): RunnerConfig;
-    __pegboardWebSocket?: WebSocket;
     runnerId?: string;
     get log(): Logger | undefined;
     constructor(config: RunnerConfig);
@@ -299,9 +265,7 @@ declare class Runner {
     setAlarm(actorId: string, alarmTs: number | null, generation?: number): void;
     clearAlarm(actorId: string, generation?: number): void;
     /** Asserts WebSocket exists and is ready. */
-    __webSocketReady(): this is this & {
-        __pegboardWebSocket: NonNullable<Runner["__pegboardWebSocket"]>;
-    };
+    getPegboardWebSocketIfReady(): WebSocket | undefined;
     __sendToServer(message: protocol.ToServer): void;
     sendHibernatableWebSocketMessageAck(gatewayId: ArrayBuffer, requestId: ArrayBuffer, index: number): void;
     /**