@aomi-labs/react 0.3.2 → 0.3.3

package/dist/index.d.cts

@@ -16,10 +16,13 @@ type UserState = {
     chainId?: number;
     isConnected: boolean;
     ensName?: string;
+    ext?: Record<string, unknown>;
 };
 declare function useUser(): {
     user: UserState;
     setUser: (data: Partial<UserState>) => void;
+    addExtValue: (key: string, value: unknown) => void;
+    removeExtValue: (key: string) => void;
     getUserState: () => UserState;
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
 };
@@ -166,6 +169,10 @@ type AomiRuntimeApi = {
     getUserState: () => UserState;
     /** Update user state (partial updates merged with existing state) */
     setUser: (data: Partial<UserState>) => void;
+    /** Add or overwrite a value in user_state.ext */
+    addExtValue: (key: string, value: unknown) => void;
+    /** Remove a value from user_state.ext */
+    removeExtValue: (key: string) => void;
     /** Subscribe to user state changes. Returns unsubscribe function. */
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
     /** ID of the currently active thread */

package/dist/index.d.ts

@@ -16,10 +16,13 @@ type UserState = {
     chainId?: number;
     isConnected: boolean;
     ensName?: string;
+    ext?: Record<string, unknown>;
 };
 declare function useUser(): {
     user: UserState;
     setUser: (data: Partial<UserState>) => void;
+    addExtValue: (key: string, value: unknown) => void;
+    removeExtValue: (key: string) => void;
     getUserState: () => UserState;
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
 };
@@ -166,6 +169,10 @@ type AomiRuntimeApi = {
     getUserState: () => UserState;
     /** Update user state (partial updates merged with existing state) */
     setUser: (data: Partial<UserState>) => void;
+    /** Add or overwrite a value in user_state.ext */
+    addExtValue: (key: string, value: unknown) => void;
+    /** Remove a value from user_state.ext */
+    removeExtValue: (key: string) => void;
     /** Subscribe to user state changes. Returns unsubscribe function. */
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
     /** ID of the currently active thread */

package/dist/index.js

@@ -832,6 +832,8 @@ function useUser() {
   return {
     user: context.user,
     setUser: context.setUser,
+    addExtValue: context.addExtValue,
+    removeExtValue: context.removeExtValue,
     getUserState: context.getUserState,
     onUserStateChange: context.onUserStateChange
   };
@@ -841,7 +843,8 @@ function UserContextProvider({ children }) {
     isConnected: false,
     address: void 0,
     chainId: void 0,
-    ensName: void 0
+    ensName: void 0,
+    ext: void 0
   });
   const userRef = useRef4(user);
   userRef.current = user;
@@ -857,6 +860,36 @@ function UserContextProvider({ children }) {
       return next;
     });
   }, []);
+  const addExtValue = useCallback4((key, value) => {
+    setUserState((prev) => {
+      var _a;
+      const next = __spreadProps(__spreadValues({}, prev), {
+        ext: __spreadProps(__spreadValues({}, (_a = prev.ext) != null ? _a : {}), {
+          [key]: value
+        })
+      });
+      StateChangeCallbacks.current.forEach((callback) => {
+        callback(next);
+      });
+      return next;
+    });
+  }, []);
+  const removeExtValue = useCallback4((key) => {
+    setUserState((prev) => {
+      if (!prev.ext || !(key in prev.ext)) {
+        return prev;
+      }
+      const nextExt = __spreadValues({}, prev.ext);
+      delete nextExt[key];
+      const next = __spreadProps(__spreadValues({}, prev), {
+        ext: Object.keys(nextExt).length > 0 ? nextExt : void 0
+      });
+      StateChangeCallbacks.current.forEach((callback) => {
+        callback(next);
+      });
+      return next;
+    });
+  }, []);
   const getUserState = useCallback4(() => userRef.current, []);
   const onUserStateChange = useCallback4(
     (callback) => {
@@ -873,6 +906,8 @@ function UserContextProvider({ children }) {
       value: {
         user,
         setUser,
+        addExtValue,
+        removeExtValue,
         getUserState,
         onUserStateChange
       },
@@ -1596,22 +1631,40 @@ function AomiRuntimeCore({
     getApp: getCurrentThreadApp,
     getApiKey: () => getControlState().apiKey
   });
+  const walletSnapshot = useCallback7(
+    (nextUser) => ({
+      address: nextUser.address,
+      chainId: nextUser.chainId,
+      isConnected: nextUser.isConnected,
+      ensName: nextUser.ensName
+    }),
+    [getUserState]
+  );
+  const lastWalletStateRef = useRef7(walletSnapshot(getUserState()));
   useEffect4(() => {
+    lastWalletStateRef.current = walletSnapshot(getUserState());
     const unsubscribe = onUserStateChange(async (newUser) => {
+      const nextWalletState = walletSnapshot(newUser);
+      const prevWalletState = lastWalletStateRef.current;
+      if (prevWalletState.address === nextWalletState.address && prevWalletState.chainId === nextWalletState.chainId && prevWalletState.isConnected === nextWalletState.isConnected && prevWalletState.ensName === nextWalletState.ensName) {
+        return;
+      }
+      lastWalletStateRef.current = nextWalletState;
       const sessionId = threadContext.currentThreadId;
       const message = JSON.stringify({
         type: "wallet:state_changed",
-        payload: {
-          address: newUser.address,
-          chainId: newUser.chainId,
-          isConnected: newUser.isConnected,
-          ensName: newUser.ensName
-        }
+        payload: nextWalletState
       });
       await aomiClientRef.current.sendSystemMessage(sessionId, message);
     });
     return unsubscribe;
-  }, [onUserStateChange, aomiClientRef, threadContext.currentThreadId]);
+  }, [
+    onUserStateChange,
+    aomiClientRef,
+    threadContext.currentThreadId,
+    getUserState,
+    walletSnapshot
+  ]);
   const threadContextRef = useRef7(threadContext);
   threadContextRef.current = threadContext;
   const currentThreadIdRef = useRef7(threadContext.currentThreadId);
@@ -1867,6 +1920,8 @@ function AomiRuntimeCore({
       user: userContext.user,
       getUserState: userContext.getUserState,
       setUser: userContext.setUser,
+      addExtValue: userContext.addExtValue,
+      removeExtValue: userContext.removeExtValue,
       onUserStateChange: userContext.onUserStateChange,
       // Thread API
       currentThreadId: threadContext.currentThreadId,