@openzeppelin/contracts-ui-builder-adapter-evm/src/wallet/utils/walletImplementationManager.ts

EVM Adapter for Contracts UI Builder

import type { UiKitConfiguration } from '@openzeppelin/contracts-ui-builder-types';
import { appConfigService, logger } from '@openzeppelin/contracts-ui-builder-utils';

import { WagmiWalletImplementation } from '../implementation/wagmi-implementation';

let walletImplementationInstance: WagmiWalletImplementation | undefined;
let walletImplementationPromise: Promise<WagmiWalletImplementation> | undefined;

const LOG_SYSTEM = 'EvmWalletImplementationManager';

/**
 * Get or create the singleton instance of WagmiWalletImplementation.
 * This function ensures that the initialization logic runs only once.
 *
 * @returns A Promise resolving to the WagmiWalletImplementation singleton
 */
export async function getEvmWalletImplementation(): Promise<WagmiWalletImplementation> {
  if (walletImplementationInstance) {
    return walletImplementationInstance;
  }

  if (walletImplementationPromise) {
    return walletImplementationPromise;
  }

  walletImplementationPromise = (async () => {
    try {
      logger.info(LOG_SYSTEM, 'Initializing WagmiWalletImplementation singleton (async)... ');
      // Get appName and projectId from appConfigService for RainbowKit
      // This assumes appConfigService is initialized before this manager is first called.
      const initialUiKitConfig = appConfigService.getTypedNestedConfig<UiKitConfiguration>(
        'walletui',
        'config'
      );

      const wcProjectId = appConfigService.getGlobalServiceParam('walletconnect', 'projectId') as
        | string
        | undefined;

      // Pass initialUiKitConfig to the constructor
      const instance = new WagmiWalletImplementation(wcProjectId, initialUiKitConfig);
      logger.info(LOG_SYSTEM, 'WagmiWalletImplementation singleton created (async).');
      walletImplementationInstance = instance;
      return instance;
    } catch (error) {
      logger.error(LOG_SYSTEM, 'Failed to initialize WagmiWalletImplementation (async):', error);
      const fallbackInstance = new WagmiWalletImplementation();
      walletImplementationInstance = fallbackInstance;
      return fallbackInstance;
    }
  })();

  return walletImplementationPromise;
}

// Optional: A synchronous getter for cases where the instance is known to be initialized.
// Use with caution, prefer the async getter.
export function getInitializedEvmWalletImplementation(): WagmiWalletImplementation | undefined {
  if (!walletImplementationInstance) {
    logger.warn(
      LOG_SYSTEM,
      'getInitializedEvmWalletImplementation called before instance was ready.'
    );
  }
  return walletImplementationInstance;
}