-
Notifications
You must be signed in to change notification settings - Fork 0
/
proposal_handler.go
86 lines (76 loc) · 2.61 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package staking
import (
sdk "github.com/furyaxyz/fuxchain/libs/cosmos-sdk/types"
"github.com/furyaxyz/fuxchain/x/common"
"github.com/furyaxyz/fuxchain/x/staking/types"
govTypes "github.com/furyaxyz/fuxchain/x/gov/types"
)
// NewProposalHandler handles "gov" type message in "staking"
func NewProposalHandler(k *Keeper) govTypes.Handler {
return func(ctx sdk.Context, proposal *govTypes.Proposal) (err sdk.Error) {
switch content := proposal.Content.(type) {
case types.ProposeValidatorProposal:
return handleProposeValidatorProposal(ctx, k, proposal)
default:
return common.ErrUnknownProposalType(types.DefaultCodespace, content.ProposalType())
}
}
}
func handleProposeValidatorProposal(ctx sdk.Context, k *Keeper, proposal *govTypes.Proposal) sdk.Error {
// check
validatorProposal, ok := proposal.Content.(types.ProposeValidatorProposal)
if !ok {
return types.ErrUnexpectedProposalType
}
validator := validatorProposal.Validator
var valKey [sdk.AddrLen]byte
copy(valKey[:], validator.ValidatorAddress[:])
// verify proposed validator with validator set
valSetCount, err := verifyProposalWithValSet(ctx, k, valKey, validatorProposal.IsAdd)
if err != nil {
return err
}
// verify validator count
if err := verifyValidatorCount(ctx, k, valSetCount, valKey, validatorProposal.IsAdd); err != nil {
return err
}
if _, found := k.GetValidator(ctx, validator.ValidatorAddress); !found {
// create validator
createValMsg := NewMsgCreateValidator(validator.ValidatorAddress, validator.PubKey,
validator.Description, validator.MinSelfDelegation)
if _, err := handleMsgCreateValidator(ctx, createValMsg, *k); err != nil {
return err
}
}
k.SetProposeValidator(ctx, validator.ValidatorAddress, validatorProposal.IsAdd)
return nil
}
func verifyProposalWithValSet(ctx sdk.Context, k *Keeper, valKey [sdk.AddrLen]byte, isAdd bool) (int, sdk.Error) {
lastValSet := k.GetLastValidatorsByAddr(ctx)
_, inValSet := lastValSet[valKey] // exist in validator set
if isAdd && inValSet {
return 0, types.ErrProposedInValSet
}
if !isAdd && !inValSet {
return 0, types.ErrProposedNotInValSet
}
return len(lastValSet), nil
}
func verifyValidatorCount(ctx sdk.Context, k *Keeper, valSetCount int, valKey [sdk.AddrLen]byte, isAdd bool) sdk.Error {
maxCount := k.MaxValidators(ctx)
// proposed validators
proposedValidators := k.GetProposeValidators(ctx)
proposedValidators[valKey] = isAdd
proposeCount := 0
for _, isAdd := range proposedValidators {
if isAdd {
proposeCount++
} else {
proposeCount--
}
}
if valSetCount+proposeCount > int(maxCount) {
return types.ErrProposedExceedMax
}
return nil
}