Allow *User object to fill bridge states manually

mautrix · Jul 14, 2023 · 82f817e · 82f817e
1 parent 1cc8ce5
commit 82f817e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bridge/status/bridgestate.go b/bridge/status/bridgestate.go
@@ -78,11 +78,20 @@ type BridgeStateFiller interface {
 	GetRemoteName() string
 }
 
+type CustomBridgeStateFiller interface {
+	BridgeStateFiller
+	FillBridgeState(BridgeState) BridgeState
+}
+
 func (pong BridgeState) Fill(user BridgeStateFiller) BridgeState {
 	if user != nil {
 		pong.UserID = user.GetMXID()
 		pong.RemoteID = user.GetRemoteID()
 		pong.RemoteName = user.GetRemoteName()
+
+		if custom, ok := user.(CustomBridgeStateFiller); ok {
+			pong = custom.FillBridgeState(pong)
+		}
 	}
 
 	pong.Timestamp = jsontime.UnixNow()