ai-publish-sdk 1.2.0 → 1.3.0

package/README.md

@@ -8,17 +8,18 @@ SDK for embedded apps to communicate with the host environment.
 npm install ai-publish-sdk
 ```
 
-## General
+## Timeout Configuration
 
-### RpcConfig
-
-Optional config passed to any function call.
+Default timeout is 15 seconds. You can change it globally or wrap a single call:
 
 ```typescript
-interface RpcConfig {
-  timeout?: number      // ms, default 15000
-  targetOrigin?: string // postMessage target origin, default '*'
-}
+import { setGlobalTimeout, withTimeout, getUserInfo } from 'ai-publish-sdk'
+
+// Change the default for all calls
+setGlobalTimeout(30_000)
+
+// Override for a single call
+await withTimeout(() => getUserInfo(), 10_000)
 ```
 
 ## Core API
@@ -28,7 +29,7 @@ interface RpcConfig {
 Get current user info.
 
 ```typescript
-getUserInfo(config?: RpcConfig): Promise<UserInfo | null>
+getUserInfo(): Promise<UserInfo | null>
 
 interface UserInfo {
   userId: string
@@ -48,7 +49,7 @@ interface UserInfo {
 Get OAuth token for an integration.
 
 ```typescript
-getToken(appName: IntegrationAppName, options?: GetTokenOptions, config?: RpcConfig): Promise<GetTokenResult | null>
+getToken(appName: IntegrationAppName, options?: GetTokenOptions): Promise<GetTokenResult | null>
 
 type IntegrationAppName = 'gmail' | 'jira' | 'googledrive' | 'salesforce' | 'googlecalendar' | 'slack'
 
@@ -67,7 +68,7 @@ interface GetTokenResult {
 Generate an AI message.
 
 ```typescript
-generateMessage(prompt: string, options: GenerateMessageOptions, config?: RpcConfig): Promise<string | null>
+generateMessage(prompt: string, options: GenerateMessageOptions): Promise<string | null>
 
 interface GenerateMessageOptions {
   withPageContext?: boolean // include current page content as context
@@ -80,7 +81,7 @@ interface GenerateMessageOptions {
 Execute JavaScript on the host page.
 
 ```typescript
-executeScript(code: string, config?: RpcConfig): Promise<void>
+executeScript(code: string): Promise<void>
 ```
 
 ## TCP API
@@ -92,12 +93,12 @@ All data is base64-encoded. Import as `import { tcp }`.
 Open a TCP connection. Returns a `TcpSocket` object for subsequent operations, or `null` if TCP is unavailable.
 
 ```typescript
-tcp.connect(host: string, port: number, config?: RpcConfig): Promise<TcpSocket | null>
+tcp.connect(host: string, port: number): Promise<TcpSocket | null>
 
 interface TcpSocket {
-  send(dataBase64: string, config?: RpcConfig): Promise<number | null>
-  receive(timeoutMs?: number, config?: RpcConfig): Promise<string | null>
-  close(config?: RpcConfig): Promise<void>
+  send(dataBase64: string): Promise<number | null>
+  receive(): Promise<string | null>
+  close(): Promise<void>
 }
 ```
 
@@ -105,11 +106,9 @@ interface TcpSocket {
 
 Send data over an open socket. Data must be base64-encoded. Returns the number of bytes sent, or `null` if TCP is unavailable.
 
-### TcpSocket.receive(timeoutMs?)
-
-Wait for and receive data from an open socket. Returns base64-encoded data, or `null` if TCP is unavailable. Defaults to a 10 second timeout.
+### TcpSocket.receive()
 
-**Two-timeout interaction:** `receive()` has two independent timeouts: `timeoutMs` (host-side, how long to wait for data) and `RpcConfig.timeout` (SDK-side postMessage round-trip, default 15 s). If `timeoutMs` is longer than `RpcConfig.timeout`, the RPC layer will reject before the host-side timeout fires. Set `RpcConfig.timeout` to at least `timeoutMs + 5000` to avoid spurious RPC timeout rejections.
+Wait for and receive data from an open socket. Returns base64-encoded data, or `null` if TCP is unavailable. Use `withTimeout(() => socket.receive(), ms)` for custom timeouts.
 
 ### TcpSocket.close()
 

package/index.d.ts

@@ -1,5 +1,5 @@
 export { generateMessage, getToken, getUserInfo } from './lib/functions';
 export { tcp } from './lib/tcp-socket';
 export type { TcpSocket } from './lib/tcp-socket';
-export type { RpcConfig } from './lib/rpc-bridge';
+export { setGlobalTimeout, withTimeout } from './lib/rpc-bridge';
 export type { GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, UserInfo } from './lib/types';

package/index.js

@@ -1 +1 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const m=15e3,u=new Map;let l=!1;function p(){l||(l=!0,window.addEventListener("message",t=>{const e=t.data;if(!e?.messageId||!("name"in e))return;const n=u.get(e.messageId);n&&(clearTimeout(n.timer),u.delete(e.messageId),e.error?n.reject(new Error(e.error.message??"RPC error")):n.resolve(e.data))}))}function s(t,e,n){p();const o=n?.timeout??m,r=n?.targetOrigin??"*";return new Promise((c,a)=>{const i=crypto.randomUUID(),g=setTimeout(()=>{u.delete(i),a(new Error(`RPC timeout: ${t}`))},o);u.set(i,{resolve:c,reject:a,timer:g});const d={action:t,params:e,messageId:i};window.parent.postMessage(d,r)})}async function f(t,e,n){return s("generateMessage",[t,e],n)}async function y(t,e,n){return s("getToken",[t,e],n)}async function w(t){return s("getUserInfo",[],t)}async function I(t,e){return s("tcpConnect",[t],e)}async function T(t,e,n){return s("tcpSend",[t,e],n)}async function v(t,e,n){return s("tcpReceive",[t,e],n)}async function C(t,e){return s("tcpClose",[t],e)}const U={async connect(t,e,n){const o=await I({host:t,port:e},n);return o===null?null:{send:async(r,c)=>(await T(o,{dataBase64:r},c??n))?.bytesSent??null,receive:async(r,c)=>(await v(o,r!==void 0?{timeoutMs:r}:void 0,c??n))?.data??null,close:r=>C(o,r??n)}}};exports.generateMessage=f;exports.getToken=y;exports.getUserInfo=w;exports.tcp=U;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});let u=15e3;const c=new Map;let a=!1;function m(e){u=e}let i;function d(e,t){i=t;try{return e()}finally{i=void 0}}function f(){a||(a=!0,window.addEventListener("message",e=>{const t=e.data;if(!t?.messageId||!("name"in t))return;const n=c.get(t.messageId);n&&(clearTimeout(n.timer),c.delete(t.messageId),t.error?n.reject(new Error(t.error.message??"RPC error")):n.resolve(t.data))}))}function r(e,t){return f(),new Promise((n,s)=>{const o=crypto.randomUUID(),l=setTimeout(()=>{c.delete(o),s(new Error(`RPC timeout: ${e}`))},i??u);c.set(o,{resolve:n,reject:s,timer:l});const g={action:e,params:t,messageId:o};window.parent.postMessage(g,"*")})}async function p(e,t){return r("generateMessage",[e,t])}async function y(e,t){return r("getToken",[e,t])}async function w(){return r("getUserInfo",[])}async function T(e){return r("tcpConnect",[e])}async function v(e,t){return r("tcpSend",[e,t])}async function I(e,t){return r("tcpReceive",[e,t])}async function C(e){return r("tcpClose",[e])}const b={async connect(e,t){const n=await T({host:e,port:t});return n===null?null:{send:async s=>(await v(n,{dataBase64:s}))?.bytesSent??null,receive:async()=>(await I(n))?.data??null,close:()=>C(n)}}};exports.generateMessage=p;exports.getToken=y;exports.getUserInfo=w;exports.setGlobalTimeout=m;exports.tcp=b;exports.withTimeout=d;

package/index.mjs

@@ -1,51 +1,62 @@
-const u = /* @__PURE__ */ new Map();
-let l = !1;
+let u = 15e3;
+const c = /* @__PURE__ */ new Map();
+let i = !1;
+function T(e) {
+  u = e;
+}
+let a;
+function v(e, t) {
+  a = t;
+  try {
+    return e();
+  } finally {
+    a = void 0;
+  }
+}
 function m() {
-  l || (l = !0, window.addEventListener("message", (t) => {
-    const e = t.data;
-    if (!e?.messageId || !("name" in e)) return;
-    const n = u.get(e.messageId);
-    n && (clearTimeout(n.timer), u.delete(e.messageId), e.error ? n.reject(new Error(e.error.message ?? "RPC error")) : n.resolve(e.data));
+  i || (i = !0, window.addEventListener("message", (e) => {
+    const t = e.data;
+    if (!t?.messageId || !("name" in t)) return;
+    const n = c.get(t.messageId);
+    n && (clearTimeout(n.timer), c.delete(t.messageId), t.error ? n.reject(new Error(t.error.message ?? "RPC error")) : n.resolve(t.data));
   }));
 }
-function s(t, e, n) {
-  m();
-  const c = n?.timeout ?? 15e3, r = n?.targetOrigin ?? "*";
-  return new Promise((o, a) => {
-    const i = crypto.randomUUID(), d = setTimeout(() => {
-      u.delete(i), a(new Error(`RPC timeout: ${t}`));
-    }, c);
-    u.set(i, {
-      resolve: o,
-      reject: a,
-      timer: d
+function r(e, t) {
+  return m(), new Promise((n, s) => {
+    const o = crypto.randomUUID(), l = setTimeout(() => {
+      c.delete(o), s(new Error(`RPC timeout: ${e}`));
+    }, a ?? u);
+    c.set(o, {
+      resolve: n,
+      reject: s,
+      timer: l
     });
-    const g = { action: t, params: e, messageId: i };
-    window.parent.postMessage(g, r);
+    const d = { action: e, params: t, messageId: o };
+    window.parent.postMessage(d, "*");
   });
 }
-async function T(t, e, n) {
-  return s("generateMessage", [t, e], n);
+async function C(e, t) {
+  return r("generateMessage", [e, t]);
 }
-async function I(t, e, n) {
-  return s("getToken", [t, e], n);
+async function I(e, t) {
+  return r("getToken", [e, t]);
 }
-async function U(t) {
-  return s("getUserInfo", [], t);
+async function M() {
+  return r("getUserInfo", []);
 }
-async function p(t, e) {
-  return s("tcpConnect", [t], e);
+async function f(e) {
+  return r("tcpConnect", [e]);
 }
-async function f(t, e, n) {
-  return s("tcpSend", [t, e], n);
+async function g(e, t) {
+  return r("tcpSend", [e, t]);
 }
-async function y(t, e, n) {
-  return s("tcpReceive", [t, e], n);
+async function p(e, t) {
+  return r("tcpReceive", [e, t]);
 }
-async function w(t, e) {
-  return s("tcpClose", [t], e);
+async function y(e) {
+  return r("tcpClose", [e]);
 }
-const v = {
+const R = {
   /**
    * Opens a TCP connection and returns a `TcpSocket` wrapper.
    *
@@ -54,18 +65,20 @@ const v = {
    * use a socket) is enforced on the host side — do not assume the closure provides
    * security guarantees.
    */
-  async connect(t, e, n) {
-    const c = await p({ host: t, port: e }, n);
-    return c === null ? null : {
-      send: async (r, o) => (await f(c, { dataBase64: r }, o ?? n))?.bytesSent ?? null,
-      receive: async (r, o) => (await y(c, r !== void 0 ? { timeoutMs: r } : void 0, o ?? n))?.data ?? null,
-      close: (r) => w(c, r ?? n)
+  async connect(e, t) {
+    const n = await f({ host: e, port: t });
+    return n === null ? null : {
+      send: async (s) => (await g(n, { dataBase64: s }))?.bytesSent ?? null,
+      receive: async () => (await p(n))?.data ?? null,
+      close: () => y(n)
     };
   }
 };
 export {
-  T as generateMessage,
+  C as generateMessage,
   I as getToken,
-  U as getUserInfo,
-  v as tcp
+  M as getUserInfo,
+  T as setGlobalTimeout,
+  R as tcp,
+  v as withTimeout
 };

package/lib/functions.d.ts

@@ -1,10 +1,9 @@
-import { RpcConfig } from './rpc-bridge';
 import { GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, TcpConnectOptions, TcpConnectionSettings, TcpReceiveOptions, TcpReceiveResult, TcpSendMessage, TcpSendResult, UserInfo } from './types';
-export declare function generateMessage(prompt: string, options: GenerateMessageOptions, config?: RpcConfig): Promise<string | null>;
-export declare function getToken(appName: IntegrationAppName, options?: GetTokenOptions, config?: RpcConfig): Promise<GetTokenResult | null>;
-export declare function getUserInfo(config?: RpcConfig): Promise<UserInfo | null>;
-export declare function executeScript(code: string, config?: RpcConfig): Promise<void>;
-export declare function tcpConnect(options: TcpConnectOptions, config?: RpcConfig): Promise<TcpConnectionSettings | null>;
-export declare function tcpSend(connection: TcpConnectionSettings, message: TcpSendMessage, config?: RpcConfig): Promise<TcpSendResult | null>;
-export declare function tcpReceive(connection: TcpConnectionSettings, options?: TcpReceiveOptions, config?: RpcConfig): Promise<TcpReceiveResult | null>;
-export declare function tcpClose(connection: TcpConnectionSettings, config?: RpcConfig): Promise<void>;
+export declare function generateMessage(prompt: string, options: GenerateMessageOptions): Promise<string | null>;
+export declare function getToken(appName: IntegrationAppName, options?: GetTokenOptions): Promise<GetTokenResult | null>;
+export declare function getUserInfo(): Promise<UserInfo | null>;
+export declare function executeScript(code: string): Promise<void>;
+export declare function tcpConnect(options: TcpConnectOptions): Promise<TcpConnectionSettings | null>;
+export declare function tcpSend(connection: TcpConnectionSettings, message: TcpSendMessage): Promise<TcpSendResult | null>;
+export declare function tcpReceive(connection: TcpConnectionSettings, options?: TcpReceiveOptions): Promise<TcpReceiveResult | null>;
+export declare function tcpClose(connection: TcpConnectionSettings): Promise<void>;

package/lib/rpc-bridge.d.ts

@@ -1,5 +1,4 @@
-export interface RpcConfig {
-    timeout?: number;
-    targetOrigin?: string;
-}
-export declare function rpcCall<T = unknown>(action: string, params: unknown[], config?: RpcConfig): Promise<T>;
+export declare function setGlobalTimeout(ms: number): void;
+export declare function resetForTesting(): void;
+export declare function withTimeout<T>(fn: () => Promise<T>, ms: number): Promise<T>;
+export declare function rpcCall<T = unknown>(action: string, params: unknown[]): Promise<T>;

package/lib/tcp-socket.d.ts

@@ -1,8 +1,7 @@
-import { RpcConfig } from './rpc-bridge';
 export interface TcpSocket {
-    send(dataBase64: string, config?: RpcConfig): Promise<number | null>;
-    receive(timeoutMs?: number, config?: RpcConfig): Promise<string | null>;
-    close(config?: RpcConfig): Promise<void>;
+    send(dataBase64: string): Promise<number | null>;
+    receive(): Promise<string | null>;
+    close(): Promise<void>;
 }
 export declare const tcp: {
     /**
@@ -13,5 +12,5 @@ export declare const tcp: {
      * use a socket) is enforced on the host side — do not assume the closure provides
      * security guarantees.
      */
-    connect(host: string, port: number, config?: RpcConfig): Promise<TcpSocket | null>;
+    connect(host: string, port: number): Promise<TcpSocket | null>;
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-publish-sdk",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "module": "./index.mjs",
   "main": "./index.js",
   "types": "./index.d.ts",