/
msg_server.go
102 lines (82 loc) · 2.76 KB
/
msg_server.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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package keeper
import (
"context"
errorsmod "cosmossdk.io/errors"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/public-awesome/stargaze/v13/x/cron/types"
)
type msgServer struct {
Keeper
}
// NewMsgServerImpl returns an implementation of the MsgServer interface
// for the provided Keeper.
func NewMsgServerImpl(keeper Keeper) types.MsgServer {
return &msgServer{Keeper: keeper}
}
var _ types.MsgServer = msgServer{}
func (m msgServer) PromoteToPrivilegedContract(goCtx context.Context, msg *types.MsgPromoteToPrivilegedContract) (*types.MsgPromoteToPrivilegedContractResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
authorityAddr, err := sdk.AccAddressFromBech32(msg.GetAuthority())
if err != nil {
return nil, err
}
if !m.isAuthorized(ctx, authorityAddr.String()) {
return nil, errorsmod.Wrap(types.ErrUnauthorized, "sender address is not authorized address to promote contracts")
}
contractAddr, err := sdk.AccAddressFromBech32(msg.GetContract())
if err != nil {
return nil, err
}
err = m.SetPrivileged(ctx, contractAddr)
if err != nil {
return nil, err
}
return &types.MsgPromoteToPrivilegedContractResponse{}, nil
}
func (m msgServer) DemoteFromPrivilegedContract(goCtx context.Context, msg *types.MsgDemoteFromPrivilegedContract) (*types.MsgDemoteFromPrivilegedContractResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
authorityAddr, err := sdk.AccAddressFromBech32(msg.GetAuthority())
if err != nil {
return nil, err
}
if !m.isAuthorized(ctx, authorityAddr.String()) {
return nil, errorsmod.Wrap(types.ErrUnauthorized, "sender address is not authorized address to demote contracts")
}
contractAddr, err := sdk.AccAddressFromBech32(msg.GetContract())
if err != nil {
return nil, err
}
err = m.UnsetPrivileged(ctx, contractAddr)
if err != nil {
return nil, err
}
return &types.MsgDemoteFromPrivilegedContractResponse{}, nil
}
func (m msgServer) UpdateParams(goCtx context.Context, msg *types.MsgUpdateParams) (*types.MsgUpdateParamsResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
_, err := sdk.AccAddressFromBech32(msg.GetAuthority())
if err != nil {
return nil, err
}
if msg.GetAuthority() != m.Keeper.GetAuthority() {
return nil, errorsmod.Wrap(types.ErrUnauthorized, "sender address is not authorized address to update module params")
}
err = msg.GetParams().Validate() // need to explicitly validate as x/gov invokes this msg and it does not validate
if err != nil {
return nil, err
}
err = m.SetParams(ctx, msg.GetParams())
if err != nil {
return nil, err
}
return &types.MsgUpdateParamsResponse{}, nil
}
func (m msgServer) isAuthorized(ctx sdk.Context, actor string) bool {
if actor == m.Keeper.GetAuthority() {
return true
}
if m.Keeper.IsAdminAddress(ctx, actor) {
return true
}
return false
}