aomi-wallet-kit

1"use client";
2 
3import { createContext, useContext, useEffect, type ReactNode } from "react";
4import { useUser } from "@aomi-labs/react";
5import { AOMI_SESSION_DISCONNECTED_IDENTITY } from "./identity";
6import type { AomiWalletKit } from "./types";
7 
8const DISCONNECTED_WALLET_KIT: AomiWalletKit = {
9  identity: AOMI_SESSION_DISCONNECTED_IDENTITY,
10  isReady: true,
11  isSwitchingChain: false,
12  canConnect: false,
13  canOpenAccountUI: false,
14  canDisconnect: false,
15  accounts: [],
16  selectAccount: async () => undefined,
17  evmWallets: [],
18  solanaWallets: [],
19  socialLoginOptions: [],
20  supportedNetworks: {
21    evm: [],
22    solana: [],
23  },
24  connect: async () => undefined,
25};
26 
27const AomiWalletKitContext =
28  createContext<AomiWalletKit>(DISCONNECTED_WALLET_KIT);
29 
30function toSvmCapabilities(
31  capabilities: AomiWalletKit["identity"]["svmCapabilities"],
32): string[] | undefined {
33  if (!capabilities) return undefined;
34  return Object.entries(capabilities)
35    .filter(([, enabled]) => enabled)
36    .map(([name]) =>
37      name.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`),
38    );
39}
40 
41function AomiWalletKitSync({ walletKit }: { walletKit: AomiWalletKit }) {
42  const { setUser } = useUser();
43  const identity = walletKit.identity;
44 
45  useEffect(() => {
46    // NOTE: aaMode / SmartAccount4337 / Delegation7702 are NOT forwarded
47    // here. They are session-owned: `session.ts` writes them on tx-complete
48    // and providers read them back via `useUser()`. Forwarding them from
49    // identity would create a write loop (UserState -> identity -> setUser
50    // -> UserState). walletKind is provider-static and forwarded normally.
51    setUser({
52      address: identity.address ?? undefined,
53      walletKind: identity.walletKind ?? undefined,
54      chainId: identity.chainId ?? undefined,
55      isConnected: identity.isConnected,
56      svm: {
57        address: identity.svmAddress ?? null,
58        cluster: identity.svmCluster ?? undefined,
59        wallet_name: identity.svmWalletName ?? null,
60        transport: identity.svmTransport ?? null,
61        capabilities: toSvmCapabilities(identity.svmCapabilities) ?? [],
62      },
63      walletProvider: identity.isConnected
64        ? (identity.sessionProvider ??
65          identity.embeddedProvider ??
66          identity.walletProvider ??
67          null)
68        : null,
69      walletProviderSubject: identity.isConnected
70        ? (identity.walletProviderSubject ?? null)
71        : null,
72// … truncated