scaffold tx set protocol version

app/app.go

@@ -659,6 +659,7 @@ func New(
 		keys[protocolmoduletypes.StoreKey],
 		keys[protocolmoduletypes.MemStoreKey],
 		app.GetSubspace(protocolmoduletypes.ModuleName),
+		authtypes.NewModuleAddress(govtypes.ModuleName).String(),
 	)
 	protocolModule := protocolmodule.NewAppModule(appCodec, app.ProtocolKeeper)
 

proto/lavanet/lava/protocol/tx.proto

@@ -2,12 +2,20 @@ syntax = "proto3";
 package lavanet.lava.protocol;
 
 // this line is used by starport scaffolding # proto/tx/import
-
+import "lavanet/lava/protocol/params.proto";
 option go_package = "github.com/lavanet/lava/x/protocol/types";
 
 // Msg defines the Msg service.
 service Msg {
+    rpc SetVersion(MsgSetVersion) returns (MsgSetVersionResponse);
     // this line is used by starport scaffolding # proto/tx/rpc
 }
 
 // this line is used by starport scaffolding # proto/tx/message
+message MsgSetVersion {
+    string authority = 1;
+    Version version = 2;
+}
+
+message MsgSetVersionResponse {
+}

testutil/keeper/keepers_init.go

@@ -255,7 +255,7 @@ func InitAllKeepers(t testing.TB) (*Servers, *Keepers, context.Context) {
 	ks.SlashingKeeper = slashingkeeper.NewKeeper(cdc, legacyCdc, slashingStoreKey, ks.StakingKeeper, authtypes.NewModuleAddress(govtypes.ModuleName).String())
 	ks.Plans = *planskeeper.NewKeeper(cdc, plansStoreKey, plansMemStoreKey, plansparamsSubspace, ks.Epochstorage, ks.Spec, ks.FixationStoreKeeper, ks.StakingKeeper)
 	ks.Projects = *projectskeeper.NewKeeper(cdc, projectsStoreKey, projectsMemStoreKey, projectsparamsSubspace, ks.Epochstorage, ks.FixationStoreKeeper)
-	ks.Protocol = *protocolkeeper.NewKeeper(cdc, protocolStoreKey, protocolMemStoreKey, protocolparamsSubspace)
+	ks.Protocol = *protocolkeeper.NewKeeper(cdc, protocolStoreKey, protocolMemStoreKey, protocolparamsSubspace, authtypes.NewModuleAddress(govtypes.ModuleName).String())
 	ks.Downtime = downtimekeeper.NewKeeper(cdc, downtimeKey, downtimeParamsSubspace, ks.Epochstorage)
 	ks.Rewards = *rewardskeeper.NewKeeper(cdc, rewardsStoreKey, rewardsMemStoreKey, rewardsparamsSubspace, ks.BankKeeper, ks.AccountKeeper, ks.Spec, ks.Epochstorage, ks.Downtime, ks.StakingKeeper, ks.Dualstaking, ks.Distribution, authtypes.FeeCollectorName, ks.TimerStoreKeeper)
 	ks.Subscription = *subscriptionkeeper.NewKeeper(cdc, subscriptionStoreKey, subscriptionMemStoreKey, subscriptionparamsSubspace, &ks.BankKeeper, &ks.AccountKeeper, &ks.Epochstorage, ks.Projects, ks.Plans, ks.Dualstaking, ks.Rewards, ks.FixationStoreKeeper, ks.TimerStoreKeeper, ks.StakingKeeper)

testutil/keeper/protocol.go

@@ -11,6 +11,8 @@ import (
 	"github.com/cosmos/cosmos-sdk/store"
 	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
+	govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
 	typesparams "github.com/cosmos/cosmos-sdk/x/params/types"
 	"github.com/lavanet/lava/x/protocol/keeper"
 	"github.com/lavanet/lava/x/protocol/types"
@@ -41,6 +43,7 @@ func ProtocolKeeper(t testing.TB) (*keeper.Keeper, sdk.Context) {
 		storeKey,
 		memStoreKey,
 		paramsSubspace,
+		authtypes.NewModuleAddress(govtypes.ModuleName).String(),
 	)
 
 	ctx := sdk.NewContext(stateStore, tmproto.Header{}, false, log.NewNopLogger())

x/protocol/keeper/keeper.go

@@ -19,6 +19,10 @@ type (
 		storeKey   storetypes.StoreKey
 		memKey     storetypes.StoreKey
 		paramstore paramtypes.Subspace
+
+		// the address capable of executing a MsgSetIprpcData message. Typically, this
+		// should be the x/gov module account.
+		authority string
 	}
 )
 
@@ -27,6 +31,7 @@ func NewKeeper(
 	storeKey,
 	memKey storetypes.StoreKey,
 	ps paramtypes.Subspace,
+	authority string,
 ) *Keeper {
 	// set KeyTable if it has not already been set
 	if !ps.HasKeyTable() {
@@ -38,6 +43,7 @@ func NewKeeper(
 		storeKey:   storeKey,
 		memKey:     memKey,
 		paramstore: ps,
+		authority:  authority,
 	}
 }
 

x/protocol/keeper/msg_server_set_version.go

@@ -0,0 +1,28 @@
+package keeper
+
+import (
+	"context"
+
+	sdkerrors "cosmossdk.io/errors"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
+	"github.com/lavanet/lava/x/protocol/types"
+)
+
+func (k msgServer) SetVersion(goCtx context.Context, msg *types.MsgSetVersion) (*types.MsgSetVersionResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	if msg.Authority != k.authority {
+		sdkerrors.Wrapf(govtypes.ErrInvalidSigner, "invalid authority; expected %s, got %s", k.authority, msg.Authority)
+	}
+
+	params := k.GetParams(ctx)
+	params.Version = *msg.Version
+	if err := params.Validate(); err != nil {
+		return &types.MsgSetVersionResponse{}, err
+	}
+
+	k.SetParams(ctx, params)
+
+	return &types.MsgSetVersionResponse{}, nil
+}

x/protocol/module.go

@@ -123,7 +123,7 @@ func (am AppModule) Name() string {
 // module-specific GRPC queries.
 func (am AppModule) RegisterServices(cfg module.Configurator) {
 	types.RegisterQueryServer(cfg.QueryServer(), am.keeper)
-	types.RegisterMsgServer(cfg.MsgServer(), am.keeper)
+	types.RegisterMsgServer(cfg.MsgServer(), keeper.NewMsgServerImpl(am.keeper))
 
 	migrator := keeper.NewMigrator(am.keeper)
 

x/protocol/types/codec.go

@@ -3,19 +3,18 @@ package types
 import (
 	"github.com/cosmos/cosmos-sdk/codec"
 	cdctypes "github.com/cosmos/cosmos-sdk/codec/types"
-
+	sdk "github.com/cosmos/cosmos-sdk/types"
 	// this line is used by starport scaffolding # 1
-	"github.com/cosmos/cosmos-sdk/types/msgservice"
 )
 
 func RegisterCodec(cdc *codec.LegacyAmino) {
+	cdc.RegisterConcrete(&MsgSetVersion{}, "protocol/MsgSetVersion", nil)
 	// this line is used by starport scaffolding # 2
 }
 
 func RegisterInterfaces(registry cdctypes.InterfaceRegistry) {
 	// this line is used by starport scaffolding # 3
-
-	msgservice.RegisterMsgServiceDesc(registry, &_Msg_serviceDesc)
+	registry.RegisterImplementations((*sdk.Msg)(nil), &MsgSetVersion{})
 }
 
 var (

x/protocol/types/genesis.go

@@ -16,5 +16,5 @@ func DefaultGenesis() *GenesisState {
 func (gs GenesisState) Validate() error {
 	// this line is used by starport scaffolding # genesis/types/validate
 
-	return gs.Params.Validate(true)
+	return gs.Params.Validate()
 }

x/protocol/types/message_set_version.go

@@ -0,0 +1,43 @@
+package types
+
+import (
+	sdkerrors "cosmossdk.io/errors"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+)
+
+const TypeMsgSetVersion = "set_version"
+
+var _ sdk.Msg = &MsgSetVersion{}
+
+func NewMsgSetVersion(authority string, version Version) *MsgSetVersion {
+	return &MsgSetVersion{
+		Authority: authority,
+		Version:   &version,
+	}
+}
+
+func (msg *MsgSetVersion) Route() string {
+	return RouterKey
+}
+
+func (msg *MsgSetVersion) Type() string {
+	return TypeMsgSetVersion
+}
+
+func (msg *MsgSetVersion) GetSigners() []sdk.AccAddress {
+	authority, _ := sdk.AccAddressFromBech32(msg.Authority)
+	return []sdk.AccAddress{authority}
+}
+
+func (msg *MsgSetVersion) GetSignBytes() []byte {
+	bz := ModuleCdc.MustMarshalJSON(msg)
+	return sdk.MustSortJSON(bz)
+}
+
+func (msg *MsgSetVersion) ValidateBasic() error {
+	if _, err := sdk.AccAddressFromBech32(msg.Authority); err != nil {
+		return sdkerrors.Wrap(err, "invalid authority address")
+	}
+
+	return nil
+}

x/protocol/types/message_set_version_test.go

@@ -0,0 +1,48 @@
+package types
+
+import (
+	"testing"
+
+	"github.com/lavanet/lava/testutil/sample"
+	"github.com/stretchr/testify/require"
+)
+
+func TestSetIprpcData_ValidateBasic(t *testing.T) {
+	tests := []struct {
+		name  string
+		msg   MsgSetVersion
+		valid bool
+	}{
+		{
+			name: "invalid authority address",
+			msg: MsgSetVersion{
+				Authority: "invalid_address",
+			},
+			valid: false,
+		},
+		{
+			name: "invalid subscription address",
+			msg: MsgSetVersion{
+				Authority: sample.AccAddress(),
+			},
+			valid: false,
+		},
+		{
+			name: "valid message",
+			msg: MsgSetVersion{
+				Authority: sample.AccAddress(),
+			},
+			valid: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := tt.msg.ValidateBasic()
+			if tt.valid {
+				require.NoError(t, err)
+				return
+			}
+			require.Error(t, err)
+		})
+	}
+}

x/protocol/types/params.go

@@ -96,7 +96,7 @@ func versionToInteger(v string) (int, error) {
 }
 
 // Validate validates the set of params
-func (p Params) Validate(genesis bool) error {
+func (p Params) Validate() error {
 	return validateVersion(p.Version)
 }