mirror of https://github.com/certusone/wasmd.git
74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
package wasm
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
"github.com/CosmWasm/wasmd/x/wasm/keeper"
|
|
"github.com/CosmWasm/wasmd/x/wasm/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
)
|
|
|
|
// NewHandler returns a handler for "wasm" type messages.
|
|
func NewHandler(k types.ContractOpsKeeper) sdk.Handler {
|
|
msgServer := keeper.NewMsgServerImpl(k)
|
|
|
|
return func(ctx sdk.Context, msg sdk.Msg) (*sdk.Result, error) {
|
|
ctx = ctx.WithEventManager(sdk.NewEventManager())
|
|
|
|
var (
|
|
res proto.Message
|
|
err error
|
|
)
|
|
switch msg := msg.(type) {
|
|
case *MsgStoreCode: //nolint:typecheck
|
|
res, err = msgServer.StoreCode(sdk.WrapSDKContext(ctx), msg)
|
|
case *MsgInstantiateContract:
|
|
res, err = msgServer.InstantiateContract(sdk.WrapSDKContext(ctx), msg)
|
|
case *MsgExecuteContract:
|
|
res, err = msgServer.ExecuteContract(sdk.WrapSDKContext(ctx), msg)
|
|
case *MsgMigrateContract:
|
|
res, err = msgServer.MigrateContract(sdk.WrapSDKContext(ctx), msg)
|
|
case *MsgUpdateAdmin:
|
|
res, err = msgServer.UpdateAdmin(sdk.WrapSDKContext(ctx), msg)
|
|
case *MsgClearAdmin:
|
|
res, err = msgServer.ClearAdmin(sdk.WrapSDKContext(ctx), msg)
|
|
default:
|
|
errMsg := fmt.Sprintf("unrecognized wasm message type: %T", msg)
|
|
return nil, sdkerrors.Wrap(sdkerrors.ErrUnknownRequest, errMsg)
|
|
}
|
|
|
|
ctx = ctx.WithEventManager(filterMessageEvents(ctx))
|
|
return sdk.WrapServiceResult(ctx, res, err)
|
|
}
|
|
}
|
|
|
|
// filterMessageEvents returns the same events with all of type == EventTypeMessage removed except
|
|
// for wasm message types.
|
|
// this is so only our top-level message event comes through
|
|
func filterMessageEvents(ctx sdk.Context) *sdk.EventManager {
|
|
m := sdk.NewEventManager()
|
|
for _, e := range ctx.EventManager().Events() {
|
|
if e.Type == sdk.EventTypeMessage &&
|
|
!hasWasmModuleAttribute(e.Attributes) {
|
|
continue
|
|
}
|
|
m.EmitEvent(e)
|
|
}
|
|
return m
|
|
}
|
|
|
|
func hasWasmModuleAttribute(attrs []abci.EventAttribute) bool {
|
|
for _, a := range attrs {
|
|
if sdk.AttributeKeyModule == string(a.Key) &&
|
|
types.ModuleName == string(a.Value) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|