Add rewards restaking option to Cosmos staking tx

kilnfi · Mar 5, 2024 · 4410cf6 · 4410cf6
1 parent eaf9ab3
commit 4410cf6
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/src/services/atom.ts b/src/services/atom.ts
@@ -27,12 +27,14 @@ export class AtomService extends Service {
    * @param pubkey wallet pubkey, this is different from the wallet address
    * @param validatorAddress validator address to delegate to
    * @param amountAtom how many tokens to stake in ATOM
+   * @param restakeRewards If enabled, the rewards will be automatically restaked
    */
   async craftStakeTx(
     accountId: string,
     pubkey: string,
     validatorAddress: string,
     amountAtom: number,
+    restakeRewards: boolean = false,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
@@ -42,6 +44,7 @@ export class AtomService extends Service {
         pubkey: pubkey,
         validator: validatorAddress,
         amount_uatom: this.atomToUatom(amountAtom.toString()),
+        restake_rewards: restakeRewards,
       });
     return data;
 
@@ -75,15 +78,13 @@ export class AtomService extends Service {
    */
   async craftRestakeRewardsTx(
     pubkey: string,
-    validatorAccount: string,
     validatorAddress: string,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
       `/v1/atom/transaction/restake-rewards`,
       {
         pubkey: pubkey,
-        validator_account: validatorAccount,
         validator_address: validatorAddress,
       });
     return data;

diff --git a/src/services/osmo.ts b/src/services/osmo.ts
@@ -30,12 +30,14 @@ export class OsmoService extends Service {
    * @param pubkey wallet pubkey, this is different from the wallet address
    * @param validatorAddress validator address to delegate to
    * @param amountOsmo how many tokens to stake in OSMO
+   * @param restakeRewards If enabled, the rewards will be automatically restaked
    */
   async craftStakeTx(
     accountId: string,
     pubkey: string,
     validatorAddress: string,
     amountOsmo: number,
+    restakeRewards: boolean = false,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
@@ -45,6 +47,7 @@ export class OsmoService extends Service {
         pubkey: pubkey,
         validator: validatorAddress,
         amount_uosmo: this.osmoToUosmo(amountOsmo.toString()),
+        restake_rewards: restakeRewards,
       });
     return data;
 
@@ -78,15 +81,13 @@ export class OsmoService extends Service {
    */
   async craftRestakeRewardsTx(
     pubkey: string,
-    validatorAccount: string,
     validatorAddress: string,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
       `/v1/osmo/transaction/restake-rewards`,
       {
         pubkey: pubkey,
-        validator_account: validatorAccount,
         validator_address: validatorAddress,
       });
     return data;

diff --git a/src/services/tia.ts b/src/services/tia.ts
@@ -26,12 +26,14 @@ export class TiaService extends Service {
    * @param pubkey wallet pubkey, this is different from the wallet address
    * @param validatorAddress validator address to delegate to
    * @param amountTia how many tokens to stake in TIA
+   * @param restakeRewards If enabled, the rewards will be automatically restaked
    */
   async craftStakeTx(
     accountId: string,
     pubkey: string,
     validatorAddress: string,
     amountTia: number,
+    restakeRewards: boolean = false,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
@@ -41,6 +43,7 @@ export class TiaService extends Service {
         pubkey: pubkey,
         validator: validatorAddress,
         amount_utia: this.tiaToUtia(amountTia.toString()),
+        restake_rewards: restakeRewards,
       });
     return data;
 
@@ -74,15 +77,13 @@ export class TiaService extends Service {
    */
   async craftRestakeRewardsTx(
     pubkey: string,
-    validatorAccount: string,
     validatorAddress: string,
   ): Promise<CosmosTx> {
 
     const { data } = await api.post<CosmosTx>(
       `/v1/tia/transaction/restake-rewards`,
       {
         pubkey: pubkey,
-        validator_account: validatorAccount,
         validator_address: validatorAddress,
       });
     return data;