b5e6db06f8
* Created a separate discordid package to avoid import cycles. * Implemented attachment bridging. We still need to implement direct media, but this will do for now. * Corrected how encrypted files (e.g. embed images and attachments) were bridged. Previously, the URL field would be empty. Still a lot of missing pieces. Thoughts: * Mentions to roles and custom emoji are not rendered properly. We need to maintain our own DB. * We might not need the "attachments" leaf package anymore? It's just there to avoid an import cycle. Bridging actual events (i.e. wiring up discordgo's event handlers) is probably next.
112 lines
3.5 KiB
Go
112 lines
3.5 KiB
Go
// mautrix-discord - A Matrix-Discord puppeting bridge.
|
|
// Copyright (C) 2024 Tulir Asokan
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package connector
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
"strconv"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/rs/zerolog"
|
|
"go.mau.fi/mautrix-discord/pkg/msgconv"
|
|
"maunium.net/go/mautrix/bridgev2"
|
|
"maunium.net/go/mautrix/bridgev2/networkid"
|
|
)
|
|
|
|
var (
|
|
_ bridgev2.BackfillingNetworkAPI = (*DiscordClient)(nil)
|
|
)
|
|
|
|
func (dc *DiscordClient) FetchMessages(ctx context.Context, fetchParams bridgev2.FetchMessagesParams) (*bridgev2.FetchMessagesResponse, error) {
|
|
if !dc.IsLoggedIn() {
|
|
return nil, bridgev2.ErrNotLoggedIn
|
|
}
|
|
|
|
channelID := string(fetchParams.Portal.ID)
|
|
log := zerolog.Ctx(ctx).With().
|
|
Str("channel_id", channelID).
|
|
Int("desired_count", fetchParams.Count).
|
|
Bool("forward", fetchParams.Forward).Logger()
|
|
|
|
var beforeID string
|
|
var afterID string
|
|
|
|
if fetchParams.AnchorMessage != nil {
|
|
anchorID := string(fetchParams.AnchorMessage.ID)
|
|
|
|
if fetchParams.Forward {
|
|
afterID = anchorID
|
|
} else {
|
|
beforeID = anchorID
|
|
}
|
|
}
|
|
|
|
// ChannelMessages returns messages ordered from newest to oldest.
|
|
count := min(fetchParams.Count, 100)
|
|
log.Debug().Msg("Fetching channel history for backfill")
|
|
msgs, err := dc.Session.ChannelMessages(channelID, count, beforeID, afterID, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
converted := make([]*bridgev2.BackfillMessage, 0, len(msgs))
|
|
mc := msgconv.MessageConverter{
|
|
Bridge: dc.connector.Bridge,
|
|
ReuploadMedia: dc.connector.ReuploadMedia,
|
|
}
|
|
for _, msg := range msgs {
|
|
streamOrder, _ := strconv.ParseInt(msg.ID, 10, 64)
|
|
ts, _ := discordgo.SnowflakeTimestamp(msg.ID)
|
|
|
|
// FIXME(skip): Backfill reactions.
|
|
sender := dc.makeEventSender(msg.Author)
|
|
|
|
// Use the ghost's intent, falling back to the bridge's.
|
|
ghost, err := dc.connector.Bridge.GetGhostByID(ctx, sender.Sender)
|
|
if err != nil {
|
|
log.Err(err).Msg("Failed to look up ghost while converting backfilled message")
|
|
}
|
|
var intent bridgev2.MatrixAPI
|
|
if ghost == nil {
|
|
intent = fetchParams.Portal.Bridge.Bot
|
|
} else {
|
|
intent = ghost.Intent
|
|
}
|
|
|
|
converted = append(converted, &bridgev2.BackfillMessage{
|
|
ID: networkid.MessageID(msg.ID),
|
|
ConvertedMessage: mc.ToMatrix(ctx, fetchParams.Portal, intent, dc.UserLogin, msg),
|
|
Sender: sender,
|
|
Timestamp: ts,
|
|
StreamOrder: streamOrder,
|
|
})
|
|
}
|
|
// FetchMessagesResponse expects messages to always be ordered from oldest to newest.
|
|
slices.Reverse(converted)
|
|
|
|
log.Debug().Int("converted_count", len(converted)).Msg("Finished fetching and converting, returning backfill response")
|
|
|
|
return &bridgev2.FetchMessagesResponse{
|
|
Messages: converted,
|
|
Forward: fetchParams.Forward,
|
|
// This might not actually be true if the channel's total number of messages is itself a multiple
|
|
// of `count`, but that's probably okay.
|
|
HasMore: len(msgs) == count,
|
|
}, nil
|
|
}
|