@sign-global/tokentable-core 1.4.0 → 1.5.0

package/src/contracts/index.ts

@@ -11,3 +11,6 @@ export { AirdropService as TonAirdropService } from './ton/AirdropService';
 
 // SUI Contracts
 export { SignatureAirdropService as SuiSignatureAirdropService } from './sui/SignatureAirdropService';
+
+// Aptos Contracts
+export { AirdropService as AptosAirdropService } from './aptos/AirdropService';

package/src/contracts/sui/DistributorWithFeeClient.ts

@@ -72,6 +72,16 @@ export class SuiDistributorWithFeeClient extends SuiContractClientBase {
 
     const feeTokenType = '0x2::sui::SUI'; // 或其他费用代币类型
 
+    // 在splitCoins之前添加余额检查
+    const gasBalance = await this.client.getBalance({
+      owner: this.wallet.accounts[0].address,
+      coinType: feeTokenType
+    });
+
+    if (BigInt(gasBalance.totalBalance) < totalFees) {
+      throw new Error(`Insufficient SUI balance. Required: ${totalFees}, Available: ${gasBalance.totalBalance}`);
+    }
+
     const [feeCoin] = tx.splitCoins(
       tx.gas, // 用 gas 对象作为源
       [tx.pure(bcs.u64().serialize(totalFees))]

package/src/types/index.ts

@@ -4,6 +4,7 @@ import { ChainConfig } from '../constants';
 import { Chain, WalletClient as ViemWalletClient } from 'viem';
 import { WalletContextState as SolanaWalletState } from '@solana/wallet-adapter-react';
 import { Wallet as SuiWalletClient } from '@mysten/wallet-standard';
+import { useWallet } from '@aptos-labs/wallet-adapter-react';
 
 export type ChainId = keyof typeof ChainConfig;
 
@@ -20,7 +21,8 @@ export enum ChainType {
   Evm = 'evm',
   Ton = 'ton',
   Solana = 'solana',
-  Sui = 'sui'
+  Sui = 'sui',
+  Aptos = 'aptos'
 }
 
 export type ThemeConfig = {
@@ -203,9 +205,16 @@ export enum EvmAirdropVersionEnum {
   V4 = '0.4.0' // zeta kyc、kyc threshold
 }
 
-export type IWalletClient = TonWalletClient | ViemWalletClient | SolanaWalletState | SuiWalletClient;
+type AptosWalletClient = ReturnType<typeof useWallet>;
 
-export type { TonWalletClient, ViemWalletClient, SolanaWalletState, SuiWalletClient };
+export type IWalletClient =
+  | TonWalletClient
+  | ViemWalletClient
+  | SolanaWalletState
+  | SuiWalletClient
+  | AptosWalletClient;
+
+export type { TonWalletClient, ViemWalletClient, SolanaWalletState, SuiWalletClient, AptosWalletClient };
 export type IAirdropService = TonAirdropService | EvmAirdropService;
 
 type BaseAirdropOptions = {