@ledgerhq/coin-aptos 2.0.0-nightly.5 → 2.1.0-next.0

package/src/network/client.ts

@@ -0,0 +1,374 @@
+import { ApolloClient, InMemoryCache } from "@apollo/client";
+
+import {
+  type AccountData,
+  Aptos,
+  AptosConfig,
+  Ed25519PublicKey,
+  type GasEstimation,
+  type InputEntryFunctionData,
+  type InputGenerateTransactionOptions,
+  MimeType,
+  type RawTransaction,
+  type SimpleTransaction,
+  type TransactionResponse,
+  type UserTransactionResponse,
+  type Block,
+  type AptosSettings,
+  Hex,
+  postAptosFullNode,
+  type PendingTransactionResponse,
+} from "@aptos-labs/ts-sdk";
+import { getEnv } from "@ledgerhq/live-env";
+import network from "@ledgerhq/live-network";
+import BigNumber from "bignumber.js";
+import isUndefined from "lodash/isUndefined";
+import {
+  APTOS_ASSET_ID,
+  DEFAULT_GAS,
+  DEFAULT_GAS_PRICE,
+  ESTIMATE_GAS_MUL,
+  TOKEN_TYPE,
+} from "../constants";
+import type { AptosBalance, AptosTransaction, TransactionOptions } from "../types";
+import { GetAccountTransactionsData, GetAccountTransactionsDataGt } from "./graphql/queries";
+import type {
+  GetAccountTransactionsDataQuery,
+  GetAccountTransactionsDataGtQueryVariables,
+} from "./graphql/types";
+import { TokenCurrency } from "@ledgerhq/types-cryptoassets";
+import {
+  BlockInfo,
+  FeeEstimation,
+  Operation,
+  Pagination,
+  TransactionIntent,
+} from "@ledgerhq/coin-framework/api/types";
+import { AptosAsset, AptosExtra, AptosFeeParameters, AptosSender } from "../types/assets";
+import { log } from "@ledgerhq/logs";
+import { transactionsToOperations } from "../logic/transactionsToOperations";
+import { isTestnet } from "../logic/isTestnet";
+import { normalizeAddress } from "../logic/normalizeAddress";
+
+const getApiEndpoint = (currencyId: string) =>
+  isTestnet(currencyId) ? getEnv("APTOS_TESTNET_API_ENDPOINT") : getEnv("APTOS_API_ENDPOINT");
+const getIndexerEndpoint = (currencyId: string) =>
+  isTestnet(currencyId)
+    ? getEnv("APTOS_TESTNET_INDEXER_ENDPOINT")
+    : getEnv("APTOS_INDEXER_ENDPOINT");
+
+export class AptosAPI {
+  private readonly aptosConfig: AptosConfig;
+  private readonly aptosClient: Aptos;
+  private readonly apolloClient: ApolloClient<object>;
+
+  constructor(currencyIdOrSettings: AptosSettings | string) {
+    if (typeof currencyIdOrSettings === "string") {
+      this.aptosConfig = new AptosConfig({
+        fullnode: getApiEndpoint(currencyIdOrSettings),
+        indexer: getIndexerEndpoint(currencyIdOrSettings),
+      });
+    } else {
+      this.aptosConfig = new AptosConfig(currencyIdOrSettings);
+    }
+
+    this.aptosClient = new Aptos(this.aptosConfig);
+    this.apolloClient = new ApolloClient({
+      uri: this.aptosConfig.indexer ?? "",
+      cache: new InMemoryCache(),
+      headers: {
+        "x-client": "ledger-live",
+      },
+    });
+  }
+
+  async getAccount(address: string): Promise<AccountData> {
+    return this.aptosClient.getAccountInfo({ accountAddress: address });
+  }
+
+  async getAccountInfo(address: string, startAt?: string) {
+    const [balance, transactions, blockHeight] = await Promise.all([
+      this.getCoinBalance(address, APTOS_ASSET_ID),
+      this.fetchTransactions(address, startAt),
+      this.getHeight(),
+    ]);
+
+    return {
+      balance,
+      transactions,
+      blockHeight,
+    };
+  }
+
+  async estimateGasPrice(): Promise<GasEstimation> {
+    return this.aptosClient.getGasPriceEstimation();
+  }
+
+  async generateTransaction(
+    address: string,
+    payload: InputEntryFunctionData,
+    options: TransactionOptions,
+  ): Promise<RawTransaction> {
+    const opts: Partial<InputGenerateTransactionOptions> = {};
+    if (!isUndefined(options.maxGasAmount)) {
+      opts.maxGasAmount = Number(options.maxGasAmount);
+    }
+
+    if (!isUndefined(options.gasUnitPrice)) {
+      opts.gasUnitPrice = Number(options.gasUnitPrice);
+    }
+
+    try {
+      const { ledger_timestamp } = await this.aptosClient.getLedgerInfo();
+      opts.expireTimestamp = Number(Math.ceil(+ledger_timestamp / 1_000_000 + 2 * 60)); // in milliseconds
+    } catch {
+      // skip
+    }
+
+    return this.aptosClient.transaction.build
+      .simple({
+        sender: address,
+        data: payload,
+        options: opts,
+      })
+      .then(t => t.rawTransaction)
+      .catch(error => {
+        throw error;
+      });
+  }
+
+  async simulateTransaction(
+    address: Ed25519PublicKey,
+    tx: RawTransaction,
+    options = {
+      estimateGasUnitPrice: true,
+      estimateMaxGasAmount: true,
+      estimatePrioritizedGasUnitPrice: false,
+    },
+  ): Promise<UserTransactionResponse[]> {
+    return this.aptosClient.transaction.simulate.simple({
+      signerPublicKey: address,
+      transaction: { rawTransaction: tx } as SimpleTransaction,
+      options,
+    });
+  }
+
+  async broadcast(tx: string): Promise<string> {
+    const txBytes = Hex.fromHexString(tx).toUint8Array();
+
+    const pendingTx = await postAptosFullNode<Uint8Array, PendingTransactionResponse>({
+      aptosConfig: this.aptosClient.config,
+      body: txBytes,
+      path: "transactions",
+      originMethod: "",
+      contentType: MimeType.BCS_SIGNED_TRANSACTION,
+    });
+
+    return pendingTx.data.hash;
+  }
+
+  async getBalance(address: string, token: TokenCurrency): Promise<BigNumber> {
+    if (token.tokenType === "coin") {
+      return await this.getCoinBalance(address, token.contractAddress);
+    }
+
+    return await this.getFABalance(address, token.contractAddress);
+  }
+
+  async getLastBlock(): Promise<BlockInfo> {
+    const { block_height } = await this.aptosClient.getLedgerInfo();
+    const block = await this.aptosClient.getBlockByHeight({ blockHeight: Number(block_height) });
+    return {
+      height: Number(block.block_height),
+      hash: block.block_hash,
+      time: new Date(Number(block.block_timestamp) / 1_000),
+    };
+  }
+
+  async getCoinBalance(address: string, contract_address: string): Promise<BigNumber> {
+    try {
+      const [balanceStr] = await this.aptosClient.view<[string]>({
+        payload: {
+          function: "0x1::coin::balance",
+          typeArguments: [contract_address],
+          functionArguments: [address],
+        },
+      });
+      const balance = parseInt(balanceStr, 10);
+      return new BigNumber(balance);
+    } catch (error) {
+      log("error", "getCoinBalance", {
+        error,
+      });
+      return new BigNumber(0);
+    }
+  }
+
+  async getFABalance(address: string, contract_address: string): Promise<BigNumber> {
+    try {
+      const [balanceStr] = await this.aptosClient.view<[string]>({
+        payload: {
+          function: "0x1::primary_fungible_store::balance",
+          typeArguments: ["0x1::object::ObjectCore"],
+          functionArguments: [address, contract_address],
+        },
+      });
+      const balance = parseInt(balanceStr, 10);
+      return new BigNumber(balance);
+    } catch (error) {
+      log("error", "getFABalance", {
+        error,
+      });
+      return new BigNumber(0);
+    }
+  }
+
+  async estimateFees(
+    transactionIntent: TransactionIntent<AptosAsset, AptosExtra, AptosSender>,
+  ): Promise<FeeEstimation<AptosFeeParameters>> {
+    const publicKeyEd = new Ed25519PublicKey(transactionIntent.sender.xpub);
+
+    const txPayload: InputEntryFunctionData = {
+      function: "0x1::aptos_account::transfer_coins",
+      typeArguments: [APTOS_ASSET_ID],
+      functionArguments: [transactionIntent.recipient, transactionIntent.amount],
+    };
+
+    if (transactionIntent.asset.type === "token") {
+      const { standard } = transactionIntent.asset;
+
+      if (standard === TOKEN_TYPE.FUNGIBLE_ASSET) {
+        txPayload.function = "0x1::primary_fungible_store::transfer";
+        txPayload.typeArguments = ["0x1::fungible_asset::Metadata"];
+        txPayload.functionArguments = [
+          transactionIntent.asset.contractAddress,
+          transactionIntent.recipient,
+          transactionIntent.amount,
+        ];
+      } else if (standard === TOKEN_TYPE.COIN) {
+        txPayload.function = "0x1::aptos_account::transfer_coins";
+        txPayload.typeArguments = [transactionIntent.asset.contractAddress];
+      }
+    }
+
+    const txOptions: TransactionOptions = {
+      maxGasAmount: DEFAULT_GAS.toString(),
+      gasUnitPrice: DEFAULT_GAS_PRICE.toString(),
+    };
+
+    const tx = await this.generateTransaction(
+      transactionIntent.sender.freshAddress,
+      txPayload,
+      txOptions,
+    );
+
+    const simulation = await this.simulateTransaction(publicKeyEd, tx);
+    const completedTx = simulation[0];
+
+    const gasLimit = new BigNumber(completedTx.gas_used).multipliedBy(ESTIMATE_GAS_MUL);
+    const gasPrice = new BigNumber(completedTx.gas_unit_price);
+
+    const expectedGas = gasPrice.multipliedBy(gasLimit);
+
+    return {
+      value: BigInt(expectedGas.toString()),
+      parameters: {
+        gasLimit: BigInt(gasLimit.toString()),
+        gasPrice: BigInt(gasPrice.toString()),
+      },
+    };
+  }
+
+  async listOperations(
+    rawAddress: string,
+    pagination: Pagination,
+  ): Promise<[Operation<AptosAsset>[], string]> {
+    const address = normalizeAddress(rawAddress);
+    const transactions = await this.getAccountInfo(address, pagination.minHeight.toString());
+    const newOperations = transactionsToOperations(address, transactions.transactions);
+
+    return [newOperations, ""];
+  }
+
+  private async fetchTransactions(address: string, gt?: string) {
+    if (!address) {
+      return [];
+    }
+
+    let query = GetAccountTransactionsData;
+    if (gt) {
+      query = GetAccountTransactionsDataGt;
+    }
+
+    const queryResponse = await this.apolloClient.query<
+      GetAccountTransactionsDataQuery,
+      GetAccountTransactionsDataGtQueryVariables
+    >({
+      query,
+      variables: {
+        address,
+        limit: 1000,
+        gt,
+      },
+      fetchPolicy: "network-only",
+    });
+
+    return Promise.all(
+      queryResponse.data.account_transactions.map(({ transaction_version }) => {
+        return this.richItemByVersion(transaction_version);
+      }),
+    );
+  }
+
+  private async richItemByVersion(version: number): Promise<AptosTransaction | null> {
+    try {
+      const tx: TransactionResponse = await this.aptosClient.getTransactionByVersion({
+        ledgerVersion: version,
+      });
+      const block = await this.getBlock(version);
+      return {
+        ...tx,
+        block,
+      } as AptosTransaction;
+    } catch (error) {
+      log("error", "richItemByVersion", {
+        error,
+      });
+      return null;
+    }
+  }
+
+  private async getHeight(): Promise<number> {
+    const { data } = await network<Block>({
+      method: "GET",
+      url: this.aptosConfig.fullnode ?? "",
+    });
+    return parseInt(data.block_height);
+  }
+
+  private async getBlock(version: number) {
+    const block = await this.aptosClient.getBlockByVersion({ ledgerVersion: version });
+    return {
+      height: parseInt(block.block_height),
+      hash: block.block_hash,
+    };
+  }
+
+  async getBalances(address: string): Promise<AptosBalance[]> {
+    const response = await this.aptosClient.getCurrentFungibleAssetBalances({
+      options: {
+        offset: 0,
+        limit: 1000,
+        where: {
+          asset_type: { _eq: APTOS_ASSET_ID }, // to return all asset balances (native / token) we should remove this filter
+          owner_address: { _eq: address },
+        },
+      },
+    });
+
+    return response.map(x => ({
+      contractAddress: x.asset_type ?? "-",
+      amount: BigNumber(x.amount),
+    }));
+  }
+}

package/src/{api → network}/graphql/queries.ts

@@ -18,8 +18,8 @@ export const GetDelegatedStakingActivities = gql`
 `;
 export const GetAccountTransactionsData = gql`
   query GetAccountTransactionsData($address: String, $limit: Int) {
-    address_version_from_move_resources(
-      where: { address: { _eq: $address } }
+    account_transactions(
+      where: { account_address: { _eq: $address } }
       order_by: { transaction_version: desc }
       limit: $limit
     ) {
@@ -30,8 +30,8 @@ export const GetAccountTransactionsData = gql`
 `;
 export const GetAccountTransactionsDataGt = gql`
   query GetAccountTransactionsDataGt($address: String, $limit: Int, $gt: bigint) {
-    address_version_from_move_resources(
-      where: { address: { _eq: $address }, transaction_version: { _gt: $gt } }
+    account_transactions(
+      where: { account_address: { _eq: $address }, transaction_version: { _gt: $gt } }
       order_by: { transaction_version: desc }
       limit: $limit
     ) {
@@ -42,8 +42,8 @@ export const GetAccountTransactionsDataGt = gql`
 `;
 export const GetAccountTransactionsDataLt = gql`
   query GetAccountTransactionsDataLt($address: String, $limit: Int, $lt: bigint) {
-    address_version_from_move_resources(
-      where: { address: { _eq: $address }, transaction_version: { _lt: $lt } }
+    account_transactions(
+      where: { account_address: { _eq: $address }, transaction_version: { _lt: $lt } }
       order_by: { transaction_version: desc }
       limit: $limit
     ) {

package/src/{api → network}/graphql/types.ts

@@ -22,9 +22,9 @@ export type GetAccountTransactionsDataQueryVariables = Exact<{
 
 export type GetAccountTransactionsDataQuery = {
   __typename?: "query_root";
-  address_version_from_move_resources: Array<{
-    __typename: "address_version_from_move_resources";
-    transaction_version?: any | null;
+  account_transactions: Array<{
+    __typename?: "account_transactions";
+    transaction_version: number;
   }>;
 };
 
@@ -36,9 +36,9 @@ export type GetAccountTransactionsDataGtQueryVariables = Exact<{
 
 export type GetAccountTransactionsDataGtQuery = {
   __typename?: "query_root";
-  address_version_from_move_resources: Array<{
-    __typename: "address_version_from_move_resources";
-    transaction_version?: any | null;
+  account_transactions: Array<{
+    __typename?: "account_transactions";
+    transaction_version: number;
   }>;
 };
 
@@ -50,9 +50,9 @@ export type GetAccountTransactionsDataLtQueryVariables = Exact<{
 
 export type GetAccountTransactionsDataLtQuery = {
   __typename?: "query_root";
-  address_version_from_move_resources: Array<{
-    __typename: "address_version_from_move_resources";
-    transaction_version?: any | null;
+  account_transactions: Array<{
+    __typename?: "account_transactions";
+    transaction_version: number;
   }>;
 };
 

package/src/network/index.ts

@@ -1,18 +1,17 @@
 import {
   AccountAddress,
-  AccountAuthenticatorEd25519,
   Ed25519PublicKey,
   Ed25519Signature,
-  generateSignedTransaction,
   generateSigningMessageForTransaction,
-  Hex,
   RawTransaction,
   SimpleTransaction,
 } from "@aptos-labs/ts-sdk";
-import { SignerContext } from "@ledgerhq/coin-framework/lib/signer";
+import { SignerContext } from "@ledgerhq/coin-framework/signer";
 import { Account } from "@ledgerhq/types-live";
 import { AptosSigner } from "../types";
 import { sha3_256 as sha3Hash } from "@noble/hashes/sha3";
+import { combineSignedTransaction } from "../logic/combine";
+export * from "./client";
 
 export async function signTransaction(
   signerContext: SignerContext<AptosSigner>,
@@ -40,15 +39,8 @@ export async function signTransaction(
     async signer => await signer.signTransaction(derivationPath, Buffer.from(signingMessage)),
   );
 
-  const sigHexStr = Hex.fromHexString(response.signature.toString("hex"));
-  const signature = new Ed25519Signature(sigHexStr.toUint8Array());
-  const authenticator = new AccountAuthenticatorEd25519(
-    new Ed25519PublicKey(publicKey.toString("hex")),
-    signature,
-  );
+  const signature = new Ed25519Signature(response.signature.toString("hex"));
+  const pubkey = new Ed25519PublicKey(publicKey.toString("hex"));
 
-  return generateSignedTransaction({
-    transaction: { rawTransaction: rawTxn } as SimpleTransaction,
-    senderAuthenticator: authenticator,
-  });
+  return combineSignedTransaction(rawTxn, signature, pubkey);
 }

package/src/test/bot-specs.ts

@@ -5,8 +5,9 @@ import BigNumber from "bignumber.js";
 import type { Transaction } from "../types";
 import { getCryptoCurrencyById } from "@ledgerhq/cryptoassets/currencies";
 import { genericTestDestination, pickSiblings, botTest } from "@ledgerhq/coin-framework/bot/specs";
-import type { AppSpec } from "@ledgerhq/coin-framework/bot/types";
-import { acceptTransaction } from "./speculos-deviceActions";
+import type { AppSpec, TransactionTestInput } from "@ledgerhq/coin-framework/bot/types";
+import { acceptTokenTransaction, acceptTransaction } from "./speculos-deviceActions";
+import { Account, TokenAccount } from "@ledgerhq/types-live";
 
 const MIN_SAFE = new BigNumber(0.0001);
 const maxAccount = 6;
@@ -19,7 +20,7 @@ const aptosSpecs: AppSpec<Transaction> = {
     appName: "Aptos",
   },
   genericDeviceAction: acceptTransaction,
-  testTimeout: 6 * 60 * 1000,
+  testTimeout: 2 * 60 * 1000,
   minViableAmount: MIN_SAFE,
   transactionCheck: ({ maxSpendable }) => {
     invariant(maxSpendable.gt(MIN_SAFE), "balance is too low");
@@ -84,9 +85,68 @@ const aptosSpecs: AppSpec<Transaction> = {
         );
       },
     },
+    {
+      name: "Send ~50% of token amount",
+      feature: "tokens",
+      maxRun: 1,
+      deviceAction: acceptTokenTransaction,
+      transaction: ({ account, bridge, siblings, maxSpendable }) => {
+        invariant(maxSpendable.gt(MIN_SAFE), "Balance is too low");
+
+        const senderTokenAcc = findTokenSubAccountWithBalance(account);
+        invariant(senderTokenAcc, "Sender token account with available balance not found");
+
+        const sibling = pickSiblings(siblings, maxAccount);
+
+        const recipientTokenAcc = findTokenSubAccountWithBalance(sibling);
+        invariant(recipientTokenAcc, "Receiver token account with available balance not found");
+
+        const amount = senderTokenAcc.spendableBalance.div(2).integerValue();
+        const recipient = sibling.freshAddress;
+        const transaction = bridge.createTransaction(account);
+        const subAccountId = senderTokenAcc.id;
+
+        return {
+          transaction,
+          updates: [{ subAccountId }, { recipient }, { amount }],
+        };
+      },
+      test: input => {
+        expectTokenAccountCorrectBalanceChange(input);
+      },
+    },
   ],
 };
 
+function findTokenSubAccountWithBalance(account: Account) {
+  return account.subAccounts?.find(acc => acc.type === "TokenAccount" && acc.balance.gt(0)) as
+    | TokenAccount
+    | undefined;
+}
+
+function expectTokenAccountCorrectBalanceChange({
+  account,
+  accountBeforeTransaction,
+  status,
+  transaction,
+}: TransactionTestInput<Transaction>) {
+  const tokenAccId = transaction.subAccountId;
+  if (!tokenAccId) throw new Error("Wrong subAccountId");
+
+  const tokenAccAfterTx = account.subAccounts?.find(acc => acc.id === tokenAccId);
+  const tokenAccBeforeTx = accountBeforeTransaction.subAccounts?.find(acc => acc.id === tokenAccId);
+
+  if (!tokenAccAfterTx || !tokenAccBeforeTx) {
+    throw new Error("Token sub accounts not found!");
+  }
+
+  botTest("Token balance decreased with operation", () =>
+    expect(tokenAccAfterTx.balance.toString()).toBe(
+      tokenAccBeforeTx.balance.minus(status.amount).toString(),
+    ),
+  );
+}
+
 export default {
   aptosSpecs,
 };