|
1 | 1 | import type { CaipAccount } from '@rango-dev/wallets-core/namespaces/common'; |
2 | 2 | import type { SolanaActions } from '@rango-dev/wallets-core/namespaces/solana'; |
| 3 | +import type { PublicKey } from '@solana/web3.js'; |
3 | 4 |
|
4 | | -import { ActionBuilder, NamespaceBuilder } from '@rango-dev/wallets-core'; |
| 5 | +import { NamespaceBuilder } from '@rango-dev/wallets-core'; |
5 | 6 | import { builders as commonBuilders } from '@rango-dev/wallets-core/namespaces/common'; |
6 | 7 | import { |
7 | 8 | builders, |
8 | 9 | CAIP_NAMESPACE, |
9 | 10 | CAIP_SOLANA_CHAIN_ID, |
10 | 11 | } from '@rango-dev/wallets-core/namespaces/solana'; |
11 | 12 | import { CAIP } from '@rango-dev/wallets-core/utils'; |
| 13 | +import { |
| 14 | + type WalletError, |
| 15 | + WalletReadyState, |
| 16 | +} from '@solana/wallet-adapter-base'; |
12 | 17 |
|
13 | 18 | import { WALLET_ID } from '../constants.js'; |
14 | 19 | import { mobileWalletAdapter } from '../utils.js'; |
15 | 20 |
|
16 | 21 | const connect = builders |
17 | 22 | .connect() |
18 | 23 | .action(async function () { |
19 | | - const phantomInstance = mobileWalletAdapter(); |
20 | | - await phantomInstance.connect(); |
| 24 | + const adapterInstance = mobileWalletAdapter(); |
| 25 | + |
| 26 | + const connectHandler = new Promise((resolve, reject) => { |
| 27 | + const handleConnect = (publicKey: PublicKey) => { |
| 28 | + const account = publicKey?.toString(); |
21 | 29 |
|
22 | | - const account = phantomInstance.publicKey?.toString(); |
23 | | - if (!account) { |
24 | | - throw new Error('Connected account is not found!'); |
| 30 | + if (!account) { |
| 31 | + return reject(new Error('Connected account is not found!')); |
| 32 | + } |
| 33 | + return resolve([ |
| 34 | + CAIP.AccountId.format({ |
| 35 | + address: account, |
| 36 | + chainId: { |
| 37 | + namespace: CAIP_NAMESPACE, |
| 38 | + reference: CAIP_SOLANA_CHAIN_ID, |
| 39 | + }, |
| 40 | + }) as CaipAccount, |
| 41 | + ]); |
| 42 | + }; |
| 43 | + const handleError = (error: WalletError) => { |
| 44 | + return reject(error); |
| 45 | + }; |
| 46 | + adapterInstance.on('connect', handleConnect); |
| 47 | + adapterInstance.on('error', handleError); |
| 48 | + }); |
| 49 | + |
| 50 | + if ( |
| 51 | + adapterInstance.readyState === WalletReadyState.Loadable || |
| 52 | + adapterInstance.connected === true |
| 53 | + ) { |
| 54 | + await adapterInstance.disconnect(); |
25 | 55 | } |
26 | | - return [ |
27 | | - CAIP.AccountId.format({ |
28 | | - address: account, |
29 | | - chainId: { |
30 | | - namespace: CAIP_NAMESPACE, |
31 | | - reference: CAIP_SOLANA_CHAIN_ID, |
32 | | - }, |
33 | | - }) as CaipAccount, |
34 | | - ]; |
| 56 | + await adapterInstance.connect(); |
| 57 | + return connectHandler; |
35 | 58 | }) |
36 | 59 | .build(); |
37 | 60 |
|
38 | | -const disconnect = commonBuilders.disconnect<SolanaActions>().build(); |
39 | | - |
40 | | -export const canEagerConnectAction = async () => { |
41 | | - const phantomInstance = mobileWalletAdapter(); |
42 | | - try { |
43 | | - await phantomInstance.autoConnect(); |
44 | | - } catch { |
45 | | - return false; |
46 | | - } |
47 | | -}; |
48 | | - |
49 | | -const canEagerConnect = new ActionBuilder<SolanaActions, 'canEagerConnect'>( |
50 | | - 'canEagerConnect' |
51 | | -) |
52 | | - .action(canEagerConnectAction) |
| 61 | +const disconnect = commonBuilders |
| 62 | + .disconnect<SolanaActions>() |
| 63 | + .before(async () => { |
| 64 | + const adapterInstance = mobileWalletAdapter(); |
| 65 | + await adapterInstance.disconnect(); |
| 66 | + }) |
53 | 67 | .build(); |
54 | 68 |
|
55 | 69 | const solana = new NamespaceBuilder<SolanaActions>('Solana', WALLET_ID) |
56 | 70 | .action(connect) |
57 | 71 | .action(disconnect) |
58 | | - .action(canEagerConnect) |
59 | 72 | .build(); |
60 | 73 |
|
61 | 74 | export { solana }; |
0 commit comments