/
proposal_handler.go
66 lines (56 loc) · 1.87 KB
/
proposal_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
package keeper
import (
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
"github.com/public-awesome/stargaze/v10/x/cron/types"
)
// govKeeper is a subset of Keeper that is needed for the gov proposal handling
type govKeeper interface {
SetPrivileged(ctx sdk.Context, contractAddr sdk.AccAddress) error
UnsetPrivileged(ctx sdk.Context, contractAddr sdk.AccAddress) error
}
// NewProposalHandler creates a new governance Handler for wasm proposals
func NewProposalHandler(k Keeper) govtypes.Handler {
return NewProposalHandlerX(k)
}
func NewProposalHandlerX(k govKeeper) govtypes.Handler {
return func(ctx sdk.Context, content govtypes.Content) error {
switch c := content.(type) {
case *types.PromoteToPrivilegedContractProposal:
return handlePromoteContractProposal(ctx, k, *c)
case *types.DemotePrivilegedContractProposal:
return handleDemoteContractProposal(ctx, k, *c)
default:
return sdkerrors.Wrapf(sdkerrors.ErrUnknownRequest, "unrecognized cron srcProposal content type: %T", c)
}
}
}
func handlePromoteContractProposal(ctx sdk.Context, k govKeeper, p types.PromoteToPrivilegedContractProposal) error {
if err := p.ValidateBasic(); err != nil {
return err
}
contractAddr, err := sdk.AccAddressFromBech32(p.Contract)
if err != nil {
return sdkerrors.Wrap(err, "contract address")
}
err = k.SetPrivileged(ctx, contractAddr)
if err != nil {
return err
}
return nil
}
func handleDemoteContractProposal(ctx sdk.Context, k govKeeper, p types.DemotePrivilegedContractProposal) error {
if err := p.ValidateBasic(); err != nil {
return err
}
contractAddr, err := sdk.AccAddressFromBech32(p.Contract)
if err != nil {
return sdkerrors.Wrap(err, "contract address")
}
err = k.UnsetPrivileged(ctx, contractAddr)
if err != nil {
return err
}
return nil
}