ai-functions 0.4.0 → 2.0.1

package/dist/rpc/local.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Local RPC target for in-memory implementations
+ *
+ * Allows using the same RPC semantics (including promise pipelining)
+ * for local in-memory implementations, useful for:
+ * - Testing
+ * - Client-side caching
+ * - Offline support
+ * - Development without a server
+ */
+import { RpcTarget } from 'capnweb';
+/**
+ * Create a local RPC target that can be used with the same API
+ * as remote targets, but executes locally.
+ *
+ * @example
+ * ```ts
+ * class MyLocalAPI extends RpcTarget {
+ *   async getData(id: string) {
+ *     return this.localStore.get(id)
+ *   }
+ * }
+ *
+ * const api = createLocalTarget(new MyLocalAPI())
+ * const data = await api.getData('123') // Executes locally
+ * ```
+ */
+export declare function createLocalTarget<T extends RpcTarget>(target: T): T;
+/**
+ * Base class for local implementations that mirror remote APIs
+ */
+export declare abstract class LocalTarget extends RpcTarget {
+    /**
+     * Override to provide initialization logic
+     */
+    initialize(): Promise<void>;
+    /**
+     * Override to provide cleanup logic
+     */
+    dispose(): Promise<void>;
+}
+//# sourceMappingURL=local.d.ts.map

package/dist/rpc/local.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"local.d.ts","sourceRoot":"","sources":["../../src/rpc/local.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAA;AAEnC;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,SAAS,SAAS,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,CAInE;AAED;;GAEG;AACH,8BAAsB,WAAY,SAAQ,SAAS;IACjD;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjC;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B"}

package/dist/rpc/local.js

@@ -0,0 +1,50 @@
+/**
+ * Local RPC target for in-memory implementations
+ *
+ * Allows using the same RPC semantics (including promise pipelining)
+ * for local in-memory implementations, useful for:
+ * - Testing
+ * - Client-side caching
+ * - Offline support
+ * - Development without a server
+ */
+import { RpcTarget } from 'capnweb';
+/**
+ * Create a local RPC target that can be used with the same API
+ * as remote targets, but executes locally.
+ *
+ * @example
+ * ```ts
+ * class MyLocalAPI extends RpcTarget {
+ *   async getData(id: string) {
+ *     return this.localStore.get(id)
+ *   }
+ * }
+ *
+ * const api = createLocalTarget(new MyLocalAPI())
+ * const data = await api.getData('123') // Executes locally
+ * ```
+ */
+export function createLocalTarget(target) {
+    // For local targets, we can return the target directly
+    // since capnweb's RpcTarget already handles method calls properly
+    return target;
+}
+/**
+ * Base class for local implementations that mirror remote APIs
+ */
+export class LocalTarget extends RpcTarget {
+    /**
+     * Override to provide initialization logic
+     */
+    async initialize() {
+        // Default: no initialization needed
+    }
+    /**
+     * Override to provide cleanup logic
+     */
+    async dispose() {
+        // Default: no cleanup needed
+    }
+}
+//# sourceMappingURL=local.js.map

package/dist/rpc/local.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local.js","sourceRoot":"","sources":["../../src/rpc/local.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAA;AAEnC;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,iBAAiB,CAAsB,MAAS;IAC9D,uDAAuD;IACvD,kEAAkE;IAClE,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,MAAM,OAAgB,WAAY,SAAQ,SAAS;IACjD;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,oCAAoC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,6BAA6B;IAC/B,CAAC;CACF"}

package/dist/rpc/server.d.ts

@@ -0,0 +1,165 @@
+/**
+ * Hono RPC Server Handlers
+ *
+ * Provides RPC endpoints for:
+ * - HTTP batch requests (POST /rpc)
+ * - WebSocket persistent connections (GET /rpc/ws)
+ * - postMessage bridge for iframe communication (GET /rpc/bridge)
+ *
+ * @packageDocumentation
+ */
+import type { Context, Hono, MiddlewareHandler } from 'hono';
+import type { WSContext } from 'hono/ws';
+/**
+ * RPC method handler function
+ */
+export type RPCMethod<TContext = unknown> = (ctx: TContext, ...args: unknown[]) => unknown | Promise<unknown>;
+/**
+ * Collection of RPC methods
+ */
+export type RPCMethods<TContext = unknown> = Record<string, RPCMethod<TContext>>;
+/**
+ * Options for creating an RPC handler
+ */
+export interface RPCHandlerOptions<TContext = unknown> {
+    /** Available RPC methods */
+    methods: RPCMethods<TContext>;
+    /** Create context from request (for auth, etc) */
+    createContext?: (c: Context) => TContext | Promise<TContext>;
+    /** Called before each method invocation */
+    onCall?: (method: string, params: unknown[], ctx: TContext) => void | Promise<void>;
+    /** Called after each method invocation */
+    onResult?: (method: string, result: unknown, ctx: TContext) => void | Promise<void>;
+    /** Called on errors */
+    onError?: (method: string, error: Error, ctx: TContext) => void | Promise<void>;
+    /** Allowed origins for CORS (default: '*') */
+    allowedOrigins?: string[] | '*';
+    /** Allowed origins for postMessage bridge */
+    bridgeOrigins?: string[];
+}
+/**
+ * Create HTTP batch RPC handler
+ */
+export declare function createHTTPHandler<TContext>(options: RPCHandlerOptions<TContext>): MiddlewareHandler;
+/**
+ * Create WebSocket RPC handler for Hono
+ *
+ * Usage with hono/ws:
+ * ```ts
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers' // or hono/bun, etc
+ *
+ * app.get('/rpc/ws', upgradeWebSocket(createWSHandler({ methods })))
+ * ```
+ */
+export declare function createWSHandler<TContext>(options: RPCHandlerOptions<TContext>): (c: Context) => {
+    onOpen(_evt: Event, ws: WSContext): Promise<void>;
+    onMessage(evt: MessageEvent, ws: WSContext): Promise<void>;
+    onClose(_evt: CloseEvent, _ws: WSContext): void;
+    onError(evt: Event, _ws: WSContext): void;
+};
+/**
+ * Generate the postMessage bridge HTML/JS
+ *
+ * This creates a lightweight page that:
+ * 1. Listens for postMessage from parent window
+ * 2. Makes authenticated fetch/WS calls to the RPC endpoint
+ * 3. Posts results back to parent
+ */
+export declare function createBridgeHandler<TContext>(options: RPCHandlerOptions<TContext> & {
+    /** RPC endpoint URL (default: '/rpc') */
+    rpcUrl?: string;
+    /** WebSocket URL (default: derived from rpcUrl) */
+    wsUrl?: string;
+}): MiddlewareHandler;
+/**
+ * Create a complete RPC middleware that handles all transports
+ *
+ * @example
+ * ```ts
+ * import { Hono } from 'hono'
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers'
+ * import { createRPCMiddleware } from 'ai-functions/rpc'
+ *
+ * const app = new Hono()
+ *
+ * const rpc = createRPCMiddleware({
+ *   methods: {
+ *     greet: (ctx, name) => `Hello, ${name}!`,
+ *     getUser: async (ctx, id) => db.users.get(id),
+ *   },
+ *   createContext: (c) => ({
+ *     user: c.get('user'),
+ *     db: c.get('db'),
+ *   }),
+ *   bridgeOrigins: ['https://myapp.com', 'https://app.myapp.com'],
+ * })
+ *
+ * // Single /rpc endpoint for both HTTP POST and WebSocket GET
+ * app.post('/rpc', rpc.http)
+ * app.get('/rpc', upgradeWebSocket(rpc.ws))
+ *
+ * // postMessage bridge HTML (embed in iframe for cross-origin auth)
+ * app.get('/rpc.html', rpc.bridge)
+ * ```
+ */
+export declare function createRPCMiddleware<TContext>(options: RPCHandlerOptions<TContext> & {
+    rpcUrl?: string;
+    wsUrl?: string;
+}): {
+    /** HTTP batch handler */
+    http: MiddlewareHandler;
+    /** WebSocket handler (use with upgradeWebSocket) */
+    ws: (c: Context) => {
+        onOpen(_evt: Event, ws: WSContext): Promise<void>;
+        onMessage(evt: MessageEvent, ws: WSContext): Promise<void>;
+        onClose(_evt: CloseEvent, _ws: WSContext): void;
+        onError(evt: Event, _ws: WSContext): void;
+    };
+    /** postMessage bridge handler */
+    bridge: MiddlewareHandler;
+    /** The methods for direct access */
+    methods: RPCMethods<TContext>;
+};
+/**
+ * Mount all RPC routes on a Hono app
+ *
+ * Single endpoint handles both HTTP POST and WebSocket upgrade:
+ * - POST /rpc → HTTP batch requests
+ * - GET /rpc (with Upgrade header) → WebSocket connection
+ * - GET /rpc.html → postMessage bridge for iframe embedding
+ *
+ * @example
+ * ```ts
+ * import { Hono } from 'hono'
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers'
+ * import { mountRPC } from 'ai-functions/rpc'
+ *
+ * const app = new Hono()
+ *
+ * mountRPC(app, '/rpc', {
+ *   methods: { ... },
+ *   upgradeWebSocket,
+ * })
+ * ```
+ */
+export declare function mountRPC<TContext>(app: Hono, basePath: string, options: RPCHandlerOptions<TContext> & {
+    rpcUrl?: string;
+    wsUrl?: string;
+    /** WebSocket upgrade function from hono adapter */
+    upgradeWebSocket?: (handler: ReturnType<typeof createWSHandler>) => MiddlewareHandler;
+}): {
+    /** HTTP batch handler */
+    http: MiddlewareHandler;
+    /** WebSocket handler (use with upgradeWebSocket) */
+    ws: (c: Context) => {
+        onOpen(_evt: Event, ws: WSContext): Promise<void>;
+        onMessage(evt: MessageEvent, ws: WSContext): Promise<void>;
+        onClose(_evt: CloseEvent, _ws: WSContext): void;
+        onError(evt: Event, _ws: WSContext): void;
+    };
+    /** postMessage bridge handler */
+    bridge: MiddlewareHandler;
+    /** The methods for direct access */
+    methods: RPCMethods<TContext>;
+};
+//# sourceMappingURL=server.d.ts.map

package/dist/rpc/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/rpc/server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,MAAM,CAAA;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAA;AAQxC;;GAEG;AACH,MAAM,MAAM,SAAS,CAAC,QAAQ,GAAG,OAAO,IAAI,CAC1C,GAAG,EAAE,QAAQ,EACb,GAAG,IAAI,EAAE,OAAO,EAAE,KACf,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAA;AAE/B;;GAEG;AACH,MAAM,MAAM,UAAU,CAAC,QAAQ,GAAG,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;AAEhF;;GAEG;AACH,MAAM,WAAW,iBAAiB,CAAC,QAAQ,GAAG,OAAO;IACnD,4BAA4B;IAC5B,OAAO,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAA;IAC7B,kDAAkD;IAClD,aAAa,CAAC,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAA;IAC5D,2CAA2C;IAC3C,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,QAAQ,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACnF,0CAA0C;IAC1C,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACnF,uBAAuB;IACvB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC/E,8CAA8C;IAC9C,cAAc,CAAC,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;IAC/B,6CAA6C;IAC7C,aAAa,CAAC,EAAE,MAAM,EAAE,CAAA;CACzB;AA0ED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EACxC,OAAO,EAAE,iBAAiB,CAAC,QAAQ,CAAC,GACnC,iBAAiB,CAwCnB;AAMD;;;;;;;;;GASG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,iBAAiB,CAAC,QAAQ,CAAC,IAGpE,GAAG,OAAO;iBAcK,KAAK,MAAM,SAAS;mBAIlB,YAAY,MAAM,SAAS;kBAyClC,UAAU,OAAO,SAAS;iBAK3B,KAAK,OAAO,SAAS;EAKvC;AAMD;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAC1C,OAAO,EAAE,iBAAiB,CAAC,QAAQ,CAAC,GAAG;IACrC,yCAAyC;IACzC,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,mDAAmD;IACnD,KAAK,CAAC,EAAE,MAAM,CAAA;CACf,GACA,iBAAiB,CA+InB;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAC1C,OAAO,EAAE,iBAAiB,CAAC,QAAQ,CAAC,GAAG;IACrC,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,KAAK,CAAC,EAAE,MAAM,CAAA;CACf;IAGC,yBAAyB;;IAEzB,oDAAoD;YAvR3C,OAAO;qBAcK,KAAK,MAAM,SAAS;uBAIlB,YAAY,MAAM,SAAS;sBAyClC,UAAU,OAAO,SAAS;qBAK3B,KAAK,OAAO,SAAS;;IAyNpC,iCAAiC;;IAEjC,oCAAoC;;EAGvC;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,QAAQ,CAAC,QAAQ,EAC/B,GAAG,EAAE,IAAI,EACT,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,iBAAiB,CAAC,QAAQ,CAAC,GAAG;IACrC,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,mDAAmD;IACnD,gBAAgB,CAAC,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC,OAAO,eAAe,CAAC,KAAK,iBAAiB,CAAA;CACtF;IAzCC,yBAAyB;;IAEzB,oDAAoD;YAvR3C,OAAO;qBAcK,KAAK,MAAM,SAAS;uBAIlB,YAAY,MAAM,SAAS;sBAyClC,UAAU,OAAO,SAAS;qBAK3B,KAAK,OAAO,SAAS;;IAyNpC,iCAAiC;;IAEjC,oCAAoC;;EAuDvC"}

package/dist/rpc/server.js

@@ -0,0 +1,405 @@
+/**
+ * Hono RPC Server Handlers
+ *
+ * Provides RPC endpoints for:
+ * - HTTP batch requests (POST /rpc)
+ * - WebSocket persistent connections (GET /rpc/ws)
+ * - postMessage bridge for iframe communication (GET /rpc/bridge)
+ *
+ * @packageDocumentation
+ */
+import { applyChain } from './deferred.js';
+// =============================================================================
+// HTTP Batch Handler
+// =============================================================================
+/**
+ * Process a single RPC call
+ */
+async function processCall(call, methods, ctx, options) {
+    const { method, params = [], chain, id } = call;
+    if (!method) {
+        return {
+            id,
+            type: 'error',
+            error: { message: 'Method name required', code: 'INVALID_REQUEST' },
+        };
+    }
+    const handler = methods[method];
+    if (!handler) {
+        return {
+            id,
+            type: 'error',
+            error: { message: `Unknown method: ${method}`, code: 'METHOD_NOT_FOUND' },
+        };
+    }
+    try {
+        // Pre-call hook
+        await options.onCall?.(method, params, ctx);
+        // Execute the method
+        let result = await handler(ctx, ...params);
+        // Apply chain operations (promise pipelining)
+        if (chain && chain.length > 0) {
+            result = applyChain(result, chain);
+        }
+        // Post-call hook
+        await options.onResult?.(method, result, ctx);
+        return { id, type: 'result', result };
+    }
+    catch (error) {
+        const err = error instanceof Error ? error : new Error(String(error));
+        await options.onError?.(method, err, ctx);
+        return {
+            id,
+            type: 'error',
+            error: {
+                message: err.message,
+                code: 'INTERNAL_ERROR',
+                stack: process.env.NODE_ENV === 'development' ? err.stack : undefined,
+            },
+        };
+    }
+}
+/**
+ * Create HTTP batch RPC handler
+ */
+export function createHTTPHandler(options) {
+    const { methods, createContext, allowedOrigins = '*' } = options;
+    return async (c) => {
+        // Handle CORS
+        const origin = c.req.header('origin');
+        if (allowedOrigins === '*') {
+            c.header('Access-Control-Allow-Origin', '*');
+        }
+        else if (origin && allowedOrigins.includes(origin)) {
+            c.header('Access-Control-Allow-Origin', origin);
+        }
+        c.header('Access-Control-Allow-Methods', 'POST, OPTIONS');
+        c.header('Access-Control-Allow-Headers', 'Content-Type, Authorization');
+        if (c.req.method === 'OPTIONS') {
+            return new Response(null, { status: 204 });
+        }
+        // Parse request body
+        let calls;
+        try {
+            const body = await c.req.json();
+            calls = Array.isArray(body) ? body : [body];
+        }
+        catch {
+            return c.json({ error: { message: 'Invalid JSON', code: 'PARSE_ERROR' } }, 400);
+        }
+        // Create context
+        const ctx = createContext ? await createContext(c) : {};
+        // Process all calls
+        const results = await Promise.all(calls.map((call) => processCall(call, methods, ctx, options)));
+        return c.json(results);
+    };
+}
+// =============================================================================
+// WebSocket Handler
+// =============================================================================
+/**
+ * Create WebSocket RPC handler for Hono
+ *
+ * Usage with hono/ws:
+ * ```ts
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers' // or hono/bun, etc
+ *
+ * app.get('/rpc/ws', upgradeWebSocket(createWSHandler({ methods })))
+ * ```
+ */
+export function createWSHandler(options) {
+    const { methods, createContext } = options;
+    return (c) => {
+        let ctx;
+        // Server-side callback registry for this connection
+        const callbackRegistry = {
+            callbacks: new Map(),
+            invoke: async (id, args) => {
+                const fn = callbackRegistry.callbacks.get(id);
+                if (!fn)
+                    throw new Error(`Callback not found: ${id}`);
+                return fn(...args);
+            },
+        };
+        return {
+            async onOpen(_evt, ws) {
+                ctx = createContext ? await createContext(c) : {};
+            },
+            async onMessage(evt, ws) {
+                let message;
+                try {
+                    message = JSON.parse(evt.data);
+                }
+                catch {
+                    ws.send(JSON.stringify({
+                        type: 'error',
+                        error: { message: 'Invalid JSON', code: 'PARSE_ERROR' },
+                    }));
+                    return;
+                }
+                // Handle ping
+                if (message.type === 'ping') {
+                    ws.send(JSON.stringify({ type: 'pong' }));
+                    return;
+                }
+                // Handle callback response
+                if (message.type === 'result' && message.callbackId) {
+                    // Client is responding to a callback we invoked
+                    // This would be handled by pending callback promises
+                    return;
+                }
+                // Handle RPC call
+                if (message.type === 'call') {
+                    const result = await processCall(message, methods, ctx, options);
+                    ws.send(JSON.stringify(result));
+                    return;
+                }
+                // Handle cancel (for streaming)
+                if (message.type === 'cancel') {
+                    // Could implement stream cancellation here
+                    return;
+                }
+            },
+            onClose(_evt, _ws) {
+                // Cleanup callback registry
+                callbackRegistry.callbacks.clear();
+            },
+            onError(evt, _ws) {
+                console.error('WebSocket error:', evt);
+            },
+        };
+    };
+}
+// =============================================================================
+// postMessage Bridge
+// =============================================================================
+/**
+ * Generate the postMessage bridge HTML/JS
+ *
+ * This creates a lightweight page that:
+ * 1. Listens for postMessage from parent window
+ * 2. Makes authenticated fetch/WS calls to the RPC endpoint
+ * 3. Posts results back to parent
+ */
+export function createBridgeHandler(options) {
+    const { bridgeOrigins = [], rpcUrl = '/rpc', wsUrl } = options;
+    // Generate allowed origins check
+    const originsCheck = bridgeOrigins.length === 0
+        ? 'true' // Allow all in dev
+        : `[${bridgeOrigins.map((o) => `'${o}'`).join(',')}].includes(event.origin)`;
+    const bridgeScript = `
+<!DOCTYPE html>
+<html>
+<head>
+  <title>RPC Bridge</title>
+  <script>
+    (function() {
+      const RPC_URL = '${rpcUrl}';
+      const WS_URL = ${wsUrl ? `'${wsUrl}'` : 'null'};
+      const pendingRequests = new Map();
+      let ws = null;
+      let wsReady = false;
+      let wsQueue = [];
+
+      // Connect WebSocket if available
+      function connectWS() {
+        if (!WS_URL) return;
+
+        ws = new WebSocket(WS_URL);
+
+        ws.onopen = () => {
+          wsReady = true;
+          // Flush queued messages
+          wsQueue.forEach(msg => ws.send(JSON.stringify(msg)));
+          wsQueue = [];
+        };
+
+        ws.onmessage = (event) => {
+          const msg = JSON.parse(event.data);
+          const pending = pendingRequests.get(msg.id);
+          if (pending) {
+            pendingRequests.delete(msg.id);
+            pending.source.postMessage({
+              type: 'rpc-response',
+              id: msg.id,
+              result: msg.result,
+              error: msg.error
+            }, pending.origin);
+          }
+        };
+
+        ws.onclose = () => {
+          wsReady = false;
+          // Reconnect after delay
+          setTimeout(connectWS, 1000);
+        };
+      }
+
+      // Make HTTP request
+      async function httpRequest(calls, source, origin) {
+        try {
+          const response = await fetch(RPC_URL, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            credentials: 'include', // Include cookies!
+            body: JSON.stringify(calls)
+          });
+
+          const results = await response.json();
+
+          results.forEach(result => {
+            source.postMessage({
+              type: 'rpc-response',
+              id: result.id,
+              result: result.result,
+              error: result.error
+            }, origin);
+          });
+        } catch (error) {
+          calls.forEach(call => {
+            source.postMessage({
+              type: 'rpc-response',
+              id: call.id,
+              error: { message: error.message }
+            }, origin);
+          });
+        }
+      }
+
+      // Handle incoming messages
+      window.addEventListener('message', (event) => {
+        // Validate origin
+        if (!(${originsCheck})) {
+          console.warn('RPC Bridge: Rejected message from', event.origin);
+          return;
+        }
+
+        const { type, calls, useWebSocket } = event.data;
+
+        if (type !== 'rpc-request') return;
+
+        // Store pending requests for response routing
+        calls.forEach(call => {
+          pendingRequests.set(call.id, {
+            source: event.source,
+            origin: event.origin
+          });
+        });
+
+        // Route to WebSocket or HTTP
+        if (useWebSocket && ws && wsReady) {
+          calls.forEach(call => ws.send(JSON.stringify(call)));
+        } else if (useWebSocket && ws) {
+          // Queue for when WS is ready
+          calls.forEach(call => wsQueue.push(call));
+        } else {
+          httpRequest(calls, event.source, event.origin);
+        }
+      });
+
+      // Initialize
+      connectWS();
+
+      // Signal ready
+      if (window.parent !== window) {
+        window.parent.postMessage({ type: 'rpc-bridge-ready' }, '*');
+      }
+    })();
+  </script>
+</head>
+<body></body>
+</html>
+`;
+    return async (c) => {
+        c.header('Content-Type', 'text/html');
+        // Security headers for iframe
+        c.header('X-Frame-Options', 'ALLOWALL'); // Allow embedding
+        c.header('Content-Security-Policy', "default-src 'self'; script-src 'unsafe-inline'; connect-src 'self' wss://*.apis.do https://*.apis.do");
+        return c.html(bridgeScript);
+    };
+}
+// =============================================================================
+// Unified Middleware
+// =============================================================================
+/**
+ * Create a complete RPC middleware that handles all transports
+ *
+ * @example
+ * ```ts
+ * import { Hono } from 'hono'
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers'
+ * import { createRPCMiddleware } from 'ai-functions/rpc'
+ *
+ * const app = new Hono()
+ *
+ * const rpc = createRPCMiddleware({
+ *   methods: {
+ *     greet: (ctx, name) => `Hello, ${name}!`,
+ *     getUser: async (ctx, id) => db.users.get(id),
+ *   },
+ *   createContext: (c) => ({
+ *     user: c.get('user'),
+ *     db: c.get('db'),
+ *   }),
+ *   bridgeOrigins: ['https://myapp.com', 'https://app.myapp.com'],
+ * })
+ *
+ * // Single /rpc endpoint for both HTTP POST and WebSocket GET
+ * app.post('/rpc', rpc.http)
+ * app.get('/rpc', upgradeWebSocket(rpc.ws))
+ *
+ * // postMessage bridge HTML (embed in iframe for cross-origin auth)
+ * app.get('/rpc.html', rpc.bridge)
+ * ```
+ */
+export function createRPCMiddleware(options) {
+    return {
+        /** HTTP batch handler */
+        http: createHTTPHandler(options),
+        /** WebSocket handler (use with upgradeWebSocket) */
+        ws: createWSHandler(options),
+        /** postMessage bridge handler */
+        bridge: createBridgeHandler(options),
+        /** The methods for direct access */
+        methods: options.methods,
+    };
+}
+/**
+ * Mount all RPC routes on a Hono app
+ *
+ * Single endpoint handles both HTTP POST and WebSocket upgrade:
+ * - POST /rpc → HTTP batch requests
+ * - GET /rpc (with Upgrade header) → WebSocket connection
+ * - GET /rpc.html → postMessage bridge for iframe embedding
+ *
+ * @example
+ * ```ts
+ * import { Hono } from 'hono'
+ * import { upgradeWebSocket } from 'hono/cloudflare-workers'
+ * import { mountRPC } from 'ai-functions/rpc'
+ *
+ * const app = new Hono()
+ *
+ * mountRPC(app, '/rpc', {
+ *   methods: { ... },
+ *   upgradeWebSocket,
+ * })
+ * ```
+ */
+export function mountRPC(app, basePath, options) {
+    const rpc = createRPCMiddleware({
+        ...options,
+        rpcUrl: options.rpcUrl ?? basePath,
+        wsUrl: options.wsUrl ?? basePath.replace(/^http/, 'ws'),
+    });
+    // HTTP batch (POST)
+    app.post(basePath, rpc.http);
+    // WebSocket upgrade (GET with Upgrade header)
+    if (options.upgradeWebSocket) {
+        app.get(basePath, options.upgradeWebSocket(rpc.ws));
+    }
+    // Bridge HTML page (for iframe embedding)
+    app.get(`${basePath}.html`, rpc.bridge);
+    return rpc;
+}
+//# sourceMappingURL=server.js.map

package/dist/rpc/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/rpc/server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAIH,OAAO,EAAE,UAAU,EAAkB,MAAM,eAAe,CAAA;AAgD1D,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF;;GAEG;AACH,KAAK,UAAU,WAAW,CACxB,IAAgB,EAChB,OAA6B,EAC7B,GAAa,EACb,OAAoC;IAEpC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,IAAI,CAAA;IAE/C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;YACL,EAAE;YACF,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAE,IAAI,EAAE,iBAAiB,EAAE;SACpE,CAAA;IACH,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAA;IAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;YACL,EAAE;YACF,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,EAAE,OAAO,EAAE,mBAAmB,MAAM,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;SAC1E,CAAA;IACH,CAAC;IAED,IAAI,CAAC;QACH,gBAAgB;QAChB,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAA;QAE3C,qBAAqB;QACrB,IAAI,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,CAAA;QAE1C,8CAA8C;QAC9C,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,GAAG,UAAU,CAAC,MAAM,EAAE,KAAoB,CAAC,CAAA;QACnD,CAAC;QAED,iBAAiB;QACjB,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAA;QAE7C,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;IACvC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;QACrE,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;QAEzC,OAAO;YACL,EAAE;YACF,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,IAAI,EAAE,gBAAgB;gBACtB,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;aACtE;SACF,CAAA;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,OAAoC;IAEpC,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,cAAc,GAAG,GAAG,EAAE,GAAG,OAAO,CAAA;IAEhE,OAAO,KAAK,EAAE,CAAU,EAAE,EAAE;QAC1B,cAAc;QACd,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QACrC,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;YAC3B,CAAC,CAAC,MAAM,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAA;QAC9C,CAAC;aAAM,IAAI,MAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACrD,CAAC,CAAC,MAAM,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAA;QACjD,CAAC;QACD,CAAC,CAAC,MAAM,CAAC,8BAA8B,EAAE,eAAe,CAAC,CAAA;QACzD,CAAC,CAAC,MAAM,CAAC,8BAA8B,EAAE,6BAA6B,CAAC,CAAA;QAEvE,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC5C,CAAC;QAED,qBAAqB;QACrB,IAAI,KAAmB,CAAA;QACvB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAA;YAC/B,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QAC7C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC,IAAI,CACX,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAC3D,GAAG,CACJ,CAAA;QACH,CAAC;QAED,iBAAiB;QACjB,MAAM,GAAG,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,EAAe,CAAA;QAErE,oBAAoB;QACpB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,CAC9D,CAAA;QAED,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IACxB,CAAC,CAAA;AACH,CAAC;AAED,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;;;;;;;;GASG;AACH,MAAM,UAAU,eAAe,CAAW,OAAoC;IAC5E,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,OAAO,CAAA;IAE1C,OAAO,CAAC,CAAU,EAAE,EAAE;QACpB,IAAI,GAAa,CAAA;QAEjB,oDAAoD;QACpD,MAAM,gBAAgB,GAA2B;YAC/C,SAAS,EAAE,IAAI,GAAG,EAAE;YACpB,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE;gBACzB,MAAM,EAAE,GAAG,gBAAgB,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;gBAC7C,IAAI,CAAC,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAA;gBACrD,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAA;YACpB,CAAC;SACF,CAAA;QAED,OAAO;YACL,KAAK,CAAC,MAAM,CAAC,IAAW,EAAE,EAAa;gBACrC,GAAG,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,EAAe,CAAA;YACjE,CAAC;YAED,KAAK,CAAC,SAAS,CAAC,GAAiB,EAAE,EAAa;gBAC9C,IAAI,OAAmB,CAAA;gBACvB,IAAI,CAAC;oBACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAc,CAAC,CAAA;gBAC1C,CAAC;gBAAC,MAAM,CAAC;oBACP,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;wBACb,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE;qBACxD,CAAC,CACH,CAAA;oBACD,OAAM;gBACR,CAAC;gBAED,cAAc;gBACd,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC5B,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;oBACzC,OAAM;gBACR,CAAC;gBAED,2BAA2B;gBAC3B,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;oBACpD,gDAAgD;oBAChD,qDAAqD;oBACrD,OAAM;gBACR,CAAC;gBAED,kBAAkB;gBAClB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC5B,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;oBAChE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAA;oBAC/B,OAAM;gBACR,CAAC;gBAED,gCAAgC;gBAChC,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC9B,2CAA2C;oBAC3C,OAAM;gBACR,CAAC;YACH,CAAC;YAED,OAAO,CAAC,IAAgB,EAAE,GAAc;gBACtC,4BAA4B;gBAC5B,gBAAgB,CAAC,SAAS,CAAC,KAAK,EAAE,CAAA;YACpC,CAAC;YAED,OAAO,CAAC,GAAU,EAAE,GAAc;gBAChC,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAA;YACxC,CAAC;SACF,CAAA;IACH,CAAC,CAAA;AACH,CAAC;AAED,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF;;;;;;;GAOG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAKC;IAED,MAAM,EAAE,aAAa,GAAG,EAAE,EAAE,MAAM,GAAG,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAA;IAE9D,iCAAiC;IACjC,MAAM,YAAY,GAChB,aAAa,CAAC,MAAM,KAAK,CAAC;QACxB,CAAC,CAAC,MAAM,CAAC,mBAAmB;QAC5B,CAAC,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,0BAA0B,CAAA;IAEhF,MAAM,YAAY,GAAG;;;;;;;yBAOE,MAAM;uBACR,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gBA0EpC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwC3B,CAAA;IAEC,OAAO,KAAK,EAAE,CAAU,EAAE,EAAE;QAC1B,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,CAAA;QACrC,8BAA8B;QAC9B,CAAC,CAAC,MAAM,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAA,CAAC,kBAAkB;QAC1D,CAAC,CAAC,MAAM,CACN,yBAAyB,EACzB,sGAAsG,CACvG,CAAA;QACD,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAC7B,CAAC,CAAA;AACH,CAAC;AAED,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAGC;IAED,OAAO;QACL,yBAAyB;QACzB,IAAI,EAAE,iBAAiB,CAAC,OAAO,CAAC;QAChC,oDAAoD;QACpD,EAAE,EAAE,eAAe,CAAC,OAAO,CAAC;QAC5B,iCAAiC;QACjC,MAAM,EAAE,mBAAmB,CAAC,OAAO,CAAC;QACpC,oCAAoC;QACpC,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAA;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,QAAQ,CACtB,GAAS,EACT,QAAgB,EAChB,OAKC;IAED,MAAM,GAAG,GAAG,mBAAmB,CAAC;QAC9B,GAAG,OAAO;QACV,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,QAAQ;QAClC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;KACxD,CAAC,CAAA;IAEF,oBAAoB;IACpB,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,CAAC,CAAA;IAE5B,8CAA8C;IAC9C,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAC7B,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;IACrD,CAAC;IAED,0CAA0C;IAC1C,GAAG,CAAC,GAAG,CAAC,GAAG,QAAQ,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,CAAA;IAEvC,OAAO,GAAG,CAAA;AACZ,CAAC"}