aomi-privy-provider

1"use client";
2 
3import {
4  useCallback,
5  useEffect,
6  useMemo,
7  useState,
8  type ReactNode,
9} from "react";
10import { serializeSignature, type Chain, type Hex } from "viem";
11import type {
12  WalletAaSignPayload,
13  WalletEip712Payload,
14} from "@aomi-labs/react";
15import { toViemSignTypedDataArgs } from "@aomi-labs/react";
16import { AomiWalletKitComposer } from "../../composer/AomiWalletKitComposer";
17import type { AuthRuntime, ExecutionRuntime } from "../../composer/types";
18import { useResolvedAccountRuntime } from "../../account/use-resolved-account-runtime";
19import { buildEvmExecutionRuntime } from "../../execution/execution-runtime";
20import { useAomiWalletNetworkPreferences } from "../../network-preferences";
21import { useEvmWalletRuntime } from "../../runtime/evm/wallet-runtime";
22import { useSvmWalletRuntime } from "../../runtime/svm/wallet-runtime";
23import { REGISTRY_STORAGE_KEY } from "../../registry/types";
24import type { AomiAccount, AomiAccountCredential } from "../../types";
25import type { AccountConfig, ExecutionConfig } from "../../config/types";
26import {
27  inferPrivyAuthMethod,
28  inferPrivyPrimaryLabel,
29  pickPrivyEmbeddedEvmWallet,
30  pickPrivyEmbeddedEvmUserWallet,
31  privyLoginMethodsToOptions,
32  useSafePrivy,
33  useSafeSmartWallets,
34  useSafeSvmWallets,
35  useSafeWallets,
36} from "./privy-auth";
37import {
38  useSign7702Authorization,
39  type PrivyClientConfig,
40} from "@privy-io/react-auth";
41import { buildPrivySvmWalletState } from "./privy-svm";
42import { sendPrivySmartWalletTransaction } from "./privy-execution";
43import { useEmbeddedSessionSource } from "../sources/embedded-session-source";
44 
45export type AomiPrivyPluginProviderProps = {
46  children: ReactNode;
47  supportedChains: readonly Chain[];
48  loginMethods?: PrivyClientConfig["loginMethods"];
49  execution?: ExecutionConfig;
50  account?: AccountConfig;
51  preferDirectSend?: boolean;
52};
53 
54export function AomiPrivyPluginProvider({
55  children,
56  supportedChains,
57  loginMethods,
58  execution,
59  account,
60  preferDirectSend = true,
61}: AomiPrivyPluginProviderProps) {
62  const privy = useSafePrivy();
63  const { client: smartWalletClient, getClientForChain } =
64    useSafeSmartWallets();
65  const { wallets: solanaWallets } = useSafeSvmWallets();
66  const { wallets: connectedWallets } = useSafeWallets();
67  const { signAuthorization } = useSign7702Authorization();
68  const [activeSolanaAddress, setActiveSolanaAddress] = useState<
69    string | undefined
70  >();
71  const {
72    selectedEvmChainId,
73    selectedSolanaNetwork,
74// … truncated