From 4bdbe67c943435dfd9380d3f6b406c0ac1cf5ee2 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Mon, 30 Dec 2019 20:40:20 +0200
Subject: [PATCH] Mark custom puppets as registered automatically. Fixes #128

---
 custompuppet.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/custompuppet.go b/custompuppet.go
index f1c335d..b7f4a3c 100644
--- a/custompuppet.go
+++ b/custompuppet.go
@@ -56,6 +56,7 @@ func (puppet *Puppet) SwitchCustomMXID(accessToken string, mxid string) error {
 	if len(puppet.CustomMXID) > 0 {
 		puppet.bridge.puppetsByCustomMXID[puppet.CustomMXID] = puppet
 	}
+	puppet.bridge.AS.StateStore.MarkRegistered(puppet.CustomMXID)
 	puppet.Update()
 	// TODO leave rooms with default puppet
 	return nil