@pooflabs/core 0.0.38 → 0.0.39

package/dist/client/operations.d.ts

@@ -1,10 +1,29 @@
+import { AuthProvider } from '../types';
 import { PublicKey, Transaction, VersionedTransaction } from '@solana/web3.js';
+/** Internal overrides for per-request auth and headers. */
+export type RequestOverrides = {
+    headers?: Record<string, string>;
+    authProvider?: AuthProvider;
+    _getAuthHeaders?: () => Promise<Record<string, string>>;
+    _clearAuth?: () => Promise<void>;
+    _walletAddress?: string;
+};
 export type SetOptions = {
     shouldSubmitTx?: boolean;
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
+/**
+ * Thrown when a user's wallet doesn't have enough SOL to cover the transaction.
+ * Apps can catch this to trigger an onramp/funding flow.
+ */
+export declare class InsufficientBalanceError extends Error {
+    address: string;
+    balanceLamports: number;
+    estimatedCostLamports: number;
+    deficitLamports: number;
+    deficitSol: number;
+    constructor(address: string, balanceLamports: number, estimatedCostLamports: number, deficitLamports: number, deficitSol: number);
+}
 /**
  * Options for the get function.
  */
@@ -21,15 +40,10 @@ export type GetOptions = {
     limit?: number;
     /** Opaque cursor for cursor-based pagination (used with limit) */
     cursor?: string;
-    /** Internal overrides for headers */
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
 export type RunQueryOptions = {
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
 /**
  * Supported aggregate operations for count/aggregate queries.
@@ -47,10 +61,7 @@ export type AggregateResult = {
 export type CountOptions = {
     /** Natural language filter prompt (e.g., "posts created in the last 7 days") */
     prompt?: string;
-    /** Internal overrides for headers */
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
 /**
  * Options for the aggregate function.
@@ -60,10 +71,7 @@ export type AggregateOptions = {
     prompt?: string;
     /** Field name to aggregate on (required for sum, avg, min, max) */
     field?: string;
-    /** Internal overrides for headers */
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
 /**
  * Count items in a collection path. Returns a numeric result.
@@ -107,9 +115,7 @@ export declare function aggregate(path: string, operation: AggregateOperation, o
 export declare function get(path: string, opts?: GetOptions): Promise<any>;
 export type RunExpressionOptions = {
     returnType?: 'Bool' | 'String' | 'Int' | 'UInt';
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 };
 export type RunExpressionResult = {
     result: any;
@@ -131,9 +137,7 @@ export declare function runExpressionMany(many: {
     expression: string;
     queryArgs: any;
     returnType?: 'Bool' | 'String' | 'Int' | 'UInt';
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 }[]): Promise<RunExpressionResult[]>;
 export declare function set(path: string, document: any, options?: SetOptions): Promise<any>;
 export declare function setMany(many: {
@@ -144,14 +148,10 @@ export declare function clearCache(path?: string, opts?: {
     prompt?: string;
 }): void;
 export declare function getFiles(path: string, options?: {
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 }): Promise<any>;
 export declare function setFile(path: string, file: File | null, options?: {
-    _overrides?: {
-        headers?: Record<string, string>;
-    };
+    _overrides?: RequestOverrides;
 }): Promise<boolean>;
 export declare function signMessage(message: string): Promise<string>;
 export declare function signTransaction(transaction: Transaction | VersionedTransaction): Promise<Transaction | VersionedTransaction>;

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 export { init } from './client/config';
 export { getConfig, ClientConfig } from './client/config';
-export { get, set, setMany, setFile, getFiles, runQuery, runQueryMany, runExpression, runExpressionMany, signMessage, signTransaction, signAndSubmitTransaction, count, aggregate, SetOptions, GetOptions, CountOptions, AggregateOptions, AggregateOperation, AggregateResult, RunExpressionOptions, RunExpressionResult } from './client/operations';
+export { get, set, setMany, setFile, getFiles, runQuery, runQueryMany, runExpression, runExpressionMany, signMessage, signTransaction, signAndSubmitTransaction, count, aggregate, RequestOverrides, SetOptions, GetOptions, RunQueryOptions, CountOptions, AggregateOptions, AggregateOperation, AggregateResult, RunExpressionOptions, RunExpressionResult, InsufficientBalanceError } from './client/operations';
 export { subscribe, closeAllSubscriptions, clearCache, getCachedData, reconnectWithNewAuth } from './client/subscription';
 export * from './types';
 export { getIdToken } from './utils/utils';

package/dist/index.js

@@ -3047,31 +3047,39 @@ function loadKeypairFromEnv() {
     }
 }
 /* ------------------------------------------------------------------ */
-/*  REAL createSession()                                              */
-/* ------------------------------------------------------------------ */
-async function createSession() {
-    const kp = loadKeypairFromEnv();
-    const address = kp.publicKey.toBase58();
-    /* fetch nonce from auth API */
-    const nonce = await genAuthNonce();
-    const message = await genSolanaMessage(address, nonce);
-    /* sign the message */
-    const sigBytes = nacl.sign.detached(new TextEncoder().encode(message), kp.secretKey);
-    const signature = bufferExports.Buffer.from(sigBytes).toString("base64");
-    /* call auth API */
-    const { accessToken, idToken, refreshToken, } = await createSessionWithSignature(address, message, signature);
-    return { address, accessToken, idToken, refreshToken };
-}
-/* ------------------------------------------------------------------ */
-/* SINGLETON                                                          */
+/* SESSION MANAGER                                                     */
 /* ------------------------------------------------------------------ */
 class ServerSessionManager {
-    /* Private constructor prevents external `new` */
-    constructor() {
-        /* Private cache (lives for the life of the process) */
+    constructor(keypair) {
+        /* Private cache (lives for the life of the instance) */
         this.session = null;
         /* Coalesce concurrent getSession() calls into a single in-flight request */
         this.pendingSession = null;
+        this.keypair = keypair !== null && keypair !== void 0 ? keypair : null;
+    }
+    /**
+     * Create a session manager for a specific keypair.
+     * Each instance has its own independent session cache.
+     */
+    static forKeypair(keypair) {
+        return new ServerSessionManager(keypair);
+    }
+    /* ---------------------------------------------- *
+     * Session creation (instance method)
+     * ---------------------------------------------- */
+    async createSession() {
+        var _a;
+        const kp = (_a = this.keypair) !== null && _a !== void 0 ? _a : loadKeypairFromEnv();
+        const address = kp.publicKey.toBase58();
+        /* fetch nonce from auth API */
+        const nonce = await genAuthNonce();
+        const message = await genSolanaMessage(address, nonce);
+        /* sign the message */
+        const sigBytes = nacl.sign.detached(new TextEncoder().encode(message), kp.secretKey);
+        const signature = bufferExports.Buffer.from(sigBytes).toString("base64");
+        /* call auth API */
+        const { accessToken, idToken, refreshToken, } = await createSessionWithSignature(address, message, signature);
+        return { address, accessToken, idToken, refreshToken };
     }
     /* ---------------------------------------------- *
      * GET (lazy-fetch)
@@ -3085,7 +3093,7 @@ class ServerSessionManager {
         if (this.pendingSession !== null) {
             return this.pendingSession;
         }
-        this.pendingSession = createSession()
+        this.pendingSession = this.createSession()
             .then((session) => {
             this.session = session;
             return session;
@@ -3102,7 +3110,7 @@ class ServerSessionManager {
         this.session = session;
     }
     /* ---------------------------------------------- *
-     * CLEAR  (e.g. after logout)
+     * CLEAR  (e.g. after logout or 401 retry)
      * ---------------------------------------------- */
     clearSession() {
         this.session = null;
@@ -3123,7 +3131,7 @@ class ServerSessionManager {
         return (_b = (_a = this.session) === null || _a === void 0 ? void 0 : _a.refreshToken) !== null && _b !== void 0 ? _b : null;
     }
 }
-/* The single, shared instance */
+/* The default singleton instance (reads keypair from env) */
 ServerSessionManager.instance = new ServerSessionManager();
 
 var serverSessionManager = /*#__PURE__*/Object.freeze({
@@ -3238,7 +3246,11 @@ async function makeApiRequest(method, urlPath, data, _overrides) {
         ServerSessionManager.instance.clearSession();
     };
     async function executeRequest() {
-        const authHeader = await createAuthHeader(config.isServer);
+        // When _getAuthHeaders is provided (wallet client), use it as the sole auth source.
+        // Otherwise use the global createAuthHeader (default path).
+        const authHeader = (_overrides === null || _overrides === void 0 ? void 0 : _overrides._getAuthHeaders)
+            ? await _overrides._getAuthHeaders()
+            : await createAuthHeader(config.isServer);
         const headers = Object.assign({ "Content-Type": "application/json", "X-Public-App-Id": config.appId, "X-App-Id": config.appId }, authHeader);
         if (typeof window !== "undefined" &&
             window.CUSTOM_TAROBASE_APP_ID_HEADER) {
@@ -3267,6 +3279,13 @@ async function makeApiRequest(method, urlPath, data, _overrides) {
     }
     catch (error) {
         if (((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 401) {
+            // Wallet client path: use client-specific retry, skip global fallback entirely
+            if (_overrides === null || _overrides === void 0 ? void 0 : _overrides._clearAuth) {
+                await _overrides._clearAuth();
+                return await executeRequest();
+                // If this throws, let it propagate — do NOT clear global ServerSessionManager
+            }
+            // Default path (no wallet client): refresh global session
             try {
                 const refreshed = await refreshAuthSessionOnce(config.appId, config.isServer);
                 if (!refreshed) {
@@ -3275,8 +3294,7 @@ async function makeApiRequest(method, urlPath, data, _overrides) {
                 return await executeRequest();
             }
             catch (_refreshError) {
-                // Server-side fallback: clear cached session and retry once to force
-                // createSession() with a fresh Cognito session.
+                // Server-side fallback: clear global session and retry once
                 if (config.isServer && !hasRetriedAfterServerSessionReset) {
                     hasRetriedAfterServerSessionReset = true;
                     await clearServerSession();
@@ -3394,6 +3412,21 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
     return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
 };
 
+/**
+ * Thrown when a user's wallet doesn't have enough SOL to cover the transaction.
+ * Apps can catch this to trigger an onramp/funding flow.
+ */
+class InsufficientBalanceError extends Error {
+    constructor(address, balanceLamports, estimatedCostLamports, deficitLamports, deficitSol) {
+        super(`Insufficient balance: wallet ${address} needs ~${deficitSol.toFixed(6)} SOL more to complete this transaction.`);
+        this.name = 'InsufficientBalanceError';
+        this.address = address;
+        this.balanceLamports = balanceLamports;
+        this.estimatedCostLamports = estimatedCostLamports;
+        this.deficitLamports = deficitLamports;
+        this.deficitSol = deficitSol;
+    }
+}
 // Cache for get responses
 const getCache = {};
 // Track in-flight requests to coalesce multiple identical requests
@@ -3554,6 +3587,7 @@ async function aggregate(path, operation, opts = {}) {
     return { value: parseAggregateValue(result) };
 }
 async function get(path, opts = {}) {
+    var _a;
     try {
         let normalizedPath = path.startsWith("/") ? path.slice(1) : path;
         // Remove last '*' if it exists in the path
@@ -3563,12 +3597,13 @@ async function get(path, opts = {}) {
         if (!normalizedPath || normalizedPath.length === 0) {
             return new Error("Invalid path provided.");
         }
-        // Create cache key combining path, prompt, includeSubPaths, shape, limit, and cursor
+        // Create cache key combining path, prompt, includeSubPaths, shape, limit, cursor, and wallet
         const shapeKey = opts.shape ? JSON.stringify(opts.shape) : '';
         const includeSubPathsKey = opts.includeSubPaths ? ':subpaths' : '';
         const limitKey = opts.limit !== undefined ? `:l${opts.limit}` : '';
         const cursorKey = opts.cursor ? `:c${hashForKey$1(opts.cursor)}` : '';
-        const cacheKey = `${normalizedPath}:${opts.prompt || ''}${includeSubPathsKey}:${shapeKey}${limitKey}${cursorKey}`;
+        const walletSuffix = ((_a = opts._overrides) === null || _a === void 0 ? void 0 : _a._walletAddress) ? `:w${opts._overrides._walletAddress}` : '';
+        const cacheKey = `${normalizedPath}:${opts.prompt || ''}${includeSubPathsKey}:${shapeKey}${limitKey}${cursorKey}${walletSuffix}`;
         const now = Date.now();
         // Check for valid cache entry if not bypassing cache
         if (!opts.bypassCache && getCache[cacheKey] && now < getCache[cacheKey].expiresAt) {
@@ -3707,10 +3742,11 @@ async function set(path, document, options) {
     return result;
 }
 async function setMany(many, options) {
+    var _a, _b, _c, _d, _e, _f, _g;
     // Returns the data that was set, or undefined if the document was already set.
     try {
         const config = await getConfig();
-        const authProvider = config.authProvider;
+        const authProvider = (_b = (_a = options === null || options === void 0 ? void 0 : options._overrides) === null || _a === void 0 ? void 0 : _a.authProvider) !== null && _b !== void 0 ? _b : config.authProvider;
         if (!authProvider) {
             throw new Error('Auth provider not initialized');
         }
@@ -3741,7 +3777,18 @@ async function setMany(many, options) {
             }
             documents.push({ destinationPath, document });
         }
-        const setResponse = await makeApiRequest('PUT', `items`, { documents }, options === null || options === void 0 ? void 0 : options._overrides);
+        let setResponse;
+        try {
+            setResponse = await makeApiRequest('PUT', `items`, { documents }, options === null || options === void 0 ? void 0 : options._overrides);
+        }
+        catch (error) {
+            if ((error === null || error === void 0 ? void 0 : error.statusCode) === 402 && (error === null || error === void 0 ? void 0 : error.error) === 'INSUFFICIENT_BALANCE') {
+                const deficitLamports = Number((_c = error.deficitLamports) !== null && _c !== void 0 ? _c : 0);
+                const deficitSol = Number((_d = error.deficitSol) !== null && _d !== void 0 ? _d : deficitLamports / 1000000000);
+                throw new InsufficientBalanceError(String((_e = error.address) !== null && _e !== void 0 ? _e : ''), Number((_f = error.balanceLamports) !== null && _f !== void 0 ? _f : 0), Number((_g = error.estimatedCostLamports) !== null && _g !== void 0 ? _g : 0), deficitLamports, deficitSol);
+            }
+            throw error;
+        }
         // Invalidate caches and pending requests for affected paths
         pathsToInvalidate.forEach(path => {
             clearCacheByPrefix(path);
@@ -3782,7 +3829,7 @@ async function setMany(many, options) {
             else if (setResponse.data &&
                 typeof setResponse.data === 'object' &&
                 setResponse.data.success === true) {
-                const _a = setResponse.data, { success: _success } = _a, rest = __rest(_a, ["success"]);
+                const _h = setResponse.data, { success: _success } = _h, rest = __rest(_h, ["success"]);
                 return Object.assign(Object.assign(Object.assign({}, documents.map(d => d.document)), rest), { transactionId: null });
             }
             else {
@@ -3849,7 +3896,7 @@ async function setMany(many, options) {
         return transactionResult;
     }
     async function handleOffchainTransaction(tx, authProvider, options) {
-        var _a, _b, _c, _d, _e;
+        var _a, _b, _c, _d, _e, _f;
         const config = await getConfig();
         // 1. Sign the transaction message using mock signing for offchain transactions
         // Use signMessageMock if available (OffchainAuthProvider), otherwise fall back to signMessage
@@ -3868,32 +3915,18 @@ async function setMany(many, options) {
                 signedTransaction: bufferExports.Buffer.from(JSON.stringify(signedTx)).toString('base64')
             };
         }
-        // 4. Submit to RPC endpoint
-        // Use appId from headers if provided, otherwise fallback to config.appId
+        // 4. Submit to RPC endpoint via makeApiRequest (gets auth headers + 401 retry)
         const appId = ((_b = (_a = options === null || options === void 0 ? void 0 : options._overrides) === null || _a === void 0 ? void 0 : _a.headers) === null || _b === void 0 ? void 0 : _b['x-app-id']) || ((_d = (_c = options === null || options === void 0 ? void 0 : options._overrides) === null || _c === void 0 ? void 0 : _c.headers) === null || _d === void 0 ? void 0 : _d['X-App-Id']) || config.appId;
-        const rpcUrl = `${config.apiUrl}/app/${appId}/rpc`;
-        // Build headers, applying overrides if provided
-        const headers = {
-            'Content-Type': 'application/json',
-        };
-        if ((_e = options === null || options === void 0 ? void 0 : options._overrides) === null || _e === void 0 ? void 0 : _e.headers) {
-            Object.assign(headers, options._overrides.headers);
-        }
-        const rpcResponse = await fetch(rpcUrl, {
-            method: 'POST',
-            headers,
-            body: JSON.stringify({
-                jsonrpc: '2.0',
-                id: 1,
-                method: 'sendTransaction',
-                params: [bufferExports.Buffer.from(JSON.stringify(signedTx)).toString('base64')]
-            })
-        });
-        const result = await rpcResponse.json();
-        if (result.error) {
-            throw new Error(result.error.message);
+        const rpcResponse = await makeApiRequest('POST', `/app/${appId}/rpc`, {
+            jsonrpc: '2.0',
+            id: 1,
+            method: 'sendTransaction',
+            params: [bufferExports.Buffer.from(JSON.stringify(signedTx)).toString('base64')]
+        }, options === null || options === void 0 ? void 0 : options._overrides);
+        if ((_e = rpcResponse.data) === null || _e === void 0 ? void 0 : _e.error) {
+            throw new Error(rpcResponse.data.error.message);
         }
-        return { signature: result.result };
+        return { signature: (_f = rpcResponse.data) === null || _f === void 0 ? void 0 : _f.result };
     }
 }
 // Helper to clear cache entries by prefix
@@ -4726,6 +4759,7 @@ async function reconnectWithNewAuth() {
     return reconnectWithNewAuthV2();
 }
 
+exports.InsufficientBalanceError = InsufficientBalanceError;
 exports.ServerSessionManager = ServerSessionManager;
 exports.WebSessionManager = WebSessionManager;
 exports.aggregate = aggregate;