@aptos-labs/cross-chain-core 4.25.0 → 5.0.1

package/dist/index.mjs

@@ -416,14 +416,92 @@ async function signAndSendTransaction3(request, wallet, chainName, options) {
   return receipt?.hash || "";
 }
 
+// src/providers/wormhole/signers/AptosSigner.ts
+import {
+  AccountAddress,
+  Aptos as Aptos2,
+  AptosConfig as AptosConfig2,
+  Network as AptosNetwork2,
+  Deserializer
+} from "@aptos-labs/ts-sdk";
+import { UserResponseStatus } from "@aptos-labs/wallet-standard";
+async function signAndSendTransaction4(request, wallet, sponsorAccount) {
+  if (!wallet) {
+    throw new Error("wallet.sendTransaction is undefined").message;
+  }
+  const payload = request.transaction;
+  payload.functionArguments = payload.functionArguments.map((a) => {
+    if (a instanceof Uint8Array) {
+      return Array.from(a);
+    } else if (typeof a === "bigint") {
+      return a.toString();
+    } else {
+      return a;
+    }
+  });
+  const aptosConfig = new AptosConfig2({
+    network: AptosNetwork2.TESTNET
+  });
+  const aptos2 = new Aptos2(aptosConfig);
+  const contractAddress = AptosNetwork2.TESTNET ? "0x5e2d961f06cd27aa07554a39d55f5ce1e58dff35d803c3529b1cd5c4fa3ab584" : "0x1";
+  const functionArguments = extractFunctionArguments(
+    payload.functionArguments
+  );
+  const transactionData = {
+    // a custom function to withdraw tokens from the aptos chain, published here on testnet:
+    // https://explorer.aptoslabs.com/account/0x5e2d961f06cd27aa07554a39d55f5ce1e58dff35d803c3529b1cd5c4fa3ab584/modules/code/withdraw?network=testnet
+    function: `${contractAddress}::withdraw::deposit_for_burn`,
+    functionArguments
+  };
+  const txnToSign = await aptos2.transaction.build.simple({
+    data: transactionData,
+    sender: (await wallet.features["aptos:account"]?.account()).address.toString(),
+    withFeePayer: sponsorAccount ? true : false
+  });
+  const response = await wallet.features["aptos:signTransaction"]?.signTransaction(txnToSign);
+  if (response?.status === UserResponseStatus.REJECTED) {
+    throw new Error("User has rejected the request");
+  }
+  const txnToSubmit = {
+    transaction: txnToSign,
+    senderAuthenticator: response.args
+  };
+  if (sponsorAccount) {
+    if (typeof sponsorAccount === "string") {
+    } else {
+      const feePayerSignerAuthenticator = aptos2.transaction.signAsFeePayer({
+        signer: sponsorAccount,
+        transaction: txnToSign
+      });
+      txnToSubmit.feePayerAuthenticator = feePayerSignerAuthenticator;
+    }
+  }
+  const txnSubmitted = await aptos2.transaction.submit.simple(txnToSubmit);
+  const tx = await aptos2.waitForTransaction({
+    transactionHash: txnSubmitted.hash
+  });
+  return tx.hash;
+}
+function extractFunctionArguments(functionArguments) {
+  const deserializer1 = new Deserializer(functionArguments[0].bcsToBytes());
+  const amount = deserializer1.deserializeU64();
+  const deserializer2 = new Deserializer(functionArguments[1].bcsToBytes());
+  const destination_domain = deserializer2.deserializeU32();
+  const mint_recipient = new AccountAddress(functionArguments[2].bcsToBytes());
+  const burn_token = new AccountAddress(functionArguments[3].bcsToBytes());
+  return [amount, destination_domain, mint_recipient, burn_token];
+}
+
 // src/providers/wormhole/signers/Signer.ts
 var Signer = class {
-  constructor(chain, address, options, wallet, crossChainCore) {
+  constructor(chain, address, options, wallet, crossChainCore, sponsorAccount) {
     this._chain = chain;
     this._address = address;
     this._options = options;
     this._wallet = wallet;
     this._crossChainCore = crossChainCore;
+    this._sponsorAccount = sponsorAccount;
+    this._claimedTransactionHashes = "";
   }
   chain() {
     return this._chain.displayName;
@@ -431,22 +509,27 @@ var Signer = class {
   address() {
     return this._address;
   }
+  claimedTransactionHashes() {
+    return this._claimedTransactionHashes;
+  }
   async signAndSend(txs) {
     const txHashes = [];
     for (const tx of txs) {
-      const txId = await signAndSendTransaction4(
+      const txId = await signAndSendTransaction5(
         this._chain,
         tx,
         this._wallet,
         this._options,
-        this._crossChainCore
+        this._crossChainCore,
+        this._sponsorAccount
       );
       txHashes.push(txId);
+      this._claimedTransactionHashes = txId;
     }
     return txHashes;
   }
 };
-var signAndSendTransaction4 = async (chain, request, wallet, options = {}, crossChainCore) => {
+var signAndSendTransaction5 = async (chain, request, wallet, options = {}, crossChainCore, sponsorAccount) => {
   if (!wallet) {
     throw new Error("wallet is undefined");
   }
@@ -466,6 +549,13 @@ var signAndSendTransaction4 = async (chain, request, wallet, options = {}, cross
       options
     );
     return tx;
+  } else if (chain.context === "Aptos") {
+    const tx = await signAndSendTransaction4(
+      request,
+      wallet,
+      sponsorAccount
+    );
+    return tx;
   } else {
     throw new Error(`Unsupported chain: ${chain}`);
   }
@@ -492,15 +582,20 @@ var WormholeProvider = class {
     const wh = await wormhole(isMainnet ? "Mainnet" : "Testnet", platforms);
     this._wormholeContext = wh;
   }
-  async getRoute(sourceChain) {
+  async getRoute(sourceChain, destinationChain) {
     if (!this._wormholeContext) {
       throw new Error("Wormhole context not initialized");
     }
-    const { sourceToken, destToken } = this.getTokenInfo(sourceChain);
+    const { sourceToken, destToken } = this.getTokenInfo(
+      sourceChain,
+      destinationChain
+    );
+    const destContext = this._wormholeContext.getPlatform(chainToPlatform(destinationChain)).getChain(destinationChain);
     const sourceContext = this._wormholeContext.getPlatform(chainToPlatform(sourceChain)).getChain(sourceChain);
     logger.log("sourceContext", sourceContext);
-    const destContext = this._wormholeContext.getPlatform(chainToPlatform("Aptos")).getChain("Aptos");
+    logger.log("sourceToken", sourceToken);
     logger.log("destContext", destContext);
+    logger.log("destToken", destToken);
     const request = await routes.RouteTransferRequest.create(
       this._wormholeContext,
       {
@@ -521,11 +616,17 @@ var WormholeProvider = class {
     return { route: cctpRoute, request };
   }
   async getQuote(input) {
-    const { amount, sourceChain } = input;
+    const { amount, originChain, type } = input;
     if (!this._wormholeContext) {
-      await this.setWormholeContext(sourceChain);
+      await this.setWormholeContext(originChain);
     }
-    const { route, request } = await this.getRoute(sourceChain);
+    logger.log("type", type);
+    const sourceChain = type === "transfer" ? originChain : "Aptos";
+    const destinationChain = type === "transfer" ? "Aptos" : originChain;
+    const { route, request } = await this.getRoute(
+      sourceChain,
+      destinationChain
+    );
     const transferParams = {
       amount,
       options: { nativeGas: 0 }
@@ -631,14 +732,15 @@ var WormholeProvider = class {
    * @param args
    * @returns
    */
-  async initiateCCTPTransfer(input) {
+  async transfer(input) {
     if (this.crossChainCore._dappConfig?.aptosNetwork === Network.DEVNET) {
       throw new Error("Devnet is not supported on Wormhole");
     }
     if (input.amount) {
       await this.getQuote({
         amount: input.amount,
-        sourceChain: input.sourceChain
+        originChain: input.sourceChain,
+        type: "transfer"
       });
     }
     let { originChainTxnId, receipt } = await this.submitCCTPTransfer(input);
@@ -649,6 +751,97 @@ var WormholeProvider = class {
     });
     return { originChainTxnId, destinationChainTxnId };
   }
+  async withdraw(input) {
+    const { sourceChain, wallet, destinationAddress, sponsorAccount } = input;
+    logger.log("sourceChain", sourceChain);
+    logger.log("wallet", wallet);
+    logger.log("destinationAddress", destinationAddress);
+    logger.log("sponsorAccount", sponsorAccount);
+    if (!this._wormholeContext) {
+      await this.setWormholeContext(sourceChain);
+    }
+    if (!this._wormholeContext) {
+      throw new Error("Wormhole context not initialized");
+    }
+    if (!this.wormholeRoute || !this.wormholeRequest || !this.wormholeQuote) {
+      throw new Error("Wormhole route, request, or quote not initialized");
+    }
+    const signer = new Signer(
+      this.getChainConfig("Aptos"),
+      (await input.wallet.features["aptos:account"].account()).address.toString(),
+      {},
+      input.wallet,
+      void 0,
+      sponsorAccount
+    );
+    logger.log("signer", signer);
+    logger.log("wormholeRequest", this.wormholeRequest);
+    logger.log("wormholeQuote", this.wormholeQuote);
+    logger.log(
+      "Wormhole.chainAddress",
+      Wormhole.chainAddress(sourceChain, input.destinationAddress.toString())
+    );
+    let receipt = await this.wormholeRoute.initiate(
+      this.wormholeRequest,
+      signer,
+      this.wormholeQuote,
+      Wormhole.chainAddress(sourceChain, input.destinationAddress.toString())
+    );
+    logger.log("receipt", receipt);
+    const originChainTxnId = "originTxs" in receipt ? receipt.originTxs[receipt.originTxs.length - 1].txid : void 0;
+    let retries = 0;
+    const maxRetries = 5;
+    const baseDelay = 1e3;
+    while (retries < maxRetries) {
+      try {
+        for await (receipt of this.wormholeRoute.track(receipt, 120 * 1e3)) {
+          if (receipt.state >= TransferState.SourceInitiated) {
+            logger.log("Receipt is on track ", receipt);
+            try {
+              const signer2 = new Signer(
+                this.getChainConfig(sourceChain),
+                destinationAddress.toString(),
+                {},
+                wallet
+              );
+              if (routes.isManual(this.wormholeRoute)) {
+                const circleAttestationReceipt = await this.wormholeRoute.complete(signer2, receipt);
+                logger.log("Claim receipt: ", circleAttestationReceipt);
+                const destinationChainTxnId = signer2.claimedTransactionHashes();
+                return {
+                  originChainTxnId: originChainTxnId || "",
+                  destinationChainTxnId
+                };
+              } else {
+                return {
+                  originChainTxnId: originChainTxnId || "",
+                  destinationChainTxnId: ""
+                };
+              }
+            } catch (e) {
+              console.error("Failed to claim", e);
+              return {
+                originChainTxnId: originChainTxnId || "",
+                destinationChainTxnId: ""
+              };
+            }
+          }
+        }
+      } catch (e) {
+        console.error(
+          `Error tracking transfer (attempt ${retries + 1} / ${maxRetries}):`,
+          e
+        );
+        const delay = baseDelay * Math.pow(2, retries);
+        await sleep2(delay);
+        retries++;
+      }
+    }
+    return {
+      originChainTxnId: originChainTxnId || "",
+      destinationChainTxnId: ""
+    };
+  }
   getChainConfig(chain) {
     const chainConfig = this.crossChainCore.CHAINS[chain];
     if (!chainConfig) {
@@ -656,14 +849,14 @@ var WormholeProvider = class {
     }
     return chainConfig;
   }
-  getTokenInfo(sourceChain) {
+  getTokenInfo(sourceChain, destinationChain) {
     const sourceToken = Wormhole.tokenId(
       this.crossChainCore.TOKENS[sourceChain].tokenId.chain,
       this.crossChainCore.TOKENS[sourceChain].tokenId.address
     );
     const destToken = Wormhole.tokenId(
-      this.crossChainCore.APTOS_TOKEN.tokenId.chain,
-      this.crossChainCore.APTOS_TOKEN.tokenId.address
+      this.crossChainCore.TOKENS[destinationChain].tokenId.chain,
+      this.crossChainCore.TOKENS[destinationChain].tokenId.address
     );
     return { sourceToken, destToken };
   }
@@ -710,6 +903,21 @@ var testnetChains = {
     symbol: "SOL",
     defaultRpc: "https://api.devnet.solana.com",
     wrappedGasToken: "So11111111111111111111111111111111111111112"
+  },
+  Aptos: {
+    key: "Aptos",
+    id: 22,
+    context: "Aptos" /* APTOS */,
+    finalityThreshold: 0,
+    displayName: "Aptos",
+    explorerUrl: "https://explorer.aptoslabs.com?network=testnet",
+    explorerName: "Aptos Explorer",
+    gasToken: "APT",
+    chainId: 0,
+    icon: "Aptos",
+    maxBlockSearch: 0,
+    symbol: "APT",
+    defaultRpc: "https://fullnode.testnet.aptos.dev"
   }
   // Sui: {
   //   key: "Sui",
@@ -727,21 +935,6 @@ var testnetChains = {
   //   sdkName: "Sui",
   // },
 };
-var AptosTestnetChain = {
-  key: "Aptos",
-  id: 22,
-  context: "Aptos" /* APTOS */,
-  finalityThreshold: 0,
-  displayName: "Aptos",
-  explorerUrl: "https://explorer.aptoslabs.com?network=testnet",
-  explorerName: "Aptos Explorer",
-  gasToken: "APT",
-  chainId: 0,
-  icon: "Aptos",
-  maxBlockSearch: 0,
-  symbol: "APT",
-  sdkName: "Aptos"
-};
 
 // src/config/testnet/tokens.ts
 var testnetTokens = {
@@ -762,6 +955,15 @@ var testnetTokens = {
     },
     icon: "USDC",
     decimals: 6
+  },
+  Aptos: {
+    symbol: "USDC",
+    decimals: 6,
+    tokenId: {
+      chain: "Aptos",
+      address: "0x69091fbab5f7d635ee7ac5098cf0c1efbe31d68fec0f2cd565e8d168daf52832"
+    },
+    icon: "USDC"
   }
   // Sui: {
   //   symbol: "USDC",
@@ -774,15 +976,6 @@ var testnetTokens = {
   //   decimals: 6,
   // },
 };
-var AptosTestnetUSDCToken = {
-  symbol: "USDC",
-  decimals: 6,
-  tokenId: {
-    chain: "Aptos",
-    address: "0x69091fbab5f7d635ee7ac5098cf0c1efbe31d68fec0f2cd565e8d168daf52832"
-  },
-  icon: "USDC"
-};
 
 // src/config/mainnet/chains.ts
 var mainnetChains = {
@@ -815,6 +1008,21 @@ var mainnetChains = {
     maxBlockSearch: 2e3,
     symbol: "SOL",
     defaultRpc: "https://solana-mainnet.rpc.extrnode.com"
+  },
+  Aptos: {
+    key: "Aptos",
+    id: 22,
+    context: "Aptos" /* APTOS */,
+    finalityThreshold: 0,
+    displayName: "Aptos",
+    explorerUrl: "https://explorer.aptoslabs.com/",
+    explorerName: "Aptos Explorer",
+    gasToken: "APT",
+    chainId: 0,
+    icon: "Aptos",
+    maxBlockSearch: 0,
+    symbol: "APT",
+    defaultRpc: "https://fullnode.mainnet.aptos.dev"
   }
   // Sui: {
   //   key: "Sui",
@@ -832,20 +1040,6 @@ var mainnetChains = {
   //   symbol: "SUI",
   // },
 };
-var AptosMainnetChain = {
-  key: "Aptos",
-  id: 22,
-  context: "Aptos",
-  finalityThreshold: 0,
-  displayName: "Aptos",
-  explorerUrl: "https://explorer.aptoslabs.com/",
-  explorerName: "Aptos Explorer",
-  gasToken: "APT",
-  chainId: 0,
-  icon: "Aptos",
-  maxBlockSearch: 0,
-  symbol: "APT"
-};
 
 // src/config/mainnet/tokens.ts
 var mainnetTokens = {
@@ -866,6 +1060,15 @@ var mainnetTokens = {
     },
     icon: "USDC",
     decimals: 6
+  },
+  Aptos: {
+    symbol: "USDC",
+    decimals: 6,
+    tokenId: {
+      chain: "Aptos",
+      address: "0xbae207659db88bea0cbead6da0ed00aac12edcdda169e591cd41c94180b46f3b"
+    },
+    icon: "USDC"
   }
   // Sui: {
   //   symbol: "USDC",
@@ -878,18 +1081,9 @@ var mainnetTokens = {
   //   icon: "USDC",
   // },
 };
-var AptosMainnetUSDCToken = {
-  symbol: "USDC",
-  tokenId: {
-    chain: "Aptos",
-    address: "0xbae207659db88bea0cbead6da0ed00aac12edcdda169e591cd41c94180b46f3b"
-  },
-  icon: "USDC",
-  decimals: 6
-};
 
 // src/utils/getUsdcBalance.ts
-import { Aptos as Aptos2, AptosConfig as AptosConfig2, Network as Network2 } from "@aptos-labs/ts-sdk";
+import { Aptos as Aptos3, AptosConfig as AptosConfig3, Network as Network2 } from "@aptos-labs/ts-sdk";
 import { Connection as Connection2, PublicKey } from "@solana/web3.js";
 import { ethers as ethers2, JsonRpcProvider } from "ethers";
 var getSolanaWalletUSDCBalance = async (walletAddress, aptosNetwork, rpc) => {
@@ -913,10 +1107,10 @@ var getEthereumWalletUSDCBalance = async (walletAddress, aptosNetwork, rpc) => {
   return ethers2.formatUnits(balance, token.decimals).toString();
 };
 var getAptosWalletUSDCBalance = async (walletAddress, aptosNetwork) => {
-  const token = aptosNetwork === Network2.MAINNET ? AptosMainnetUSDCToken : AptosTestnetUSDCToken;
+  const token = aptosNetwork === Network2.MAINNET ? mainnetTokens["Aptos"] : testnetTokens["Aptos"];
   const tokenAddress = token.tokenId.address;
-  const aptosConfig = new AptosConfig2({ network: aptosNetwork });
-  const connection = new Aptos2(aptosConfig);
+  const aptosConfig = new AptosConfig3({ network: aptosNetwork });
+  const connection = new Aptos3(aptosConfig);
   const response = await connection.getCurrentFungibleAssetBalances({
     options: {
       where: {
@@ -925,6 +1119,9 @@ var getAptosWalletUSDCBalance = async (walletAddress, aptosNetwork) => {
       }
     }
   });
+  if (response.length === 0) {
+    return "0";
+  }
   const balance = (Number(response[0].amount) / 10 ** token.decimals).toString();
   return balance;
 };
@@ -938,16 +1135,13 @@ var CrossChainCore = class {
     };
     this.CHAINS = testnetChains;
     this.TOKENS = testnetTokens;
-    this.APTOS_TOKEN = AptosTestnetUSDCToken;
     this._dappConfig = args.dappConfig;
     if (args.dappConfig?.aptosNetwork === Network3.MAINNET) {
       this.CHAINS = mainnetChains;
       this.TOKENS = mainnetTokens;
-      this.APTOS_TOKEN = AptosMainnetUSDCToken;
     } else {
       this.CHAINS = testnetChains;
       this.TOKENS = testnetTokens;
-      this.APTOS_TOKEN = AptosTestnetUSDCToken;
     }
   }
   getProvider(providerType) {
@@ -993,10 +1187,6 @@ var CrossChainCore = class {
 import { Network as Network4 } from "@aptos-labs/ts-sdk";
 export {
   AptosLocalSigner,
-  AptosMainnetChain,
-  AptosMainnetUSDCToken,
-  AptosTestnetChain,
-  AptosTestnetUSDCToken,
   Context,
   CrossChainCore,
   Network4 as Network,