feat(operator): Voting power snapshot

app/ante/cosmos/min_price_test.go

@@ -51,7 +51,7 @@ func (suite *AnteTestSuite) TestMinGasPriceDecorator() {
 				err := suite.app.FeeMarketKeeper.SetParams(suite.ctx, params)
 				suite.Require().NoError(err)
 
-				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.NewInt(0), denom, &testMsg)
+				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.ZeroInt(), denom, &testMsg)
 				return txBuilder.GetTx()
 			},
 			true,
@@ -96,7 +96,7 @@ func (suite *AnteTestSuite) TestMinGasPriceDecorator() {
 				err := suite.app.FeeMarketKeeper.SetParams(suite.ctx, params)
 				suite.Require().NoError(err)
 
-				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.NewInt(0), denom, &testMsg)
+				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.ZeroInt(), denom, &testMsg)
 				return txBuilder.GetTx()
 			},
 			false,

app/ante/evm/fees_test.go

@@ -60,7 +60,7 @@ func (suite *AnteTestSuite) TestEthMinGasPriceDecorator() {
 					ToAddress:   "evmos1dx67l23hz9l0k9hcher8xz04uj7wf3yu26l2yn",
 					Amount:      sdk.Coins{sdk.Coin{Amount: sdkmath.NewInt(10), Denom: denom}},
 				}
-				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.NewInt(0), denom, &testMsg)
+				txBuilder := suite.CreateTestCosmosTxBuilder(sdkmath.ZeroInt(), denom, &testMsg)
 				return txBuilder.GetTx()
 			},
 			false,

app/app.go

@@ -1179,7 +1179,6 @@ func (app *ExocoreApp) InitChainer(
 	if err := json.Unmarshal(req.AppStateBytes, &genesisState); err != nil {
 		panic(err)
 	}
-
 	app.UpgradeKeeper.SetModuleVersionMap(ctx, app.mm.GetVersionMap())
 
 	return app.mm.InitGenesis(ctx, app.appCodec, genesisState)

precompiles/assets/assets_test.go

@@ -376,7 +376,6 @@ func (s *AssetsPrecompileSuite) TestRunWithdrawPrincipal() {
 			readOnly:    false,
 			expPass:     true,
 			extra: func() {
-
 				stakerID, assetID := assetstype.GetStakerIDAndAssetID(s.ClientChains[0].LayerZeroChainID, s.Address.Bytes(), NSTAddress)
 				// check depositNST successfully updated stakerAssetInfo in assets_module
 				stakerAssetInfo, _ := s.App.AssetsKeeper.GetStakerSpecifiedAssetInfo(s.Ctx, stakerID, assetID)

precompiles/assets/tx.go

@@ -156,7 +156,7 @@ func (p Precompile) RegisterToken(
 
 	stakingAsset := &assetstypes.StakingAssetInfo{
 		AssetBasicInfo:     asset,
-		StakingTotalAmount: sdkmath.NewInt(0),
+		StakingTotalAmount: sdkmath.ZeroInt(),
 	}
 
 	if err := p.assetsKeeper.RegisterNewTokenAndSetTokenFeeder(ctx, &oInfo); err != nil {

precompiles/avs/avs_test.go

@@ -1,13 +1,14 @@
 package avs_test
 
 import (
-	"cosmossdk.io/math"
-	assetstypes "github.com/ExocoreNetwork/exocore/x/assets/types"
-	"github.com/ExocoreNetwork/exocore/x/avs/types"
 	"math/big"
 	"strconv"
 	"time"
 
+	"cosmossdk.io/math"
+	assetstypes "github.com/ExocoreNetwork/exocore/x/assets/types"
+	"github.com/ExocoreNetwork/exocore/x/avs/types"
+
 	sdkmath "cosmossdk.io/math"
 	operatorKeeper "github.com/ExocoreNetwork/exocore/x/operator/keeper"
 
@@ -650,7 +651,6 @@ func (suite *AVSManagerPrecompileSuite) TestRegisterOperatorToAVS() {
 }
 
 func (suite *AVSManagerPrecompileSuite) TestDeregisterOperatorFromAVS() {
-
 	// from := s.Address
 	operatorAddress := sdk.AccAddress(suite.Address.Bytes())
 	assetID := suite.AssetIDs
@@ -835,7 +835,7 @@ func (suite *AVSManagerPrecompileSuite) TestRunRegTaskInfo() {
 			suite.Ctx,
 			&assetstypes.StakingAssetInfo{
 				AssetBasicInfo:     usdcClientChainAsset,
-				StakingTotalAmount: sdkmath.NewInt(0),
+				StakingTotalAmount: sdkmath.ZeroInt(),
 			},
 		)
 		suite.NoError(err)

precompiles/avs/query_test.go

@@ -2,11 +2,12 @@ package avs_test
 
 import (
 	"fmt"
+	"math/big"
+	"time"
+
 	epochstypes "github.com/ExocoreNetwork/exocore/x/epochs/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/prysmaticlabs/prysm/v4/crypto/bls/blst"
-	"math/big"
-	"time"
 
 	utiltx "github.com/ExocoreNetwork/exocore/testutil/tx"
 
@@ -144,7 +145,7 @@ func (suite *AVSManagerPrecompileSuite) TestGetOptedInOperatorAccAddrs() {
 
 func (suite *AVSManagerPrecompileSuite) TestAVSUSDValue() {
 	method := s.precompile.Methods[avsManagerPrecompile.MethodGetAVSUSDValue]
-	expectedUSDvalue := sdkmath.LegacyNewDec(0)
+	expectedUSDvalue := sdkmath.LegacyZeroDec()
 
 	setUp := func() {
 		suite.prepare()
@@ -162,7 +163,7 @@ func (suite *AVSManagerPrecompileSuite) TestAVSUSDValue() {
 			suite.Ctx,
 			&assetstype.StakingAssetInfo{
 				AssetBasicInfo:     usdcClientChainAsset,
-				StakingTotalAmount: sdkmath.NewInt(0),
+				StakingTotalAmount: sdkmath.ZeroInt(),
 			},
 		)
 		suite.NoError(err)
@@ -232,7 +233,7 @@ func (suite *AVSManagerPrecompileSuite) TestAVSUSDValue() {
 
 func (suite *AVSManagerPrecompileSuite) TestGetOperatorOptedUSDValue() {
 	method := s.precompile.Methods[avsManagerPrecompile.MethodGetOperatorOptedUSDValue]
-	expectedUSDvalue := sdkmath.LegacyNewDec(0)
+	expectedUSDvalue := sdkmath.LegacyZeroDec()
 
 	setUp := func() {
 		suite.prepare()
@@ -250,7 +251,7 @@ func (suite *AVSManagerPrecompileSuite) TestGetOperatorOptedUSDValue() {
 			suite.Ctx,
 			&assetstype.StakingAssetInfo{
 				AssetBasicInfo:     usdcClientChainAsset,
-				StakingTotalAmount: sdkmath.NewInt(0),
+				StakingTotalAmount: sdkmath.ZeroInt(),
 			},
 		)
 		suite.NoError(err)
@@ -494,6 +495,7 @@ func (suite *AVSManagerPrecompileSuite) TestIsoperator() {
 		})
 	}
 }
+
 func (suite *AVSManagerPrecompileSuite) TestGetTaskInfo() {
 	method := suite.precompile.Methods[avsManagerPrecompile.MethodGetTaskInfo]
 	taskAddress := "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48"
@@ -507,7 +509,7 @@ func (suite *AVSManagerPrecompileSuite) TestGetTaskInfo() {
 			TaskResponsePeriod:    10,
 			StartingEpoch:         5,
 			TaskStatisticalPeriod: 60,
-			TaskTotalPower:        sdk.Dec(sdkmath.NewInt(0)),
+			TaskTotalPower:        sdk.Dec(sdkmath.ZeroInt()),
 		}
 		err := suite.App.AVSManagerKeeper.SetTaskInfo(suite.Ctx, info)
 		suite.NoError(err)
@@ -553,6 +555,7 @@ func (suite *AVSManagerPrecompileSuite) TestGetTaskInfo() {
 		})
 	}
 }
+
 func (suite *AVSManagerPrecompileSuite) TestGetCurrentEpoch() {
 	method := suite.precompile.Methods[avsManagerPrecompile.MethodGetCurrentEpoch]
 	testCases := []avsTestCases{

precompiles/avs/utils_test.go

@@ -202,8 +202,8 @@ func (suite *AVSManagerPrecompileSuite) TestOptOut() {
 			OptedInHeight:  uint64(optInHeight),
 			OptedOutHeight: uint64(suite.Ctx.BlockHeight()),
 		},
-		AVSTotalShare:    sdkmath.LegacyNewDec(0),
-		AVSOperatorShare: sdkmath.LegacyNewDec(0),
+		AVSTotalShare:    sdkmath.LegacyZeroDec(),
+		AVSOperatorShare: sdkmath.LegacyZeroDec(),
 		AssetState:       nil,
 		OperatorShare:    sdkmath.LegacyDec{},
 		StakerShare:      sdkmath.LegacyDec{},

proto/exocore/operator/v1/genesis.proto

@@ -70,7 +70,7 @@ message AVSUSDValue {
 // it's corresponding to the kvStore `KeyPrefixUSDValueForOperator`
 message OperatorUSDValue {
   // key is used for storing the voting power of specified operator and AVS,
-  // which is the combination of operator and AVS address.
+  // which is the combination of AVS and operator address.
   string key = 1;
   // value is the USD value states for the AVS address
   OperatorOptedUSDValue opted_usd_value = 2 [(gogoproto.nullable) = false, (gogoproto.customname) = "OptedUSDValue"];

proto/exocore/operator/v1/query.proto

@@ -19,7 +19,8 @@ option go_package = "github.com/ExocoreNetwork/exocore/x/operator/types";
 // QueryOperatorInfoReq is the request to obtain the operator information.
 message GetOperatorInfoReq {
   // operator_addr is the operator address,its type should be a sdk.AccAddress
-  string operator_addr = 1 [(cosmos_proto.scalar) = "cosmos.AddressString"];
+  string operator_addr = 1
+  [(cosmos_proto.scalar) = "cosmos.AddressString"];
 }
 
 // QueryAllOperatorsRequest is the request to obtain all operators.
@@ -53,13 +54,15 @@ message QueryOperatorUSDValueRequest {
 // QueryOperatorUSDValueResponse is the response to obtain the USD value for operator.
 message QueryOperatorUSDValueResponse {
   // usd_info includes the self and total staking for the operator and AVS
-  OperatorOptedUSDValue usd_values = 1 [(gogoproto.customname) = "USDValues"];
+  OperatorOptedUSDValue usd_values = 1
+  [(gogoproto.customname) = "USDValues"];
 }
 
 // QueryAVSUSDValueRequest is the request to obtain the USD value for AVS.
 message QueryAVSUSDValueRequest {
   // avs_address is the AVS address opted-in by the operator
-  string avs_address = 1 [(gogoproto.customname) = "AVSAddress"];
+  string avs_address = 1
+  [(gogoproto.customname) = "AVSAddress"];
 }
 
 // QueryOperatorSlashInfoRequest is the request to obtain the slash information for the specified
@@ -223,6 +226,55 @@ message QueryOptInfoRequest {
   OperatorAVSAddress operator_and_avs = 1 [(gogoproto.embed) = true];
 }
 
+// QuerySnapshotHelperRequest is the request to obtain the voting power snapshot
+// helper information.
+message QuerySnapshotHelperRequest {
+  // avs address
+  string avs = 1;
+}
+
+// QuerySpecifiedSnapshotRequest is the request to obtain the voting power snapshot
+// at the specified height.
+message QuerySpecifiedSnapshotRequest {
+  // avs address
+  string avs = 1;
+  // height is used to specify the snapshot height you want to query.
+  int64 height = 2;
+}
+
+// VotingPowerSnapshotKeyHeight is used in the response of QuerySpecifiedSnapshot
+// and QueryAllSnapshot
+message VotingPowerSnapshotKeyHeight {
+  // snapshot_key_height when it is used in QuerySpecifiedSnapshot, it's the latest
+  // height with a snapshot key found based on the input height; this height is typically
+  // the start height of the epoch in which the input height is located.
+  // when it is used in QueryAllSnapshot, it's the height in the current snapshot key.
+  int64 snapshot_key_height =1;
+  // snapshot when it is used in QuerySpecifiedSnapshot, it is the final retrieved information
+  // containing the voting power set.
+  // when it is used in QueryAllSnapshot, It is the snapshot stored under `snapshot_key_height`,
+  // and its voting power set may be nil.
+  VotingPowerSnapshot snapshot = 2;
+}
+
+// QueryAllSnapshotRequest is the request to obtain all voting power snapshot
+// for the specified AVS
+message QueryAllSnapshotRequest {
+  // avs address
+  string avs = 1;
+  // pagination related options.
+  cosmos.base.query.v1beta1.PageRequest pagination = 2;
+}
+
+// QueryAllSnapshotResponse is the response to obtain all voting power snapshot
+// for the specified AVS
+message QueryAllSnapshotResponse {
+  // snapshots is a list of all snapshots for the specified AVS.
+  repeated VotingPowerSnapshotKeyHeight snapshots = 1;
+  // pagination related response.
+  cosmos.base.query.v1beta1.PageResponse pagination = 2;
+}
+
 // Query defines the gRPC querier service.
 service Query {
   // QueryOperatorInfo queries the operator information.
@@ -261,20 +313,22 @@ service Query {
   }
 
   // QueryOperatorUSDValue queries the opted-in USD value for the operator
-  rpc QueryOperatorUSDValue(QueryOperatorUSDValueRequest) returns (QueryOperatorUSDValueResponse) {
-    option (google.api.http).get = "/exocore/operator/v1/query_operator_usd_value/{operator_and_avs.operator_addr}/"
-                                   "{operator_and_avs.avs_address}";
+  rpc QueryOperatorUSDValue(QueryOperatorUSDValueRequest) returns(QueryOperatorUSDValueResponse){
+    option (google.api.http).get =
+    "/exocore/operator/v1/operator_usd_value/{operator_and_avs.operator_addr}/"
+        "{operator_and_avs.avs_address}";
   }
 
   // QueryAVSUSDValue queries the USD value for the AVS
-  rpc QueryAVSUSDValue(QueryAVSUSDValueRequest) returns (DecValueField) {
-    option (google.api.http).get = "/exocore/operator/v1/QueryAVSUSDValue";
+  rpc QueryAVSUSDValue(QueryAVSUSDValueRequest) returns(DecValueField){
+    option (google.api.http).get = "/exocore/operator/v1/avs_usd_value";
   }
 
   // QueryOperatorSlashInfo queries the slash information for the specified operator and AVS
-  rpc QueryOperatorSlashInfo(QueryOperatorSlashInfoRequest) returns (QueryOperatorSlashInfoResponse) {
-    option (google.api.http).get = "/exocore/operator/v1/query_operator_slash_info/{operator_and_avs.operator_addr}/"
-                                   "{operator_and_avs.avs_address}";
+  rpc QueryOperatorSlashInfo(QueryOperatorSlashInfoRequest) returns(QueryOperatorSlashInfoResponse){
+    option (google.api.http).get =
+    "/exocore/operator/v1/operator_slash_info/{operator_and_avs.operator_addr}/"
+        "{operator_and_avs.avs_address}";
   }
 
   // QueryAllOperatorConsAddrsByChainID queries all operators and their consensus addresses
@@ -287,7 +341,8 @@ service Query {
   }
 
   // QueryAllOperatorsWithOptInAVS queries  operator list by avs.
-  rpc QueryAllOperatorsWithOptInAVS(QueryAllOperatorsByOptInAVSRequest) returns (QueryAllOperatorsByOptInAVSResponse) {
+  rpc QueryAllOperatorsWithOptInAVS(QueryAllOperatorsByOptInAVSRequest) returns (
+      QueryAllOperatorsByOptInAVSResponse) {
     option (google.api.http) = {
       get: "/exocore/operator/v1/opt/operator_list/{avs}"
     };
@@ -307,6 +362,21 @@ service Query {
     };
   }
 
+  // QuerySnapshotHelper queries the snapshot helper of the AVS
+  rpc QuerySnapshotHelper(QuerySnapshotHelperRequest) returns (SnapshotHelper) {
+    option (google.api.http).get = "/exocore/operator/v1/snapshot_helper/{avs}";
+  }
+
+  // QuerySpecifiedSnapshot queries the voting power snapshot of specified AVS and height
+  rpc QuerySpecifiedSnapshot(QuerySpecifiedSnapshotRequest) returns (VotingPowerSnapshotKeyHeight) {
+    option (google.api.http).get = "/exocore/operator/v1/snapshot/{avs}/{height}";
+  }
+
+  // QueryAllSnapshot queries all voting power snapshot for the specified AVS
+  rpc QueryAllSnapshot(QueryAllSnapshotRequest) returns (QueryAllSnapshotResponse) {
+    option (google.api.http).get = "/exocore/operator/v1/all_snapshot/{avs}";
+  }
+
   // Validators queries all validators that match the given status.
   // When called from another module, this query might consume a high amount of
   // gas if the pagination field is incorrectly set.