@reown/appkit-core-react-native 2.0.0-alpha.0 → 2.0.0-alpha.2

package/src/controllers/ConnectionsController.ts

@@ -1,119 +1,221 @@
 import { proxy, ref } from 'valtio';
-import { derive } from 'valtio/utils';
-import type {
-  AppKitNetwork,
-  BlockchainAdapter,
-  CaipAddress,
-  CaipNetworkId,
-  ChainNamespace,
-  GetBalanceResponse,
-  WalletInfo
+import { derive } from 'derive-valtio';
+import {
+  EVMAdapter,
+  type AppKitNetwork,
+  type BlockchainAdapter,
+  type CaipAddress,
+  type CaipNetworkId,
+  type ChainNamespace,
+  type Balance,
+  type WalletInfo,
+  type ConnectionProperties,
+  type AccountType,
+  type Connection,
+  SolanaBaseAdapter,
+  type Identity
 } from '@reown/appkit-common-react-native';
 import { StorageUtil } from '../utils/StorageUtil';
+import { BlockchainApiController } from './BlockchainApiController';
+import { SnackController } from './SnackController';
+import { OptionsController } from './OptionsController';
+import { CoreHelperUtil } from '../utils/CoreHelperUtil';
 
 // -- Types --------------------------------------------- //
-type Balance = GetBalanceResponse;
-
-//TODO: balance could be elsewhere
-interface Connection {
-  accounts: CaipAddress[];
-  balances: Record<CaipAddress, Balance>; //TODO: make this an array of balances
-  adapter: BlockchainAdapter;
-  chains: CaipNetworkId[];
-  activeChain: CaipNetworkId;
-  wallet?: WalletInfo;
-}
-
 export interface ConnectionsControllerState {
   activeNamespace?: ChainNamespace;
-  connections: Record<string, Connection>;
+  connections: Map<ChainNamespace, Connection>;
   networks: AppKitNetwork[];
 }
 
 // -- State --------------------------------------------- //
 const baseState = proxy<ConnectionsControllerState>({
   activeNamespace: undefined,
-  connections: {},
+  connections: new Map<ChainNamespace, Connection>(),
   networks: []
 });
 
+// -- Helper Functions --------------------------------------------- //
+const getActiveConnection = (snap: ConnectionsControllerState): Connection | undefined => {
+  if (!snap.activeNamespace) return undefined;
+
+  return snap.connections.get(snap.activeNamespace);
+};
+
+const hasValidAccounts = (connection: Connection): boolean => {
+  return connection?.accounts && connection.accounts.length > 0;
+};
+
+const findSmartAccountForNetwork = (connection: Connection): CaipAddress | undefined => {
+  return connection.properties?.smartAccounts?.find(account =>
+    account.startsWith(connection.caipNetwork)
+  );
+};
+
+const findEOAForNetwork = (connection: Connection): CaipAddress | undefined => {
+  const smartAccounts = connection.properties?.smartAccounts || [];
+
+  return connection.accounts.find(
+    account => account.startsWith(connection.caipNetwork) && !smartAccounts.includes(account)
+  );
+};
+
+const getActiveAddress = (connection: Connection): CaipAddress | undefined => {
+  if (!hasValidAccounts(connection)) {
+    return undefined;
+  }
+
+  // For smart accounts, prioritize smart account addresses
+  if (connection.type === 'smartAccount') {
+    const smartAccount = findSmartAccountForNetwork(connection);
+    if (smartAccount) {
+      return smartAccount;
+    }
+  }
+
+  // Fall back to EOA or any account that matches the network
+  return findEOAForNetwork(connection);
+};
+
+const updateConnection = (
+  namespace: ChainNamespace,
+  connection: Connection,
+  updates: Partial<Connection>
+) => {
+  if (!connection) return;
+  const newConnectionsMap = new Map(baseState.connections);
+  newConnectionsMap.set(namespace, { ...connection, ...updates });
+  baseState.connections = newConnectionsMap;
+};
+
+const getActiveIdentity = (connection: Connection): Identity | undefined => {
+  const activeAddress = getActiveAddress(connection);
+  if (!activeAddress) return undefined;
+
+  return connection.identities?.get(activeAddress);
+};
+
 const derivedState = derive(
   {
-    activeAddress: (get): CaipAddress | undefined => {
+    isConnected: (get): boolean => {
       const snap = get(baseState);
 
-      if (!snap.activeNamespace) return undefined;
-
-      const connection = snap.connections[snap.activeNamespace];
-
-      if (!connection || !connection.accounts || connection.accounts.length === 0) {
-        return undefined;
-      }
+      return !!snap.activeNamespace && !!snap.connections.size;
+    },
+    activeAddress: (get): CaipAddress | undefined => {
+      const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      //TODO: what happens if there are several accounts on the same chain?
-      const activeAccount = connection.accounts.find(account =>
-        account.startsWith(connection.activeChain)
-      );
+      return connection ? getActiveAddress(connection) : undefined;
+    },
+    identity: (get): Identity | undefined => {
+      const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      return activeAccount;
+      return connection ? getActiveIdentity(connection) : undefined;
     },
     activeBalance: (get): Balance | undefined => {
       const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      if (!snap.activeNamespace) return undefined;
-      const connection = snap.connections[snap.activeNamespace];
+      if (!connection) {
+        return undefined;
+      }
 
-      if (!connection || !connection.accounts || connection.accounts.length === 0) {
+      const activeAddress = getActiveAddress(connection);
+      if (!activeAddress || !connection.balances || connection.balances.size === 0) {
         return undefined;
       }
 
-      const activeAccount = connection.accounts.find(account =>
-        account.startsWith(connection.activeChain)
+      const addressBalances = connection.balances.get(activeAddress);
+      if (!addressBalances || addressBalances.length === 0) {
+        return undefined;
+      }
+
+      // Check if there's a specific token configured in OptionsController
+      const configuredTokens = OptionsController.state.tokens;
+      const activeNetwork = snap.networks.find(
+        network =>
+          network.chainNamespace === snap.activeNamespace &&
+          network.id?.toString() === connection.caipNetwork?.split(':')[1]
       );
 
-      if (
-        !connection ||
-        !connection.balances ||
-        !activeAccount ||
-        Object.keys(connection.balances).length === 0
-      ) {
-        return undefined;
+      if (configuredTokens && activeNetwork) {
+        const configuredToken = configuredTokens[activeNetwork.caipNetworkId];
+        if (configuredToken) {
+          // Find the configured token in the balances
+          const specificToken = addressBalances.find(
+            balance => balance.address === configuredToken.address
+          );
+          if (specificToken) {
+            return specificToken;
+          }
+        }
       }
 
-      return connection.balances[activeAccount];
+      // Return the native token (first balance without address)
+      const nativeToken = addressBalances.find(balance => !balance.address);
+      if (nativeToken) {
+        return nativeToken;
+      }
+
+      // Fallback to first available balance
+      return addressBalances[0];
     },
     activeNetwork: (get): AppKitNetwork | undefined => {
       const snap = get(baseState);
-
-      if (!snap.activeNamespace) return undefined;
-
-      const connection = snap.connections[snap.activeNamespace];
+      const connection = getActiveConnection(snap);
 
       if (!connection) return undefined;
 
       return snap.networks.find(
         network =>
           network.chainNamespace === snap.activeNamespace &&
-          network.id?.toString() === connection.activeChain?.split(':')[1]
+          network.id?.toString() === connection.caipNetwork?.split(':')[1]
       );
     },
     activeCaipNetworkId: (get): CaipNetworkId | undefined => {
       const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      if (!snap.activeNamespace) return undefined;
+      return connection?.caipNetwork;
+    },
+    accountType: (get): AccountType | undefined => {
+      const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      const connection = snap.connections[snap.activeNamespace];
+      return connection?.type;
+    },
+    connection: (get): Connection | undefined => {
+      const snap = get(baseState);
 
-      if (!connection) return undefined;
+      return getActiveConnection(snap);
+    },
+    balances: (get): Balance[] | undefined => {
+      const snap = get(baseState);
+
+      const _connection = getActiveConnection(snap);
+
+      if (!_connection) {
+        return undefined;
+      }
+
+      const _activeAddress = getActiveAddress(_connection);
 
-      return connection.activeChain;
+      if (!_activeAddress) return [];
+
+      return (
+        _connection?.balances
+          .get(_activeAddress)
+          // Filter out tokens with no quantity
+          ?.filter(balance => balance?.quantity?.numeric)
+      );
     },
     walletInfo: (get): WalletInfo | undefined => {
       const snap = get(baseState);
+      const connection = getActiveConnection(snap);
 
-      if (!snap.activeNamespace) return undefined;
-
-      return snap.connections[snap.activeNamespace]?.wallet;
+      return connection?.wallet;
     }
   },
   {
@@ -130,55 +232,127 @@ export const ConnectionsController = {
     StorageUtil.setActiveNamespace(namespace);
   },
 
-  storeConnection({
-    namespace,
-    adapter,
+  setConnection({
     accounts,
-    chains,
-    wallet,
-    activeChain
+    adapter,
+    caipNetwork,
+    namespace,
+    properties,
+    wallet
   }: {
-    namespace: string;
-    adapter: BlockchainAdapter;
     accounts: CaipAddress[];
-    chains: CaipNetworkId[];
+    adapter: BlockchainAdapter;
+    caipNetwork: CaipNetworkId;
+    namespace: ChainNamespace;
+    properties?: ConnectionProperties;
     wallet?: WalletInfo;
-    activeChain?: CaipNetworkId;
   }) {
-    baseState.connections[namespace] = {
-      balances: {},
-      activeChain: activeChain ?? chains[0]!,
+    const type: AccountType =
+      properties?.smartAccounts?.length &&
+      properties.smartAccounts.find(account => account.startsWith(caipNetwork))
+        ? 'smartAccount'
+        : 'eoa';
+
+    const newConnectionEntry: Connection = {
+      balances: new Map<CaipAddress, Balance[]>(),
+      caipNetwork,
       adapter: ref(adapter),
       accounts,
-      chains,
-      wallet
+      wallet,
+      properties,
+      type
     };
+
+    // Create a new Map to ensure Valtio detects the change
+    const newConnectionsMap = new Map(baseState.connections);
+    newConnectionsMap.set(namespace, newConnectionEntry);
+    baseState.connections = newConnectionsMap;
   },
 
-  updateAccounts(namespace: string, accounts: CaipAddress[]) {
-    const connection = baseState.connections[namespace];
+  updateAccounts(namespace: ChainNamespace, accounts: CaipAddress[]) {
+    const connection = baseState.connections.get(namespace);
     if (!connection) {
       return;
     }
-    connection.accounts = accounts;
+
+    const newConnectionsMap = new Map(baseState.connections);
+    const updatedConnection = { ...connection, accounts };
+    newConnectionsMap.set(namespace, updatedConnection);
+    baseState.connections = newConnectionsMap;
   },
 
-  updateBalance(namespace: string, address: CaipAddress, balance: Balance) {
-    const connection = baseState.connections[namespace];
+  updateBalance(namespace: ChainNamespace, address: CaipAddress, balance: Balance) {
+    const connection = baseState.connections.get(namespace);
     if (!connection) {
+      console.warn(`No connection found for namespace: ${namespace}`);
+
       return;
     }
-    connection.balances[address] = balance;
+    const newBalances = new Map(connection.balances);
+    const existingBalances = connection.balances.get(address) || [];
+    // Check if this token already exists by contract address or symbol
+    const existingIndex = existingBalances.findIndex(existingBalance => {
+      if (balance.address) {
+        return existingBalance.address === balance.address;
+      }
+
+      return existingBalance.symbol === balance.symbol;
+    });
+    let updatedBalances: Balance[];
+    if (existingIndex >= 0) {
+      // Update existing token
+      updatedBalances = [...existingBalances];
+      updatedBalances[existingIndex] = {
+        ...updatedBalances[existingIndex],
+        ...balance
+      };
+    } else {
+      // Add new token
+      updatedBalances = [...existingBalances, balance];
+    }
+    newBalances.set(address, updatedBalances);
+    updateConnection(namespace, connection, { balances: newBalances });
   },
 
-  setActiveChain(namespace: string, chain: CaipNetworkId) {
-    const connection = baseState.connections[namespace];
+  updateIdentity(
+    namespace: ChainNamespace,
+    connection: Connection,
+    plainAddress: string,
+    identity: Identity
+  ) {
+    const accounts = connection.accounts.filter(
+      account => CoreHelperUtil.getPlainAddress(account) === plainAddress
+    );
+
+    if (accounts.length > 0) {
+      const newIdentities = new Map(connection.identities || []);
+      let hasChanges = false;
+
+      accounts.forEach(account => {
+        const existingIdentity = newIdentities.get(account);
+        if (!existingIdentity || existingIdentity.name !== identity.name) {
+          newIdentities.set(account, identity);
+          hasChanges = true;
+        }
+      });
+
+      if (hasChanges) {
+        updateConnection(namespace, connection, { identities: newIdentities });
+      }
+    }
+  },
+
+  setActiveNetwork(namespace: ChainNamespace, networkId: CaipNetworkId) {
+    const connection = baseState.connections.get(namespace);
 
     if (!connection) {
       return;
     }
 
-    connection.activeChain = chain;
+    baseState.connections.set(namespace, {
+      ...connection,
+      caipNetwork: networkId
+    });
   },
 
   setNetworks(networks: AppKitNetwork[]) {
@@ -188,12 +362,24 @@ export const ConnectionsController = {
   getConnectedNetworks() {
     return baseState.networks.filter(
       network =>
-        baseState.connections[network.chainNamespace]?.chains.includes(network.caipNetworkId)
+        baseState.connections
+          .get(network.chainNamespace)
+          ?.accounts.some(account => account.startsWith(network.caipNetworkId))
     );
   },
 
-  async disconnect(namespace: string, isInternal = true) {
-    const connection = baseState.connections[namespace];
+  setAccountType(namespace: ChainNamespace, type: AccountType) {
+    const connection = baseState.connections.get(namespace);
+    if (!connection) return;
+
+    const newConnectionsMap = new Map(baseState.connections);
+    const newConnection = { ...connection, type };
+    newConnectionsMap.set(namespace, newConnection);
+    baseState.connections = newConnectionsMap;
+  },
+
+  async disconnect(namespace: ChainNamespace, isInternal = true) {
+    const connection = baseState.connections.get(namespace);
     if (!connection) return;
 
     // Get the current connector from the adapter
@@ -201,13 +387,13 @@ export const ConnectionsController = {
     if (!connector) return;
 
     // Find all namespaces that use the same connector
-    const namespacesUsingConnector = Object.keys(baseState.connections).filter(
-      ns => baseState.connections[ns]?.adapter.connector === connector
+    const namespacesUsingConnector = Array.from(baseState.connections.keys()).filter(
+      ns => baseState.connections.get(ns)?.adapter.connector === connector
     );
 
     // Unsubscribe all event listeners from the adapter
     namespacesUsingConnector.forEach(ns => {
-      const _connection = baseState.connections[ns];
+      const _connection = baseState.connections.get(ns);
       if (_connection?.adapter) {
         _connection.adapter.removeAllListeners();
       }
@@ -219,9 +405,11 @@ export const ConnectionsController = {
     }
 
     // Remove all namespaces that used this connector
+    const newConnectionsMap = new Map(baseState.connections);
     namespacesUsingConnector.forEach(ns => {
-      delete baseState.connections[ns];
+      newConnectionsMap.delete(ns);
     });
+    baseState.connections = newConnectionsMap;
 
     // Remove activeNamespace if it is in the list of namespaces using the connector
     if (
@@ -232,5 +420,124 @@ export const ConnectionsController = {
       baseState.activeNamespace = undefined;
       StorageUtil.setActiveNamespace(undefined);
     }
+  },
+
+  parseUnits(value: string, decimals: number) {
+    if (!baseState.activeNamespace) return undefined;
+
+    return baseState.connections
+      .get(baseState.activeNamespace)
+      ?.adapter.parseUnits(value, decimals);
+  },
+
+  async signMessage(address: CaipAddress, message: string) {
+    if (!baseState.activeNamespace) return undefined;
+
+    const [namespace, chainId, plainAddress] = address.split(':');
+
+    if (!namespace || namespace !== baseState.activeNamespace || !chainId || !plainAddress) {
+      return undefined;
+    }
+
+    const adapter = baseState.connections.get(baseState.activeNamespace)?.adapter;
+
+    if (adapter instanceof EVMAdapter && plainAddress && chainId) {
+      return adapter.signMessage(plainAddress, message, chainId);
+    }
+
+    return undefined;
+  },
+
+  async sendTransaction(args: any) {
+    if (!baseState.activeNamespace) return undefined;
+
+    const adapter = baseState.connections.get(baseState.activeNamespace)?.adapter;
+    if (adapter instanceof EVMAdapter || adapter instanceof SolanaBaseAdapter) {
+      return adapter.sendTransaction(args);
+    }
+
+    return undefined;
+  },
+
+  async estimateGas(args: any) {
+    if (!baseState.activeNamespace || baseState.activeNamespace !== 'eip155') return undefined;
+
+    const adapter = baseState.connections.get(baseState.activeNamespace)?.adapter;
+    if (adapter instanceof EVMAdapter) {
+      return adapter.estimateGas(args);
+    }
+
+    return undefined;
+  },
+
+  async writeContract(args: any) {
+    if (!baseState.activeNamespace) return undefined;
+
+    const adapter = baseState.connections.get(baseState.activeNamespace)?.adapter;
+    if (adapter instanceof EVMAdapter) {
+      return adapter.writeContract(args);
+    }
+
+    return undefined;
+  },
+
+  async fetchBalance(forceUpdateAddresses?: CaipAddress[]) {
+    const connection = getActiveConnection(baseState);
+    if (!connection) {
+      throw new Error('No active connection found for balance fetch');
+    }
+
+    const chainId = connection.caipNetwork;
+    const address = getActiveAddress(connection);
+    const namespace = baseState.activeNamespace;
+    if (!namespace || !address || !chainId) {
+      throw new Error('Missing required data for balance fetch');
+    }
+
+    try {
+      const response = await BlockchainApiController.getBalance(address, forceUpdateAddresses);
+      if (!response) {
+        throw new Error('Failed to fetch token balance');
+      }
+      // Update balances for each token in the response
+      response.balances.forEach(balance => {
+        this.updateBalance(namespace, address, {
+          name: balance.name,
+          symbol: balance.symbol,
+          amount: balance.quantity.numeric,
+          address: balance.address,
+          quantity: balance.quantity,
+          price: balance.price,
+          value: balance.value,
+          iconUrl: balance.iconUrl
+        });
+      });
+    } catch (error) {
+      SnackController.showError('Failed to get account balance');
+    }
+  },
+
+  getSmartAccountEnabledNetworks(): AppKitNetwork[] {
+    const activeConnection = getActiveConnection(baseState);
+    if (!activeConnection) {
+      return [];
+    }
+    if (!activeConnection.properties?.smartAccounts?.length) {
+      return [];
+    }
+    const smartAccountNetworks = new Set<CaipNetworkId>();
+    activeConnection.properties.smartAccounts.forEach(smartAccount => {
+      const parts = smartAccount.split(':');
+      if (parts.length >= 2) {
+        const networkId: CaipNetworkId = `${parts[0]}:${parts[1]}`;
+        smartAccountNetworks.add(networkId);
+      }
+    });
+
+    return baseState.networks.filter(network => {
+      const networkId: CaipNetworkId = `${network.chainNamespace}:${network.id}`;
+
+      return smartAccountNetworks.has(networkId);
+    });
   }
 };

package/src/controllers/EnsController.ts

@@ -1,5 +1,5 @@
-import { subscribeKey as subKey } from 'valtio/vanilla/utils';
-import { proxy, subscribe as sub } from 'valtio/vanilla';
+import { subscribeKey as subKey } from 'valtio/utils';
+import { proxy, subscribe as sub } from 'valtio';
 import { BlockchainApiController } from './BlockchainApiController';
 import type { BlockchainApiEnsError } from '../utils/TypeUtil';
 

package/src/controllers/EventsController.ts

@@ -1,4 +1,4 @@
-import { proxy, subscribe as sub } from 'valtio/vanilla';
+import { proxy, subscribe as sub } from 'valtio';
 import { ApiController } from './ApiController';
 import { OptionsController } from './OptionsController';
 import { CoreHelperUtil } from '../utils/CoreHelperUtil';

package/src/controllers/ModalController.ts

@@ -1,11 +1,9 @@
 import { proxy } from 'valtio';
-import { AccountController } from './AccountController';
-import type { RouterControllerState } from './RouterController';
-import { RouterController } from './RouterController';
+import { RouterController, type RouterControllerState } from './RouterController';
 import { PublicStateController } from './PublicStateController';
 import { EventsController } from './EventsController';
 import { ApiController } from './ApiController';
-import { ConnectorController } from './ConnectorController';
+import { ConnectionsController } from './ConnectionsController';
 
 // -- Types --------------------------------------------- //
 export interface ModalControllerState {
@@ -31,11 +29,11 @@ export const ModalController = {
 
   async open(options?: ModalControllerArguments['open']) {
     await ApiController.state.prefetchPromise;
-    const connected = AccountController.state.isConnected;
+    const isConnected = ConnectionsController.state.isConnected;
     if (options?.view) {
       RouterController.reset(options.view);
-    } else if (AccountController.state.isConnected) {
-      const isUniversalWallet = ConnectorController.state.connectedConnector === 'AUTH';
+    } else if (isConnected) {
+      const isUniversalWallet = !!ConnectionsController.state.connection?.properties?.provider;
       RouterController.reset(isUniversalWallet ? 'Account' : 'AccountDefault');
     } else {
       RouterController.reset('Connect');
@@ -45,12 +43,12 @@ export const ModalController = {
     EventsController.sendEvent({
       type: 'track',
       event: 'MODAL_OPEN',
-      properties: { connected }
+      properties: { connected: isConnected }
     });
   },
 
   close() {
-    const connected = AccountController.state.isConnected;
+    const connected = ConnectionsController.state.isConnected;
     state.open = false;
     PublicStateController.set({ open: false });
     EventsController.sendEvent({