@aomi-labs/react 0.3.10 → 0.3.12

package/dist/index.d.cts

@@ -135,10 +135,12 @@ type WalletHandlerApi = {
     pendingRequests: WalletRequest[];
     /** Replace pending requests with the session's authoritative snapshot. */
     setRequests: (requests: WalletRequest[]) => void;
+    /** Mark a request as in-flight so it is not replayed while awaiting backend ack. */
+    startRequest: (id: string) => void;
     /** Complete a request successfully — sends response to backend via ClientSession */
-    resolveRequest: (id: string, result: WalletRequestResult) => void;
+    resolveRequest: (id: string, result: WalletRequestResult) => Promise<void>;
     /** Fail a request — sends error to backend via ClientSession */
-    rejectRequest: (id: string, error?: string) => void;
+    rejectRequest: (id: string, error?: string) => Promise<void>;
 };
 declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
@@ -193,10 +195,10 @@ type AomiRuntimeApi = {
     pendingWalletRequests: WalletRequest[];
     /** Mark a wallet request as being processed */
     startWalletRequest: (id: string) => void;
-    /** Complete a wallet request — dequeues + sends response to backend */
-    resolveWalletRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a wallet request — dequeues + sends error to backend */
-    rejectWalletRequest: (id: string, error?: string) => void;
+    /** Complete a wallet request after the backend acknowledges the response */
+    resolveWalletRequest: (id: string, result: WalletRequestResult) => Promise<void>;
+    /** Fail a wallet request after the backend acknowledges the error */
+    rejectWalletRequest: (id: string, error?: string) => Promise<void>;
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */

package/dist/index.d.ts

@@ -135,10 +135,12 @@ type WalletHandlerApi = {
     pendingRequests: WalletRequest[];
     /** Replace pending requests with the session's authoritative snapshot. */
     setRequests: (requests: WalletRequest[]) => void;
+    /** Mark a request as in-flight so it is not replayed while awaiting backend ack. */
+    startRequest: (id: string) => void;
     /** Complete a request successfully — sends response to backend via ClientSession */
-    resolveRequest: (id: string, result: WalletRequestResult) => void;
+    resolveRequest: (id: string, result: WalletRequestResult) => Promise<void>;
     /** Fail a request — sends error to backend via ClientSession */
-    rejectRequest: (id: string, error?: string) => void;
+    rejectRequest: (id: string, error?: string) => Promise<void>;
 };
 declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
@@ -193,10 +195,10 @@ type AomiRuntimeApi = {
     pendingWalletRequests: WalletRequest[];
     /** Mark a wallet request as being processed */
     startWalletRequest: (id: string) => void;
-    /** Complete a wallet request — dequeues + sends response to backend */
-    resolveWalletRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a wallet request — dequeues + sends error to backend */
-    rejectWalletRequest: (id: string, error?: string) => void;
+    /** Complete a wallet request after the backend acknowledges the response */
+    resolveWalletRequest: (id: string, result: WalletRequestResult) => Promise<void>;
+    /** Fail a wallet request after the backend acknowledges the error */
+    rejectWalletRequest: (id: string, error?: string) => Promise<void>;
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */

package/dist/index.js

@@ -1143,14 +1143,28 @@ function useRuntimeOrchestrator(aomiClient, options) {
     (threadId) => {
       var _a, _b, _c, _d, _e;
       const manager = sessionManagerRef.current;
+      const nextApp = options.getApp();
+      const nextPublicKey = (_a = options.getPublicKey) == null ? void 0 : _a.call(options);
+      const nextApiKey = (_c = (_b = options.getApiKey) == null ? void 0 : _b.call(options)) != null ? _c : void 0;
+      const nextClientId = (_d = options.getClientId) == null ? void 0 : _d.call(options);
+      const nextUserState = (_e = options.getUserState) == null ? void 0 : _e.call(options);
       const existing = manager.get(threadId);
-      if (existing) return existing;
+      if (existing) {
+        existing.syncRuntimeOptions({
+          app: nextApp,
+          publicKey: nextPublicKey,
+          apiKey: nextApiKey,
+          clientId: nextClientId,
+          userState: nextUserState
+        });
+        return existing;
+      }
       const session = manager.getOrCreate(threadId, {
-        app: options.getApp(),
-        publicKey: (_a = options.getPublicKey) == null ? void 0 : _a.call(options),
-        apiKey: (_c = (_b = options.getApiKey) == null ? void 0 : _b.call(options)) != null ? _c : void 0,
-        clientId: (_d = options.getClientId) == null ? void 0 : _d.call(options),
-        userState: (_e = options.getUserState) == null ? void 0 : _e.call(options)
+        app: nextApp,
+        publicKey: nextPublicKey,
+        apiKey: nextApiKey,
+        clientId: nextClientId,
+        userState: nextUserState
       });
       const cleanups = [];
       cleanups.push(
@@ -1442,41 +1456,84 @@ function useWalletHandler({
 }) {
   const [pendingRequests, setPendingRequests] = useState5([]);
   const requestsRef = useRef6(pendingRequests);
-  const setRequests = useCallback6((requests) => {
-    requestsRef.current = [...requests];
-    setPendingRequests(requestsRef.current);
+  const inFlightRequestSetRef = useRef6(/* @__PURE__ */ new Set());
+  const suppressedRequestSetRef = useRef6(/* @__PURE__ */ new Set());
+  const syncVisibleRequests = useCallback6(() => {
+    setPendingRequests(
+      requestsRef.current.filter(
+        (request) => !suppressedRequestSetRef.current.has(request.id)
+      )
+    );
   }, []);
+  const setRequests = useCallback6((requests) => {
+    const incomingIds = new Set(requests.map((request) => request.id));
+    for (const id of suppressedRequestSetRef.current) {
+      if (!incomingIds.has(id) && !inFlightRequestSetRef.current.has(id)) {
+        suppressedRequestSetRef.current.delete(id);
+      }
+    }
+    const preservedInFlight = requestsRef.current.filter(
+      (request) => inFlightRequestSetRef.current.has(request.id) && !incomingIds.has(request.id)
+    );
+    requestsRef.current = [...requests, ...preservedInFlight];
+    syncVisibleRequests();
+  }, [syncVisibleRequests]);
+  const startRequest = useCallback6((id) => {
+    if (!requestsRef.current.some((request) => request.id === id)) {
+      return;
+    }
+    inFlightRequestSetRef.current.add(id);
+    suppressedRequestSetRef.current.add(id);
+    syncVisibleRequests();
+  }, [syncVisibleRequests]);
   const resolveRequest = useCallback6(
-    (id, result) => {
+    async (id, result) => {
       const session = getSession();
       if (!session) {
         console.error("[wallet-handler] No session available to resolve request");
         return;
       }
-      setRequests(requestsRef.current.filter((request) => request.id !== id));
-      void session.resolve(id, result).catch((err) => {
+      startRequest(id);
+      try {
+        await session.resolve(id, result);
+      } catch (err) {
         console.error("[wallet-handler] Failed to resolve request:", err);
-      });
+      } finally {
+        requestsRef.current = requestsRef.current.filter(
+          (request) => request.id !== id
+        );
+        inFlightRequestSetRef.current.delete(id);
+        syncVisibleRequests();
+      }
     },
-    [getSession, setRequests]
+    [getSession, startRequest, syncVisibleRequests]
   );
   const rejectRequest = useCallback6(
-    (id, error) => {
+    async (id, error) => {
       const session = getSession();
       if (!session) {
         console.error("[wallet-handler] No session available to reject request");
         return;
       }
-      setRequests(requestsRef.current.filter((request) => request.id !== id));
-      void session.reject(id, error).catch((err) => {
+      startRequest(id);
+      try {
+        await session.reject(id, error);
+      } catch (err) {
         console.error("[wallet-handler] Failed to reject request:", err);
-      });
+      } finally {
+        requestsRef.current = requestsRef.current.filter(
+          (request) => request.id !== id
+        );
+        inFlightRequestSetRef.current.delete(id);
+        syncVisibleRequests();
+      }
     },
-    [getSession, setRequests]
+    [getSession, startRequest, syncVisibleRequests]
   );
   return {
     pendingRequests,
     setRequests,
+    startRequest,
     resolveRequest,
     rejectRequest
   };
@@ -1822,9 +1879,7 @@ function AomiRuntimeCore({
       clearAllNotifications: notificationContext.clearAll,
       // Wallet API
       pendingWalletRequests: walletHandler.pendingRequests,
-      startWalletRequest: () => {
-      },
-      // No-op: ClientSession manages processing state
+      startWalletRequest: walletHandler.startRequest,
       resolveWalletRequest: walletHandler.resolveRequest,
       rejectWalletRequest: walletHandler.rejectRequest,
       // Event API