@dynamic-labs/ethereum 4.8.2-preview.0 → 4.8.3

package/src/walletConnect/walletConnect.js

@@ -0,0 +1,495 @@
+'use client'
+import { __awaiter } from '../../_virtual/_tslib.js';
+import EthereumProvider from '@walletconnect/ethereum-provider';
+import EventEmitter from 'eventemitter3';
+import { createWalletClient, custom } from 'viem';
+import { EthereumWalletConnector, chainsMap } from '@dynamic-labs/ethereum-core';
+import { parseIntSafe, sleep, DynamicError, isMobile, PlatformService } from '@dynamic-labs/utils';
+import { logger, performPlatformSpecificConnectionMethod, getDeepLink } from '@dynamic-labs/wallet-connector-core';
+
+const activeAccountKey = (walletName) => `dynamic-wc2-active-account-${walletName}`;
+const sessionTopicKey = (walletName) => `dynamic-wc2-session-topic-${walletName}`;
+const currentChainKey = (walletName) => `dynamic-wc2-current-chain-${walletName}`;
+const ee = new EventEmitter();
+class WalletConnect extends EthereumWalletConnector {
+    supportsNetworkSwitching() {
+        return true;
+    }
+    constructor(opts) {
+        super(opts);
+        this.isInitialized = false;
+        this.canConnectViaQrCode = true;
+        this.isWalletConnect = true;
+        this.preferredChains = [];
+        this.sessionEventHandler = () => { };
+        this.sessionDeleteHandler = () => { };
+        this.name = opts.walletName;
+        this.projectId = opts.projectId;
+        this.deepLinkPreference = opts.deepLinkPreference || 'native';
+        this.preferredChains = opts.walletConnectPreferredChains || [];
+        const lsCurrentChain = localStorage.getItem(this.currentChainKey);
+        this.currentChainId = lsCurrentChain
+            ? parseIntSafe(lsCurrentChain)
+            : undefined;
+    }
+    /**
+     * This method is used to get the address of the active account.
+     * It will re-initialize the provider if the provider is not found.
+     * @param opts
+     * @returns
+     */
+    getAddress(opts) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const activeAccount = this.getActiveAccount();
+            if (activeAccount === null || activeAccount === void 0 ? void 0 : activeAccount.address) {
+                return activeAccount.address;
+            }
+            if (!WalletConnect.provider || !((_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.signer.uri)) {
+                logger.debug('No WC2 provider found, re-initializing...');
+                yield this.endSession();
+                yield this.init();
+                // sleep 1 s to wait for connect call to finish
+                // the connect call isn't await-ed because it only resolves once
+                // the connection is established, but we need to wait for it to
+                // finish setting up the connection URI and making it available
+                // on the provider
+                yield sleep(1000);
+                if (!WalletConnect.provider || !((_b = WalletConnect.provider) === null || _b === void 0 ? void 0 : _b.signer.uri)) {
+                    logger.debug('No WC2 provider was found after init attempt, aborting');
+                    throw new DynamicError('No provider found');
+                }
+                else {
+                    logger.debug('provider was successfully initialized, continuing');
+                }
+            }
+            // Set Deep links
+            performPlatformSpecificConnectionMethod(WalletConnect.provider.signer.uri, this.metadata.deepLinks, {
+                onDesktopUri: opts === null || opts === void 0 ? void 0 : opts.onDesktopUri,
+                onDisplayUri: opts === null || opts === void 0 ? void 0 : opts.onDisplayUri,
+            }, this.deepLinkPreference);
+            return new Promise((resolve, reject) => {
+                if (!WalletConnect.provider) {
+                    reject(new DynamicError('No provider found'));
+                    return;
+                }
+                const onFail = () => {
+                    logger.debug('onConnection faile re-initializing provider');
+                    const error = new DynamicError('Connection rejected. Please try again.');
+                    error.code = 'connection_rejected';
+                    if (WalletConnect.provider) {
+                        WalletConnect.provider.signer.uri = undefined;
+                        // this is needed for mobile to work when using universal links.
+                        // if the user cancels the connection, we need to re-initialize the provider
+                        // so that the async work is done ahead of time, before the user tries to connect again,
+                        // otherwise they will trigger the iOS bug where they are redirected to the app store
+                        this.init();
+                        logger.debug('provider was re-initialized in onConnection fail, continuing');
+                    }
+                    reject(error);
+                    // We must clean up the onConnect and onFail listeners
+                    // whenever the connection attempt either succeeds or fails
+                    cleanupListeners();
+                };
+                const onConnect = () => {
+                    var _a;
+                    logger.debug('onConnection success, setting session');
+                    const session = (_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.session;
+                    if (!session) {
+                        logger.debug('onConnection success, but no session found rejecting');
+                        reject(new DynamicError('No session found'));
+                        return;
+                    }
+                    this.setSession(session);
+                    this.setWCActiveAccount(session.namespaces.eip155.accounts[0].split(':')[2]);
+                    this.getNetwork().then((chainId) => {
+                        var _a, _b;
+                        logger.debug('onConnection success, resolving, chainId:', chainId);
+                        this.currentChainId = chainId;
+                        logger.debug('onConnection success, resolving, activeAccountAddress:', (_a = this.getActiveAccount()) === null || _a === void 0 ? void 0 : _a.address);
+                        resolve((_b = this.getActiveAccount()) === null || _b === void 0 ? void 0 : _b.address);
+                    });
+                    // We must clean up the onConnect and onFail listeners
+                    // whenever the connection attempt either succeeds or fails
+                    cleanupListeners();
+                };
+                const cleanupListeners = () => {
+                    var _a;
+                    ee.off('walletconnect_connection_failed', onFail);
+                    (_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.off('connect', onConnect);
+                };
+                ee.on('walletconnect_connection_failed', onFail);
+                WalletConnect.provider.on('connect', onConnect);
+            });
+        });
+    }
+    init() {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger.debug('init was called', this.name);
+            yield this.initProvider();
+            yield this.initConnection();
+            this.isInitialized = true;
+        });
+    }
+    // We need to add a gate to this method since we will be calling it asynchronously
+    // from different places (such as setShowAuthFlow), which means there's a chance for
+    // a race condition to happen where createInitProviderPromise is called multiple times
+    initProvider() {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger.debug('initProvider was called', this.name);
+            const { provider } = WalletConnect;
+            if (!provider) {
+                logger.debug('provider is undefined', this.name);
+                if (this.initializePromise === undefined) {
+                    this.initializePromise = this.createInitProviderPromise();
+                }
+                yield this.initializePromise;
+            }
+        });
+    }
+    createInitProviderPromise() {
+        return __awaiter(this, void 0, void 0, function* () {
+            WalletConnect.provider = yield this.createProvider();
+            this.teardownEventListeners();
+            this.setupEventListeners();
+        });
+    }
+    initConnection() {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger.debug('initConnection was called', this.name);
+            const { provider } = WalletConnect;
+            if (!provider) {
+                throw new DynamicError('No provider found (init connection)');
+            }
+            // this means there is already a connection in progress, so don't call connect again
+            if (provider === null || provider === void 0 ? void 0 : provider.signer.uri) {
+                return;
+            }
+            provider.connect().catch((e) => {
+                logger.error(e);
+                ee.emit('walletconnect_connection_failed', e);
+            });
+        });
+    }
+    createProvider() {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger.debug('createProvider was called', this.name);
+            return EthereumProvider.init({
+                events: ['chainChanged', 'accountsChanged'],
+                methods: [],
+                optionalChains: this.getMappedChainsByPreferredOrder(),
+                optionalMethods: [
+                    'eth_chainId',
+                    'eth_signTypedData',
+                    'eth_signTransaction',
+                    'eth_sign',
+                    'personal_sign',
+                    'eth_sendTransaction',
+                    'eth_signTypedData_v4',
+                    'wallet_switchEthereumChain',
+                    'wallet_addEthereumChain',
+                ],
+                projectId: this.projectId,
+                rpcMap: this.evmNetworkRpcMap(),
+                showQrModal: false,
+            });
+        });
+    }
+    refreshSession() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f;
+            logger.debug('refreshSession was called', this.name);
+            if (!((_b = (_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.session) === null || _b === void 0 ? void 0 : _b.topic) ||
+                localStorage.getItem(this.sessionTopicKey) !==
+                    ((_d = (_c = WalletConnect.provider) === null || _c === void 0 ? void 0 : _c.session) === null || _d === void 0 ? void 0 : _d.topic)) {
+                const chain = (_e = this.getMappedChainsByPreferredOrder()) === null || _e === void 0 ? void 0 : _e[0];
+                logger.debug('provider is not connected, skipping refreshSession, just setting the default chain', chain);
+                this.currentChainId = chain;
+                return;
+            }
+            this.session = WalletConnect.provider.session;
+            this.setActiveAccount(((_f = localStorage.getItem(this.activeAccountKey)) !== null && _f !== void 0 ? _f : undefined));
+            const walletClient = this.getWalletClient();
+            const walletChainId = yield walletClient.getChainId();
+            logger.debug('checking selected chain in refreshSession', this.name, walletChainId, this.currentChainId);
+            if (this.currentChainId && this.currentChainId !== walletChainId) {
+                logger.debug('switching to selected chain', this.currentChainId);
+                yield walletClient.switchChain({ id: this.currentChainId });
+            }
+        });
+    }
+    getMappedChainsByPreferredOrder() {
+        const allChains = this.evmNetworks.map((network) => `eip155:${network.chainId}`);
+        const reorderedChains = this.preferredChains.filter((chain) => allChains.includes(chain));
+        const remainingChains = allChains.filter((chain) => !this.preferredChains.includes(chain));
+        return [...reorderedChains, ...remainingChains].map((chain) => Number(chain.split(':')[1]));
+    }
+    getWalletClient(chainId) {
+        logger.debug('getWalletClient was called', this.name);
+        const walletConnect = WalletConnect.provider || this.createProvider();
+        const walletClient = createWalletClient({
+            account: this.getActiveAccount(),
+            chain: chainsMap[chainId !== null && chainId !== void 0 ? chainId : String(this.currentChainId)],
+            transport: custom({
+                request: (args) => __awaiter(this, void 0, void 0, function* () {
+                    const provider = yield walletConnect;
+                    return provider.request(args);
+                }),
+            }),
+        });
+        return walletClient;
+    }
+    get currentChainId() {
+        return this._currentChainId;
+    }
+    set currentChainId(value) {
+        this._currentChainId = value;
+        if (value) {
+            localStorage.setItem(this.currentChainKey, value.toString());
+        }
+        else {
+            localStorage.removeItem(this.currentChainKey);
+        }
+    }
+    get sessionTopicKey() {
+        return sessionTopicKey(this.key);
+    }
+    get activeAccountKey() {
+        return activeAccountKey(this.key);
+    }
+    get currentChainKey() {
+        return currentChainKey(this.key);
+    }
+    setupEventListeners() {
+        if (!WalletConnect.provider) {
+            return;
+        }
+        this.sessionEventHandler = ({ params, }) => {
+            logger.debug('session_event was called', { params });
+            if (!params || !params.event) {
+                logger.debug('session_event was called without params or params.event');
+                return;
+            }
+            const { name, data } = params.event;
+            if (name === 'chainChanged') {
+                const chainId = parseIntSafe(data);
+                if (chainId === this.currentChainId) {
+                    logger.debug(`ignoring chainChanged event with same chain id as current chain id: ${chainId}`);
+                    return;
+                }
+                if (chainId === undefined) {
+                    logger.debug(`received unexpected data for chainChanged: ${data} with type ${typeof data}}`);
+                    return;
+                }
+                this.currentChainId = chainId;
+                this.emit('chainChange', { chain: String(chainId) });
+                // When a user switches network from their wallet, we need the provider to change network
+                // such that any future calls to `getNetwork` will return the correct network
+                this.switchNetwork({ networkChainId: chainId });
+            }
+            else if (name === 'accountsChanged') {
+                if (!Array.isArray(data)) {
+                    logger.debug(`received unexpected data for accountsChanged: ${data} with type ${typeof data}}`);
+                    return;
+                }
+                // eslint-disable-next-line prefer-destructuring
+                const account = data[0].split(':')[2];
+                this.setWCActiveAccount(account);
+            }
+        };
+        WalletConnect.provider.on('session_event', this.sessionEventHandler);
+        this.sessionDeleteHandler = () => __awaiter(this, void 0, void 0, function* () {
+            this.endSession();
+            this.emit('disconnect');
+        });
+        WalletConnect.provider.on('session_delete', this.sessionDeleteHandler);
+    }
+    teardownEventListeners() {
+        if (!WalletConnect.provider) {
+            return;
+        }
+        WalletConnect.provider.off('session_event', this.sessionEventHandler);
+        WalletConnect.provider.off('session_delete', this.sessionDeleteHandler);
+    }
+    /**
+     * WalletConnect V2 will fail to send the sign message request if the chainId
+     * is not the same as the one in the session. This method will wait for the
+     * chainId to change and then retry the sign message request.
+     *
+     * Otherwise it will just return the result of the sign message request.
+     *
+     * @param signMessageFn - Function to sign message with provider
+     * @param messageToSign - Message to sign
+     * @returns
+     */
+    waitForSignMessage(signMessageFn, messageToSign) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const raceConditionPromise = new Promise((resolve, reject) => {
+                // Create listener for chain change event
+                this.on('chainChange', () => resolve({ success: false }));
+                signMessageFn(messageToSign)
+                    .then((result) => resolve({ signedMessage: result, success: true }))
+                    .catch(reject);
+            });
+            const signedMessageResult = yield raceConditionPromise;
+            if (signedMessageResult.success === false) {
+                return signMessageFn(messageToSign);
+            }
+            return signedMessageResult.signedMessage;
+        });
+    }
+    getDeepLink() {
+        var _a;
+        if (!this.session) {
+            return;
+        }
+        const deepLink = getDeepLink({
+            deepLinks: this.metadata.deepLinks,
+            mode: 'regular',
+            preference: this.deepLinkPreference,
+            uri: (_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.signer.uri,
+        });
+        if (!deepLink) {
+            return;
+        }
+        // we need to include the session topic here because it helps the wallet
+        // auto redirect back to the dapp after signing
+        return `${deepLink}?sessionTopic=${this.session.topic}`;
+    }
+    signMessage(messageToSign) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.session) {
+                throw new DynamicError('no session');
+            }
+            const deepLink = this.getDeepLink();
+            if (isMobile() && deepLink) {
+                PlatformService.openURL(deepLink);
+            }
+            const signMessageFn = (messageToSign) => __awaiter(this, void 0, void 0, function* () {
+                const activeAccount = this.getActiveAccount();
+                if (!activeAccount) {
+                    return;
+                }
+                const walletClient = yield this.getWalletClient();
+                return walletClient.signMessage({
+                    account: activeAccount,
+                    message: messageToSign,
+                });
+            });
+            const response = yield this.waitForSignMessage(signMessageFn, messageToSign);
+            return response;
+        });
+    }
+    clearActiveAccount() {
+        localStorage.removeItem(this.activeAccountKey);
+        this.setActiveAccount(undefined);
+    }
+    clearSession() {
+        localStorage.removeItem(this.sessionTopicKey);
+        this.session = undefined;
+    }
+    setWCActiveAccount(account) {
+        localStorage.setItem(this.activeAccountKey, account);
+        this.setActiveAccount(account);
+        this.emit('accountChange', { accounts: [account] });
+    }
+    setSession(session) {
+        localStorage.setItem(this.sessionTopicKey, session.topic);
+        this.session = session;
+    }
+    endSession() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            this.clearActiveAccount();
+            this.clearSession();
+            this.currentChainId = undefined;
+            if (!((_a = WalletConnect.provider) === null || _a === void 0 ? void 0 : _a.session)) {
+                return;
+            }
+            try {
+                yield WalletConnect.provider.disconnect();
+                // We must unset provider on logout so that a new session can be established
+                // If we don't then the provider will still have the old session and will hang
+                WalletConnect.provider = undefined;
+            }
+            catch (e) {
+                logger.debug(e);
+            }
+        });
+    }
+    getNetwork() {
+        const _super = Object.create(null, {
+            getNetwork: { get: () => super.getNetwork }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.currentChainId) {
+                return this.currentChainId;
+            }
+            yield this.initProvider();
+            return _super.getNetwork.call(this);
+        });
+    }
+    providerSwitchNetwork(_a) {
+        const _super = Object.create(null, {
+            providerSwitchNetwork: { get: () => super.providerSwitchNetwork }
+        });
+        return __awaiter(this, arguments, void 0, function* ({ network, }) {
+            const supportedNetworks = yield this.getSupportedNetworks();
+            if (!(supportedNetworks === null || supportedNetworks === void 0 ? void 0 : supportedNetworks.includes(network.chainId.toString()))) {
+                const error = new DynamicError('Network switching is not available at this time. The user should manually switch network in their wallet');
+                error.code = 'network_switching_only_available_in_wallet';
+                throw error;
+            }
+            const currentNetworkId = yield this.getNetwork();
+            if (currentNetworkId && currentNetworkId === network.chainId) {
+                return;
+            }
+            if (this.switchNetworkOnlyFromWallet) {
+                throw new DynamicError('Network switching is only supported through the wallet');
+            }
+            const walletClient = yield this.getWalletClient();
+            yield _super.providerSwitchNetwork.call(this, { network, provider: walletClient });
+            this.currentChainId = network.chainId;
+            this.emit('chainChange', { chain: String(network.chainId) });
+        });
+    }
+    getConnectedAccounts() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.isInitialized === false) {
+                // TODO why not just call init()?
+                yield this.initProvider();
+                yield this.refreshSession();
+                this.isInitialized = true;
+            }
+            const activeAccount = this.getActiveAccount();
+            if (!(activeAccount === null || activeAccount === void 0 ? void 0 : activeAccount.address)) {
+                return [];
+            }
+            return [activeAccount.address];
+        });
+    }
+    getSupportedNetworks() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            yield this.initProvider();
+            yield this.refreshSession();
+            if (!this.session) {
+                return [];
+            }
+            const chains = [];
+            // Some wallet (i.e ZenGo) use namespaces.account to list supported chains
+            // while others use keys within the namespaces object
+            Object.keys(this.session.namespaces).forEach((key) => {
+                if (key.startsWith('eip155:')) {
+                    chains.push(key.split(':')[1]);
+                }
+            });
+            (_a = this.session.namespaces.eip155) === null || _a === void 0 ? void 0 : _a.accounts.forEach((account) => chains.push(account.split(':')[1]));
+            return chains.length
+                ? chains
+                : this.evmNetworks.map((network) => network.chainId.toString());
+        });
+    }
+}
+
+export { WalletConnect };