@sign-global/tokentable-core 1.4.0 → 1.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sign-global/tokentable-core",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -38,7 +38,9 @@
     "nanoid": "^5.0.9",
     "@mysten/dapp-kit": "^0.16.3",
     "@mysten/sui": "^1.29.1",
-    "@mysten/wallet-standard": "^0.16.10"
+    "@mysten/wallet-standard": "^0.16.10",
+    "@aptos-labs/ts-sdk": "^4.0.0",
+    "@aptos-labs/wallet-adapter-react": "^7.0.4"
   },
   "publishConfig": {
     "access": "public"

package/src/constants/chain-config.ts

@@ -13,6 +13,7 @@ import {
   arbitrum
 } from 'viem/chains';
 import {
+  aptosTestNet,
   mevmDevNet,
   signBrBTestnet,
   solanaDevNet,
@@ -113,6 +114,10 @@ export const ChainConfig = {
   [suiMainNet.id]: {
     contractAddress: '0x3f73ec78d3e8a99277496edb6183d6075490983c2a29542a1c3f709aa31169b5',
     rpcUrl: suiMainNet.rpcUrls.default.http[0]
+  },
+  [aptosTestNet.id]: {
+    contractAddress: '0x0ee860c2c1dc9e6d7c013cfc78cecfc226dfffa403b0ca212b38d958bc1048c9',
+    rpcUrl: aptosTestNet.rpcUrls.default.http[0]
   }
 };
 

package/src/constants/network.ts

@@ -89,10 +89,10 @@ export const solanaDevNet = {
   },
   rpcUrls: {
     public: {
-      http: ['https://devnet.helius-rpc.com/?api-key=92a10901-e587-4f65-a180-4a63b7305966']
+      http: ['https://serene-empty-film.solana-devnet.quiknode.pro/7fcc62232b14dc8a6a154597ee028ebf00e0c195']
     },
     default: {
-      http: ['https://devnet.helius-rpc.com/?api-key=92a10901-e587-4f65-a180-4a63b7305966']
+      http: ['https://serene-empty-film.solana-devnet.quiknode.pro/7fcc62232b14dc8a6a154597ee028ebf00e0c195']
     }
   },
   blockExplorers: {
@@ -115,10 +115,10 @@ export const solanaMainNet = {
   },
   rpcUrls: {
     public: {
-      http: ['https://neat-falling-dream.solana-mainnet.quiknode.pro/55dbcc2244d010e047c4ab1e3b7964cc6204505d']
+      http: ['https://serene-empty-film.solana-mainnet.quiknode.pro/7fcc62232b14dc8a6a154597ee028ebf00e0c195']
     },
     default: {
-      http: ['https://neat-falling-dream.solana-mainnet.quiknode.pro/55dbcc2244d010e047c4ab1e3b7964cc6204505d']
+      http: ['https://serene-empty-film.solana-mainnet.quiknode.pro/7fcc62232b14dc8a6a154597ee028ebf00e0c195']
     }
   },
   blockExplorers: {
@@ -203,3 +203,29 @@ export const suiMainNet = {
   },
   testnet: false
 };
+
+export const aptosTestNet = {
+  id: 2,
+  name: 'Aptos Testnet',
+  network: 'Aptos Testnet',
+  nativeCurrency: {
+    decimals: 8,
+    name: 'Aptos',
+    symbol: 'APT'
+  },
+  rpcUrls: {
+    public: {
+      http: ['https://api.testnet.aptoslabs.com/v1']
+    },
+    default: {
+      http: ['https://api.testnet.aptoslabs.com/v1']
+    }
+  },
+  blockExplorers: {
+    default: {
+      name: 'aptos scan',
+      url: 'hhttps://explorer.aptoslabs.com/'
+    }
+  },
+  testnet: true
+};

package/src/contracts/aptos/AirdropService.ts

@@ -0,0 +1,55 @@
+import { AptosAirdropContractClient } from './AptosAirdropClient';
+import { AptosBaseDistributorContractClient } from './AptosBaseDistributorClient';
+import { AptosContractInfo } from './AptosContractClient';
+
+export class AirdropService {
+  options: Omit<AptosContractInfo, 'module'>;
+  constructor(options: Omit<AptosContractInfo, 'module'>) {
+    this.options = options;
+  }
+
+  private get airdropClient(): AptosAirdropContractClient {
+    if (!this.options) {
+      throw new Error('options is empty');
+    }
+
+    return new AptosAirdropContractClient(this.options);
+  }
+
+  private get baseClient(): AptosBaseDistributorContractClient {
+    if (!this.options) {
+      throw new Error('options is empty');
+    }
+    return new AptosBaseDistributorContractClient(this.options);
+  }
+
+  async isLeafUsed(leaf: string): Promise<boolean> {
+    return this.baseClient.isLeafUsed(leaf);
+  }
+
+  async batchFetchClaimed(leafs: string[]) {
+    return this.baseClient.batchFetchClaimed(leafs);
+  }
+
+  async claim({ proof, group, data, value }: { proof: string[]; group: string; data: string; value?: bigint }) {
+    return this.airdropClient.claim({
+      proof,
+      group,
+      data,
+      value
+    });
+  }
+
+  async getVersion() {
+    const res = await this.baseClient.getContractInfo();
+    return res.version;
+  }
+
+  async getClaimFee(contractAddress: string, amount: bigint) {
+    return this.baseClient.getClaimFee(contractAddress, amount);
+  }
+
+  async getPaused() {
+    return this.baseClient.getPaused();
+  }
+}

package/src/contracts/aptos/AptosAirdropClient.ts

@@ -0,0 +1,27 @@
+import { AptosContractClientBase, AptosContractInfo } from './AptosContractClient';
+import { getChainConfig } from '../../common';
+
+export class AptosAirdropContractClient extends AptosContractClientBase {
+  airdropAddress: string;
+  constructor(options: Omit<AptosContractInfo, 'module'>) {
+    const chainConfig = getChainConfig(options.chainId as any);
+    super({
+      ...options,
+      address: chainConfig.contractAddress,
+      module: 'merkle_distributor_with_fees'
+    });
+    this.airdropAddress = options.address;
+  }
+
+  private hexToBytes(hexStr: string): number[] {
+    const hex = hexStr.startsWith('0x') ? hexStr.slice(2) : hexStr;
+    return Array.from(Buffer.from(hex, 'hex'));
+  }
+
+  async claim({ proof, group, data, value }: { proof: string[]; group: string; data: string; value?: bigint }) {
+    const dataBytes = this.hexToBytes(data);
+    const proofBytes = proof.map((p) => this.hexToBytes(p));
+
+    return await this.invokeContractWrite('claim', [this.airdropAddress, proofBytes, dataBytes]);
+  }
+}

package/src/contracts/aptos/AptosBaseDistributorClient.ts

@@ -0,0 +1,81 @@
+import { AptosContractClientBase, AptosContractInfo } from './AptosContractClient';
+import { getChainConfig } from '../../common';
+
+export class AptosBaseDistributorContractClient extends AptosContractClientBase {
+  airdropAddress: string;
+  constructor(options: Omit<AptosContractInfo, 'module'>) {
+    const chainConfig = getChainConfig(options.chainId as any);
+    super({
+      ...options,
+      address: chainConfig.contractAddress,
+      module: 'base_distributor'
+    });
+    this.airdropAddress = options.address;
+  }
+
+  async getContractInfo() {
+    const data = await this.invokeContractRead('get_distribution_info', [this.airdropAddress]);
+    console.log('data', data);
+    const [root, token, startTime, endTime, owner, version, paused] = data;
+    return {
+      root,
+      token,
+      startTime: Number(startTime),
+      endTime: Number(endTime),
+      owner,
+      version: version,
+      paused
+    };
+  }
+
+  async deposit({ amount, token }: { amount: bigint; token: string }): Promise<string> {
+    return await this.signTransaction({
+      functionName: `0x1::primary_fungible_store::transfer`,
+      functionArguments: [token, this.airdropAddress, amount],
+      typeArguments: ['0x1::fungible_asset::Metadata']
+    });
+  }
+
+  async isLeafUsed(leaf: string): Promise<boolean> {
+    const leafBytes = leaf.startsWith('0x')
+      ? Array.from(Buffer.from(leaf.slice(2), 'hex'))
+      : Array.from(Buffer.from(leaf, 'hex'));
+    const res = await this.invokeContractRead('is_leaf_used', [this.airdropAddress, leafBytes]);
+    console.log('res', res);
+    return res[0];
+  }
+
+  async batchFetchClaimed(leafs: string[]) {
+    const res = await Promise.all(
+      leafs.map(async (leaf) => {
+        return this.isLeafUsed(leaf);
+      })
+    );
+    return res.map((info) => ({
+      result: info || false,
+      status: 'success'
+    }));
+  }
+
+  private hexToBytes(hexStr: string): number[] {
+    const hex = hexStr.startsWith('0x') ? hexStr.slice(2) : hexStr;
+    return Array.from(Buffer.from(hex, 'hex'));
+  }
+
+  async claim({ proof, data }: { proof: string[]; group: string; data: string; value?: bigint; extraData?: string }) {
+    const groupBytes = this.hexToBytes('');
+    const dataBytes = this.hexToBytes(data);
+    const proofBytes = proof.map((p) => this.hexToBytes(p));
+
+    return await this.invokeContractWrite('claim', [this.airdropAddress, proofBytes, groupBytes, dataBytes]);
+  }
+
+  async getPaused() {
+    const res = await this.getContractInfo();
+    return res.paused;
+  }
+
+  async getClaimFee(contractAddress: string, amount: bigint) {
+    return 0;
+  }
+}

package/src/contracts/aptos/AptosContractClient.ts

@@ -0,0 +1,109 @@
+import { Aptos, AptosConfig, Network } from '@aptos-labs/ts-sdk';
+import { useWallet } from '@aptos-labs/wallet-adapter-react';
+
+export type AptosContractInfo = {
+  address: string;
+  chainId: string;
+  account?: ReturnType<typeof useWallet>;
+  module: string;
+  chainRpc: string;
+};
+
+export class AptosContractClientBase {
+  protected address: string;
+  protected chainId: string;
+  protected wallet?: ReturnType<typeof useWallet>;
+  protected client: Aptos;
+  protected module: string;
+
+  constructor(contractInfo: AptosContractInfo) {
+    this.address = contractInfo.address;
+    this.chainId = contractInfo.chainId;
+    this.wallet = contractInfo.account;
+    this.module = contractInfo.module;
+    this.client = this.initClient(contractInfo.chainRpc);
+  }
+
+  private initClient(chainRpc: string): Aptos {
+    const chainIdMap = {
+      '2': Network.TESTNET,
+      '1': Network.MAINNET
+    };
+    const config = new AptosConfig({
+      network: chainIdMap[this.chainId as keyof typeof chainIdMap],
+      fullnode: chainRpc
+    });
+
+    return new Aptos(config);
+  }
+
+  protected async invokeContractRead(method: string, args: any[]): Promise<any> {
+    try {
+      const viewPayload = {
+        function: `${this.address}::${this.module}::${method}` as `${string}::${string}::${string}`,
+        functionArguments: args
+      };
+
+      return await this.client.view({ payload: viewPayload });
+    } catch (error) {
+      console.error(`Error invoking read method ${method}:`, error);
+      throw error;
+    }
+  }
+
+  protected async invokeContractWrite(method: string, args: any[]): Promise<string> {
+    if (!this.wallet) {
+      throw new Error('Account not connected');
+    }
+
+    try {
+      const response = await this.wallet.signAndSubmitTransaction({
+        sender: this.wallet.account?.address,
+        data: {
+          function: `${this.address}::${this.module}::${method}`,
+          functionArguments: args
+        }
+      });
+      await this.client.waitForTransaction({
+        transactionHash: response.hash
+      });
+      return response.hash;
+    } catch (error) {
+      console.error(`Error invoking write method ${method}:`, error);
+
+      throw error;
+    }
+  }
+
+  async signTransaction({
+    functionName,
+    functionArguments,
+    typeArguments
+  }: {
+    functionName: `${string}::${string}::${string}`;
+    functionArguments: any[];
+    typeArguments?: string[];
+  }) {
+    if (!this.wallet) {
+      throw new Error('Account not connected');
+    }
+
+    try {
+      const response = await this.wallet.signAndSubmitTransaction({
+        sender: this.wallet.account?.address,
+        data: {
+          function: functionName,
+          functionArguments: functionArguments,
+          typeArguments: typeArguments || []
+        }
+      });
+      const res = await this.client.waitForTransaction({
+        transactionHash: response.hash
+      });
+
+      return response.hash;
+    } catch (error) {
+      throw error;
+    }
+  }
+}

package/src/contracts/aptos/abis/fee_collector.ts

@@ -0,0 +1,223 @@
+export const FEE_COLLECTOR_ABI = {
+  "address": "0x90f5a927d597d49b4fb64a48f3dc34695687e389136b04c42f9b9e7fe7fe59a0",
+  "name": "fee_collector",
+  "friends": [],
+  "exposed_functions": [
+    {
+      "name": "version",
+      "visibility": "public",
+      "is_entry": false,
+      "is_view": true,
+      "generic_type_params": [],
+      "params": [],
+      "return": [
+        "0x1::string::String"
+      ]
+    },
+    {
+      "name": "get_fee",
+      "visibility": "public",
+      "is_entry": false,
+      "is_view": true,
+      "generic_type_params": [],
+      "params": [
+        "address"
+      ],
+      "return": [
+        "u64"
+      ]
+    },
+    {
+      "name": "get_fee_token",
+      "visibility": "public",
+      "is_entry": false,
+      "is_view": true,
+      "generic_type_params": [],
+      "params": [
+        "address"
+      ],
+      "return": [
+        "address"
+      ]
+    },
+    {
+      "name": "set_custom_fee_fixed",
+      "visibility": "public",
+      "is_entry": true,
+      "is_view": false,
+      "generic_type_params": [],
+      "params": [
+        "&signer",
+        "address",
+        "u64"
+      ],
+      "return": []
+    },
+    {
+      "name": "set_custom_fee_token",
+      "visibility": "public",
+      "is_entry": true,
+      "is_view": false,
+      "generic_type_params": [],
+      "params": [
+        "&signer",
+        "address",
+        "address"
+      ],
+      "return": []
+    },
+    {
+      "name": "set_default_fee",
+      "visibility": "public",
+      "is_entry": true,
+      "is_view": false,
+      "generic_type_params": [],
+      "params": [
+        "&signer",
+        "u64"
+      ],
+      "return": []
+    },
+    {
+      "name": "set_default_fee_token",
+      "visibility": "public",
+      "is_entry": true,
+      "is_view": false,
+      "generic_type_params": [],
+      "params": [
+        "&signer",
+        "address"
+      ],
+      "return": []
+    },
+    {
+      "name": "withdraw_fee",
+      "visibility": "public",
+      "is_entry": true,
+      "is_view": false,
+      "generic_type_params": [],
+      "params": [
+        "&signer",
+        "0x1::object::Object<0x1::fungible_asset::Metadata>",
+        "u64"
+      ],
+      "return": []
+    }
+  ],
+  "structs": [
+    {
+      "name": "CustomFeeSetFixed",
+      "is_native": false,
+      "is_event": true,
+      "abilities": [
+        "drop",
+        "store"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "distributor_address",
+          "type": "address"
+        },
+        {
+          "name": "fixed_fee",
+          "type": "u64"
+        }
+      ]
+    },
+    {
+      "name": "CustomFeeSetToken",
+      "is_native": false,
+      "is_event": true,
+      "abilities": [
+        "drop",
+        "store"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "distributor_address",
+          "type": "address"
+        },
+        {
+          "name": "token_address",
+          "type": "address"
+        }
+      ]
+    },
+    {
+      "name": "DefaultFeeSet",
+      "is_native": false,
+      "is_event": true,
+      "abilities": [
+        "drop",
+        "store"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "fee",
+          "type": "u64"
+        }
+      ]
+    },
+    {
+      "name": "DefaultFeeTokenSet",
+      "is_native": false,
+      "is_event": true,
+      "abilities": [
+        "drop",
+        "store"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "token_address",
+          "type": "address"
+        }
+      ]
+    },
+    {
+      "name": "FeeCollectorData",
+      "is_native": false,
+      "is_event": false,
+      "abilities": [
+        "key"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "default_fee",
+          "type": "u64"
+        },
+        {
+          "name": "default_fee_token",
+          "type": "address"
+        },
+        {
+          "name": "custom_fee_tokens",
+          "type": "0x1::table::Table<address, address>"
+        },
+        {
+          "name": "custom_fees_fixed",
+          "type": "0x1::table::Table<address, u64>"
+        }
+      ]
+    },
+    {
+      "name": "ModuleData",
+      "is_native": false,
+      "is_event": false,
+      "abilities": [
+        "key"
+      ],
+      "generic_type_params": [],
+      "fields": [
+        {
+          "name": "signer_cap",
+          "type": "0x1::account::SignerCapability"
+        }
+      ]
+    }
+  ]
+} as const

package/src/contracts/aptos/abis/index.ts

@@ -0,0 +1,18 @@
+// 统一导出所有ABI模块
+export { OWNABLE_ABI } from './ownable';
+export { MD_CREATE2_ABI } from './md_create2';
+export { FEE_COLLECTOR_ABI } from './fee_collector';
+export { MERKLE_VERIFIER_ABI } from './merkle_verifier';
+export { MERKLE_DISTRIBUTOR_ABI } from './merkle_distributor';
+
+// 合约地址常量
+export const CONTRACT_ADDRESS = '0x90f5a927d597d49b4fb64a48f3dc34695687e389136b04c42f9b9e7fe7fe59a0';
+
+// 模块名称映射
+export const MODULE_NAMES = {
+  OWNABLE: 'ownable',
+  MD_CREATE2: 'md_create2',
+  FEE_COLLECTOR: 'fee_collector',
+  MERKLE_VERIFIER: 'merkle_verifier',
+  MERKLE_DISTRIBUTOR: 'merkle_distributor'
+} as const;