-
Notifications
You must be signed in to change notification settings - Fork 38
/
handler.go
34 lines (29 loc) · 935 Bytes
/
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
package account
import (
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/auth"
common "github.com/bnb-chain/node/common/types"
)
// NewHandler creates a set account flags handler
func NewHandler(accKeeper auth.AccountKeeper) sdk.Handler {
return func(ctx sdk.Context, msg sdk.Msg) sdk.Result {
switch msg := msg.(type) {
case SetAccountFlagsMsg:
return handleSetAccountFlags(ctx, accKeeper, msg)
default:
errMsg := fmt.Sprintf("unrecognized message type: %T", msg)
return sdk.ErrUnknownRequest(errMsg).Result()
}
}
}
func handleSetAccountFlags(ctx sdk.Context, accKeeper auth.AccountKeeper, msg SetAccountFlagsMsg) sdk.Result {
acc := accKeeper.GetAccount(ctx, msg.From)
account, ok := acc.(common.NamedAccount)
if !ok {
return sdk.ErrInternal("unexpected account type").Result()
}
account.SetFlags(msg.Flags)
accKeeper.SetAccount(ctx, account)
return sdk.Result{}
}