[total-rewards] add new output to getStakeBalance

Author: Cast0001

changelog/next-release.md

@@ -0,0 +1,18 @@
+# New
+---
+
+# Updates
+### 1. [sdk.vault.getStakeBalance](https://sdk.stakewise.io/vault/requests/getstakebalance)
+
+#### New output:
+
+```ts
+type Output = {
+  assets: bigint
+  totalEarnedAssets: bigint
+}
+```
+| Name                | Description             |
+|---------------------|-------------------------|
+| `assets`            | Balance in ETH          |
+| `totalEarnedAssets` | Total earned rewards    |

src/graphql/subgraph/allocator/allocatorsQuery.graphql

@@ -6,5 +6,6 @@ query Allocators($address: Bytes!, $vaultAddress: String!) {
     }
   ) {
     assets
+    totalEarnedAssets
   }
 }

src/methods/vault/requests/getStakeBalance/getStakeBalance.md

@@ -18,14 +18,15 @@ Getting user's balance in the vault
 
 ```ts
 type Output = {
-  shares: bigint
   assets: bigint
+  totalEarnedAssets: bigint
 }
 ```
 
-| Name     | Description             |
-|----------|-------------------------|
-| `assets` | Balance in ETH          |
+| Name                | Description             |
+|---------------------|-------------------------|
+| `assets`            | Balance in ETH          |
+| `totalEarnedAssets` | Total earned rewards    |
 
 #### Example:
 

src/methods/vault/requests/getStakeBalance/index.ts

@@ -22,6 +22,7 @@ const getStakeBalance = (values: GetStakeBalanceInput) => {
     },
     modifyResult: (data) => ({
       assets: BigInt(data?.allocators?.[0]?.assets || 0),
+      totalEarnedAssets: BigInt(data?.allocators?.[0]?.totalEarnedAssets || 0),
     }),
   })
 }