@arbidocs/sdk 0.3.109 → 0.3.110

package/dist/{browser-C4gqnXaX.d.cts → browser-DgApY8QV.d.cts}

@@ -1,4 +1,4 @@
-import { SessionStorageProvider, ArbiClient, LoginResult, components, WebSocketServerMessage, WsTaskUpdateMessage, KeyPair } from '@arbidocs/client';
+import { SessionStorageProvider, ArbiClient, LoginResult, components, WebSocketServerMessage, WsAuthResultMessage, WsTaskUpdateMessage, KeyPair } from '@arbidocs/client';
 
 /**
  * Core error types and throwing error utilities.
@@ -470,6 +470,12 @@ interface ConnectOptions {
     accessToken: string;
     onMessage: (msg: WebSocketServerMessage) => void;
     onClose?: (code: number, reason: string) => void;
+    /**
+     * Called once the server accepts auth (`auth_result { success: true }`). The
+     * full message is passed so callers can read fields like `new_messages`. The
+     * auth handshake itself is owned by the transport — this is a notification.
+     */
+    onAuthenticated?: (msg: WsAuthResultMessage) => void;
 }
 interface ReconnectOptions extends ConnectOptions {
     /**
@@ -497,6 +503,12 @@ interface ReconnectOptions extends ConnectOptions {
     onReconnecting?: (attempt: number, maxRetries: number) => void;
     onReconnected?: () => void;
     onReconnectFailed?: () => void;
+    /**
+     * Decide whether to reconnect after a given close. Return `false` to stop —
+     * e.g. a server-initiated single-session eviction (code 1008), where
+     * reconnecting would just get kicked again. Default: always reconnect.
+     */
+    shouldReconnect?: (code: number, reason: string) => boolean;
 }
 interface ReconnectableWsConnection {
     close: () => void;
@@ -511,7 +523,10 @@ declare function connectWebSocket(options: ConnectOptions): Promise<WsConnection
 /**
  * Connect WebSocket with automatic reconnection on disconnect.
  *
- * Initial connection throws on failure (same as connectWebSocket).
+ * The initial connection is auth-symmetric with the reconnect loop: if the first
+ * attempt is rejected for auth reasons and `refreshAuth` is provided, it re-logs
+ * in and retries once with the fresh token before throwing. Any other failure —
+ * or a still-failing retry — throws to the caller.
  * After successful auth, disconnects trigger exponential backoff reconnection:
  * - a fresh token is resolved before every attempt (see `getAccessToken`), so a
  *   token rotated elsewhere is picked up rather than reusing a stale one;
@@ -1630,6 +1645,18 @@ declare class Arbi {
     get isLoggedIn(): boolean;
     /** Check if a workspace is selected. */
     get hasWorkspace(): boolean;
+    /**
+     * Live connection info for opening a raw transport (e.g. a WebSocket).
+     *
+     * Reads the access token from the session on each call, so a token refreshed
+     * elsewhere is reflected on the next (re)connect. Returns `null` when there is
+     * no client/token yet (not logged in) rather than throwing, so callers can
+     * poll for readiness instead of guarding against exceptions.
+     */
+    getConnectionInfo(): {
+        baseUrl: string;
+        accessToken: string;
+    } | null;
     private getAuthHeaders;
     readonly projects: {
         list: () => Promise<{

package/dist/{browser-C4gqnXaX.d.ts → browser-DgApY8QV.d.ts}

@@ -1,4 +1,4 @@
-import { SessionStorageProvider, ArbiClient, LoginResult, components, WebSocketServerMessage, WsTaskUpdateMessage, KeyPair } from '@arbidocs/client';
+import { SessionStorageProvider, ArbiClient, LoginResult, components, WebSocketServerMessage, WsAuthResultMessage, WsTaskUpdateMessage, KeyPair } from '@arbidocs/client';
 
 /**
  * Core error types and throwing error utilities.
@@ -470,6 +470,12 @@ interface ConnectOptions {
     accessToken: string;
     onMessage: (msg: WebSocketServerMessage) => void;
     onClose?: (code: number, reason: string) => void;
+    /**
+     * Called once the server accepts auth (`auth_result { success: true }`). The
+     * full message is passed so callers can read fields like `new_messages`. The
+     * auth handshake itself is owned by the transport — this is a notification.
+     */
+    onAuthenticated?: (msg: WsAuthResultMessage) => void;
 }
 interface ReconnectOptions extends ConnectOptions {
     /**
@@ -497,6 +503,12 @@ interface ReconnectOptions extends ConnectOptions {
     onReconnecting?: (attempt: number, maxRetries: number) => void;
     onReconnected?: () => void;
     onReconnectFailed?: () => void;
+    /**
+     * Decide whether to reconnect after a given close. Return `false` to stop —
+     * e.g. a server-initiated single-session eviction (code 1008), where
+     * reconnecting would just get kicked again. Default: always reconnect.
+     */
+    shouldReconnect?: (code: number, reason: string) => boolean;
 }
 interface ReconnectableWsConnection {
     close: () => void;
@@ -511,7 +523,10 @@ declare function connectWebSocket(options: ConnectOptions): Promise<WsConnection
 /**
  * Connect WebSocket with automatic reconnection on disconnect.
  *
- * Initial connection throws on failure (same as connectWebSocket).
+ * The initial connection is auth-symmetric with the reconnect loop: if the first
+ * attempt is rejected for auth reasons and `refreshAuth` is provided, it re-logs
+ * in and retries once with the fresh token before throwing. Any other failure —
+ * or a still-failing retry — throws to the caller.
  * After successful auth, disconnects trigger exponential backoff reconnection:
  * - a fresh token is resolved before every attempt (see `getAccessToken`), so a
  *   token rotated elsewhere is picked up rather than reusing a stale one;
@@ -1630,6 +1645,18 @@ declare class Arbi {
     get isLoggedIn(): boolean;
     /** Check if a workspace is selected. */
     get hasWorkspace(): boolean;
+    /**
+     * Live connection info for opening a raw transport (e.g. a WebSocket).
+     *
+     * Reads the access token from the session on each call, so a token refreshed
+     * elsewhere is reflected on the next (re)connect. Returns `null` when there is
+     * no client/token yet (not logged in) rather than throwing, so callers can
+     * poll for readiness instead of guarding against exceptions.
+     */
+    getConnectionInfo(): {
+        baseUrl: string;
+        accessToken: string;
+    } | null;
     private getAuthHeaders;
     readonly projects: {
         list: () => Promise<{

package/dist/browser.cjs

@@ -4043,7 +4043,7 @@ var WebSocketAuthError = class extends Error {
   }
 };
 function connectWebSocket(options) {
-  const { baseUrl, accessToken, onMessage, onClose } = options;
+  const { baseUrl, accessToken, onMessage, onClose, onAuthenticated } = options;
   const url = client.buildWebSocketUrl(baseUrl);
   return new Promise((resolve, reject) => {
     const ws = new WebSocket(url);
@@ -4066,6 +4066,7 @@ function connectWebSocket(options) {
           clearTimeout(timeout);
           if (msg.success) {
             authenticated = true;
+            onAuthenticated?.(msg);
             resolve({ close: () => ws.close() });
           } else {
             ws.close();
@@ -4099,6 +4100,8 @@ async function connectWithReconnect(options) {
     onReconnectFailed,
     onClose,
     onMessage,
+    onAuthenticated,
+    shouldReconnect,
     baseUrl,
     accessToken
   } = options;
@@ -4113,10 +4116,11 @@ async function connectWithReconnect(options) {
       baseUrl,
       accessToken: token,
       onMessage,
+      onAuthenticated,
       onClose: (code, reason) => {
         currentConnection = null;
         onClose?.(code, reason);
-        if (!closed) void scheduleReconnect();
+        if (!closed && (shouldReconnect?.(code, reason) ?? true)) void scheduleReconnect();
       }
     });
   };
@@ -4147,7 +4151,14 @@ async function connectWithReconnect(options) {
     }
     if (!closed) onReconnectFailed?.();
   };
-  currentConnection = await connectOnce();
+  try {
+    currentConnection = await connectOnce();
+  } catch (err) {
+    if (!(err instanceof WebSocketAuthError) || !refreshAuth) throw err;
+    await refreshAuth().catch(() => {
+    });
+    currentConnection = await connectOnce();
+  }
   return {
     close: () => {
       closed = true;
@@ -5556,6 +5567,19 @@ var Arbi = class {
   get hasWorkspace() {
     return this.currentWorkspaceId !== null;
   }
+  /**
+   * Live connection info for opening a raw transport (e.g. a WebSocket).
+   *
+   * Reads the access token from the session on each call, so a token refreshed
+   * elsewhere is reflected on the next (re)connect. Returns `null` when there is
+   * no client/token yet (not logged in) rather than throwing, so callers can
+   * poll for readiness instead of guarding against exceptions.
+   */
+  getConnectionInfo() {
+    const accessToken = this.client?.session.getState().accessToken;
+    if (!accessToken) return null;
+    return { baseUrl: this.options.url, accessToken };
+  }
   // ── Auth headers (for raw fetch operations) ───────────────────────────
   getAuthHeaders() {
     const client = this.requireClient();