From baf4b6e841482ba30170a11ea4cecf06f6600a67 Mon Sep 17 00:00:00 2001 From: Julien Robert Date: Mon, 26 Jan 2026 17:21:29 +0100 Subject: [PATCH 1/2] refactor(staking): add msg server wrapper --- modules/staking/keeper/msg_server.go | 28 ++++++++++++++++++++++++++++ modules/staking/module.go | 8 ++++++++ 2 files changed, 36 insertions(+) create mode 100644 modules/staking/keeper/msg_server.go diff --git a/modules/staking/keeper/msg_server.go b/modules/staking/keeper/msg_server.go new file mode 100644 index 00000000..e9c9d1f3 --- /dev/null +++ b/modules/staking/keeper/msg_server.go @@ -0,0 +1,28 @@ +package keeper + +import ( + "context" + "fmt" + + "github.com/cosmos/cosmos-sdk/x/staking/types" +) + +var _ types.MsgServer = MsgServer{} + +type MsgServer struct { + types.MsgServer +} + +func NewMsgServerImpl(ms types.MsgServer) MsgServer { + return MsgServer{ + MsgServer: ms, + } +} + +func (MsgServer) CreateValidator(ctx context.Context, msg *types.MsgCreateValidator) (*types.MsgCreateValidatorResponse, error) { + return &types.MsgCreateValidatorResponse{}, fmt.Errorf("CreateValidator is disabled") +} + +func (MsgServer) BeginRedelegate(ctx context.Context, msg *types.MsgBeginRedelegate) (*types.MsgBeginRedelegateResponse, error) { + return &types.MsgBeginRedelegateResponse{}, fmt.Errorf("BeginRedelegate is disabled") +} diff --git a/modules/staking/module.go b/modules/staking/module.go index 5f6d28f8..380dbf56 100644 --- a/modules/staking/module.go +++ b/modules/staking/module.go @@ -14,6 +14,7 @@ import ( "github.com/cosmos/cosmos-sdk/types/module" "github.com/cosmos/cosmos-sdk/x/staking" "github.com/cosmos/cosmos-sdk/x/staking/exported" + ogkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper" "github.com/cosmos/cosmos-sdk/x/staking/types" "github.com/evstack/ev-abci/modules/staking/keeper" @@ -65,3 +66,10 @@ func (am AppModule) InitGenesis(ctx sdk.Context, cdc codec.JSONCodec, data json. cdc.MustUnmarshalJSON(data, &genesisState) return am.keeper.InitGenesis(ctx, &genesisState) } + +// RegisterServices registers module services. +func (am AppModule) RegisterServices(cfg module.Configurator) { + types.RegisterMsgServer(cfg.MsgServer(), keeper.NewMsgServerImpl(ogkeeper.NewMsgServerImpl(am.keeper.Keeper))) + querier := ogkeeper.Querier{Keeper: am.keeper.Keeper} + types.RegisterQueryServer(cfg.QueryServer(), querier) +} From ca96083c1c457b7bccce0147298bf0b2e4f20a90 Mon Sep 17 00:00:00 2001 From: Julien Robert Date: Mon, 26 Jan 2026 17:57:08 +0100 Subject: [PATCH 2/2] update --- modules/staking/keeper/msg_server.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/staking/keeper/msg_server.go b/modules/staking/keeper/msg_server.go index e9c9d1f3..6621523d 100644 --- a/modules/staking/keeper/msg_server.go +++ b/modules/staking/keeper/msg_server.go @@ -26,3 +26,7 @@ func (MsgServer) CreateValidator(ctx context.Context, msg *types.MsgCreateValida func (MsgServer) BeginRedelegate(ctx context.Context, msg *types.MsgBeginRedelegate) (*types.MsgBeginRedelegateResponse, error) { return &types.MsgBeginRedelegateResponse{}, fmt.Errorf("BeginRedelegate is disabled") } + +func (MsgServer) Delegate(ctx context.Context, msg *types.MsgDelegate) (*types.MsgDelegateResponse, error) { + return &types.MsgDelegateResponse{}, fmt.Errorf("Delegate is disabled") +}