Merge pull request #91 from kilnfi/cosmos-rewards-restaking

Add rewards restaking option to Cosmos staking tx

Author: goulinkh

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;

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;

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;