From 82f817eff669c5ac9250eca2829c2a411054cbe1 Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Sat, 15 Jul 2023 02:32:18 +0300 Subject: [PATCH] Allow *User object to fill bridge states manually --- bridge/status/bridgestate.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/bridge/status/bridgestate.go b/bridge/status/bridgestate.go index 4d22befe..a19798c7 100644 --- 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()