feat: protocol-wide restaking cap

milkyway-labs · Dec 12, 2024 · cd0043d · cd0043d
1 parent fccb49b
commit cd0043d
Show file tree

Hide file tree

Showing 37 changed files with 1,985 additions and 239 deletions.
diff --git a/api/milkyway/restaking/v1/genesis.pulsar.go b/api/milkyway/restaking/v1/genesis.pulsar.go
diff --git a/api/milkyway/restaking/v1/models.pulsar.go b/api/milkyway/restaking/v1/models.pulsar.go
diff --git a/api/milkyway/restaking/v1/params.pulsar.go b/api/milkyway/restaking/v1/params.pulsar.go
diff --git a/app/keepers/keepers.go b/app/keepers/keepers.go
@@ -510,6 +510,11 @@ func NewAppKeeper(
 		runtime.NewKVStoreService(appKeepers.keys[poolstypes.StoreKey]),
 		appKeepers.AccountKeeper,
 	)
+	appKeepers.AssetsKeeper = assetskeeper.NewKeeper(
+		appCodec,
+		runtime.NewKVStoreService(appKeepers.keys[assetstypes.StoreKey]),
+		govAuthority,
+	)
 	appKeepers.RestakingKeeper = restakingkeeper.NewKeeper(
 		appCodec,
 		runtime.NewKVStoreService(appKeepers.keys[restakingtypes.StoreKey]),
@@ -518,11 +523,8 @@ func NewAppKeeper(
 		appKeepers.PoolsKeeper,
 		appKeepers.OperatorsKeeper,
 		appKeepers.ServicesKeeper,
-		govAuthority,
-	)
-	appKeepers.AssetsKeeper = assetskeeper.NewKeeper(
-		appCodec,
-		runtime.NewKVStoreService(appKeepers.keys[assetstypes.StoreKey]),
+		appKeepers.OracleKeeper,
+		appKeepers.AssetsKeeper,
 		govAuthority,
 	)
 	appKeepers.RewardsKeeper = rewardskeeper.NewKeeper(

diff --git a/proto/milkyway/restaking/v1/genesis.proto b/proto/milkyway/restaking/v1/genesis.proto
@@ -2,6 +2,7 @@ syntax = "proto3";
 package milkyway.restaking.v1;
 
 import "gogoproto/gogo.proto";
+import "cosmos/base/v1beta1/coin.proto";
 import "milkyway/restaking/v1/models.proto";
 import "milkyway/restaking/v1/params.proto";
 
@@ -76,4 +77,10 @@ message GenesisState {
   // UserPreferences represents the user preferences.
   repeated UserPreferencesEntry users_preferences = 7
       [ (gogoproto.nullable) = false ];
+
+  // TotalRestakedAssets represents the total restaked assets.
+  repeated cosmos.base.v1beta1.Coin total_restaked_assets = 8 [
+    (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.Coins",
+    (gogoproto.nullable) = false
+  ];
 }
diff --git a/proto/milkyway/restaking/v1/models.proto b/proto/milkyway/restaking/v1/models.proto
@@ -153,4 +153,14 @@ message UserPreferences {
   // accredited and non-accredited).
   repeated uint32 trusted_services_ids = 3
       [ (gogoproto.customname) = "TrustedServicesIDs" ];
+}
+
+// TotalRestakedAssets is a struct that contains the total amount of assets
+// that have been restaked inside the chain.
+message TotalRestakedAssets {
+  // Coins is the amount of coins that have been restaked.
+  repeated cosmos.base.v1beta1.Coin coins = 1 [
+    (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.Coins",
+    (gogoproto.nullable) = false
+  ];
 }
diff --git a/proto/milkyway/restaking/v1/params.proto b/proto/milkyway/restaking/v1/params.proto
@@ -2,6 +2,7 @@ syntax = "proto3";
 package milkyway.restaking.v1;
 
 import "gogoproto/gogo.proto";
+import "cosmos_proto/cosmos.proto";
 
 option go_package = "github.com/milkyway-labs/milkyway/v3/x/restaking/types";
 
@@ -17,4 +18,12 @@ message Params {
   // and that will be considered when computing rewards. If no denoms are set,
   // all denoms will be considered as restakable.
   repeated string allowed_denoms = 2;
+
+  // RestakingCap represents the maximum USD value of overall restaked assets
+  // inside the chain.
+  string restaking_cap = 3 [
+    (cosmos_proto.scalar) = "cosmos.Dec",
+    (gogoproto.customtype) = "cosmossdk.io/math.LegacyDec",
+    (gogoproto.nullable) = false
+  ];
 }
diff --git a/x/liquidvesting/keeper/end_blocker_test.go b/x/liquidvesting/keeper/end_blocker_test.go
@@ -34,7 +34,11 @@ func (suite *KeeperTestSuite) TestKeeper_EndBlocker() {
 			},
 			store: func(ctx sdk.Context) {
 				// Set the unbonding delegation time to 7 days
-				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(7*24*time.Hour, nil))
+				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(
+					7*24*time.Hour,
+					nil,
+					restakingtypes.DefaultRestakingCap,
+				))
 				suite.Require().NoError(err)
 
 				// Add some tokens to the user's insurance fund so they can restake the locked representation
@@ -117,7 +121,11 @@ func (suite *KeeperTestSuite) TestKeeper_EndBlocker() {
 			},
 			store: func(ctx sdk.Context) {
 				// Set the unbonding delegation time to 7 days
-				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(7*24*time.Hour, nil))
+				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(
+					7*24*time.Hour,
+					nil,
+					restakingtypes.DefaultRestakingCap,
+				))
 				suite.Require().NoError(err)
 
 				// Add some tokens to the user's insurance fund so they can restake

diff --git a/x/liquidvesting/keeper/genesis_test.go b/x/liquidvesting/keeper/genesis_test.go
@@ -66,7 +66,11 @@ func (suite *KeeperTestSuite) TestKeeper_ExportGenesis() {
 			},
 			store: func(ctx sdk.Context) {
 				// Set the unbonding delegation time to 7 days
-				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(7*24*time.Hour, nil))
+				err = suite.rk.SetParams(ctx, restakingtypes.NewParams(
+					7*24*time.Hour,
+					nil,
+					restakingtypes.DefaultRestakingCap,
+				))
 				suite.Require().NoError(err)
 
 				// Fund the users' insurance fund