@aptos-labs/ts-sdk 1.35.0-zeta.2 → 1.36.0-zeta.0

package/src/transactions/transactionBuilder/remoteAbi.ts

@@ -23,9 +23,9 @@ import {
   FunctionABI,
   TypeArgument,
 } from "../types";
-import { Bool, MoveOption, MoveString, MoveVector, U128, U16, U256, U32, U64, U8 } from "../../bcs";
+import { Bool, FixedBytes, MoveOption, MoveString, MoveVector, U128, U16, U256, U32, U64, U8 } from "../../bcs";
 import { AccountAddress } from "../../core";
-import { getModule } from "../../internal/utils";
+import { getModule } from "../../internal/account";
 import {
   findFirstNonSignerArg,
   isBcsAddress,
@@ -45,7 +45,7 @@ import {
   throwTypeMismatch,
   convertNumber,
 } from "./helpers";
-import { CallArgument, MoveFunction } from "../../types";
+import { CallArgument, MoveFunction, MoveModule } from "../../types";
 
 const TEXT_ENCODER = new TextEncoder();
 
@@ -69,6 +69,24 @@ export function standardizeTypeTags(typeArguments?: Array<TypeArgument>): Array<
   );
 }
 
+/**
+ * Fetches the ABI of a specified module from the on-chain module ABI.
+ *
+ * @param moduleAddress - The address of the module from which to fetch the ABI.
+ * @param moduleName - The name of the module containing the ABI.
+ * @param aptosConfig - The configuration settings for Aptos.
+ * @group Implementation
+ * @category Transactions
+ */
+export async function fetchModuleAbi(
+  moduleAddress: string,
+  moduleName: string,
+  aptosConfig: AptosConfig,
+): Promise<MoveModule | undefined> {
+  const moduleBytecode = await getModule({ aptosConfig, accountAddress: moduleAddress, moduleName });
+  return moduleBytecode.abi;
+}
+
 /**
  * Fetches the ABI of a specified function from the on-chain module ABI. This function allows you to access the details of a
  * specific function within a module.
@@ -86,22 +104,13 @@ export async function fetchFunctionAbi(
   functionName: string,
   aptosConfig: AptosConfig,
 ): Promise<MoveFunction | undefined> {
-  // This fetch from the API is currently cached
-  const module = await getModule({ aptosConfig, accountAddress: moduleAddress, moduleName });
-
-  if (module.abi) {
-    return module.abi.exposed_functions.find((func) => func.name === functionName);
-  }
-
-  return undefined;
+  const moduleAbi = await fetchModuleAbi(moduleAddress, moduleName, aptosConfig);
+  if (!moduleAbi) throw new Error(`Could not find module ABI for '${moduleAddress}::${moduleName}'`);
+  return moduleAbi.exposed_functions.find((func) => func.name === functionName);
 }
 
 /**
- * Fetches a function ABI from the on-chain module ABI.  It doesn't validate whether it's a view or entry function.
- * @param moduleAddress
- * @param moduleName
- * @param functionName
- * @param aptosConfig
+ * @deprecated Use `fetchFunctionAbi` instead and manually parse the type tags.
  */
 export async function fetchMoveFunctionAbi(
   moduleAddress: string,
@@ -220,15 +229,14 @@ export async function fetchViewFunctionAbi(
 }
 
 /**
- * Converts a entry function argument into CallArgument, if necessary.
- * This function checks the provided argument against the expected parameter type and converts it accordingly.
+ * @deprecated Handle this inline
  *
- * @param functionName - The name of the function for which the argument is being converted.
- * @param functionAbi - The ABI (Application Binary Interface) of the function, which defines its parameters.
- * @param argument - The argument to be converted, which can be of various types. If the argument is already
- *                   CallArgument returned from TransactionComposer it would be returned immediately.
- * @param position - The index of the argument in the function's parameter list.
- * @param genericTypeParams - An array of type tags for any generic type parameters.
+ * @example
+ * ```typescript
+ * const callArgument = argument instanceof CallArgument ? argument : CallArgument.newBytes(
+ *   convertArgument(functionName, functionAbi, argument, position, genericTypeParams).bcsToBytes()
+ * );
+ * ```
  */
 export function convertCallArgument(
   argument: CallArgument | EntryFunctionArgumentTypes | SimpleEntryFunctionArgumentTypes,
@@ -250,27 +258,54 @@ export function convertCallArgument(
  * This function checks the provided argument against the expected parameter type and converts it accordingly.
  *
  * @param functionName - The name of the function for which the argument is being converted.
- * @param functionAbi - The ABI (Application Binary Interface) of the function, which defines its parameters.
+ * @param functionAbiOrModuleAbi - The ABI (Application Binary Interface) of the function, which defines its parameters.
  * @param arg - The argument to be converted, which can be of various types.
  * @param position - The index of the argument in the function's parameter list.
  * @param genericTypeParams - An array of type tags for any generic type parameters.
+ * @param options - Options for the conversion process.
+ * @param options.allowUnknownStructs - If true, unknown structs will be allowed and converted to a `FixedBytes`.
  * @group Implementation
  * @category Transactions
  */
 export function convertArgument(
   functionName: string,
-  functionAbi: FunctionABI,
+  functionAbiOrModuleAbi: MoveModule | FunctionABI,
   arg: EntryFunctionArgumentTypes | SimpleEntryFunctionArgumentTypes,
   position: number,
   genericTypeParams: Array<TypeTag>,
+  options?: { allowUnknownStructs?: boolean },
 ) {
-  // Ensure not too many arguments
-  if (position >= functionAbi.parameters.length) {
-    throw new Error(`Too many arguments for '${functionName}', expected ${functionAbi.parameters.length}`);
+  let param: TypeTag;
+
+  if ("exposed_functions" in functionAbiOrModuleAbi) {
+    const functionAbi = functionAbiOrModuleAbi.exposed_functions.find((func) => func.name === functionName);
+    if (!functionAbi) {
+      throw new Error(
+        `Could not find function ABI for '${functionAbiOrModuleAbi.address}::${functionAbiOrModuleAbi.name}::${functionName}'`,
+      );
+    }
+
+    if (position >= functionAbi.params.length) {
+      throw new Error(`Too many arguments for '${functionName}', expected ${functionAbi.params.length}`);
+    }
+
+    param = parseTypeTag(functionAbi.params[position], { allowGenerics: true });
+  } else {
+    if (position >= functionAbiOrModuleAbi.parameters.length) {
+      throw new Error(`Too many arguments for '${functionName}', expected ${functionAbiOrModuleAbi.parameters.length}`);
+    }
+
+    param = functionAbiOrModuleAbi.parameters[position];
   }
 
-  const param = functionAbi.parameters[position];
-  return checkOrConvertArgument(arg, param, position, genericTypeParams);
+  return checkOrConvertArgument(
+    arg,
+    param,
+    position,
+    genericTypeParams,
+    "exposed_functions" in functionAbiOrModuleAbi ? functionAbiOrModuleAbi : undefined,
+    options,
+  );
 }
 
 /**
@@ -289,6 +324,8 @@ export function checkOrConvertArgument(
   param: TypeTag,
   position: number,
   genericTypeParams: Array<TypeTag>,
+  moduleAbi?: MoveModule,
+  options?: { allowUnknownStructs?: boolean },
 ) {
   // If the argument is bcs encoded, we can just use it directly
   if (isEncodedEntryFunctionArgument(arg)) {
@@ -301,6 +338,8 @@ export function checkOrConvertArgument(
      * @param typeArgs - The expected type arguments.
      * @param arg - The argument to be checked.
      * @param position - The position of the argument in the context of the check.
+     * @param moduleAbi - The ABI of the module containing the function, used for type checking.
+     *                    This will typically have information about structs, enums, and other types.
      * @group Implementation
      * @category Transactions
      */
@@ -309,7 +348,7 @@ export function checkOrConvertArgument(
   }
 
   // If it is not BCS encoded, we will need to convert it with the ABI
-  return parseArg(arg, param, position, genericTypeParams);
+  return parseArg(arg, param, position, genericTypeParams, moduleAbi, options);
 }
 
 /**
@@ -321,6 +360,10 @@ export function checkOrConvertArgument(
  * @param param - The type tag that defines the expected type of the argument.
  * @param position - The position of the argument in the function call, used for error reporting.
  * @param genericTypeParams - An array of type tags for generic type parameters, used when the parameter type is generic.
+ * @param moduleAbi - The ABI of the module containing the function, used for type checking.
+ *                    This will typically have information about structs, enums, and other types.
+ * @param options - Options for the conversion process.
+ * @param options.allowUnknownStructs - If true, unknown structs will be allowed and converted to a `FixedBytes`.
  * @group Implementation
  * @category Transactions
  */
@@ -329,6 +372,8 @@ function parseArg(
   param: TypeTag,
   position: number,
   genericTypeParams: Array<TypeTag>,
+  moduleAbi?: MoveModule,
+  options?: { allowUnknownStructs?: boolean },
 ): EntryFunctionArgumentTypes {
   if (param.isBool()) {
     if (isBool(arg)) {
@@ -403,7 +448,7 @@ function parseArg(
       throw new Error(`Generic argument ${param.toString()} is invalid for argument ${position}`);
     }
 
-    return checkOrConvertArgument(arg, genericTypeParams[genericIndex], position, genericTypeParams);
+    return checkOrConvertArgument(arg, genericTypeParams[genericIndex], position, genericTypeParams, moduleAbi);
   }
 
   // We have to special case some vectors for Vector<u8>
@@ -433,7 +478,9 @@ function parseArg(
     // TODO: Support Uint16Array, Uint32Array, BigUint64Array?
 
     if (Array.isArray(arg)) {
-      return new MoveVector(arg.map((item) => checkOrConvertArgument(item, param.value, position, genericTypeParams)));
+      return new MoveVector(
+        arg.map((item) => checkOrConvertArgument(item, param.value, position, genericTypeParams, moduleAbi)),
+      );
     }
 
     throw new Error(`Type mismatch for argument ${position}, type '${param.toString()}'`);
@@ -454,6 +501,13 @@ function parseArg(
       }
       throwTypeMismatch("string | AccountAddress", position);
     }
+    // Handle known enum types from Aptos framework
+    if (param.isDelegationKey() || param.isRateLimiter()) {
+      if (arg instanceof Uint8Array) {
+        return new FixedBytes(arg);
+      }
+      throwTypeMismatch("Uint8Array", position);
+    }
 
     if (param.isOption()) {
       if (isEmptyOption(arg)) {
@@ -490,7 +544,25 @@ function parseArg(
         return new MoveOption<MoveString>(null);
       }
 
-      return new MoveOption(checkOrConvertArgument(arg, param.value.typeArgs[0], position, genericTypeParams));
+      return new MoveOption(
+        checkOrConvertArgument(arg, param.value.typeArgs[0], position, genericTypeParams, moduleAbi),
+      );
+    }
+
+    // We are assuming that fieldless structs are enums, and therefore we cannot typecheck any further due
+    // to limited information from the ABI. This does not work for structs on other modules.
+    const structDefinition = moduleAbi?.structs.find((s) => s.name === param.value.name.identifier);
+    if (structDefinition?.fields.length === 0 && arg instanceof Uint8Array) {
+      return new FixedBytes(arg);
+    }
+
+    if (arg instanceof Uint8Array && options?.allowUnknownStructs) {
+      // eslint-disable-next-line no-console
+      console.warn(
+        // eslint-disable-next-line max-len
+        `Unsupported struct input type for argument ${position}. Continuing since 'allowUnknownStructs' is enabled.`,
+      );
+      return new FixedBytes(arg);
     }
 
     throw new Error(`Unsupported struct input type for argument ${position}, type '${param.toString()}'`);

package/src/transactions/typeTag/index.ts

@@ -639,6 +639,28 @@ export class TypeTagStruct extends TypeTag {
   isObject(): boolean {
     return this.isTypeTag(AccountAddress.ONE, "object", "Object");
   }
+
+  /**
+   * Checks if the provided value is a 'DelegationKey' for permissioned signers.
+   *
+   * @returns {boolean} Returns true if the value is a DelegationKey, otherwise false.
+   * @group Implementation
+   * @category Transactions
+   */
+  isDelegationKey(): boolean {
+    return this.isTypeTag(AccountAddress.ONE, "permissioned_delegation", "DelegationKey");
+  }
+
+  /**
+   * Checks if the provided value is of type `RateLimiter`.
+   *
+   * @returns {boolean} Returns true if the value is a RateLimiter, otherwise false.
+   * @group Implementation
+   * @category Transactions
+   */
+  isRateLimiter(): boolean {
+    return this.isTypeTag(AccountAddress.ONE, "rate_limiter", "RateLimiter");
+  }
 }
 
 /**

package/src/types/codegen.yaml

@@ -1,6 +1,6 @@
 overwrite: true
 documents: src/internal/queries/**/*.graphql
-schema: https://api.devnet.aptoslabs.com/v1/graphql
+schema: https://api.mainnet.aptoslabs.com/v1/graphql
 generates:
   src/types/generated/types.ts:
     plugins:

package/src/types/generated/operations.ts

@@ -77,19 +77,6 @@ export type CurrentTokenOwnershipFieldsFragment = {
   } | null;
 };
 
-export type GetAccountAddressesForAuthKeyQueryVariables = Types.Exact<{
-  where_condition?: Types.InputMaybe<Types.AuthKeyAccountAddressesBoolExp>;
-}>;
-
-export type GetAccountAddressesForAuthKeyQuery = {
-  auth_key_account_addresses: Array<{
-    auth_key: string;
-    address: string;
-    last_transaction_version: any;
-    verified: boolean;
-  }>;
-};
-
 export type GetAccountCoinsCountQueryVariables = Types.Exact<{
   address?: Types.InputMaybe<Types.Scalars["String"]["input"]>;
 }>;
@@ -109,15 +96,15 @@ export type GetAccountCoinsDataQueryVariables = Types.Exact<{
 
 export type GetAccountCoinsDataQuery = {
   current_fungible_asset_balances: Array<{
-    amount: any;
-    asset_type: string;
+    amount?: any | null;
+    asset_type?: string | null;
     is_frozen: boolean;
-    is_primary: boolean;
+    is_primary?: boolean | null;
     last_transaction_timestamp?: any | null;
     last_transaction_version?: any | null;
     owner_address: string;
     storage_id: string;
-    token_standard: string;
+    token_standard?: string | null;
     metadata?: {
       token_standard: string;
       symbol: string;
@@ -359,20 +346,6 @@ export type GetAccountTransactionsCountQuery = {
   account_transactions_aggregate: { aggregate?: { count: number } | null };
 };
 
-export type GetAuthKeysForPublicKeyQueryVariables = Types.Exact<{
-  where_condition?: Types.InputMaybe<Types.PublicKeyAuthKeysBoolExp>;
-}>;
-
-export type GetAuthKeysForPublicKeyQuery = {
-  public_key_auth_keys: Array<{
-    public_key: string;
-    public_key_type: string;
-    auth_key: string;
-    last_transaction_version: any;
-    verified: boolean;
-  }>;
-};
-
 export type GetChainTopUserTransactionsQueryVariables = Types.Exact<{
   limit?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;
 }>;
@@ -421,15 +394,15 @@ export type GetCurrentFungibleAssetBalancesQueryVariables = Types.Exact<{
 
 export type GetCurrentFungibleAssetBalancesQuery = {
   current_fungible_asset_balances: Array<{
-    amount: any;
-    asset_type: string;
+    amount?: any | null;
+    asset_type?: string | null;
     is_frozen: boolean;
-    is_primary: boolean;
+    is_primary?: boolean | null;
     last_transaction_timestamp?: any | null;
     last_transaction_version?: any | null;
     owner_address: string;
     storage_id: string;
-    token_standard: string;
+    token_standard?: string | null;
   }>;
 };
 
@@ -522,20 +495,6 @@ export type GetFungibleAssetMetadataQuery = {
   }>;
 };
 
-export type GetMultiKeyForAuthKeyQueryVariables = Types.Exact<{
-  where_condition?: Types.InputMaybe<Types.AuthKeyMultikeyLayoutBoolExp>;
-}>;
-
-export type GetMultiKeyForAuthKeyQuery = {
-  auth_key_multikey_layout: Array<{
-    auth_key: string;
-    last_transaction_version: any;
-    multikey_layout_with_prefixes: any;
-    multikey_type: string;
-    signatures_required: any;
-  }>;
-};
-
 export type GetNamesQueryVariables = Types.Exact<{
   offset?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;
   limit?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;
@@ -593,26 +552,6 @@ export type GetProcessorStatusQuery = {
   processor_status: Array<{ last_success_version: any; processor: string; last_updated: any }>;
 };
 
-export type GetSignaturesQueryVariables = Types.Exact<{
-  where_condition?: Types.InputMaybe<Types.SignaturesBoolExp>;
-  offset?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;
-  limit?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;
-  order_by?: Types.InputMaybe<Array<Types.SignaturesOrderBy> | Types.SignaturesOrderBy>;
-}>;
-
-export type GetSignaturesQuery = {
-  signatures: Array<{
-    signature: string;
-    public_key: string;
-    public_key_indices: any;
-    type: string;
-    signer: string;
-    transaction_version: any;
-    threshold: any;
-    is_sender_primary: boolean;
-  }>;
-};
-
 export type GetTableItemsDataQueryVariables = Types.Exact<{
   where_condition: Types.TableItemsBoolExp;
   offset?: Types.InputMaybe<Types.Scalars["Int"]["input"]>;

package/src/types/generated/queries.ts

@@ -82,16 +82,6 @@ export const CurrentTokenOwnershipFieldsFragmentDoc = `
   }
 }
     `;
-export const GetAccountAddressesForAuthKey = `
-    query getAccountAddressesForAuthKey($where_condition: auth_key_account_addresses_bool_exp) {
-  auth_key_account_addresses(where: $where_condition) {
-    auth_key
-    address
-    last_transaction_version
-    verified
-  }
-}
-    `;
 export const GetAccountCoinsCount = `
     query getAccountCoinsCount($address: String) {
   current_fungible_asset_balances_aggregate(
@@ -230,17 +220,6 @@ export const GetAccountTransactionsCount = `
   }
 }
     `;
-export const GetAuthKeysForPublicKey = `
-    query getAuthKeysForPublicKey($where_condition: public_key_auth_keys_bool_exp) {
-  public_key_auth_keys(where: $where_condition) {
-    public_key
-    public_key_type
-    auth_key
-    last_transaction_version
-    verified
-  }
-}
-    `;
 export const GetChainTopUserTransactions = `
     query getChainTopUserTransactions($limit: Int) {
   user_transactions(limit: $limit, order_by: {version: desc}) {
@@ -378,17 +357,6 @@ export const GetFungibleAssetMetadata = `
   }
 }
     `;
-export const GetMultiKeyForAuthKey = `
-    query getMultiKeyForAuthKey($where_condition: auth_key_multikey_layout_bool_exp) {
-  auth_key_multikey_layout(where: $where_condition) {
-    auth_key
-    last_transaction_version
-    multikey_layout_with_prefixes
-    multikey_type
-    signatures_required
-  }
-}
-    `;
 export const GetNames = `
     query getNames($offset: Int, $limit: Int, $where_condition: current_aptos_names_bool_exp, $order_by: [current_aptos_names_order_by!]) {
   current_aptos_names(
@@ -436,25 +404,6 @@ export const GetProcessorStatus = `
   }
 }
     `;
-export const GetSignatures = `
-    query getSignatures($where_condition: signatures_bool_exp, $offset: Int, $limit: Int, $order_by: [signatures_order_by!]) {
-  signatures(
-    where: $where_condition
-    offset: $offset
-    limit: $limit
-    order_by: $order_by
-  ) {
-    signature
-    public_key
-    public_key_indices
-    type
-    signer
-    transaction_version
-    threshold
-    is_sender_primary
-  }
-}
-    `;
 export const GetTableItemsData = `
     query getTableItemsData($where_condition: table_items_bool_exp!, $offset: Int, $limit: Int, $order_by: [table_items_order_by!]) {
   table_items(
@@ -563,21 +512,6 @@ const defaultWrapper: SdkFunctionWrapper = (action, _operationName, _operationTy
 
 export function getSdk(client: GraphQLClient, withWrapper: SdkFunctionWrapper = defaultWrapper) {
   return {
-    getAccountAddressesForAuthKey(
-      variables?: Types.GetAccountAddressesForAuthKeyQueryVariables,
-      requestHeaders?: GraphQLClientRequestHeaders,
-    ): Promise<Types.GetAccountAddressesForAuthKeyQuery> {
-      return withWrapper(
-        (wrappedRequestHeaders) =>
-          client.request<Types.GetAccountAddressesForAuthKeyQuery>(GetAccountAddressesForAuthKey, variables, {
-            ...requestHeaders,
-            ...wrappedRequestHeaders,
-          }),
-        "getAccountAddressesForAuthKey",
-        "query",
-        variables,
-      );
-    },
     getAccountCoinsCount(
       variables?: Types.GetAccountCoinsCountQueryVariables,
       requestHeaders?: GraphQLClientRequestHeaders,
@@ -700,21 +634,6 @@ export function getSdk(client: GraphQLClient, withWrapper: SdkFunctionWrapper =
         variables,
       );
     },
-    getAuthKeysForPublicKey(
-      variables?: Types.GetAuthKeysForPublicKeyQueryVariables,
-      requestHeaders?: GraphQLClientRequestHeaders,
-    ): Promise<Types.GetAuthKeysForPublicKeyQuery> {
-      return withWrapper(
-        (wrappedRequestHeaders) =>
-          client.request<Types.GetAuthKeysForPublicKeyQuery>(GetAuthKeysForPublicKey, variables, {
-            ...requestHeaders,
-            ...wrappedRequestHeaders,
-          }),
-        "getAuthKeysForPublicKey",
-        "query",
-        variables,
-      );
-    },
     getChainTopUserTransactions(
       variables?: Types.GetChainTopUserTransactionsQueryVariables,
       requestHeaders?: GraphQLClientRequestHeaders,
@@ -817,21 +736,6 @@ export function getSdk(client: GraphQLClient, withWrapper: SdkFunctionWrapper =
         variables,
       );
     },
-    getMultiKeyForAuthKey(
-      variables?: Types.GetMultiKeyForAuthKeyQueryVariables,
-      requestHeaders?: GraphQLClientRequestHeaders,
-    ): Promise<Types.GetMultiKeyForAuthKeyQuery> {
-      return withWrapper(
-        (wrappedRequestHeaders) =>
-          client.request<Types.GetMultiKeyForAuthKeyQuery>(GetMultiKeyForAuthKey, variables, {
-            ...requestHeaders,
-            ...wrappedRequestHeaders,
-          }),
-        "getMultiKeyForAuthKey",
-        "query",
-        variables,
-      );
-    },
     getNames(
       variables?: Types.GetNamesQueryVariables,
       requestHeaders?: GraphQLClientRequestHeaders,
@@ -889,21 +793,6 @@ export function getSdk(client: GraphQLClient, withWrapper: SdkFunctionWrapper =
         variables,
       );
     },
-    getSignatures(
-      variables?: Types.GetSignaturesQueryVariables,
-      requestHeaders?: GraphQLClientRequestHeaders,
-    ): Promise<Types.GetSignaturesQuery> {
-      return withWrapper(
-        (wrappedRequestHeaders) =>
-          client.request<Types.GetSignaturesQuery>(GetSignatures, variables, {
-            ...requestHeaders,
-            ...wrappedRequestHeaders,
-          }),
-        "getSignatures",
-        "query",
-        variables,
-      );
-    },
     getTableItemsData(
       variables: Types.GetTableItemsDataQueryVariables,
       requestHeaders?: GraphQLClientRequestHeaders,