From 51f0be415befb67c10f84bbd74b747c61aadb8c2 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Tue, 28 Jul 2020 15:58:36 +0300
Subject: [PATCH] Disconnect and reconnect automatically if post-login ping
 fails

---
 user.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/user.go b/user.go
index b4e8c0d..ed3be3b 100644
--- a/user.go
+++ b/user.go
@@ -456,7 +456,19 @@ func (user *User) intPostLogin() {
 
 	err := user.Conn.AdminTest()
 	if err != nil {
-		user.sendMarkdownBridgeAlert("Post-connection ping failed: %v", err)
+		user.log.Errorfln("Post-connection ping failed: %v. Disconnecting and then reconnecting after a second", err)
+		sess, disconnectErr := user.Conn.Disconnect()
+		if disconnectErr != nil {
+			user.log.Warnln("Error while disconnecting after failed post-login ping:", disconnectErr)
+		} else {
+			user.Session = &sess
+		}
+		user.bridge.Metrics.TrackDisconnection(user.MXID)
+		go func() {
+			time.Sleep(1 * time.Second)
+			user.tryReconnect(fmt.Sprintf("Post-connection ping failed: %v", err))
+		}()
+		return
 	} else {
 		user.log.Debugln("Post-login ping OK")
 	}