feat: add OperatorParams and ServiceParams

proto/milkyway/pools/v1/genesis.proto

@@ -4,19 +4,22 @@ package milkyway.pools.v1;
 import "gogoproto/gogo.proto";
 import "google/protobuf/timestamp.proto";
 import "milkyway/pools/v1/models.proto";
+import "milkyway/pools/v1/params.proto";
 
 option go_package = "github.com/milkyway-labs/milkyway/x/pools/types";
 
 // GenesisState defines the pools module's genesis state.
 message GenesisState {
+  // Params defines the parameters of the module.
+  Params params = 1 [ (gogoproto.nullable) = false ];
 
   // NextPoolID represents the id to be used when creating the next pool.
-  uint32 next_pool_id = 1 [
+  uint32 next_pool_id = 2 [
     (gogoproto.customname) = "NextPoolID",
     (gogoproto.moretags) = "yaml:\"next_pool_id\""
   ];
 
   // Operators defines the list of pools.
-  repeated Pool pools = 2
+  repeated Pool pools = 3
       [ (gogoproto.moretags) = "yaml:\"pools\"", (gogoproto.nullable) = false ];
 }

proto/milkyway/pools/v1/params.proto

@@ -0,0 +1,16 @@
+syntax = "proto3";
+package milkyway.pools.v1;
+
+import "gogoproto/gogo.proto";
+import "cosmos_proto/cosmos.proto";
+import "cosmos/base/v1beta1/coin.proto";
+
+option go_package = "github.com/milkyway-labs/milkyway/x/pools/types";
+
+// Params defines the parameters for the pools module.
+message Params {
+  // AllowedServiceIDs defines the list of service IDs that the module allows
+  // to join by pools.
+  repeated uint32 allowed_service_ids = 1
+      [ (gogoproto.customname) = "AllowedServiceIDs" ];
+}

proto/milkyway/restaking/v1/genesis.proto

@@ -8,11 +8,29 @@ import "milkyway/restaking/v1/params.proto";
 
 option go_package = "github.com/milkyway-labs/milkyway/x/restaking/types";
 
+message OperatorParamsRecord {
+  uint32 operator_id = 1 [ (gogoproto.customname) = "OperatorID" ];
+
+  OperatorParams params = 2 [ (gogoproto.nullable) = false ];
+}
+
+message ServiceParamsRecord {
+  uint32 service_id = 1 [ (gogoproto.customname) = "ServiceID" ];
+
+  ServiceParams params = 2 [ (gogoproto.nullable) = false ];
+}
+
 // GenesisState defines the restaking module's genesis state.
 message GenesisState {
   // Params defines the parameters of the module.
   Params params = 1 [ (gogoproto.nullable) = false ];
 
+  repeated OperatorParamsRecord operator_params = 2
+      [ (gogoproto.nullable) = false ];
+
+  repeated ServiceParamsRecord service_params = 3
+      [ (gogoproto.nullable) = false ];
+
   // Delegations represents the delegations.
-  repeated Delegation delegations = 2 [ (gogoproto.nullable) = false ];
+  repeated Delegation delegations = 4 [ (gogoproto.nullable) = false ];
 }

proto/milkyway/restaking/v1/messages.proto

@@ -8,13 +8,22 @@ import "cosmos/bank/v1beta1/bank.proto";
 import "cosmos/msg/v1/msg.proto";
 import "gogoproto/gogo.proto";
 import "milkyway/restaking/v1/params.proto";
+import "milkyway/restaking/v1/models.proto";
 
 option go_package = "github.com/milkyway-labs/milkyway/x/restaking/types";
 
 // Msg defines the restaking module's gRPC message service.
 service Msg {
   option (cosmos.msg.v1.service) = true;
 
+  // UpdateOperatorParams defines the operation for joining a service as an
+  // operator.
+  rpc UpdateOperatorParams(MsgUpdateOperatorParams)
+      returns (MsgUpdateOperatorParamsResponse);
+
+  rpc UpdateServiceParams(MsgUpdateServiceParams)
+      returns (MsgUpdateServiceParamsResponse);
+
   // DelegatePool defines the operation that allows users to delegate any amount
   // of an asset to a pool that can then be used to provide services with
   // cryptoeconomic security.
@@ -35,6 +44,31 @@ service Msg {
   rpc UpdateParams(MsgUpdateParams) returns (MsgUpdateParamsResponse);
 }
 
+// MsgUpdateOperatorParams defines the message structure for the
+// UpdateOperatorParams gRPC service method. It allows the operator admin to
+// update the operator's parameters.
+message MsgUpdateOperatorParams {
+  string sender = 1 [ (cosmos_proto.scalar) = "cosmos.AddressString" ];
+
+  uint32 operator_id = 2 [ (gogoproto.customname) = "OperatorID" ];
+
+  OperatorParams operator_params = 3 [ (gogoproto.nullable) = false ];
+}
+
+// MsgUpdateOperatorParamsResponse is the return value of
+// MsgUpdateOperatorParams.
+message MsgUpdateOperatorParamsResponse {}
+
+message MsgUpdateServiceParams {
+  string sender = 1 [ (cosmos_proto.scalar) = "cosmos.AddressString" ];
+
+  uint32 service_id = 2 [ (gogoproto.customname) = "ServiceID" ];
+
+  ServiceParams service_params = 3 [ (gogoproto.nullable) = false ];
+}
+
+message MsgUpdateServiceParamsResponse {}
+
 // MsgDelegatePool defines the message structure for the DelegatePool gRPC
 // service method. It allows a user to put their assets into a restaking pool
 // that will later be used to provide cryptoeconomic security to services that

proto/milkyway/restaking/v1/models.proto

@@ -8,6 +8,31 @@ import "cosmos/base/v1beta1/coin.proto";
 
 option go_package = "github.com/milkyway-labs/milkyway/x/restaking/types";
 
+message OperatorParams {
+  // CommissionRate defines the commission rate charged to delegators, as a
+  // fraction.
+  string commission_rate = 1 [
+    (gogoproto.customtype) = "cosmossdk.io/math.LegacyDec",
+    (gogoproto.nullable) = false
+  ];
+
+  repeated uint32 joined_service_ids = 2
+      [ (gogoproto.customname) = "JoinedServiceIDs" ];
+}
+
+message ServiceParams {
+  string slash_fraction = 1 [
+    (gogoproto.customtype) = "cosmossdk.io/math.LegacyDec",
+    (gogoproto.nullable) = false
+  ];
+
+  repeated uint32 whitelisted_pool_ids = 2
+      [ (gogoproto.customname) = "WhitelistedPoolIDs" ];
+
+  repeated uint32 whitelisted_operator_ids = 3
+      [ (gogoproto.customname) = "WhitelistedOperatorIDs" ];
+}
+
 // DelegationType defines the type of delegation.
 enum DelegationType {
   option (gogoproto.goproto_enum_prefix) = false;

proto/milkyway/restaking/v1/query.proto

@@ -17,6 +17,19 @@ option go_package = "github.com/milkyway-labs/milkyway/x/restaking/types";
 
 // Query defines the gRPC querier service.
 service Query {
+  rpc OperatorParams(QueryOperatorParamsRequest)
+      returns (QueryOperatorParamsResponse) {
+    option (cosmos.query.v1.module_query_safe) = true;
+    option (google.api.http).get =
+        "/milkyway/restaking/v1/operator_params/{operator_id}";
+  }
+
+  rpc ServiceParams(QueryServiceParamsRequest)
+      returns (QueryServiceParamsResponse) {
+    option (cosmos.query.v1.module_query_safe) = true;
+    option (google.api.http).get =
+        "/milkyway/restaking/v1/service_params/{service_id}";
+  }
 
   // PoolDelegations queries the delegations info for the given pool.
   rpc PoolDelegations(QueryPoolDelegationsRequest)
@@ -157,6 +170,32 @@ service Query {
   }
 }
 
+// QueryOperatorParamsRequest is request type for the Query/OperatorParams
+// RPC method.
+message QueryOperatorParamsRequest {
+  // OperatorId is the ID of the operator to query
+  uint32 operator_id = 1;
+}
+
+// QueryOperatorParamsResponse is response type for the Query/OperatorParams
+// RPC method.
+message QueryOperatorParamsResponse {
+  OperatorParams operator_params = 1 [ (gogoproto.nullable) = false ];
+}
+
+// QueryServiceParamsRequest is request type for the Query/ServiceParams
+// RPC method.
+message QueryServiceParamsRequest {
+  // ServiceId is the ID of the service to query
+  uint32 service_id = 1;
+}
+
+// QueryServiceParamsResponse is response type for the Query/ServiceParams
+// RPC method.
+message QueryServiceParamsResponse {
+  ServiceParams service_params = 1 [ (gogoproto.nullable) = false ];
+}
+
 // QueryPoolDelegationsRequest is request type for the Query/PoolDelegations RPC
 // method.
 message QueryPoolDelegationsRequest {

utils/key.go

@@ -0,0 +1,26 @@
+package utils
+
+import (
+	"bytes"
+	"encoding/binary"
+)
+
+func CompositeKey(parts ...[]byte) []byte {
+	return bytes.Join(parts, nil)
+}
+
+// Uint32ToBigEndian marshals uint32 to a bigendian byte slice so it can be sorted
+func Uint32ToBigEndian(i uint32) []byte {
+	b := make([]byte, 4)
+	binary.BigEndian.PutUint32(b, i)
+	return b
+}
+
+// BigEndianToUint32 returns an uint32 from big endian encoded bytes. If encoding
+// is empty, zero is returned.
+func BigEndianToUint32(bz []byte) uint32 {
+	if len(bz) == 0 {
+		return 0
+	}
+	return binary.BigEndian.Uint32(bz)
+}

utils/utils.go

@@ -307,3 +307,24 @@ func VerifyTxHash(txHash string) (err error) {
 	}
 	return nil
 }
+
+func ParseUint32Slice(s string) ([]uint32, error) {
+	ss := strings.Split(s, ",")
+	out := make([]uint32, len(ss))
+	for i, d := range ss {
+		u, err := strconv.ParseUint(d, 10, 32)
+		if err != nil {
+			return nil, err
+		}
+		out[i] = uint32(u)
+	}
+	return out, nil
+}
+
+func FormatUint32Slice(s []uint32) string {
+	ss := make([]string, len(s))
+	for i, u := range s {
+		ss[i] = fmt.Sprint(u)
+	}
+	return strings.Join(ss, ",")
+}

x/operators/types/keys.go

@@ -35,7 +35,7 @@ func GetOperatorIDFromBytes(bz []byte) (operatorID uint32) {
 	return binary.BigEndian.Uint32(bz)
 }
 
-// OperatorStoreKey turns a operator ID into a key used to store a operator in the KVStore
+// OperatorStoreKey returns a operator ID into a key used to store a operator in the KVStore
 func OperatorStoreKey(operatorID uint32) []byte {
 	return append(OperatorPrefix, GetOperatorIDBytes(operatorID)...)
 }