Merge pull request #521 from bandprotocol/extra/rest-fix-tss-member

[Extra/rest] fix tss member
bandprotocol · Dec 12, 2024 · 4c0651e · 4c0651e
2 parents 763a900 + b199d29
commit 4c0651e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/hooks/emitter/handler.go b/hooks/emitter/handler.go
@@ -178,6 +178,8 @@ func (h *Hook) handleMsgEvent(ctx sdk.Context, txHash []byte, event abci.Event)
 		h.handleTunnelEventTypeActivateTunnel(ctx, evMap)
 	case tunneltypes.EventTypeDeactivateTunnel:
 		h.handleTunnelEventTypeDeactivateTunnel(ctx, evMap)
+	case tsstypes.EventTypeSetMemberIsActive:
+		h.handleTSSSetMember(ctx, evMap)
 	}
 }
 
@@ -237,6 +239,8 @@ func (h *Hook) handleBeginBlockEndBlockEvent(
 		for _, gid := range groupIDs {
 			h.handleTSSSetGroup(ctx, tss.GroupID(common.Atoi(gid)))
 		}
+	case tsstypes.EventTypeSetMemberIsActive:
+		h.handleTSSSetMember(ctx, evMap)
 	case tunneltypes.EventTypeProducePacketSuccess:
 		h.handleTunnelEventTypeProducePacketSuccess(ctx, evMap)
 	case tunneltypes.EventTypeActivateTunnel:

diff --git a/hooks/emitter/tss.go b/hooks/emitter/tss.go
@@ -2,6 +2,7 @@ package emitter
 
 import (
 	"encoding/hex"
+	"fmt"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
@@ -210,6 +211,18 @@ func (h *Hook) handleTSSSetGroup(ctx sdk.Context, gid tss.GroupID) {
 	}
 }
 
+func (h *Hook) handleTSSSetMember(ctx sdk.Context, evMap common.EvMap) {
+	groupID := tss.GroupID(common.Atoui(evMap[types.EventTypeSetMemberIsActive+"."+types.AttributeKeyGroupID][0]))
+	memberID := tss.MemberID(common.Atoui(evMap[types.EventTypeSetMemberIsActive+"."+types.AttributeKeyMemberID][0]))
+
+	member, err := h.tssKeeper.GetMember(ctx, groupID, memberID)
+	if err != nil {
+		panic(fmt.Sprintf("invalid member; groupID: %d, memberID: %d", groupID, memberID))
+	}
+
+	h.emitSetTSSMember(member)
+}
+
 // handleTSSEventSubmitSignature implements emitter handler for SubmitSignature event.
 func (h *Hook) handleTSSEventSubmitSignature(ctx sdk.Context, evMap common.EvMap) {
 	sids := evMap[types.EventTypeSubmitSignature+"."+types.AttributeKeySigningID]