refactor(staking): add msg server wrapper

modules/staking/keeper/msg_server.go

@@ -0,0 +1,32 @@
+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")
+}
+
+func (MsgServer) Delegate(ctx context.Context, msg *types.MsgDelegate) (*types.MsgDelegateResponse, error) {
+	return &types.MsgDelegateResponse{}, fmt.Errorf("Delegate is disabled")
+}

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)
+}