feat: Add affiliates queries and transaction

v4-client-js/src/clients/constants.ts

@@ -97,6 +97,9 @@ export const TYPE_URL_MSG_WITHDRAW_FROM_SUBACCOUNT =
   '/dydxprotocol.sending.MsgWithdrawFromSubaccount';
 export const TYPE_URL_MSG_DEPOSIT_TO_SUBACCOUNT = '/dydxprotocol.sending.MsgDepositToSubaccount';
 
+// x/affiliates
+export const TYPE_URL_MSG_REGISTER_AFFILIATE = '/dydxprotocol.affiliates.MsgRegisterAffiliate';
+
 // x/vault
 export const TYPE_URL_MSG_DEPOSIT_TO_MEGAVAULT = '/dydxprotocol.vault.MsgDepositToMegavault';
 export const TYPE_URL_MSG_WITHDRAW_FROM_MEGAVAULT = '/dydxprotocol.vault.MsgWithdrawFromMegavault';

v4-client-js/src/clients/lib/registry.ts

@@ -1,5 +1,6 @@
 import { GeneratedType, Registry } from '@cosmjs/proto-signing';
 import { defaultRegistryTypes } from '@cosmjs/stargate';
+import { MsgRegisterAffiliate } from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/affiliates/tx';
 import {
   MsgPlaceOrder,
   MsgCancelOrder,
@@ -34,6 +35,7 @@ import {
   TYPE_URL_BATCH_CANCEL,
   TYPE_URL_MSG_DEPOSIT_TO_MEGAVAULT,
   TYPE_URL_MSG_WITHDRAW_FROM_MEGAVAULT,
+  TYPE_URL_MSG_REGISTER_AFFILIATE,
 } from '../constants';
 
 export const registry: ReadonlyArray<[string, GeneratedType]> = [];
@@ -58,11 +60,15 @@ export function generateRegistry(): Registry {
     // vaults
     [TYPE_URL_MSG_DEPOSIT_TO_MEGAVAULT, MsgDepositToMegavault as GeneratedType],
     [TYPE_URL_MSG_WITHDRAW_FROM_MEGAVAULT, MsgWithdrawFromMegavault as GeneratedType],
+
     // sending
     [TYPE_URL_MSG_CREATE_TRANSFER, MsgCreateTransfer as GeneratedType],
     [TYPE_URL_MSG_WITHDRAW_FROM_SUBACCOUNT, MsgWithdrawFromSubaccount as GeneratedType],
     [TYPE_URL_MSG_DEPOSIT_TO_SUBACCOUNT, MsgDepositToSubaccount as GeneratedType],
 
+    // affiliates
+    [TYPE_URL_MSG_REGISTER_AFFILIATE, MsgRegisterAffiliate as GeneratedType],
+
     // default types
     ...defaultRegistryTypes,
   ]);

v4-client-js/src/clients/modules/composer.ts

@@ -7,6 +7,7 @@ import {
   MsgDelegate,
   MsgUndelegate,
 } from '@dydxprotocol/v4-proto/src/codegen/cosmos/staking/v1beta1/tx';
+import { MsgRegisterAffiliate } from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/affiliates/tx';
 import { ClobPair_Status } from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/clob/clob_pair';
 import {
   MsgBatchCancel,
@@ -47,6 +48,7 @@ import {
   TYPE_URL_MSG_UNDELEGATE,
   TYPE_URL_MSG_WITHDRAW_DELEGATOR_REWARD,
   TYPE_URL_BATCH_CANCEL,
+  TYPE_URL_MSG_REGISTER_AFFILIATE,
   TYPE_URL_MSG_DEPOSIT_TO_MEGAVAULT,
   TYPE_URL_MSG_WITHDRAW_FROM_MEGAVAULT,
 } from '../constants';
@@ -536,6 +538,19 @@ export class Composer {
     };
   }
 
+  // ------------ x/affiliates ------------
+  public composeMsgRegisterAffiliate(referee: string, affiliate: string): EncodeObject {
+    const msg: MsgRegisterAffiliate = {
+      referee,
+      affiliate,
+    };
+
+    return {
+      typeUrl: TYPE_URL_MSG_REGISTER_AFFILIATE,
+      value: msg,
+    };
+  }
+
   // ------------ util ------------
   public validateGoodTilBlockAndTime(
     orderFlags: number,

v4-client-js/src/clients/modules/get.ts

@@ -17,6 +17,7 @@ import { bigIntToBytes } from '../../lib/helpers';
 import { PAGE_REQUEST } from '../constants';
 import { UnexpectedClientError } from '../lib/errors';
 import {
+  AffiliateModule,
   BridgeModule,
   ClobModule,
   DistributionModule,
@@ -559,6 +560,56 @@ export class Get {
     return VaultModule.QueryMegavaultWithdrawalInfoResponse.decode(data);
   }
 
+  async getAffiliateInfo(address: string): Promise<AffiliateModule.AffiliateInfoResponse> {
+    const requestData = Uint8Array.from(
+      AffiliateModule.AffiliateInfoRequest.encode({
+        address,
+      }).finish(),
+    );
+
+    const data = await this.sendQuery('/dydxprotocol.affiliates.Query/AffiliateInfo', requestData);
+
+    return AffiliateModule.AffiliateInfoResponse.decode(data);
+  }
+
+  async getReferredBy(address: string): Promise<AffiliateModule.ReferredByResponse> {
+    const requestData = Uint8Array.from(
+      AffiliateModule.ReferredByRequest.encode({
+        address,
+      }).finish(),
+    );
+
+    const data = await this.sendQuery('/dydxprotocol.affiliates.Query/ReferredBy', requestData);
+
+    return AffiliateModule.ReferredByResponse.decode(data);
+  }
+
+  async getAllAffiliateTiers(): Promise<AffiliateModule.AllAffiliateTiersResponse> {
+    const requestData = Uint8Array.from(
+      AffiliateModule.AllAffiliateTiersRequest.encode({}).finish(),
+    );
+
+    const data = await this.sendQuery(
+      '/dydxprotocol.affiliates.Query/AllAffiliateTiers',
+      requestData,
+    );
+
+    return AffiliateModule.AllAffiliateTiersResponse.decode(data);
+  }
+
+  async getAffiliateWhitelist(): Promise<AffiliateModule.AffiliateWhitelistResponse> {
+    const requestData = Uint8Array.from(
+      AffiliateModule.AffiliateWhitelistRequest.encode({}).finish(),
+    );
+
+    const data = await this.sendQuery(
+      '/dydxprotocol.affiliates.Query/AffiliateWhitelist',
+      requestData,
+    );
+
+    return AffiliateModule.AffiliateWhitelistResponse.decode(data);
+  }
+
   private async sendQuery(requestUrl: string, requestData: Uint8Array): Promise<Uint8Array> {
     // eslint-disable-next-line max-len
     const resp: QueryAbciResponse = await this.stargateQueryClient.queryAbci(

v4-client-js/src/clients/modules/post.ts

@@ -881,4 +881,24 @@ export class Post {
   ): EncodeObject {
     return this.composer.composeMsgWithdrawFromMegavault(...args);
   }
+
+  async registerAffiliate(
+    subaccount: SubaccountInfo,
+    affiliate: string,
+    broadcastMode?: BroadcastMode,
+  ): Promise<BroadcastTxAsyncResponse | BroadcastTxSyncResponse | IndexedTx> {
+    const msg = this.registerAffiliateMsg(subaccount.address, affiliate);
+    return this.send(
+      subaccount.wallet,
+      () => Promise.resolve([msg]),
+      false,
+      undefined,
+      undefined,
+      broadcastMode,
+    );
+  }
+
+  registerAffiliateMsg(...args: Parameters<Composer['composeMsgRegisterAffiliate']>): EncodeObject {
+    return this.composer.composeMsgRegisterAffiliate(...args);
+  }
 }

v4-client-js/src/clients/modules/proto-includes.ts

@@ -11,6 +11,7 @@ export * as RewardsModule from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/
 export * as StakingModule from '@dydxprotocol/v4-proto/src/codegen/cosmos/staking/v1beta1/query';
 export * as BridgeModule from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/bridge/query';
 export * as DistributionModule from '@dydxprotocol/v4-proto/src/codegen/cosmos/distribution/v1beta1/query';
+export * as AffiliateModule from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/affiliates/query';
 export * as VaultModule from '@dydxprotocol/v4-proto/src/codegen/dydxprotocol/vault/query';
 
 export * from '@dydxprotocol/v4-proto/src/codegen/cosmos/base/abci/v1beta1/abci';

v4-client-js/src/lib/constants.ts

@@ -12,7 +12,7 @@ export const BROADCAST_TIMEOUT_MS: number = 8_000;
 export const API_TIMEOUT_DEFAULT_MS: number = 5_000;
 
 // Gas
-export const GAS_MULTIPLIER: number = 1.4;
+export const GAS_MULTIPLIER: number = 1.5;
 
 export const ZERO_FEE: StdFee = {
   amount: [],