-
Notifications
You must be signed in to change notification settings - Fork 0
/
handler.go
86 lines (76 loc) · 2.59 KB
/
handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package wasm
import (
"fmt"
"github.com/gogo/protobuf/proto"
sdk "github.com/furyaxyz/fuxchain/libs/cosmos-sdk/types"
sdkerrors "github.com/furyaxyz/fuxchain/libs/cosmos-sdk/types/errors"
"github.com/furyaxyz/fuxchain/libs/tendermint/libs/kv"
types2 "github.com/furyaxyz/fuxchain/libs/tendermint/types"
"github.com/furyaxyz/fuxchain/x/wasm/keeper"
"github.com/furyaxyz/fuxchain/x/wasm/types"
"github.com/furyaxyz/fuxchain/x/wasm/watcher"
)
// 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.SetEventManager(sdk.NewEventManager())
if !types2.HigherThanEarth(ctx.BlockHeight()) {
errMsg := fmt.Sprintf("wasm not support at height %d", ctx.BlockHeight())
return nil, sdkerrors.Wrap(sdkerrors.ErrUnknownRequest, errMsg)
}
var (
res proto.Message
err error
)
// update watcher
defer func() {
// update watchDB when delivering tx
if ctx.IsDeliver() || ctx.ParaMsg() != nil {
watcher.Save(err)
}
}()
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.SetEventManager(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 []kv.Pair) bool {
for _, a := range attrs {
if sdk.AttributeKeyModule == string(a.Key) &&
types.ModuleName == string(a.Value) {
return true
}
}
return false
}