@sign-global/tokentable-wallets 1.2.6 → 1.4.0

package/src/providers/cosmos/index.tsx

@@ -0,0 +1,138 @@
+import { ReactNode, useEffect } from 'react';
+import { ChainProvider, useChain } from '@cosmos-kit/react';
+import { chains, assets } from 'chain-registry';
+import { wallets } from '@cosmos-kit/keplr';
+import { wallets as okxWallets } from '@cosmos-kit/okxwallet-extension';
+import { wallets as compassWallets } from '@cosmos-kit/compass-extension';
+import '@interchain-ui/react/styles';
+import EventEmitter from 'events';
+import { WalletBase } from '../../WalletBase';
+import { ISignResult } from '../../types';
+import { get4361Message, prepareSignMessage } from '../../utils';
+
+interface CosmosStoreType {
+  chainContext: ReturnType<typeof useChain> | null;
+  counter: number;
+}
+
+const cosmosStore: CosmosStoreType = {
+  chainContext: null,
+  counter: 0
+};
+
+export const eventBus = new EventEmitter();
+
+const COSMOS_EVENT_KEY = 'cosmos_signin';
+
+export const getEventKey = () => {
+  return `${COSMOS_EVENT_KEY}_${cosmosStore.counter}`;
+};
+
+export class CosmosWallet extends WalletBase<CosmosStoreType> {
+  getStore(): CosmosStoreType {
+    return cosmosStore;
+  }
+  get isConnected() {
+    return !!cosmosStore.chainContext?.isWalletConnected;
+  }
+  connect(): void {
+    cosmosStore.chainContext?.connect();
+  }
+
+  disconnect() {
+    cosmosStore.chainContext?.disconnect();
+  }
+
+  async sign(message: string): Promise<ISignResult> {
+    try {
+      const provider = cosmosStore.chainContext!;
+      if (!provider.address) {
+        throw new Error('Cosmos wallet not connected');
+      }
+      const result = await provider.signArbitrary?.(provider.address, message);
+
+      this.publicKey = result?.pub_key?.value;
+
+      return {
+        message,
+        signature: result?.signature
+      };
+    } catch (error) {
+      console.error(error);
+      throw error;
+    }
+  }
+
+  signin(statement: string, prepare = true): Promise<ISignResult> {
+    return new Promise((resolve, reject) => {
+      try {
+        const signCallback = async (data: ReturnType<typeof useChain>) => {
+          try {
+            this.address = data.address;
+            this.chainId = 1;
+            let fullMessage = statement;
+            if (prepare) {
+              const msg = prepareSignMessage({ statement, chainId: data.chain.chain_id, address: this.address! });
+              fullMessage = get4361Message(msg);
+            }
+
+            const res = await this.sign(fullMessage);
+
+            resolve(res);
+          } catch (error) {
+            console.error('Error in signCallback:', error);
+            reject(error);
+          }
+        };
+
+        if (this.isConnected) {
+          signCallback(cosmosStore.chainContext!);
+          return;
+        }
+        cosmosStore.counter++;
+        this.connect();
+        const eventKey = getEventKey();
+        eventBus.once(eventKey, (data) => {
+          console.log('listen event key success, event key = %s, data = %j', eventKey, data);
+          signCallback(data);
+        });
+      } catch (error) {
+        console.error('Error in signin:', error);
+        reject(error);
+      }
+    });
+  }
+}
+
+export const CosmosConnector = ({ chainName = 'cosmoshub' }: { chainName?: string }) => {
+  const chainContext = useChain(chainName);
+
+  useEffect(() => {
+    if (chainContext.address) {
+      if (chainContext.isWalletConnected) {
+        const eventKey = getEventKey();
+        eventBus.emit(eventKey, chainContext);
+      }
+    }
+  }, [chainContext]);
+
+  cosmosStore.chainContext = chainContext;
+  return null;
+};
+
+export const CosmosProvider = ({ children, chainName = 'cosmoshub' }: { children: ReactNode; chainName?: string }) => {
+  const assetList = assets.find(({ chain_name }) => chain_name === chainName);
+  return (
+    <ChainProvider
+      chains={chains} // supported chains
+      assetLists={assetList ? [assetList] : undefined} // supported asset lists
+      wallets={[...wallets, ...compassWallets, ...okxWallets]}
+      signerOptions={{}}
+      endpointOptions={{ endpoints: {} }}
+      throwErrors={true}
+    >
+      <CosmosConnector chainName={chainName} />
+      {children}
+    </ChainProvider>
+  );
+};

package/src/providers/index.ts

@@ -2,3 +2,4 @@ export { EvmWallet, EvmProvider } from './evm';
 export { SolWallet, SolanaProvider } from './solana';
 // export { StarknetWallet } from './starknet';
 export { TonWallet, TonProvider } from './ton';
+export { CosmosWallet, CosmosProvider } from './cosmos';

package/src/types.ts

@@ -1,31 +1,3 @@
-export enum WalletType {
-  EVM = 'evm',
-  BITCOIN = 'bitcoin',
-  SOLANA = 'solana',
-  TON = 'ton',
-  APTOS = 'aptos',
-  PARTICLE = 'particle-evm',
-  OKX = 'okx',
-  Cyber = 'cyber',
-  StarkNet = 'starknet',
-  WALLETCONNECT = 'walletconnect',
-  Cardano = 'cardano'
-}
-
-export enum ChainType {
-  Evm = 'evm',
-  Ton = 'ton',
-  // Starknet = 'starknet',
-  Solana = 'solana',
-  Sui = 'sui'
-}
-export enum ExtendedWalletType {
-  UNISAT = 'unisat',
-  OKXBTC = 'okx_btc'
-}
-
-export type ProviderType = WalletType | ExtendedWalletType;
-
 export type ISignResult =
   | {
       message: string;