Add leveraged Amount fields for query (#1179)

* add leverageamount

* migration

---------

Co-authored-by: Abhinav Kumar <57705190+avkr003@users.noreply.github.com>

Author: amityadav0

api/elys/leveragelp/pool.pulsar.go

@@ -28,6 +28,7 @@ message Pool {
     (gogoproto.customtype) = "cosmossdk.io/math.LegacyDec",
     (gogoproto.nullable) = false
   ];
+  repeated AssetLeverageAmount asset_leverage_amounts = 6;
 }
 
 message LegacyPool {
@@ -48,3 +49,12 @@ message LegacyPool {
     (gogoproto.nullable) = false
   ];
 }
+
+message AssetLeverageAmount {
+  string denom = 1;
+  string leveraged_amount = 2 [
+    (cosmos_proto.scalar) = "cosmos.Int",
+    (gogoproto.customtype) = "cosmossdk.io/math.Int",
+    (gogoproto.nullable) = false
+  ];
+}

proto/elys/leveragelp/pool.proto

@@ -2,9 +2,10 @@ package keeper
 
 import (
 	"context"
-	sdkmath "cosmossdk.io/math"
 	"fmt"
 
+	sdkmath "cosmossdk.io/math"
+
 	errorsmod "cosmossdk.io/errors"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
@@ -37,6 +38,12 @@ func (k msgServer) AddPool(goCtx context.Context, msg *types.MsgAddPool) (*types
 	leverage := sdkmath.LegacyMinDec(msg.Pool.LeverageMax, maxLeverageAllowed)
 
 	newPool := types.NewPool(ammPool.PoolId, leverage)
+	for _, asset := range ammPool.PoolAssets {
+		newPool.AssetLeverageAmounts = append(newPool.AssetLeverageAmounts, &types.AssetLeverageAmount{
+			Denom:           asset.Token.Denom,
+			LeveragedAmount: sdkmath.ZeroInt(),
+		})
+	}
 	k.SetPool(ctx, newPool)
 
 	if k.hooks != nil {

x/leveragelp/keeper/msg_server_add_pool.go

@@ -1,6 +1,7 @@
 package keeper
 
 import (
+	sdkmath "cosmossdk.io/math"
 	"cosmossdk.io/store/prefix"
 	storetypes "cosmossdk.io/store/types"
 	"github.com/cosmos/cosmos-sdk/runtime"
@@ -56,3 +57,34 @@ func (k Keeper) GetPool(ctx sdk.Context, poolId uint64) (val types.Pool, found b
 	k.cdc.MustUnmarshal(b, &val)
 	return val, true
 }
+
+func (k Keeper) SetLeveragedAmount(ctx sdk.Context) {
+	pools := k.GetAllPools(ctx)
+	for _, pool := range pools {
+		ammPool, found := k.amm.GetPool(ctx, pool.AmmPoolId)
+		if !found {
+			continue
+		}
+		for _, asset := range ammPool.PoolAssets {
+			pool.AssetLeverageAmounts = append(pool.AssetLeverageAmounts, &types.AssetLeverageAmount{
+				Denom:           asset.Token.Denom,
+				LeveragedAmount: sdkmath.ZeroInt(),
+			})
+		}
+		k.SetPool(ctx, pool)
+	}
+
+	iterator := k.GetPositionIterator(ctx)
+	defer iterator.Close()
+
+	for ; iterator.Valid(); iterator.Next() {
+		var position types.Position
+		k.cdc.MustUnmarshal(iterator.Value(), &position)
+		pool, found := k.GetPool(ctx, position.AmmPoolId)
+		if !found {
+			continue
+		}
+		pool.UpdateAssetLeveragedAmount(ctx, position.Collateral.Denom, position.LeveragedLpAmount, true)
+		k.SetPool(ctx, pool)
+	}
+}

x/leveragelp/keeper/pool.go

@@ -109,6 +109,7 @@ func (k Keeper) CheckHealthStopLossThenRepayAndClose(ctx sdk.Context, position *
 
 	// Subtract amount that is being reduced from lp pool as it gets checked in CheckAmmUsdcBalance
 	pool.LeveragedLpAmount = pool.LeveragedLpAmount.Sub(lpSharesForRepay)
+	pool.UpdateAssetLeveragedAmount(ctx, position.Collateral.Denom, lpSharesForRepay, false)
 	// pool is set here
 	k.UpdatePoolHealth(ctx, pool)
 
@@ -166,6 +167,7 @@ func (k Keeper) CheckHealthStopLossThenRepayAndClose(ctx sdk.Context, position *
 	// Update the pool health.
 	if sharesLeft.IsPositive() {
 		pool.LeveragedLpAmount = pool.LeveragedLpAmount.Sub(sharesLeft)
+		pool.UpdateAssetLeveragedAmount(ctx, position.Collateral.Denom, sharesLeft, false)
 	}
 
 	var coinsForAmm sdk.Coins

x/leveragelp/keeper/position_close.go

@@ -107,6 +107,7 @@ func (k Keeper) ProcessOpenLong(ctx sdk.Context, position *types.Position, poolI
 
 	// Update the pool health.
 	pool.LeveragedLpAmount = pool.LeveragedLpAmount.Add(shares)
+	pool.UpdateAssetLeveragedAmount(ctx, position.Collateral.Denom, shares, true)
 	k.UpdatePoolHealth(ctx, &pool)
 
 	position.LeveragedLpAmount = position.LeveragedLpAmount.Add(shares)

x/leveragelp/keeper/position_open.go

@@ -6,5 +6,6 @@ import (
 
 func (m Migrator) V19Migration(ctx sdk.Context) error {
 	m.keeper.SetAllPositions(ctx)
+	m.keeper.SetLeveragedAmount(ctx)
 	return nil
 }

x/leveragelp/migrations/v19_migration.go

@@ -28,3 +28,18 @@ func (p *Pool) InitiatePool(ctx sdk.Context, ammPool *ammtypes.Pool) error {
 	p.AmmPoolId = ammPool.PoolId
 	return nil
 }
+
+func (p *Pool) UpdateAssetLeveragedAmount(ctx sdk.Context, denom string, amount sdkmath.Int, isIncrease bool) {
+	newAssetLevAmounts := make([]*AssetLeverageAmount, 0)
+	for _, asset := range p.AssetLeverageAmounts {
+		if asset.Denom == denom {
+			if isIncrease {
+				asset.LeveragedAmount = asset.LeveragedAmount.Add(amount)
+			} else {
+				asset.LeveragedAmount = asset.LeveragedAmount.Sub(amount)
+			}
+		}
+		newAssetLevAmounts = append(newAssetLevAmounts, asset)
+	}
+	p.AssetLeverageAmounts = newAssetLevAmounts
+}