[Masterchef]: add query, transfer funds (#1279)

* add query, transfer funds

* coverage

* log

Author: amityadav0

api/elys/masterchef/query.pulsar.go

@@ -8,7 +8,11 @@ import (
 	storetypes "cosmossdk.io/store/types"
 	upgradetypes "cosmossdk.io/x/upgrade/types"
 
+	errorsmod "cosmossdk.io/errors"
+	sdkmath "cosmossdk.io/math"
+
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/elys-network/elys/x/masterchef/types"
 
 	m "github.com/cosmos/cosmos-sdk/types/module"
 	"github.com/cosmos/cosmos-sdk/version"
@@ -72,6 +76,25 @@ func (app *ElysApp) setUpgradeHandler() {
 			//	}
 			//}
 
+			// 250USDC from protocol account to masterchef
+			params := app.MasterchefKeeper.GetParams(ctx)
+			protocolRevenueAddress, err := sdk.AccAddressFromBech32(params.ProtocolRevenueAddress)
+			if err != nil {
+				return vm, errorsmod.Wrapf(err, "invalid protocol revenue address")
+			}
+
+			// Create 250 USDC coin
+			// get usdc denom
+			usdcDenom, _ := app.AssetprofileKeeper.GetUsdcDenom(ctx)
+			usdcAmount := sdk.NewCoin(usdcDenom, sdkmath.NewInt(250000000)) // 250 USDC with 6 decimals
+
+			// Send coins from protocol revenue address to masterchef module
+			err = app.BankKeeper.SendCoinsFromAccountToModule(ctx, protocolRevenueAddress, types.ModuleName, sdk.NewCoins(usdcAmount))
+			if err != nil {
+				// log error
+				app.Logger().Error("failed to send USDC to masterchef", "error", err)
+			}
+
 			return vm, vmErr
 		},
 	)

api/elys/masterchef/query_grpc.pb.go

@@ -96,6 +96,18 @@ service Query {
   rpc ChainTVL(QueryChainTVLRequest) returns (QueryChainTVLResponse) {
     option (google.api.http).get = "/elys-network/elys/masterchef/chain_tvl";
   }
+
+  rpc TotalPendingRewards(QueryTotalPendingRewardsRequest)
+      returns (QueryTotalPendingRewardsResponse) {
+    option (google.api.http).get =
+        "/elys-network/elys/masterchef/total_pending_rewards";
+  }
+
+  rpc PendingRewards(QueryPendingRewardsRequest)
+      returns (QueryPendingRewardsResponse) {
+    option (google.api.http).get =
+        "/elys-network/elys/masterchef/pending_rewards";
+  }
 }
 
 message QueryAllLiquidityPoolTVLRequest {}
@@ -382,3 +394,26 @@ message PoolRewards {
 message QueryPoolRewardsResponse {
   repeated PoolRewards pools = 1 [ (gogoproto.nullable) = false ];
 }
+
+message QueryTotalPendingRewardsRequest {
+  cosmos.base.query.v1beta1.PageRequest pagination = 1;
+}
+
+message QueryTotalPendingRewardsResponse {
+  repeated cosmos.base.v1beta1.Coin total_pending_rewards = 1 [
+    (gogoproto.nullable) = false,
+    (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.Coins"
+  ];
+  uint64 count = 2;
+  cosmos.base.query.v1beta1.PageResponse pagination = 3;
+}
+
+message QueryPendingRewardsRequest {}
+
+message QueryPendingRewardsResponse {
+  repeated cosmos.base.v1beta1.Coin total_pending_rewards = 1 [
+    (gogoproto.nullable) = false,
+    (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.Coins"
+  ];
+  uint64 count = 2;
+}

app/setup_handlers.go

@@ -113,6 +113,17 @@ func (am AppModule) AutoCLIOptions() *autocliv1.ModuleOptions {
 					Example:        "elysd q masterchef pool-rewards [ids]",
 					PositionalArgs: []*autocliv1.PositionalArgDescriptor{{ProtoField: "pool_ids", Varargs: true}},
 				},
+				{
+					RpcMethod: "TotalPendingRewards",
+					Use:       "total-pending-rewards",
+					Short:     "show total pending rewards",
+				},
+				{
+					RpcMethod: "PendingRewards",
+					Use:       "pending-rewards",
+					Short:     "show pending rewards",
+					Example:   "elysd q masterchef pending-rewards",
+				},
 			},
 		},
 		Tx: &autocliv1.ServiceCommandDescriptor{

proto/elys/masterchef/query.proto

@@ -2,7 +2,9 @@ package keeper
 
 import (
 	"context"
+
 	"cosmossdk.io/store/prefix"
+	storetypes "cosmossdk.io/store/types"
 	"github.com/cosmos/cosmos-sdk/runtime"
 	"github.com/cosmos/cosmos-sdk/types/query"
 
@@ -178,3 +180,69 @@ func (k Keeper) PoolAprs(goCtx context.Context, req *types.QueryPoolAprsRequest)
 	data := k.CalculatePoolAprs(ctx, req.PoolIds)
 	return &types.QueryPoolAprsResponse{Data: data}, nil
 }
+
+func (k Keeper) TotalPendingRewards(goCtx context.Context, req *types.QueryTotalPendingRewardsRequest) (*types.QueryTotalPendingRewardsResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	if req == nil {
+		return nil, status.Error(codes.InvalidArgument, "invalid request")
+	}
+
+	var totalRewards sdk.Coins
+	store := runtime.KVStoreAdapter(k.storeService.OpenKVStore(ctx))
+	positionStore := prefix.NewStore(store, types.UserRewardInfoKeyPrefix)
+
+	if req.Pagination == nil {
+		req.Pagination = &query.PageRequest{
+			Limit: 100000,
+		}
+	}
+
+	count := uint64(0)
+
+	pageRes, err := query.Paginate(positionStore, req.Pagination, func(key []byte, value []byte) error {
+		var reward types.UserRewardInfo
+		k.cdc.MustUnmarshal(value, &reward)
+		k.AfterWithdraw(ctx, reward.PoolId, sdk.MustAccAddressFromBech32(reward.User), sdkmath.ZeroInt())
+		if reward.RewardPending.IsPositive() {
+			totalRewards = totalRewards.Add(sdk.NewCoin(reward.RewardDenom, reward.RewardPending.TruncateInt()))
+		}
+		count++
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &types.QueryTotalPendingRewardsResponse{
+		TotalPendingRewards: totalRewards,
+		Count:               count,
+		Pagination:          pageRes,
+	}, nil
+}
+
+func (k Keeper) PendingRewards(goCtx context.Context, req *types.QueryPendingRewardsRequest) (*types.QueryPendingRewardsResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	store := runtime.KVStoreAdapter(k.storeService.OpenKVStore(ctx))
+	iterator := storetypes.KVStorePrefixIterator(store, types.UserRewardInfoKeyPrefix)
+
+	defer iterator.Close()
+
+	var totalRewards sdk.Coins
+	count := uint64(0)
+
+	for ; iterator.Valid(); iterator.Next() {
+		var val types.UserRewardInfo
+		k.cdc.MustUnmarshal(iterator.Value(), &val)
+		if val.RewardPending.IsPositive() {
+			totalRewards = totalRewards.Add(sdk.NewCoin(val.RewardDenom, val.RewardPending.TruncateInt()))
+		}
+		count++
+	}
+
+	return &types.QueryPendingRewardsResponse{
+		TotalPendingRewards: totalRewards,
+		Count:               count,
+	}, nil
+}