feat: addVote

docs/core/proto-docs.md

@@ -767,6 +767,7 @@
     - [VoteOption](#lbm.fbridge.v1.VoteOption)
 
 - [lbm/fbridge/v1/event.proto](#lbm/fbridge/v1/event.proto)
+    - [EventAddVoteForRole](#lbm.fbridge.v1.EventAddVoteForRole)
     - [EventClaim](#lbm.fbridge.v1.EventClaim)
     - [EventConfirmProvision](#lbm.fbridge.v1.EventConfirmProvision)
     - [EventProvision](#lbm.fbridge.v1.EventProvision)
@@ -11560,6 +11561,23 @@ VoteOption enumerates the valid vote options for a given role proposal.
 
 
 
+<a name="lbm.fbridge.v1.EventAddVoteForRole"></a>
+
+### EventAddVoteForRole
+
+
+
+| Field | Type | Label | Description |
+| ----- | ---- | ----- | ----------- |
+| `voter` | [string](#string) |  | the voter address |
+| `proposal_id` | [uint64](#uint64) |  | the role proposal id |
+| `option` | [VoteOption](#lbm.fbridge.v1.VoteOption) |  | the vote option |
+
+
+
+
+
+
 <a name="lbm.fbridge.v1.EventClaim"></a>
 
 ### EventClaim

proto/lbm/fbridge/v1/event.proto

@@ -21,6 +21,15 @@ message EventSuggestRole {
   RoleProposal proposal = 1 [(gogoproto.nullable) = false];
 }
 
+message EventAddVoteForRole {
+  // the voter address
+  string voter = 1;
+  // the role proposal id
+  uint64 proposal_id = 2;
+  // the vote option
+  VoteOption option = 3;
+}
+
 message EventProvision {
   // the sequence number of the bridge request
   uint64 seq = 1;

x/fbridge/keeper/auth.go

@@ -36,6 +36,25 @@ func (k Keeper) RegisterRoleProposal(ctx sdk.Context, proposer, target sdk.AccAd
 	return proposal, nil
 }
 
+func (k Keeper) addVote(ctx sdk.Context, proposalID uint64, voter sdk.AccAddress, option types.VoteOption) error {
+	proposal, found := k.GetRoleProposal(ctx, proposalID)
+	if !found {
+		return types.ErrUnknownProposal.Wrapf("#%d not found", proposalID)
+	}
+
+	if ctx.BlockTime().After(proposal.ExpiredAt) {
+		return types.ErrInactiveProposal.Wrapf("#%d already expired", proposalID)
+	}
+
+	if err := k.IsValidVoteOption(option); err != nil {
+		return err
+	}
+
+	k.setVote(ctx, proposalID, voter, option)
+
+	return nil
+}
+
 func (k Keeper) IsValidRole(role types.Role) error {
 	switch role {
 	case types.RoleGuardian, types.RoleOperator, types.RoleJudge:
@@ -45,6 +64,15 @@ func (k Keeper) IsValidRole(role types.Role) error {
 	return errors.New("unsupported role")
 }
 
+func (k Keeper) IsValidVoteOption(option types.VoteOption) error {
+	switch option {
+	case types.OptionYes, types.OptionNo:
+		return nil
+	}
+
+	return errors.New("unsupported vote option")
+}
+
 func (k Keeper) setNextProposalID(ctx sdk.Context, seq uint64) {
 	store := ctx.KVStore(k.storeKey)
 	bz := make([]byte, 8)
@@ -112,17 +140,32 @@ func (k Keeper) GetProposals(ctx sdk.Context) (proposals []types.RoleProposal) {
 	return
 }
 
-func (k Keeper) setRole(ctx sdk.Context, role types.Role, addr sdk.AccAddress) {
+func (k Keeper) setVote(ctx sdk.Context, proposalID uint64, voter sdk.AccAddress, option types.VoteOption) {
 	store := ctx.KVStore(k.storeKey)
+	bz := make([]byte, 4)
+	binary.BigEndian.PutUint32(bz, uint32(option))
+	store.Set(types.VoterVoteKey(proposalID, voter), bz)
+}
 
+func (k Keeper) GetVote(ctx sdk.Context, proposalID uint64, voter sdk.AccAddress) (types.VoteOption, error) {
+	store := ctx.KVStore(k.storeKey)
+	bz := store.Get(types.VoterVoteKey(proposalID, voter))
+	if bz == nil {
+		return types.OptionEmpty, types.ErrUnknownVote
+	}
+
+	return types.VoteOption(binary.BigEndian.Uint32(bz)), nil
+}
+
+func (k Keeper) setRole(ctx sdk.Context, role types.Role, addr sdk.AccAddress) {
+	store := ctx.KVStore(k.storeKey)
 	bz := make([]byte, 4)
 	binary.BigEndian.PutUint32(bz, uint32(role))
 	store.Set(types.RoleKey(addr), bz)
 }
 
 func (k Keeper) GetRole(ctx sdk.Context, addr sdk.AccAddress) types.Role {
 	store := ctx.KVStore(k.storeKey)
-
 	bz := store.Get(types.RoleKey(addr))
 	if bz == nil {
 		return types.RoleEmpty

x/fbridge/keeper/genesis.go

@@ -1,6 +1,7 @@
 package keeper
 
 import (
+	"encoding/binary"
 	sdk "github.com/Finschia/finschia-sdk/types"
 	"github.com/Finschia/finschia-sdk/x/fbridge/types"
 )
@@ -28,5 +29,32 @@ func (k Keeper) ExportGenesis(ctx sdk.Context) *types.GenesisState {
 		NextRoleProposalId: k.GetNextProposalID(ctx),
 		RoleMetadata:       k.GetRoleMetadata(ctx),
 		RoleProposals:      k.GetProposals(ctx),
+		Votes:              k.GetAllVotes(ctx),
 	}
 }
+
+// IterateVotes iterates over the all the votes for role proposals and performs a callback function
+func (k Keeper) IterateVotes(ctx sdk.Context, cb func(proposal types.Vote) (stop bool)) {
+	store := ctx.KVStore(k.storeKey)
+
+	iterator := sdk.KVStorePrefixIterator(store, types.KeyProposalPrefix)
+	defer iterator.Close()
+	for ; iterator.Valid(); iterator.Next() {
+		id, voter := types.SplitVoterVoteKey(iterator.Key())
+		opt := types.VoteOption(binary.BigEndian.Uint32(iterator.Value()))
+		v := types.Vote{ProposalId: id, Voter: voter.String(), Option: opt}
+		k.cdc.MustUnmarshal(iterator.Value(), &v)
+		if cb(v) {
+			break
+		}
+	}
+}
+
+// GetAllVotes returns all the votes from the store
+func (k Keeper) GetAllVotes(ctx sdk.Context) (votes []types.Vote) {
+	k.IterateVotes(ctx, func(vote types.Vote) bool {
+		votes = append(votes, vote)
+		return false
+	})
+	return
+}

x/fbridge/keeper/msg_server.go

@@ -103,7 +103,30 @@ func (m msgServer) SuggestRole(goCtx context.Context, msg *types.MsgSuggestRole)
 }
 
 func (m msgServer) AddVoteForRole(goCtx context.Context, msg *types.MsgAddVoteForRole) (*types.MsgAddVoteForRoleResponse, error) {
-	panic("implement me")
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	voter, err := sdk.AccAddressFromBech32(msg.From)
+	if err != nil {
+		return nil, sdkerrors.Wrapf(sdkerrors.ErrInvalidAddress, "invalid proposer address (%s)", err)
+	}
+
+	if err := m.IsValidVoteOption(msg.Option); err != nil {
+		return nil, sdkerrors.Wrap(sdkerrors.ErrInvalidRequest, err.Error())
+	}
+
+	if err := m.addVote(ctx, msg.ProposalId, voter, msg.Option); err != nil {
+		return nil, err
+	}
+
+	if err := ctx.EventManager().EmitTypedEvent(&types.EventAddVoteForRole{
+		Voter:      msg.From,
+		ProposalId: msg.ProposalId,
+		Option:     msg.Option,
+	}); err != nil {
+		panic(err)
+	}
+
+	return &types.MsgAddVoteForRoleResponse{}, nil
 }
 
 func (m msgServer) Halt(ctx context.Context, msg *types.MsgHalt) (*types.MsgHaltResponse, error) {

x/fbridge/types/errors.go

@@ -0,0 +1,9 @@
+package types
+
+import sdkerrors "github.com/Finschia/finschia-sdk/types/errors"
+
+var (
+	ErrUnknownProposal  = sdkerrors.Register(ModuleName, 2, "unknown proposal")
+	ErrInactiveProposal = sdkerrors.Register(ModuleName, 3, "inactive proposal")
+	ErrUnknownVote      = sdkerrors.Register(ModuleName, 4, "unknown vote")
+)