@sign-global/tokentable-wallets 1.3.0 → 1.5.0

package/src/providers/sui/index.tsx

@@ -0,0 +1,139 @@
+import '@mysten/dapp-kit/dist/index.css';
+import EventEmitter from 'events';
+import { ReactNode, useEffect, useState } from 'react';
+import { WalletBase } from '../../WalletBase';
+import { ISignResult } from '../../types';
+import { get4361Message, prepareSignMessage } from '../../utils';
+import {
+  createNetworkConfig,
+  SuiClientProvider,
+  WalletProvider,
+  ConnectModal,
+  useCurrentAccount,
+  useCurrentWallet
+} from '@mysten/dapp-kit';
+import { getFullnodeUrl } from '@mysten/sui/client';
+
+interface SuiStoreType {
+  connectModal: {
+    open: boolean;
+    setOpen: (open: boolean) => void;
+  } | null;
+  account: ReturnType<typeof useCurrentAccount> | null;
+  wallet: ReturnType<typeof useCurrentWallet> | null;
+  counter: number;
+}
+
+const suiStore: SuiStoreType = {
+  connectModal: null,
+  account: null,
+  wallet: null,
+  counter: 0
+};
+
+export const eventBus = new EventEmitter();
+
+const Sui_EVENT_KEY = 'sui_signin';
+
+export const getEventKey = () => {
+  return `${Sui_EVENT_KEY}_${suiStore.counter}`;
+};
+
+export class SuiWallet extends WalletBase<SuiStoreType> {
+  getStore(): SuiStoreType {
+    return suiStore;
+  }
+  get isConnected(): boolean {
+    return !!this.getStore().wallet?.isConnected;
+  }
+
+  connect() {
+    suiStore.connectModal?.setOpen(true);
+  }
+
+  disconnect() {
+    suiStore.wallet?.currentWallet?.features['standard:disconnect']?.disconnect();
+  }
+  async sign(message: string): Promise<ISignResult> {
+    const signPersonalFeature = suiStore.wallet?.currentWallet?.features['sui:signPersonalMessage'];
+    const res = await signPersonalFeature?.signPersonalMessage({
+      message: new Uint8Array(Buffer.from(message)),
+      account: suiStore.account!
+    });
+
+    return {
+      message: message,
+      signature: res!.signature
+    };
+  }
+
+  signin(statement: string, prepare = true): Promise<ISignResult> {
+    return new Promise((resolve) => {
+      const signCallback = (data: ReturnType<typeof useCurrentAccount>) => {
+        this.publicKey = Buffer.from(data!.publicKey).toString('hex');
+        this.address = data?.address;
+        this.chainId = 1;
+        let fullMessage = statement;
+        if (prepare) {
+          const msg = prepareSignMessage({ statement, chainId: this.chainId, address: this.address! });
+          fullMessage = get4361Message(msg);
+        }
+
+        const res = this.sign(JSON.stringify(fullMessage, null, '  '));
+
+        resolve(res);
+      };
+
+      if (this.isConnected && suiStore.account) {
+        signCallback(suiStore.account!);
+        return;
+      }
+      suiStore.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);
+      });
+    });
+  }
+}
+
+export const SuiConnector = () => {
+  const [showModal, setShowModal] = useState(false);
+  const account = useCurrentAccount();
+  const wallet = useCurrentWallet();
+
+  useEffect(() => {
+    if (wallet) {
+      if (wallet.isConnected && account) {
+        const eventKey = getEventKey();
+        eventBus.emit(eventKey, account);
+      }
+    }
+  }, [wallet, account]);
+
+  suiStore.connectModal = {
+    open: showModal,
+    setOpen: setShowModal
+  };
+  suiStore.account = account;
+  suiStore.wallet = wallet;
+
+  return <ConnectModal open={showModal} onOpenChange={(open) => setShowModal(open)} trigger={<div></div>} />;
+};
+
+export const SuiProvider = ({ children, network }: { children: ReactNode; network: 'testnet' | 'mainnet' }) => {
+  const { networkConfig } = createNetworkConfig({
+    testnet: { url: getFullnodeUrl('testnet') },
+    mainnet: { url: getFullnodeUrl('mainnet') }
+  });
+  return (
+    <SuiClientProvider networks={networkConfig} network={network}>
+      <WalletProvider autoConnect>
+        <SuiConnector />
+        {children}
+      </WalletProvider>
+    </SuiClientProvider>
+  );
+};

package/src/types.ts

@@ -1,24 +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 ExtendedWalletType {
-  UNISAT = 'unisat',
-  OKXBTC = 'okx_btc'
-}
-
-export type ProviderType = WalletType | ExtendedWalletType;
-
 export type ISignResult =
   | {
       message: string;