add x/staking queries

dydxprotocol · Sep 7, 2023 · c5f340e · c5f340e
1 parent 4efe64d
commit c5f340e
Show file tree

Hide file tree

Showing 7 changed files with 3,591 additions and 11 deletions.
diff --git a/v4-client-js/__native__/__ios__/v4-native-client.js b/v4-client-js/__native__/__ios__/v4-native-client.js
diff --git a/v4-client-js/__native__/__ios__/v4-native-client.js.map b/v4-client-js/__native__/__ios__/v4-native-client.js.map
diff --git a/v4-client-js/examples/validator_get_example.ts b/v4-client-js/examples/validator_get_example.ts
@@ -100,6 +100,23 @@ async function test(): Promise<void> {
   } catch (error) {
     console.log(JSON.stringify(error.message));
   }
+
+  try {
+    const delegationsParams = await client.get.getDelegatorDelegations(DYDX_TEST_ADDRESS);
+    console.log('Rewards Params');
+    console.log(JSON.stringify(delegationsParams));
+  } catch (error) {
+    console.log(JSON.stringify(error.message));
+  }
+
+  try {
+    const unbondingDelegationsParams = await client
+      .get.getDelegatorUnbondingDelegations(DYDX_TEST_ADDRESS);
+    console.log('Rewards Params');
+    console.log(JSON.stringify(unbondingDelegationsParams));
+  } catch (error) {
+    console.log(JSON.stringify(error.message));
+  }
 }
 
 test().then(() => {

diff --git a/v4-client-js/src/clients/constants.ts b/v4-client-js/src/clients/constants.ts
@@ -126,8 +126,8 @@ export const MAX_MEMO_CHARACTERS: number = 256;
 // Querying
 export const PAGE_REQUEST: PageRequest = {
   key: new Uint8Array(),
-  offset: new Long(0),
-  limit: new Long(-1),
+  offset: Long.UZERO,
+  limit: Long.MAX_UNSIGNED_VALUE,
   countTotal: true,
   reverse: false,
 };

diff --git a/v4-client-js/src/clients/modules/get.ts b/v4-client-js/src/clients/modules/get.ts
@@ -23,6 +23,7 @@ import {
   PricesModule,
   RewardsModule,
   SubaccountsModule,
+  StakingModule,
   StatsModule,
 } from './proto-includes';
 import { TendermintClient } from './tendermintClient';
@@ -195,6 +196,11 @@ export class Get {
     return SubaccountsModule.QuerySubaccountResponse.decode(data);
   }
 
+  /**
+   * @description Get the params for the rewards module.
+   *
+   * @returns Params for the rewards module.
+   */
   async getRewardsParams(): Promise<RewardsModule.QueryParamsResponse> {
     const requestData = Uint8Array.from(
       RewardsModule.QueryParamsRequest.encode({}).finish(),
@@ -354,6 +360,52 @@ export class Get {
     return ClobModule.QueryEquityTierLimitConfigurationResponse.decode(data);
   }
 
+  /**
+   *
+   * @description Get all delegations from a delegator.
+   *
+   * @returns All delegations from a delegator.
+   */
+  async getDelegatorDelegations(
+    delegatorAddr: string,
+  ): Promise<StakingModule.QueryDelegatorDelegationsResponse> {
+    const requestData = Uint8Array.from(
+      StakingModule.QueryDelegatorDelegationsRequest.encode({
+        delegatorAddr,
+        pagination: PAGE_REQUEST,
+      }).finish(),
+    );
+
+    const data: Uint8Array = await this.sendQuery(
+      '/cosmos.staking.v1beta1.Query/DelegatorDelegations',
+      requestData,
+    );
+    return StakingModule.QueryDelegatorDelegationsResponse.decode(data);
+  }
+
+  /**
+   *
+   * @description Get all unbonding delegations from a delegator.
+   *
+   * @returns All unbonding delegations from a delegator.
+   */
+  async getDelegatorUnbondingDelegations(
+    delegatorAddr: string,
+  ): Promise<StakingModule.QueryDelegatorUnbondingDelegationsResponse> {
+    const requestData = Uint8Array.from(
+      StakingModule.QueryDelegatorUnbondingDelegationsRequest.encode({
+        delegatorAddr,
+        pagination: PAGE_REQUEST,
+      }).finish(),
+    );
+
+    const data: Uint8Array = await this.sendQuery(
+      '/cosmos.staking.v1beta1.Query/DelegatorUnbondingDelegations',
+      requestData,
+    );
+    return StakingModule.QueryDelegatorUnbondingDelegationsResponse.decode(data);
+  }
+
   private async sendQuery(requestUrl: string, requestData: Uint8Array): Promise<Uint8Array> {
     return this.stargateQueryClient.queryUnverified(requestUrl, requestData);
   }

diff --git a/v4-client-js/src/clients/modules/proto-includes.ts b/v4-client-js/src/clients/modules/proto-includes.ts
@@ -6,6 +6,7 @@ from '@dydxprotocol/dydxjs/src/codegen/dydxprotocol/subaccounts/query';
 export * as FeeTierModule from '@dydxprotocol/dydxjs/src/codegen/dydxprotocol/feetiers/query';
 export * as StatsModule from '@dydxprotocol/dydxjs/src/codegen/dydxprotocol/stats/query';
 export * as RewardsModule from '@dydxprotocol/dydxjs/src/codegen/dydxprotocol/rewards/query';
+export * as StakingModule from '@dydxprotocol/dydxjs/src/codegen/cosmos/staking/v1beta1/query';
 
 export * from '@dydxprotocol/dydxjs/src/codegen/cosmos/base/abci/v1beta1/abci';
 export * from '@dydxprotocol/dydxjs/src/codegen/dydxprotocol/clob/order';

diff --git a/v4-client-js/src/clients/native.ts b/v4-client-js/src/clients/native.ts
@@ -899,3 +899,35 @@ export async function getRewardsParams(): Promise<string> {
     return wrappedError(e);
   }
 }
+
+export async function getDelegatorDelegations(
+  delegatorAddress: string,
+): Promise<string> {
+  try {
+    const client = globalThis.client;
+    if (client === undefined) {
+      throw new UserError('client is not connected. Call connectClient() first');
+    }
+    const delegations = await globalThis
+      .client?.validatorClient.get.getDelegatorDelegations(delegatorAddress);
+    return encodeJson(delegations);
+  } catch (e) {
+    return wrappedError(e);
+  }
+}
+
+export async function getDelegatorUnbondingDelegations(
+  delegatorAddress: string,
+): Promise<string> {
+  try {
+    const client = globalThis.client;
+    if (client === undefined) {
+      throw new UserError('client is not connected. Call connectClient() first');
+    }
+    const delegations = await globalThis
+      .client?.validatorClient.get.getDelegatorUnbondingDelegations(delegatorAddress);
+    return encodeJson(delegations);
+  } catch (e) {
+    return wrappedError(e);
+  }
+}