From 46ef2b4793c762b28124c9d0495ea1fa205a8837 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Thu, 25 Nov 2021 23:14:31 +0200
Subject: [PATCH] Send bridge step message tracking checkpoint on websocket
 bridges

---
 appservice/http.go | 18 +++---------------
 1 file changed, 3 insertions(+), 15 deletions(-)

diff --git a/appservice/http.go b/appservice/http.go
index 3b39dbfb..6a05673b 100644
--- a/appservice/http.go
+++ b/appservice/http.go
@@ -198,7 +198,9 @@ func (as *AppService) handleEvents(evts []*event.Event, defaultTypeClass event.T
 			as.Log.Debugfln("Failed to parse content of %s (type %s): %v", evt.ID, evt.Type.Type, err)
 		}
 
-		go as.sendMessageSendCheckpoint(evt)
+		if _, ok := CheckpointTypes[evt.Type]; ok {
+			go as.SendMessageSendCheckpoint(evt, StepBridge)
+		}
 
 		if evt.Type.IsState() {
 			// TODO remove this check after https://github.com/matrix-org/synapse/pull/11265
@@ -211,20 +213,6 @@ func (as *AppService) handleEvents(evts []*event.Event, defaultTypeClass event.T
 	}
 }
 
-func (as *AppService) sendMessageSendCheckpoint(evt *event.Event) {
-	endpoint := as.MessageSendCheckpointEndpoint
-	if endpoint == "" {
-		return
-	}
-
-	if _, ok := CheckpointTypes[evt.Type]; !ok {
-		return
-	}
-
-	as.Log.Debugfln("Sending message send checkpoint for %s to API server", evt.ID)
-	as.SendMessageSendCheckpoint(evt, StepBridge)
-}
-
 // GetRoom handles a /rooms GET call from the homeserver.
 func (as *AppService) GetRoom(w http.ResponseWriter, r *http.Request) {
 	if !as.CheckServerToken(w, r) {