legacyprovision: logout other logins when making new one

mautrix · Sep 11, 2024 · e7d0842 · e7d0842
1 parent 25f13a3
commit e7d0842
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cmd/mautrix-whatsapp/legacyprovision.go b/cmd/mautrix-whatsapp/legacyprovision.go
@@ -12,6 +12,7 @@ import (
 	"go.mau.fi/util/exhttp"
 	"go.mau.fi/whatsmeow/types"
 	"maunium.net/go/mautrix"
+	"maunium.net/go/mautrix/bridge/status"
 	"maunium.net/go/mautrix/bridgev2"
 	"maunium.net/go/mautrix/bridgev2/matrix"
 	"maunium.net/go/mautrix/id"
@@ -189,10 +190,19 @@ func legacyProvLogin(w http.ResponseWriter, r *http.Request) {
 				"platform": step.CompleteParams.UserLogin.Client.(*connector.WhatsAppClient).Device.Platform,
 				"phone":    step.CompleteParams.UserLogin.RemoteProfile.Phone,
 			})
+			go handleLoginComplete(context.WithoutCancel(ctx), user, step.CompleteParams.UserLogin)
 		}
 		break
 	}
 }
+func handleLoginComplete(ctx context.Context, user *bridgev2.User, newLogin *bridgev2.UserLogin) {
+	allLogins := user.GetUserLogins()
+	for _, login := range allLogins {
+		if login.ID != newLogin.ID {
+			login.Delete(ctx, status.BridgeState{StateEvent: status.StateLoggedOut, Reason: "LOGIN_OVERRIDDEN"}, bridgev2.DeleteOpts{})
+		}
+	}
+}
 
 func legacyProvLogout(w http.ResponseWriter, r *http.Request) {
 	user := m.Matrix.Provisioning.GetUser(r)