refactor: improve XMPP client/bridge architecture separation

- Move protocol normalization methods to XMPP client (ExtractChannelID, ExtractUserInfo, ExtractMessageBody)
- Replace message channel aggregation with direct handler delegation pattern
- XMPP client now focuses purely on protocol concerns (connection, deduplication, normalization)
- XMPP bridge handles business logic (BridgeMessage creation, routing)
- Add SourceRemoteID field to BridgeMessage for better message tracking
- Remove unused message channel infrastructure in favor of mux.MessageHandlerFunc pattern

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Felipe M 2025-08-06 12:31:28 +02:00
parent 8e9d87b176
commit 11a32afc53
No known key found for this signature in database
GPG key ID: 52E5D65FCF99808A
3 changed files with 127 additions and 125 deletions

View file

@ -16,6 +16,8 @@ import (
"github.com/mattermost/mattermost-plugin-bridge-xmpp/server/store/kvstore"
xmppClient "github.com/mattermost/mattermost-plugin-bridge-xmpp/server/xmpp"
"github.com/mattermost/mattermost/server/public/plugin"
"mellium.im/xmlstream"
"mellium.im/xmpp/stanza"
)
const (
@ -179,8 +181,6 @@ func (b *xmppBridge) Start() error {
// Start connection monitor
go b.connectionMonitor()
// Start message aggregation
go b.startMessageAggregation()
b.logger.LogInfo("Mattermost to XMPP bridge started successfully")
return nil
@ -230,6 +230,8 @@ func (b *xmppBridge) connectToXMPP() error {
b.logger.LogDebug("Set bridge client online presence")
}
b.bridgeClient.SetMessageHandler(b.handleIncomingXMPPMessage)
return nil
}
@ -572,35 +574,6 @@ func (b *xmppBridge) GetUserManager() pluginModel.BridgeUserManager {
return b.userManager
}
// startMessageAggregation starts the message aggregation goroutine
func (b *xmppBridge) startMessageAggregation() {
clientChannel := b.bridgeClient.GetMessageChannel()
for {
select {
case <-b.ctx.Done():
b.logger.LogDebug("Stopping XMPP message aggregation")
return
case msg, ok := <-clientChannel:
if !ok {
b.logger.LogDebug("Bridge client message channel closed")
return
}
// Forward to our bridge's message channel
select {
case b.incomingMessages <- msg:
// Message forwarded successfully
case <-b.ctx.Done():
return
default:
b.logger.LogWarn("Bridge message channel full, dropping message",
"source_channel", msg.SourceChannelID,
"user_id", msg.SourceUserID)
}
}
}
}
// GetMessageChannel returns the channel for incoming messages from XMPP
func (b *xmppBridge) GetMessageChannel() <-chan *pluginModel.DirectionalMessage {
@ -631,3 +604,72 @@ func (b *xmppBridge) GetRemoteID() string {
func (b *xmppBridge) ID() string {
return b.bridgeID
}
// handleIncomingXMPPMessage handles incoming XMPP messages and converts them to bridge messages
func (b *xmppBridge) handleIncomingXMPPMessage(msg stanza.Message, t xmlstream.TokenReadEncoder) error {
b.logger.LogDebug("XMPP bridge handling incoming message",
"from", msg.From.String(),
"to", msg.To.String(),
"type", fmt.Sprintf("%v", msg.Type))
// Only process groupchat messages for now (MUC messages from channels)
if msg.Type != stanza.GroupChatMessage {
b.logger.LogDebug("Ignoring non-groupchat message", "type", fmt.Sprintf("%v", msg.Type))
return nil
}
// Extract message body using client method
messageBody, err := b.bridgeClient.ExtractMessageBody(t)
if err != nil {
b.logger.LogWarn("Failed to extract message body", "error", err)
return nil
}
if messageBody == "" {
b.logger.LogDebug("Ignoring message with empty body")
return nil
}
// Use client methods for protocol normalization
channelID, err := b.bridgeClient.ExtractChannelID(msg.From)
if err != nil {
return fmt.Errorf("failed to extract channel ID: %w", err)
}
userID, displayName := b.bridgeClient.ExtractUserInfo(msg.From)
// Create bridge message
bridgeMessage := &pluginModel.BridgeMessage{
SourceBridge: b.bridgeID,
SourceChannelID: channelID,
SourceUserID: userID,
SourceUserName: displayName,
SourceRemoteID: b.remoteID,
Content: messageBody,
MessageType: "text",
Timestamp: time.Now(), // TODO: Parse timestamp from message if available
MessageID: msg.ID,
TargetBridges: []string{"mattermost"}, // Route to Mattermost
}
// Create directional message for incoming (XMPP -> Mattermost)
directionalMessage := &pluginModel.DirectionalMessage{
BridgeMessage: bridgeMessage,
Direction: pluginModel.DirectionIncoming,
}
// Send to bridge's message channel
select {
case b.incomingMessages <- directionalMessage:
b.logger.LogDebug("XMPP message queued for processing",
"channel_id", channelID,
"user_id", userID,
"message_id", msg.ID)
default:
b.logger.LogWarn("Bridge message channel full, dropping message",
"channel_id", channelID,
"user_id", userID)
}
return nil
}