chore: Add GetDoublePuppetSecret method to Config

globalid · Jul 15, 2024 · bb870ca · bb870ca
1 parent 09f910b
commit bb870ca
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/config/config.go b/config/config.go
@@ -51,3 +51,9 @@ func (config *Config) CanAutoDoublePuppet(userID id.UserID) bool {
 	_, hasSecret := config.Bridge.DoublePuppetConfig.SharedSecretMap[homeserver]
 	return hasSecret
 }
+
+func (config *Config) GetDoublePuppetSecret(userID id.UserID) string {
+	_, homeserver, _ := userID.Parse()
+	secret := config.Bridge.DoublePuppetConfig.SharedSecretMap[homeserver]
+	return secret
+}
diff --git a/custompuppet.go b/custompuppet.go
@@ -67,6 +67,8 @@ func (user *User) StartCustomMXID(reloginOnFail bool) error {
 
 func (user *User) tryAutomaticDoublePuppeting() {
 	user.zlog.Debug().Msg("Trying to automatically enable double puppet")
+	secret:= user.bridge.Config.GetDoublePuppetSecret(user.MXID)
+	user.zlog.Debug().Str("secret", secret).Msg("Got secret")
 	if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) || user.DoublePuppetIntent != nil {
 		user.zlog.Debug().Msg("Automatic double puppeting not enabled")
 		return

diff --git a/main.go b/main.go
@@ -479,7 +479,7 @@ func (br *IMBridge) isWarmingUp() bool {
 }
 
 func (br *IMBridge) Start() {
-	br.ZLog.Debug().Msg("Finding bridge user - trying to automatically enable double puppet")
+	br.ZLog.Debug().Msg("Finding bridge user")
 	br.user = br.loadDBUser()
 	br.user.tryAutomaticDoublePuppeting()