Compare commits
24 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
00465bb715 | ||
|
|
cf640ac83d | ||
|
|
67c8d9237e | ||
|
|
b2d7077e8d | ||
|
|
d5db336eee | ||
|
|
b153e70f2a | ||
|
|
cc30353075 | ||
|
|
4c62fe8b12 | ||
|
|
8c57b7a69b | ||
|
|
a265d03319 | ||
|
|
1c606e97a6 | ||
|
|
e6108cb25d | ||
|
|
d004aea9cb | ||
|
|
0fd88fedea | ||
|
|
1e9099e989 | ||
|
|
52fa4da8b2 | ||
|
|
4393772ccc | ||
|
|
824dea4745 | ||
|
|
07182efddd | ||
|
|
280e01969a | ||
|
|
084cde0162 | ||
|
|
434f27c8b4 | ||
|
|
75181741da | ||
|
|
e85f50633d |
22
CHANGELOG.md
22
CHANGELOG.md
@@ -1,3 +1,20 @@
|
||||
# v0.5.0 (2023-06-16)
|
||||
|
||||
* Added support for intentional mentions in Matrix (MSC3952).
|
||||
* Added `GlobalName` variable to displayname templates and updated the default
|
||||
template to prefer it over usernames.
|
||||
* Added `Webhook` variable to displayname templates to allow determining if a
|
||||
ghost user is a webhook.
|
||||
* Added guild profiles and webhook profiles as a custom field in Matrix
|
||||
message events.
|
||||
* Added support for bulk message delete from Discord.
|
||||
* Added support for appservice websockets.
|
||||
* Enabled parsing headers (`#`) in Discord markdown.
|
||||
* Messages that consist of a single image link are now bridged as images to
|
||||
closer match Discord.
|
||||
* Stopped bridging incoming typing notifications from users who are logged into
|
||||
the bridge to prevent echoes.
|
||||
|
||||
# v0.4.0 (2023-05-16)
|
||||
|
||||
* Added bridging of friend nicks into DM room names.
|
||||
@@ -6,12 +23,17 @@
|
||||
* Added conversion of replies to embeds when sending messages via webhook.
|
||||
* Added option to disable caching reuploaded media. This may be necessary when
|
||||
using a media repo that doesn't create a unique mxc URI for each upload.
|
||||
* Added option to disable uploading files directly to the Discord CDN
|
||||
(and send as form parts in the message send request instead).
|
||||
* Improved formatting of error messages returned by Discord.
|
||||
* Enabled discordgo info logs by default.
|
||||
* Fixed limited backfill always stopping after 50 messages
|
||||
(thanks to [@odrling] in [#81]).
|
||||
* Fixed startup sync to sync most recent private channels first.
|
||||
* Fixed syncing group DM participants when they change.
|
||||
* Fixed bridging animated emojis in messages.
|
||||
* Stopped handling all message edits from relay webhook to prevent incorrect
|
||||
edits.
|
||||
* Possibly fixed inviting to portal rooms when multiple Matrix users use the
|
||||
bridge.
|
||||
|
||||
|
||||
@@ -128,17 +128,17 @@ func (br *DiscordBridge) uploadMatrixAttachment(intent *appservice.IntentAPI, da
|
||||
ContentType: uploadMime,
|
||||
}
|
||||
if br.Config.Homeserver.AsyncMedia {
|
||||
resp, err := intent.UnstableCreateMXC()
|
||||
resp, err := intent.CreateMXC()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
dbFile.MXC = resp.ContentURI
|
||||
req.UnstableMXC = resp.ContentURI
|
||||
req.UploadURL = resp.UploadURL
|
||||
req.MXC = resp.ContentURI
|
||||
req.UnstableUploadURL = resp.UnstableUploadURL
|
||||
go func() {
|
||||
_, err = intent.UploadMedia(req)
|
||||
if err != nil {
|
||||
br.Log.Errorfln("Failed to upload %s: %v", req.UnstableMXC, err)
|
||||
br.Log.Errorfln("Failed to upload %s: %v", req.MXC, err)
|
||||
dbFile.Delete()
|
||||
}
|
||||
}()
|
||||
|
||||
40
avatar.go
40
avatar.go
@@ -1,40 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
|
||||
"maunium.net/go/mautrix/appservice"
|
||||
"maunium.net/go/mautrix/id"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
)
|
||||
|
||||
func uploadAvatar(intent *appservice.IntentAPI, url string) (id.ContentURI, error) {
|
||||
req, err := http.NewRequest(http.MethodGet, url, nil)
|
||||
if err != nil {
|
||||
return id.ContentURI{}, fmt.Errorf("failed to prepare request: %w", err)
|
||||
}
|
||||
for key, value := range discordgo.DroidImageHeaders {
|
||||
req.Header.Set(key, value)
|
||||
}
|
||||
getResp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return id.ContentURI{}, fmt.Errorf("failed to download avatar: %w", err)
|
||||
}
|
||||
|
||||
data, err := io.ReadAll(getResp.Body)
|
||||
_ = getResp.Body.Close()
|
||||
if err != nil {
|
||||
return id.ContentURI{}, fmt.Errorf("failed to read avatar data: %w", err)
|
||||
}
|
||||
|
||||
mime := http.DetectContentType(data)
|
||||
resp, err := intent.UploadBytes(data, mime)
|
||||
if err != nil {
|
||||
return id.ContentURI{}, fmt.Errorf("failed to upload avatar to Matrix: %w", err)
|
||||
}
|
||||
|
||||
return resp.ContentURI, nil
|
||||
}
|
||||
15
backfill.go
15
backfill.go
@@ -134,7 +134,7 @@ func (portal *Portal) backfillLimited(log zerolog.Logger, source *User, limit in
|
||||
Bool("found_all", foundAll).
|
||||
Msg("Collected messages to backfill")
|
||||
sort.Sort(MessageSlice(messages))
|
||||
if !foundAll {
|
||||
if !foundAll && after != "" {
|
||||
_, err = portal.sendMatrixMessage(portal.MainIntent(), event.EventMessage, &event.MessageEventContent{
|
||||
MsgType: event.MsgNotice,
|
||||
Body: "Some messages may have been missed here while the bridge was offline.",
|
||||
@@ -211,13 +211,14 @@ func (portal *Portal) convertMessageBatch(log zerolog.Logger, source *User, mess
|
||||
for _, msg := range messages {
|
||||
for _, mention := range msg.Mentions {
|
||||
puppet := portal.bridge.GetPuppetByID(mention.ID)
|
||||
puppet.UpdateInfo(nil, mention)
|
||||
puppet.UpdateInfo(nil, mention, "")
|
||||
}
|
||||
|
||||
puppet := portal.bridge.GetPuppetByID(msg.Author.ID)
|
||||
puppet.UpdateInfo(source, msg.Author)
|
||||
puppet.UpdateInfo(source, msg.Author, msg.WebhookID)
|
||||
intent := puppet.IntentFor(portal)
|
||||
replyTo := portal.getReplyTarget(source, "", msg.MessageReference, msg.Embeds, true)
|
||||
mentions := portal.convertDiscordMentions(msg, false)
|
||||
|
||||
ts, _ := discordgo.SnowflakeTimestamp(msg.ID)
|
||||
log := log.With().
|
||||
@@ -225,13 +226,18 @@ func (portal *Portal) convertMessageBatch(log zerolog.Logger, source *User, mess
|
||||
Int("message_type", int(msg.Type)).
|
||||
Str("author_id", msg.Author.ID).
|
||||
Logger()
|
||||
parts := portal.convertDiscordMessage(log.WithContext(ctx), intent, msg)
|
||||
parts := portal.convertDiscordMessage(log.WithContext(ctx), puppet, intent, msg)
|
||||
for i, part := range parts {
|
||||
if replyTo != nil {
|
||||
part.Content.RelatesTo = &event.RelatesTo{InReplyTo: replyTo}
|
||||
// Only set reply for first event
|
||||
replyTo = nil
|
||||
}
|
||||
|
||||
part.Content.Mentions = mentions
|
||||
// Only set mentions for first event, but keep empty object for rest
|
||||
mentions = &event.Mentions{}
|
||||
|
||||
partName := part.AttachmentID
|
||||
// Always use blank part name for first part so that replies and other things
|
||||
// can reference it without knowing about attachments.
|
||||
@@ -262,6 +268,7 @@ func (portal *Portal) convertMessageBatch(log zerolog.Logger, source *User, mess
|
||||
SenderID: msg.Author.ID,
|
||||
Timestamp: ts,
|
||||
AttachmentID: part.AttachmentID,
|
||||
SenderMXID: intent.UserID,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
@@ -51,6 +51,8 @@ type BridgeConfig struct {
|
||||
DeletePortalOnChannelDelete bool `yaml:"delete_portal_on_channel_delete"`
|
||||
DeleteGuildOnLeave bool `yaml:"delete_guild_on_leave"`
|
||||
FederateRooms bool `yaml:"federate_rooms"`
|
||||
PrefixWebhookMessages bool `yaml:"prefix_webhook_messages"`
|
||||
EnableWebhookAvatars bool `yaml:"enable_webhook_avatars"`
|
||||
UseDiscordCDNUpload bool `yaml:"use_discord_cdn_upload"`
|
||||
|
||||
CacheMedia string `yaml:"cache_media"`
|
||||
@@ -287,9 +289,14 @@ func (bc BridgeConfig) FormatUsername(userID string) string {
|
||||
return buffer.String()
|
||||
}
|
||||
|
||||
func (bc BridgeConfig) FormatDisplayname(user *discordgo.User) string {
|
||||
type DisplaynameParams struct {
|
||||
*discordgo.User
|
||||
Webhook bool
|
||||
}
|
||||
|
||||
func (bc BridgeConfig) FormatDisplayname(user *discordgo.User, webhook bool) string {
|
||||
var buffer strings.Builder
|
||||
_ = bc.displaynameTemplate.Execute(&buffer, user)
|
||||
_ = bc.displaynameTemplate.Execute(&buffer, &DisplaynameParams{user, webhook})
|
||||
return buffer.String()
|
||||
}
|
||||
|
||||
|
||||
@@ -55,6 +55,8 @@ func DoUpgrade(helper *up.Helper) {
|
||||
helper.Copy(up.Bool, "bridge", "delete_portal_on_channel_delete")
|
||||
helper.Copy(up.Bool, "bridge", "delete_guild_on_leave")
|
||||
helper.Copy(up.Bool, "bridge", "federate_rooms")
|
||||
helper.Copy(up.Bool, "bridge", "prefix_webhook_messages")
|
||||
helper.Copy(up.Bool, "bridge", "enable_webhook_avatars")
|
||||
helper.Copy(up.Bool, "bridge", "use_discord_cdn_upload")
|
||||
helper.Copy(up.Bool, "bridge", "media_patterns", "enabled")
|
||||
helper.Copy(up.Str, "bridge", "cache_media")
|
||||
@@ -85,6 +87,7 @@ func DoUpgrade(helper *up.Helper) {
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "require")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "appservice")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "allow_key_sharing")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "plaintext_mentions")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "delete_keys", "delete_outbound_on_ack")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "delete_keys", "dont_store_outbound")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "delete_keys", "ratchet_on_decrypt")
|
||||
@@ -98,6 +101,7 @@ func DoUpgrade(helper *up.Helper) {
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "rotation", "enable_custom")
|
||||
helper.Copy(up.Int, "bridge", "encryption", "rotation", "milliseconds")
|
||||
helper.Copy(up.Int, "bridge", "encryption", "rotation", "messages")
|
||||
helper.Copy(up.Bool, "bridge", "encryption", "rotation", "disable_device_change_key_rotation")
|
||||
|
||||
helper.Copy(up.Str, "bridge", "provisioning", "prefix")
|
||||
if secret, ok := helper.Get(up.Str, "bridge", "provisioning", "shared_secret"); !ok || secret == "generate" {
|
||||
|
||||
@@ -39,8 +39,8 @@ func (fq *FileQuery) Get(url string, encrypted bool) *File {
|
||||
return fq.New().Scan(fq.db.QueryRow(query, url, encrypted))
|
||||
}
|
||||
|
||||
func (fq *FileQuery) GetByMXC(mxc id.ContentURI) *File {
|
||||
query := fileSelect + " WHERE mxc=$1"
|
||||
func (fq *FileQuery) GetEmojiByMXC(mxc id.ContentURI) *File {
|
||||
query := fileSelect + " WHERE mxc=$1 AND emoji_name<>'' LIMIT 1"
|
||||
return fq.New().Scan(fq.db.QueryRow(query, mxc.String()))
|
||||
}
|
||||
|
||||
|
||||
@@ -19,7 +19,7 @@ type MessageQuery struct {
|
||||
}
|
||||
|
||||
const (
|
||||
messageSelect = "SELECT dcid, dc_attachment_id, dc_chan_id, dc_chan_receiver, dc_sender, timestamp, dc_edit_timestamp, dc_thread_id, mxid FROM message"
|
||||
messageSelect = "SELECT dcid, dc_attachment_id, dc_chan_id, dc_chan_receiver, dc_sender, timestamp, dc_edit_timestamp, dc_thread_id, mxid, sender_mxid FROM message"
|
||||
)
|
||||
|
||||
func (mq *MessageQuery) New() *Message {
|
||||
@@ -99,11 +99,11 @@ func (mq *MessageQuery) MassInsert(key PortalKey, msgs []Message) {
|
||||
if len(msgs) == 0 {
|
||||
return
|
||||
}
|
||||
valueStringFormat := "($%d, $%d, $1, $2, $%d, $%d, $%d, $%d, $%d)"
|
||||
valueStringFormat := "($%d, $%d, $1, $2, $%d, $%d, $%d, $%d, $%d, $%d)"
|
||||
if mq.db.Dialect == dbutil.SQLite {
|
||||
valueStringFormat = strings.ReplaceAll(valueStringFormat, "$", "?")
|
||||
}
|
||||
params := make([]interface{}, 2+len(msgs)*7)
|
||||
params := make([]interface{}, 2+len(msgs)*8)
|
||||
placeholders := make([]string, len(msgs))
|
||||
params[0] = key.ChannelID
|
||||
params[1] = key.Receiver
|
||||
@@ -116,7 +116,8 @@ func (mq *MessageQuery) MassInsert(key PortalKey, msgs []Message) {
|
||||
params[baseIndex+4] = msg.editTimestampVal()
|
||||
params[baseIndex+5] = msg.ThreadID
|
||||
params[baseIndex+6] = msg.MXID
|
||||
placeholders[i] = fmt.Sprintf(valueStringFormat, baseIndex+1, baseIndex+2, baseIndex+3, baseIndex+4, baseIndex+5, baseIndex+6, baseIndex+7)
|
||||
params[baseIndex+7] = msg.SenderMXID.String()
|
||||
placeholders[i] = fmt.Sprintf(valueStringFormat, baseIndex+1, baseIndex+2, baseIndex+3, baseIndex+4, baseIndex+5, baseIndex+6, baseIndex+7, baseIndex+8)
|
||||
}
|
||||
_, err := mq.db.Exec(fmt.Sprintf(messageMassInsertTemplate, strings.Join(placeholders, ", ")), params...)
|
||||
if err != nil {
|
||||
@@ -137,7 +138,8 @@ type Message struct {
|
||||
EditTimestamp time.Time
|
||||
ThreadID string
|
||||
|
||||
MXID id.EventID
|
||||
MXID id.EventID
|
||||
SenderMXID id.UserID
|
||||
}
|
||||
|
||||
func (m *Message) DiscordProtoChannelID() string {
|
||||
@@ -151,7 +153,7 @@ func (m *Message) DiscordProtoChannelID() string {
|
||||
func (m *Message) Scan(row dbutil.Scannable) *Message {
|
||||
var ts, editTS int64
|
||||
|
||||
err := row.Scan(&m.DiscordID, &m.AttachmentID, &m.Channel.ChannelID, &m.Channel.Receiver, &m.SenderID, &ts, &editTS, &m.ThreadID, &m.MXID)
|
||||
err := row.Scan(&m.DiscordID, &m.AttachmentID, &m.Channel.ChannelID, &m.Channel.Receiver, &m.SenderID, &ts, &editTS, &m.ThreadID, &m.MXID, &m.SenderMXID)
|
||||
if err != nil {
|
||||
if !errors.Is(err, sql.ErrNoRows) {
|
||||
m.log.Errorln("Database scan failed:", err)
|
||||
@@ -173,12 +175,12 @@ func (m *Message) Scan(row dbutil.Scannable) *Message {
|
||||
|
||||
const messageInsertQuery = `
|
||||
INSERT INTO message (
|
||||
dcid, dc_attachment_id, dc_chan_id, dc_chan_receiver, dc_sender, timestamp, dc_edit_timestamp, dc_thread_id, mxid
|
||||
dcid, dc_attachment_id, dc_chan_id, dc_chan_receiver, dc_sender, timestamp, dc_edit_timestamp, dc_thread_id, mxid, sender_mxid
|
||||
)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
||||
`
|
||||
|
||||
var messageMassInsertTemplate = strings.Replace(messageInsertQuery, "($1, $2, $3, $4, $5, $6, $7, $8, $9)", "%s", 1)
|
||||
var messageMassInsertTemplate = strings.Replace(messageInsertQuery, "($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)", "%s", 1)
|
||||
|
||||
type MessagePart struct {
|
||||
AttachmentID string
|
||||
@@ -196,11 +198,11 @@ func (m *Message) MassInsertParts(msgs []MessagePart) {
|
||||
if len(msgs) == 0 {
|
||||
return
|
||||
}
|
||||
valueStringFormat := "($1, $%d, $2, $3, $4, $5, $6, $7, $%d)"
|
||||
valueStringFormat := "($1, $%d, $2, $3, $4, $5, $6, $7, $%d, $8)"
|
||||
if m.db.Dialect == dbutil.SQLite {
|
||||
valueStringFormat = strings.ReplaceAll(valueStringFormat, "$", "?")
|
||||
}
|
||||
params := make([]interface{}, 7+len(msgs)*2)
|
||||
params := make([]interface{}, 8+len(msgs)*2)
|
||||
placeholders := make([]string, len(msgs))
|
||||
params[0] = m.DiscordID
|
||||
params[1] = m.Channel.ChannelID
|
||||
@@ -209,10 +211,11 @@ func (m *Message) MassInsertParts(msgs []MessagePart) {
|
||||
params[4] = m.Timestamp.UnixMilli()
|
||||
params[5] = m.editTimestampVal()
|
||||
params[6] = m.ThreadID
|
||||
params[7] = m.SenderMXID.String()
|
||||
for i, msg := range msgs {
|
||||
params[7+i*2] = msg.AttachmentID
|
||||
params[7+i*2+1] = msg.MXID
|
||||
placeholders[i] = fmt.Sprintf(valueStringFormat, 7+i*2+1, 7+i*2+2)
|
||||
params[8+i*2] = msg.AttachmentID
|
||||
params[8+i*2+1] = msg.MXID
|
||||
placeholders[i] = fmt.Sprintf(valueStringFormat, 8+i*2+1, 8+i*2+2)
|
||||
}
|
||||
_, err := m.db.Exec(fmt.Sprintf(messageMassInsertTemplate, strings.Join(placeholders, ", ")), params...)
|
||||
if err != nil {
|
||||
@@ -224,7 +227,7 @@ func (m *Message) MassInsertParts(msgs []MessagePart) {
|
||||
func (m *Message) Insert() {
|
||||
_, err := m.db.Exec(messageInsertQuery,
|
||||
m.DiscordID, m.AttachmentID, m.Channel.ChannelID, m.Channel.Receiver, m.SenderID,
|
||||
m.Timestamp.UnixMilli(), m.editTimestampVal(), m.ThreadID, m.MXID)
|
||||
m.Timestamp.UnixMilli(), m.editTimestampVal(), m.ThreadID, m.MXID, m.SenderMXID.String())
|
||||
|
||||
if err != nil {
|
||||
m.log.Warnfln("Failed to insert %s@%s: %v", m.DiscordID, m.Channel, err)
|
||||
|
||||
@@ -11,7 +11,7 @@ import (
|
||||
|
||||
const (
|
||||
puppetSelect = "SELECT id, name, name_set, avatar, avatar_url, avatar_set," +
|
||||
" contact_info_set, username, discriminator, is_bot, custom_mxid, access_token, next_batch" +
|
||||
" contact_info_set, global_name, username, discriminator, is_bot, is_webhook, custom_mxid, access_token, next_batch" +
|
||||
" FROM puppet "
|
||||
)
|
||||
|
||||
@@ -75,9 +75,11 @@ type Puppet struct {
|
||||
|
||||
ContactInfoSet bool
|
||||
|
||||
GlobalName string
|
||||
Username string
|
||||
Discriminator string
|
||||
IsBot bool
|
||||
IsWebhook bool
|
||||
|
||||
CustomMXID id.UserID
|
||||
AccessToken string
|
||||
@@ -89,7 +91,7 @@ func (p *Puppet) Scan(row dbutil.Scannable) *Puppet {
|
||||
var customMXID, accessToken, nextBatch sql.NullString
|
||||
|
||||
err := row.Scan(&p.ID, &p.Name, &p.NameSet, &p.Avatar, &avatarURL, &p.AvatarSet, &p.ContactInfoSet,
|
||||
&p.Username, &p.Discriminator, &p.IsBot, &customMXID, &accessToken, &nextBatch)
|
||||
&p.GlobalName, &p.Username, &p.Discriminator, &p.IsBot, &p.IsWebhook, &customMXID, &accessToken, &nextBatch)
|
||||
|
||||
if err != nil {
|
||||
if err != sql.ErrNoRows {
|
||||
@@ -110,11 +112,16 @@ func (p *Puppet) Scan(row dbutil.Scannable) *Puppet {
|
||||
|
||||
func (p *Puppet) Insert() {
|
||||
query := `
|
||||
INSERT INTO puppet (id, name, name_set, avatar, avatar_url, avatar_set, contact_info_set, username, discriminator, is_bot, custom_mxid, access_token, next_batch)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
|
||||
INSERT INTO puppet (
|
||||
id, name, name_set, avatar, avatar_url, avatar_set, contact_info_set,
|
||||
global_name, username, discriminator, is_bot, is_webhook,
|
||||
custom_mxid, access_token, next_batch
|
||||
)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15)
|
||||
`
|
||||
_, err := p.db.Exec(query, p.ID, p.Name, p.NameSet, p.Avatar, p.AvatarURL.String(), p.AvatarSet, p.ContactInfoSet,
|
||||
p.Username, p.Discriminator, p.IsBot, strPtr(p.CustomMXID), strPtr(p.AccessToken), strPtr(p.NextBatch))
|
||||
p.GlobalName, p.Username, p.Discriminator, p.IsBot, p.IsWebhook,
|
||||
strPtr(p.CustomMXID), strPtr(p.AccessToken), strPtr(p.NextBatch))
|
||||
|
||||
if err != nil {
|
||||
p.log.Warnfln("Failed to insert %s: %v", p.ID, err)
|
||||
@@ -125,12 +132,17 @@ func (p *Puppet) Insert() {
|
||||
func (p *Puppet) Update() {
|
||||
query := `
|
||||
UPDATE puppet SET name=$1, name_set=$2, avatar=$3, avatar_url=$4, avatar_set=$5, contact_info_set=$6,
|
||||
username=$7, discriminator=$8, is_bot=$9, custom_mxid=$10, access_token=$11, next_batch=$12
|
||||
WHERE id=$13
|
||||
global_name=$7, username=$8, discriminator=$9, is_bot=$10, is_webhook=$11,
|
||||
custom_mxid=$12, access_token=$13, next_batch=$14
|
||||
WHERE id=$15
|
||||
`
|
||||
_, err := p.db.Exec(query, p.Name, p.NameSet, p.Avatar, p.AvatarURL.String(), p.AvatarSet, p.ContactInfoSet,
|
||||
p.Username, p.Discriminator, p.IsBot, strPtr(p.CustomMXID), strPtr(p.AccessToken), strPtr(p.NextBatch),
|
||||
p.ID)
|
||||
_, err := p.db.Exec(
|
||||
query,
|
||||
p.Name, p.NameSet, p.Avatar, p.AvatarURL.String(), p.AvatarSet, p.ContactInfoSet,
|
||||
p.GlobalName, p.Username, p.Discriminator, p.IsBot, p.IsWebhook,
|
||||
strPtr(p.CustomMXID), strPtr(p.AccessToken), strPtr(p.NextBatch),
|
||||
p.ID,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
p.log.Warnfln("Failed to update %s: %v", p.ID, err)
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
-- v0 -> v19: Latest revision
|
||||
-- v0 -> v22 (compatible with v19+): Latest revision
|
||||
|
||||
CREATE TABLE guild (
|
||||
dcid TEXT PRIMARY KEY,
|
||||
@@ -71,9 +71,11 @@ CREATE TABLE puppet (
|
||||
|
||||
contact_info_set BOOLEAN NOT NULL DEFAULT false,
|
||||
|
||||
global_name TEXT NOT NULL DEFAULT '',
|
||||
username TEXT NOT NULL DEFAULT '',
|
||||
discriminator TEXT NOT NULL DEFAULT '',
|
||||
is_bot BOOLEAN NOT NULL DEFAULT false,
|
||||
is_webhook BOOLEAN NOT NULL DEFAULT false,
|
||||
|
||||
custom_mxid TEXT,
|
||||
access_token TEXT,
|
||||
@@ -113,7 +115,8 @@ CREATE TABLE message (
|
||||
dc_edit_timestamp BIGINT NOT NULL,
|
||||
dc_thread_id TEXT NOT NULL,
|
||||
|
||||
mxid TEXT NOT NULL UNIQUE,
|
||||
mxid TEXT NOT NULL UNIQUE,
|
||||
sender_mxid TEXT NOT NULL DEFAULT '',
|
||||
|
||||
PRIMARY KEY (dcid, dc_attachment_id, dc_chan_id, dc_chan_receiver),
|
||||
CONSTRAINT message_portal_fkey FOREIGN KEY (dc_chan_id, dc_chan_receiver) REFERENCES portal (dcid, receiver) ON DELETE CASCADE
|
||||
@@ -157,7 +160,7 @@ CREATE TABLE role (
|
||||
CREATE TABLE discord_file (
|
||||
url TEXT,
|
||||
encrypted BOOLEAN,
|
||||
mxc TEXT NOT NULL UNIQUE,
|
||||
mxc TEXT NOT NULL,
|
||||
|
||||
id TEXT,
|
||||
emoji_name TEXT,
|
||||
@@ -171,3 +174,5 @@ CREATE TABLE discord_file (
|
||||
|
||||
PRIMARY KEY (url, encrypted)
|
||||
);
|
||||
|
||||
CREATE INDEX discord_file_mxc_idx ON discord_file (mxc);
|
||||
|
||||
2
database/upgrades/20-message-sender-mxid.sql
Normal file
2
database/upgrades/20-message-sender-mxid.sql
Normal file
@@ -0,0 +1,2 @@
|
||||
-- v20 (compatible with v19+): Store message sender Matrix user ID
|
||||
ALTER TABLE message ADD COLUMN sender_mxid TEXT NOT NULL DEFAULT '';
|
||||
3
database/upgrades/21-more-puppet-info.sql
Normal file
3
database/upgrades/21-more-puppet-info.sql
Normal file
@@ -0,0 +1,3 @@
|
||||
-- v21 (compatible with v19+): Store global displayname and is webhook status for puppets
|
||||
ALTER TABLE puppet ADD COLUMN global_name TEXT NOT NULL DEFAULT '';
|
||||
ALTER TABLE puppet ADD COLUMN is_webhook BOOLEAN NOT NULL DEFAULT false;
|
||||
26
database/upgrades/22-file-cache-duplicate-mxc.sql
Normal file
26
database/upgrades/22-file-cache-duplicate-mxc.sql
Normal file
@@ -0,0 +1,26 @@
|
||||
-- v22 (compatible with v19+): Allow non-unique mxc URIs in file cache
|
||||
CREATE TABLE new_discord_file (
|
||||
url TEXT,
|
||||
encrypted BOOLEAN,
|
||||
mxc TEXT NOT NULL,
|
||||
|
||||
id TEXT,
|
||||
emoji_name TEXT,
|
||||
|
||||
size BIGINT NOT NULL,
|
||||
width INTEGER,
|
||||
height INTEGER,
|
||||
mime_type TEXT NOT NULL,
|
||||
decryption_info jsonb,
|
||||
timestamp BIGINT NOT NULL,
|
||||
|
||||
PRIMARY KEY (url, encrypted)
|
||||
);
|
||||
|
||||
INSERT INTO new_discord_file (url, encrypted, mxc, id, emoji_name, size, width, height, mime_type, decryption_info, timestamp)
|
||||
SELECT url, encrypted, mxc, id, emoji_name, size, width, height, mime_type, decryption_info, timestamp FROM discord_file;
|
||||
|
||||
DROP TABLE discord_file;
|
||||
ALTER TABLE new_discord_file RENAME TO discord_file;
|
||||
|
||||
CREATE INDEX discord_file_mxc_idx ON discord_file (mxc);
|
||||
@@ -20,6 +20,13 @@ homeserver:
|
||||
# Does the homeserver support https://github.com/matrix-org/matrix-spec-proposals/pull/2246?
|
||||
async_media: false
|
||||
|
||||
# Should the bridge use a websocket for connecting to the homeserver?
|
||||
# The server side is currently not documented anywhere and is only implemented by mautrix-wsproxy,
|
||||
# mautrix-asmux (deprecated), and hungryserv (proprietary).
|
||||
websocket: false
|
||||
# How often should the websocket be pinged? Pinging will be disabled if this is zero.
|
||||
ping_interval_seconds: 0
|
||||
|
||||
# Application service host/registration related details.
|
||||
# Changing these values requires regeneration of the registration.
|
||||
appservice:
|
||||
@@ -80,11 +87,13 @@ bridge:
|
||||
# Displayname template for Discord users. This is also used as the room name in DMs if private_chat_portal_meta is enabled.
|
||||
# Available variables:
|
||||
# .ID - Internal user ID
|
||||
# .Username - User's displayname on Discord
|
||||
# .Username - Legacy display/username on Discord
|
||||
# .GlobalName - New displayname on Discord
|
||||
# .Discriminator - The 4 numbers after the name on Discord
|
||||
# .Bot - Whether the user is a bot
|
||||
# .System - Whether the user is an official system user
|
||||
displayname_template: '{{.Username}}#{{.Discriminator}}{{if .Bot}} (bot){{end}}'
|
||||
# .Webhook - Whether the user is a webhook
|
||||
displayname_template: '{{or .GlobalName .Username}}{{if .Bot}} (bot){{end}}'
|
||||
# Displayname template for Discord channels (bridged as rooms, or spaces when type=4).
|
||||
# Available variables:
|
||||
# .Name - Channel name, or user displayname (pre-formatted with displayname_template) in DMs.
|
||||
@@ -145,6 +154,11 @@ bridge:
|
||||
# Whether or not created rooms should have federation enabled.
|
||||
# If false, created portal rooms will never be federated.
|
||||
federate_rooms: true
|
||||
# Prefix messages from webhooks with the profile info? This can be used along with a custom displayname_template
|
||||
# to better handle webhooks that change their name all the time (like ones used by bridges).
|
||||
prefix_webhook_messages: false
|
||||
# Bridge webhook avatars?
|
||||
enable_webhook_avatars: true
|
||||
# Should the bridge upload media to the Discord CDN directly before sending the message when using a user token,
|
||||
# like the official client does? The other option is sending the media in the message send request as a form part
|
||||
# (which is always used by bots and webhooks).
|
||||
@@ -246,6 +260,8 @@ bridge:
|
||||
# Enable key sharing? If enabled, key requests for rooms where users are in will be fulfilled.
|
||||
# You must use a client that supports requesting keys from other users to use this feature.
|
||||
allow_key_sharing: false
|
||||
# Should users mentions be in the event wire content to enable the server to send push notifications?
|
||||
plaintext_mentions: false
|
||||
# Options for deleting megolm sessions from the bridge.
|
||||
delete_keys:
|
||||
# Beeper-specific: delete outbound sessions when hungryserv confirms
|
||||
@@ -298,6 +314,10 @@ bridge:
|
||||
# default.
|
||||
messages: 100
|
||||
|
||||
# Disable rotating keys when a user's devices change?
|
||||
# You should not enable this option unless you understand all the implications.
|
||||
disable_device_change_key_rotation: false
|
||||
|
||||
# Settings for provisioning API
|
||||
provisioning:
|
||||
# Prefix for the provisioning API paths.
|
||||
|
||||
12
formatter.go
12
formatter.go
@@ -26,6 +26,7 @@ import (
|
||||
"github.com/yuin/goldmark/extension"
|
||||
"github.com/yuin/goldmark/parser"
|
||||
"github.com/yuin/goldmark/util"
|
||||
"golang.org/x/exp/slices"
|
||||
|
||||
"maunium.net/go/mautrix/event"
|
||||
"maunium.net/go/mautrix/format"
|
||||
@@ -58,7 +59,7 @@ func (b *indentableParagraphParser) CanAcceptIndentedLine() bool {
|
||||
|
||||
var removeFeaturesExceptLinks = []any{
|
||||
parser.NewListParser(), parser.NewListItemParser(), parser.NewHTMLBlockParser(), parser.NewRawHTMLParser(),
|
||||
parser.NewSetextHeadingParser(), parser.NewATXHeadingParser(), parser.NewThematicBreakParser(),
|
||||
parser.NewSetextHeadingParser(), parser.NewThematicBreakParser(),
|
||||
parser.NewCodeBlockParser(),
|
||||
}
|
||||
var removeFeaturesAndLinks = append(removeFeaturesExceptLinks, parser.NewLinkParser())
|
||||
@@ -93,6 +94,7 @@ func (portal *Portal) renderDiscordMarkdownOnlyHTML(text string, allowInlineLink
|
||||
|
||||
const formatterContextPortalKey = "fi.mau.discord.portal"
|
||||
const formatterContextAllowedMentionsKey = "fi.mau.discord.allowed_mentions"
|
||||
const formatterContextInputAllowedMentionsKey = "fi.mau.discord.input_allowed_mentions"
|
||||
|
||||
func appendIfNotContains(arr []string, newItem string) []string {
|
||||
for _, item := range arr {
|
||||
@@ -135,6 +137,10 @@ func (br *DiscordBridge) pillConverter(displayname, mxid, eventID string, ctx fo
|
||||
}
|
||||
}
|
||||
} else if mxid[0] == '@' {
|
||||
allowedMentions, _ := ctx.ReturnData[formatterContextInputAllowedMentionsKey].([]id.UserID)
|
||||
if allowedMentions != nil && !slices.Contains(allowedMentions, id.UserID(mxid)) {
|
||||
return displayname
|
||||
}
|
||||
mentions := ctx.ReturnData[formatterContextAllowedMentionsKey].(*discordgo.MessageAllowedMentions)
|
||||
parsedID, ok := br.ParsePuppetMXID(id.UserID(mxid))
|
||||
if ok {
|
||||
@@ -164,6 +170,7 @@ var discordMarkdownEscaper = strings.NewReplacer(
|
||||
"`", "\\`",
|
||||
`|`, `\|`,
|
||||
`<`, `\<`,
|
||||
`#`, `\#`,
|
||||
)
|
||||
|
||||
func escapeDiscordMarkdown(s string) string {
|
||||
@@ -219,6 +226,9 @@ func (portal *Portal) parseMatrixHTML(content *event.MessageEventContent) (strin
|
||||
ctx := format.NewContext()
|
||||
ctx.ReturnData[formatterContextPortalKey] = portal
|
||||
ctx.ReturnData[formatterContextAllowedMentionsKey] = allowedMentions
|
||||
if content.Mentions != nil {
|
||||
ctx.ReturnData[formatterContextInputAllowedMentionsKey] = content.Mentions.UserIDs
|
||||
}
|
||||
return variationselector.FullyQualify(matrixHTMLParser.Parse(content.FormattedBody, ctx)), allowedMentions
|
||||
} else {
|
||||
return variationselector.FullyQualify(escapeDiscordMarkdown(content.Body)), allowedMentions
|
||||
|
||||
16
go.mod
16
go.mod
@@ -9,13 +9,14 @@ require (
|
||||
github.com/gorilla/mux v1.8.0
|
||||
github.com/gorilla/websocket v1.5.0
|
||||
github.com/lib/pq v1.10.9
|
||||
github.com/mattn/go-sqlite3 v1.14.16
|
||||
github.com/mattn/go-sqlite3 v1.14.17
|
||||
github.com/rs/zerolog v1.29.1
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e
|
||||
github.com/stretchr/testify v1.8.2
|
||||
github.com/stretchr/testify v1.8.4
|
||||
github.com/yuin/goldmark v1.5.4
|
||||
golang.org/x/exp v0.0.0-20230522175609-2e198f4a06a1
|
||||
maunium.net/go/maulogger/v2 v2.4.1
|
||||
maunium.net/go/mautrix v0.15.2
|
||||
maunium.net/go/mautrix v0.15.3
|
||||
)
|
||||
|
||||
require (
|
||||
@@ -29,13 +30,12 @@ require (
|
||||
github.com/tidwall/pretty v1.2.0 // indirect
|
||||
github.com/tidwall/sjson v1.2.5 // indirect
|
||||
go.mau.fi/zeroconfig v0.1.2 // indirect
|
||||
golang.org/x/crypto v0.9.0 // indirect
|
||||
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea // indirect
|
||||
golang.org/x/net v0.10.0 // indirect
|
||||
golang.org/x/sys v0.8.0 // indirect
|
||||
golang.org/x/crypto v0.10.0 // indirect
|
||||
golang.org/x/net v0.11.0 // indirect
|
||||
golang.org/x/sys v0.9.0 // indirect
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
maunium.net/go/mauflag v1.0.0 // indirect
|
||||
)
|
||||
|
||||
replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20230426184739-79aea97f6660
|
||||
replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20230512133900-5b12693331c0
|
||||
|
||||
39
go.sum
39
go.sum
@@ -1,9 +1,8 @@
|
||||
github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
|
||||
github.com/beeper/discordgo v0.0.0-20230426184739-79aea97f6660 h1:5LFnUY/Aj/0k/UqeEmW2GS4ql1vxmivkrckPxUHf8oc=
|
||||
github.com/beeper/discordgo v0.0.0-20230426184739-79aea97f6660/go.mod h1:59+AOzzjmL6onAh62nuLXmn7dJCaC/owDLWbGtjTcFA=
|
||||
github.com/beeper/discordgo v0.0.0-20230512133900-5b12693331c0 h1:ECBEbC4ruaXzcVJJ4UurkGpT/Xlm9ZnwsHiHn9gjPZw=
|
||||
github.com/beeper/discordgo v0.0.0-20230512133900-5b12693331c0/go.mod h1:59+AOzzjmL6onAh62nuLXmn7dJCaC/owDLWbGtjTcFA=
|
||||
github.com/coreos/go-systemd/v22 v22.5.0 h1:RrqgGjYQKalulkV8NGVIfkXQf6YYmOyiJKk8iXXhfZs=
|
||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
||||
@@ -21,8 +20,8 @@ github.com/mattn/go-colorable v0.1.12 h1:jF+Du6AlPIjs2BiUiQlKOX0rt3SujHxPnksPKZb
|
||||
github.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
|
||||
github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=
|
||||
github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
|
||||
github.com/mattn/go-sqlite3 v1.14.16 h1:yOQRA0RpS5PFz/oikGwBEqvAWhWg5ufRz4ETLjwpU1Y=
|
||||
github.com/mattn/go-sqlite3 v1.14.16/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
|
||||
github.com/mattn/go-sqlite3 v1.14.17 h1:mCRHCLDUBXgpKAqIKsaAaAsrAlbkeomtRFKXh2L6YIM=
|
||||
github.com/mattn/go-sqlite3 v1.14.17/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
@@ -31,13 +30,8 @@ github.com/rs/zerolog v1.29.1 h1:cO+d60CHkknCbvzEWxP0S9K6KqyTjrCNUy1LdQLCGPc=
|
||||
github.com/rs/zerolog v1.29.1/go.mod h1:Le6ESbR7hc+DP6Lt1THiV8CQSdkkNrd3R0XbEgp3ZBU=
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e h1:MRM5ITcdelLK2j1vwZ3Je0FKVCfqOLp5zO6trqMLYs0=
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e/go.mod h1:XV66xRDqSt+GTGFMVlhk3ULuV0y9ZmzeVGR4mloJI3M=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||
github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
|
||||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||
github.com/tidwall/gjson v1.14.2/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
|
||||
github.com/tidwall/gjson v1.14.4 h1:uo0p8EbA09J7RQaflQ1aBRffTR7xedD2bcIVSYxLnkM=
|
||||
github.com/tidwall/gjson v1.14.4/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
|
||||
@@ -51,26 +45,25 @@ github.com/yuin/goldmark v1.5.4 h1:2uY/xC0roWy8IBEGLgB1ywIoEJFGmRrX21YQcvGZzjU=
|
||||
github.com/yuin/goldmark v1.5.4/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||
go.mau.fi/zeroconfig v0.1.2 h1:DKOydWnhPMn65GbXZOafgkPm11BvFashZWLct0dGFto=
|
||||
go.mau.fi/zeroconfig v0.1.2/go.mod h1:NcSJkf180JT+1IId76PcMuLTNa1CzsFFZ0nBygIQM70=
|
||||
golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g=
|
||||
golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=
|
||||
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea h1:vLCWI/yYrdEHyN2JzIzPO3aaQJHQdp89IZBA/+azVC4=
|
||||
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea/go.mod h1:V1LtkGg67GoY2N1AnLN78QLrzxkLyJw7RJb1gzOOz9w=
|
||||
golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
|
||||
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
|
||||
golang.org/x/crypto v0.10.0 h1:LKqV2xt9+kDzSTfOhx4FrkEBcMrAgHSYgzywV9zcGmM=
|
||||
golang.org/x/crypto v0.10.0/go.mod h1:o4eNf7Ede1fv+hwOwZsTHl9EsPFO6q6ZvYR8vYfY45I=
|
||||
golang.org/x/exp v0.0.0-20230522175609-2e198f4a06a1 h1:k/i9J1pBpvlfR+9QsetwPyERsqu1GIbi967PQMq3Ivc=
|
||||
golang.org/x/exp v0.0.0-20230522175609-2e198f4a06a1/go.mod h1:V1LtkGg67GoY2N1AnLN78QLrzxkLyJw7RJb1gzOOz9w=
|
||||
golang.org/x/net v0.11.0 h1:Gi2tvZIJyBtO9SDr1q9h5hEQCp/4L2RQ+ar0qjx2oNU=
|
||||
golang.org/x/net v0.11.0/go.mod h1:2L/ixqYpgIVXmeoSA/4Lu7BzTG4KIyPIryS4IsOd1oQ=
|
||||
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU=
|
||||
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.9.0 h1:KS/R3tvhPqvJvwcKfnBHJwwthS11LRhmM5D59eEXa0s=
|
||||
golang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
|
||||
gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
maunium.net/go/mauflag v1.0.0 h1:YiaRc0tEI3toYtJMRIfjP+jklH45uDHtT80nUamyD4M=
|
||||
maunium.net/go/mauflag v1.0.0/go.mod h1:nLivPOpTpHnpzEh8jEdSL9UqO9+/KBJFmNRlwKfkPeA=
|
||||
maunium.net/go/maulogger/v2 v2.4.1 h1:N7zSdd0mZkB2m2JtFUsiGTQQAdP0YeFWT7YMc80yAL8=
|
||||
maunium.net/go/maulogger/v2 v2.4.1/go.mod h1:omPuYwYBILeVQobz8uO3XC8DIRuEb5rXYlQSuqrbCho=
|
||||
maunium.net/go/mautrix v0.15.2 h1:fUiVajeoOR92uJoSShHbCvh7uG6lDY4ZO4Mvt90LbjU=
|
||||
maunium.net/go/mautrix v0.15.2/go.mod h1:h4NwfKqE4YxGTLSgn/gawKzXAb2sF4qx8agL6QEFtGg=
|
||||
maunium.net/go/mautrix v0.15.3 h1:C9BHSUM0gYbuZmAtopuLjIcH5XHLb/ZjTEz7nN+0jN0=
|
||||
maunium.net/go/mautrix v0.15.3/go.mod h1:zLrQqdxJlLkurRCozTc9CL6FySkgZlO/kpCYxBILSLE=
|
||||
|
||||
@@ -272,12 +272,15 @@ func (guild *Guild) UpdateAvatar(iconID string) bool {
|
||||
guild.Avatar = iconID
|
||||
guild.AvatarURL = id.ContentURI{}
|
||||
if guild.Avatar != "" {
|
||||
var err error
|
||||
guild.AvatarURL, err = uploadAvatar(guild.bridge.Bot, discordgo.EndpointGuildIcon(guild.ID, iconID))
|
||||
// TODO direct media support
|
||||
copied, err := guild.bridge.copyAttachmentToMatrix(guild.bridge.Bot, discordgo.EndpointGuildIcon(guild.ID, iconID), false, AttachmentMeta{
|
||||
AttachmentID: fmt.Sprintf("guild_avatar/%s/%s", guild.ID, iconID),
|
||||
})
|
||||
if err != nil {
|
||||
guild.log.Warnfln("Failed to reupload guild avatar %s: %v", guild.Avatar, err)
|
||||
guild.log.Warnfln("Failed to reupload guild avatar %s: %v", iconID, err)
|
||||
return true
|
||||
}
|
||||
guild.AvatarURL = copied.MXC
|
||||
}
|
||||
if guild.MXID != "" {
|
||||
_, err := guild.bridge.Bot.SetRoomAvatar(guild.MXID, guild.AvatarURL)
|
||||
|
||||
4
main.go
4
main.go
@@ -102,7 +102,7 @@ func (br *DiscordBridge) Start() {
|
||||
if br.Config.Bridge.Provisioning.SharedSecret != "disable" {
|
||||
br.provisioning = newProvisioningAPI(br)
|
||||
}
|
||||
go br.updatePuppetsContactInfo()
|
||||
br.WaitWebsocketConnected()
|
||||
go br.startUsers()
|
||||
}
|
||||
|
||||
@@ -176,7 +176,7 @@ func main() {
|
||||
Name: "mautrix-discord",
|
||||
URL: "https://github.com/mautrix/discord",
|
||||
Description: "A Matrix-Discord puppeting bridge.",
|
||||
Version: "0.4.0",
|
||||
Version: "0.5.0",
|
||||
ProtocolName: "Discord",
|
||||
BeeperServiceName: "discordgo",
|
||||
BeeperNetworkName: "discord",
|
||||
|
||||
78
portal.go
78
portal.go
@@ -571,6 +571,8 @@ func (portal *Portal) handleDiscordMessages(msg portalDiscordMessage) {
|
||||
portal.handleDiscordMessageUpdate(msg.user, convertedMsg.Message)
|
||||
case *discordgo.MessageDelete:
|
||||
portal.handleDiscordMessageDelete(msg.user, convertedMsg.Message)
|
||||
case *discordgo.MessageDeleteBulk:
|
||||
portal.handleDiscordMessageDeleteBulk(msg.user, convertedMsg.Messages)
|
||||
case *discordgo.MessageReactionAdd:
|
||||
portal.handleDiscordReaction(msg.user, convertedMsg.MessageReaction, true, msg.thread, convertedMsg.Member)
|
||||
case *discordgo.MessageReactionRemove:
|
||||
@@ -584,13 +586,14 @@ func (portal *Portal) ensureUserInvited(user *User, ignoreCache bool) bool {
|
||||
return user.ensureInvited(portal.MainIntent(), portal.MXID, portal.IsPrivateChat(), ignoreCache)
|
||||
}
|
||||
|
||||
func (portal *Portal) markMessageHandled(discordID string, authorID string, timestamp time.Time, threadID string, parts []database.MessagePart) {
|
||||
func (portal *Portal) markMessageHandled(discordID string, authorID string, timestamp time.Time, threadID string, senderMXID id.UserID, parts []database.MessagePart) {
|
||||
msg := portal.bridge.DB.Message.New()
|
||||
msg.Channel = portal.Key
|
||||
msg.DiscordID = discordID
|
||||
msg.SenderID = authorID
|
||||
msg.Timestamp = timestamp
|
||||
msg.ThreadID = threadID
|
||||
msg.SenderMXID = senderMXID
|
||||
msg.MassInsertParts(parts)
|
||||
}
|
||||
|
||||
@@ -618,13 +621,8 @@ func (portal *Portal) handleDiscordMessageCreate(user *User, msg *discordgo.Mess
|
||||
}
|
||||
log.Debug().Msg("Starting handling of Discord message")
|
||||
|
||||
for _, mention := range msg.Mentions {
|
||||
puppet := portal.bridge.GetPuppetByID(mention.ID)
|
||||
puppet.UpdateInfo(nil, mention)
|
||||
}
|
||||
|
||||
puppet := portal.bridge.GetPuppetByID(msg.Author.ID)
|
||||
puppet.UpdateInfo(user, msg.Author)
|
||||
puppet.UpdateInfo(user, msg.Author, msg.WebhookID)
|
||||
intent := puppet.IntentFor(portal)
|
||||
|
||||
var discordThreadID string
|
||||
@@ -639,9 +637,10 @@ func (portal *Portal) handleDiscordMessageCreate(user *User, msg *discordgo.Mess
|
||||
}
|
||||
}
|
||||
replyTo := portal.getReplyTarget(user, discordThreadID, msg.MessageReference, msg.Embeds, false)
|
||||
mentions := portal.convertDiscordMentions(msg, true)
|
||||
|
||||
ts, _ := discordgo.SnowflakeTimestamp(msg.ID)
|
||||
parts := portal.convertDiscordMessage(ctx, intent, msg)
|
||||
parts := portal.convertDiscordMessage(ctx, puppet, intent, msg)
|
||||
dbParts := make([]database.MessagePart, 0, len(parts))
|
||||
for i, part := range parts {
|
||||
if (replyTo != nil || threadRootEvent != "") && part.Content.RelatesTo == nil {
|
||||
@@ -658,6 +657,11 @@ func (portal *Portal) handleDiscordMessageCreate(user *User, msg *discordgo.Mess
|
||||
// Only set reply for first event
|
||||
replyTo = nil
|
||||
}
|
||||
|
||||
part.Content.Mentions = mentions
|
||||
// Only set mentions for first event, but keep empty object for rest
|
||||
mentions = &event.Mentions{}
|
||||
|
||||
resp, err := portal.sendMatrixMessage(intent, part.Type, part.Content, part.Extra, ts.UnixMilli())
|
||||
if err != nil {
|
||||
log.Err(err).
|
||||
@@ -674,7 +678,7 @@ func (portal *Portal) handleDiscordMessageCreate(user *User, msg *discordgo.Mess
|
||||
} else if len(dbParts) == 0 {
|
||||
log.Warn().Msg("All parts of message failed to send to Matrix")
|
||||
} else {
|
||||
portal.markMessageHandled(msg.ID, msg.Author.ID, ts, discordThreadID, dbParts)
|
||||
portal.markMessageHandled(msg.ID, msg.Author.ID, ts, discordThreadID, intent.UserID, dbParts)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -842,7 +846,8 @@ func (portal *Portal) handleDiscordMessageUpdate(user *User, msg *discordgo.Mess
|
||||
return
|
||||
}
|
||||
|
||||
intent := portal.bridge.GetPuppetByID(msg.Author.ID).IntentFor(portal)
|
||||
puppet := portal.bridge.GetPuppetByID(msg.Author.ID)
|
||||
intent := puppet.IntentFor(portal)
|
||||
|
||||
attachmentMap := map[string]*database.Message{}
|
||||
for _, existingPart := range existing {
|
||||
@@ -862,7 +867,7 @@ func (portal *Portal) handleDiscordMessageUpdate(user *User, msg *discordgo.Mess
|
||||
}
|
||||
for _, remainingEmbed := range msg.Embeds {
|
||||
// Other types of embeds are sent inline with the text message part
|
||||
if getEmbedType(remainingEmbed) != EmbedVideo {
|
||||
if getEmbedType(nil, remainingEmbed) != EmbedVideo {
|
||||
continue
|
||||
}
|
||||
embedID := "video_" + remainingEmbed.URL
|
||||
@@ -895,7 +900,12 @@ func (portal *Portal) handleDiscordMessageUpdate(user *User, msg *discordgo.Mess
|
||||
Msg("Dropping non-text edit")
|
||||
return
|
||||
}
|
||||
puppet.addWebhookMeta(converted, msg)
|
||||
puppet.addMemberMeta(converted, msg)
|
||||
converted.Content.Mentions = portal.convertDiscordMentions(msg, false)
|
||||
converted.Content.SetEdit(existing[0].MXID)
|
||||
// Never actually mention new users of edits, only include mentions inside m.new_content
|
||||
converted.Content.Mentions = &event.Mentions{}
|
||||
if converted.Extra != nil {
|
||||
converted.Extra = map[string]any{
|
||||
"m.new_content": converted.Extra,
|
||||
@@ -921,14 +931,33 @@ func (portal *Portal) handleDiscordMessageUpdate(user *User, msg *discordgo.Mess
|
||||
}
|
||||
|
||||
func (portal *Portal) handleDiscordMessageDelete(user *User, msg *discordgo.Message) {
|
||||
existing := portal.bridge.DB.Message.GetByDiscordID(portal.Key, msg.ID)
|
||||
lastResp := portal.redactAllParts(portal.MainIntent(), msg.ID)
|
||||
if lastResp != "" {
|
||||
portal.sendDeliveryReceipt(lastResp)
|
||||
}
|
||||
}
|
||||
|
||||
func (portal *Portal) handleDiscordMessageDeleteBulk(user *User, messages []string) {
|
||||
intent := portal.MainIntent()
|
||||
var lastResp id.EventID
|
||||
for _, msgID := range messages {
|
||||
newLastResp := portal.redactAllParts(intent, msgID)
|
||||
if newLastResp != "" {
|
||||
lastResp = newLastResp
|
||||
}
|
||||
}
|
||||
if lastResp != "" {
|
||||
portal.sendDeliveryReceipt(lastResp)
|
||||
}
|
||||
}
|
||||
|
||||
func (portal *Portal) redactAllParts(intent *appservice.IntentAPI, msgID string) (lastResp id.EventID) {
|
||||
existing := portal.bridge.DB.Message.GetByDiscordID(portal.Key, msgID)
|
||||
for _, dbMsg := range existing {
|
||||
resp, err := intent.RedactEvent(portal.MXID, dbMsg.MXID)
|
||||
if err != nil {
|
||||
portal.log.Err(err).
|
||||
Str("message_id", msg.ID).
|
||||
Str("message_id", msgID).
|
||||
Str("event_id", dbMsg.MXID.String()).
|
||||
Msg("Failed to redact Matrix message")
|
||||
} else if resp != nil && resp.EventID != "" {
|
||||
@@ -936,9 +965,7 @@ func (portal *Portal) handleDiscordMessageDelete(user *User, msg *discordgo.Mess
|
||||
}
|
||||
dbMsg.Delete()
|
||||
}
|
||||
if lastResp != "" {
|
||||
portal.sendDeliveryReceipt(lastResp)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (portal *Portal) handleDiscordTyping(evt *discordgo.TypingStart) {
|
||||
@@ -965,7 +992,7 @@ func (portal *Portal) handleDiscordTyping(evt *discordgo.TypingStart) {
|
||||
|
||||
func (portal *Portal) syncParticipant(source *User, participant *discordgo.User, remove bool) {
|
||||
puppet := portal.bridge.GetPuppetByID(participant.ID)
|
||||
puppet.UpdateInfo(source, participant)
|
||||
puppet.UpdateInfo(source, participant, "")
|
||||
log := portal.log.With().
|
||||
Str("participant_id", participant.ID).
|
||||
Str("ghost_mxid", puppet.MXID.String()).
|
||||
@@ -992,7 +1019,7 @@ func (portal *Portal) syncParticipant(source *User, participant *discordgo.User,
|
||||
func (portal *Portal) syncParticipants(source *User, participants []*discordgo.User) {
|
||||
for _, participant := range participants {
|
||||
puppet := portal.bridge.GetPuppetByID(participant.ID)
|
||||
puppet.UpdateInfo(source, participant)
|
||||
puppet.UpdateInfo(source, participant, "")
|
||||
|
||||
user := portal.bridge.GetUserByID(participant.ID)
|
||||
if user != nil {
|
||||
@@ -1585,6 +1612,7 @@ func (portal *Portal) handleMatrixMessage(sender *User, evt *event.Event) {
|
||||
} else {
|
||||
dbMsg.SenderID = portal.RelayWebhookID
|
||||
}
|
||||
dbMsg.SenderMXID = sender.MXID
|
||||
dbMsg.Timestamp, _ = discordgo.SnowflakeTimestamp(msg.ID)
|
||||
dbMsg.ThreadID = threadID
|
||||
dbMsg.Insert()
|
||||
@@ -1781,7 +1809,7 @@ func (portal *Portal) handleMatrixReaction(sender *User, evt *event.Event) {
|
||||
emojiID := reaction.RelatesTo.Key
|
||||
if strings.HasPrefix(emojiID, "mxc://") {
|
||||
uri, _ := id.ParseContentURI(emojiID)
|
||||
emojiFile := portal.bridge.DB.File.GetByMXC(uri)
|
||||
emojiFile := portal.bridge.DB.File.GetEmojiByMXC(uri)
|
||||
if emojiFile == nil || emojiFile.ID == "" || emojiFile.EmojiName == "" {
|
||||
go portal.sendMessageMetrics(evt, fmt.Errorf("%w %s", errUnknownEmoji, emojiID), "Ignoring")
|
||||
return
|
||||
@@ -1820,7 +1848,7 @@ func (portal *Portal) handleMatrixReaction(sender *User, evt *event.Event) {
|
||||
func (portal *Portal) handleDiscordReaction(user *User, reaction *discordgo.MessageReaction, add bool, thread *Thread, member *discordgo.Member) {
|
||||
puppet := portal.bridge.GetPuppetByID(reaction.UserID)
|
||||
if member != nil {
|
||||
puppet.UpdateInfo(user, member.User)
|
||||
puppet.UpdateInfo(user, member.User, "")
|
||||
}
|
||||
intent := puppet.IntentFor(portal)
|
||||
|
||||
@@ -2143,13 +2171,15 @@ func (portal *Portal) UpdateGroupDMAvatar(iconID string) bool {
|
||||
portal.AvatarSet = false
|
||||
portal.AvatarURL = id.ContentURI{}
|
||||
if portal.Avatar != "" {
|
||||
uri, err := uploadAvatar(portal.MainIntent(), discordgo.EndpointGroupIcon(portal.Key.ChannelID, portal.Avatar))
|
||||
// TODO direct media support
|
||||
copied, err := portal.bridge.copyAttachmentToMatrix(portal.MainIntent(), discordgo.EndpointGroupIcon(portal.Key.ChannelID, portal.Avatar), false, AttachmentMeta{
|
||||
AttachmentID: fmt.Sprintf("private_channel_avatar/%s/%s", portal.Key.ChannelID, iconID),
|
||||
})
|
||||
if err != nil {
|
||||
portal.log.Err(err).Str("avatar_id", portal.Avatar).Msg("Failed to reupload channel avatar")
|
||||
portal.log.Err(err).Str("avatar_id", iconID).Msg("Failed to reupload channel avatar")
|
||||
return true
|
||||
} else {
|
||||
portal.AvatarURL = uri
|
||||
}
|
||||
portal.AvatarURL = copied.MXC
|
||||
}
|
||||
portal.updateRoomAvatar()
|
||||
return true
|
||||
|
||||
@@ -26,6 +26,8 @@ import (
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"github.com/rs/zerolog"
|
||||
"golang.org/x/exp/slices"
|
||||
"maunium.net/go/mautrix/id"
|
||||
|
||||
"maunium.net/go/mautrix"
|
||||
"maunium.net/go/mautrix/appservice"
|
||||
@@ -193,7 +195,13 @@ func (portal *Portal) convertDiscordAttachment(ctx context.Context, intent *apps
|
||||
|
||||
func (portal *Portal) convertDiscordVideoEmbed(ctx context.Context, intent *appservice.IntentAPI, embed *discordgo.MessageEmbed) *ConvertedMessage {
|
||||
attachmentID := fmt.Sprintf("video_%s", embed.URL)
|
||||
dbFile, err := portal.bridge.copyAttachmentToMatrix(intent, embed.Video.ProxyURL, portal.Encrypted, NoMeta)
|
||||
var proxyURL string
|
||||
if embed.Video != nil {
|
||||
proxyURL = embed.Video.ProxyURL
|
||||
} else {
|
||||
proxyURL = embed.Thumbnail.ProxyURL
|
||||
}
|
||||
dbFile, err := portal.bridge.copyAttachmentToMatrix(intent, proxyURL, portal.Encrypted, NoMeta)
|
||||
if err != nil {
|
||||
zerolog.Ctx(ctx).Err(err).Msg("Failed to copy video embed to Matrix")
|
||||
return &ConvertedMessage{
|
||||
@@ -204,16 +212,21 @@ func (portal *Portal) convertDiscordVideoEmbed(ctx context.Context, intent *apps
|
||||
}
|
||||
|
||||
content := &event.MessageEventContent{
|
||||
MsgType: event.MsgVideo,
|
||||
Body: embed.URL,
|
||||
Body: embed.URL,
|
||||
Info: &event.FileInfo{
|
||||
Width: embed.Video.Width,
|
||||
Height: embed.Video.Height,
|
||||
MimeType: dbFile.MimeType,
|
||||
|
||||
Size: dbFile.Size,
|
||||
Size: dbFile.Size,
|
||||
},
|
||||
}
|
||||
if embed.Video != nil {
|
||||
content.MsgType = event.MsgVideo
|
||||
content.Info.Width = embed.Video.Width
|
||||
content.Info.Height = embed.Video.Height
|
||||
} else {
|
||||
content.MsgType = event.MsgImage
|
||||
content.Info.Width = embed.Thumbnail.Width
|
||||
content.Info.Height = embed.Thumbnail.Height
|
||||
}
|
||||
if content.Info.Width == 0 && content.Info.Height == 0 {
|
||||
content.Info.Width = dbFile.Width
|
||||
content.Info.Height = dbFile.Height
|
||||
@@ -227,7 +240,7 @@ func (portal *Portal) convertDiscordVideoEmbed(ctx context.Context, intent *apps
|
||||
content.URL = dbFile.MXC.CUString()
|
||||
}
|
||||
extra := map[string]any{}
|
||||
if embed.Type == discordgo.EmbedTypeGifv {
|
||||
if content.MsgType == event.MsgVideo && embed.Type == discordgo.EmbedTypeGifv {
|
||||
extra["info"] = map[string]any{
|
||||
"fi.mau.discord.gifv": true,
|
||||
"fi.mau.loop": true,
|
||||
@@ -244,7 +257,7 @@ func (portal *Portal) convertDiscordVideoEmbed(ctx context.Context, intent *apps
|
||||
}
|
||||
}
|
||||
|
||||
func (portal *Portal) convertDiscordMessage(ctx context.Context, intent *appservice.IntentAPI, msg *discordgo.Message) []*ConvertedMessage {
|
||||
func (portal *Portal) convertDiscordMessage(ctx context.Context, puppet *Puppet, intent *appservice.IntentAPI, msg *discordgo.Message) []*ConvertedMessage {
|
||||
predictedLength := len(msg.Attachments) + len(msg.StickerItems)
|
||||
if msg.Content != "" {
|
||||
predictedLength++
|
||||
@@ -277,7 +290,7 @@ func (portal *Portal) convertDiscordMessage(ctx context.Context, intent *appserv
|
||||
}
|
||||
for i, embed := range msg.Embeds {
|
||||
// Ignore non-video embeds, they're handled in convertDiscordTextMessage
|
||||
if getEmbedType(embed) != EmbedVideo {
|
||||
if getEmbedType(msg, embed) != EmbedVideo {
|
||||
continue
|
||||
}
|
||||
// Discord deduplicates embeds by URL. It makes things easier for us too.
|
||||
@@ -295,9 +308,90 @@ func (portal *Portal) convertDiscordMessage(ctx context.Context, intent *appserv
|
||||
parts = append(parts, part)
|
||||
}
|
||||
}
|
||||
for _, part := range parts {
|
||||
puppet.addWebhookMeta(part, msg)
|
||||
puppet.addMemberMeta(part, msg)
|
||||
}
|
||||
return parts
|
||||
}
|
||||
|
||||
func (puppet *Puppet) addMemberMeta(part *ConvertedMessage, msg *discordgo.Message) {
|
||||
if msg.Member == nil {
|
||||
return
|
||||
}
|
||||
if part.Extra == nil {
|
||||
part.Extra = make(map[string]any)
|
||||
}
|
||||
var avatarURL id.ContentURI
|
||||
if msg.Member.Avatar != "" {
|
||||
var err error
|
||||
avatarURL, err = puppet.bridge.reuploadUserAvatar(puppet.DefaultIntent(), msg.GuildID, msg.Author.ID, msg.Author.Avatar)
|
||||
if err != nil {
|
||||
puppet.log.Warn().Err(err).
|
||||
Str("avatar_id", msg.Author.Avatar).
|
||||
Msg("Failed to reupload guild user avatar")
|
||||
}
|
||||
}
|
||||
var discordAvararURL string
|
||||
if msg.Member.Avatar != "" {
|
||||
msg.Member.User = msg.Author
|
||||
discordAvararURL = msg.Member.AvatarURL("")
|
||||
}
|
||||
part.Extra["fi.mau.discord.guild_member_metadata"] = map[string]any{
|
||||
"nick": msg.Member.Nick,
|
||||
"avatar_id": msg.Member.Avatar,
|
||||
"avatar_url": discordAvararURL,
|
||||
"avatar_mxc": avatarURL.String(),
|
||||
}
|
||||
if msg.Member.Nick != "" || !avatarURL.IsEmpty() {
|
||||
perMessageProfile := map[string]any{
|
||||
"is_multiple_users": false,
|
||||
|
||||
"displayname": msg.Member.Nick,
|
||||
"avatar_url": avatarURL.String(),
|
||||
}
|
||||
if msg.Member.Nick == "" {
|
||||
perMessageProfile["displayname"] = puppet.Name
|
||||
}
|
||||
if avatarURL.IsEmpty() {
|
||||
perMessageProfile["avatar_url"] = puppet.AvatarURL.String()
|
||||
}
|
||||
part.Extra["com.beeper.per_message_profile"] = perMessageProfile
|
||||
}
|
||||
}
|
||||
|
||||
func (puppet *Puppet) addWebhookMeta(part *ConvertedMessage, msg *discordgo.Message) {
|
||||
if msg.WebhookID == "" {
|
||||
return
|
||||
}
|
||||
if part.Extra == nil {
|
||||
part.Extra = make(map[string]any)
|
||||
}
|
||||
var avatarURL id.ContentURI
|
||||
if msg.Author.Avatar != "" {
|
||||
var err error
|
||||
avatarURL, err = puppet.bridge.reuploadUserAvatar(puppet.DefaultIntent(), "", msg.Author.ID, msg.Author.Avatar)
|
||||
if err != nil {
|
||||
puppet.log.Warn().Err(err).
|
||||
Str("avatar_id", msg.Author.Avatar).
|
||||
Msg("Failed to reupload webhook avatar")
|
||||
}
|
||||
}
|
||||
part.Extra["fi.mau.discord.webhook_metadata"] = map[string]any{
|
||||
"id": msg.WebhookID,
|
||||
"name": msg.Author.Username,
|
||||
"avatar_id": msg.Author.Avatar,
|
||||
"avatar_url": msg.Author.AvatarURL(""),
|
||||
"avatar_mxc": avatarURL.String(),
|
||||
}
|
||||
part.Extra["com.beeper.per_message_profile"] = map[string]any{
|
||||
"is_multiple_users": true,
|
||||
|
||||
"avatar_url": avatarURL.String(),
|
||||
"displayname": msg.Author.Username,
|
||||
}
|
||||
}
|
||||
|
||||
const (
|
||||
embedHTMLWrapper = `<blockquote class="discord-embed">%s</blockquote>`
|
||||
embedHTMLWrapperColor = `<blockquote class="discord-embed" background-color="#%06X">%s</blockquote>`
|
||||
@@ -496,7 +590,7 @@ func isActuallyLinkPreview(embed *discordgo.MessageEmbed) bool {
|
||||
return embed.Video != nil && embed.Video.ProxyURL == ""
|
||||
}
|
||||
|
||||
func getEmbedType(embed *discordgo.MessageEmbed) BridgeEmbedType {
|
||||
func getEmbedType(msg *discordgo.Message, embed *discordgo.MessageEmbed) BridgeEmbedType {
|
||||
switch embed.Type {
|
||||
case discordgo.EmbedTypeLink, discordgo.EmbedTypeArticle:
|
||||
return EmbedLinkPreview
|
||||
@@ -507,7 +601,14 @@ func getEmbedType(embed *discordgo.MessageEmbed) BridgeEmbedType {
|
||||
return EmbedVideo
|
||||
case discordgo.EmbedTypeGifv:
|
||||
return EmbedVideo
|
||||
case discordgo.EmbedTypeRich, discordgo.EmbedTypeImage:
|
||||
case discordgo.EmbedTypeImage:
|
||||
if msg != nil && isPlainGifMessage(msg) {
|
||||
return EmbedVideo
|
||||
} else if embed.Image == nil && embed.Thumbnail != nil {
|
||||
return EmbedLinkPreview
|
||||
}
|
||||
return EmbedRich
|
||||
case discordgo.EmbedTypeRich:
|
||||
return EmbedRich
|
||||
default:
|
||||
return EmbedUnknown
|
||||
@@ -515,7 +616,31 @@ func getEmbedType(embed *discordgo.MessageEmbed) BridgeEmbedType {
|
||||
}
|
||||
|
||||
func isPlainGifMessage(msg *discordgo.Message) bool {
|
||||
return len(msg.Embeds) == 1 && msg.Embeds[0].Video != nil && msg.Embeds[0].URL == msg.Content && msg.Embeds[0].Type == discordgo.EmbedTypeGifv
|
||||
return len(msg.Embeds) == 1 && msg.Embeds[0].URL == msg.Content &&
|
||||
((msg.Embeds[0].Type == discordgo.EmbedTypeGifv && msg.Embeds[0].Video != nil) ||
|
||||
(msg.Embeds[0].Type == discordgo.EmbedTypeImage && msg.Embeds[0].Image == nil && msg.Embeds[0].Thumbnail != nil))
|
||||
}
|
||||
|
||||
func (portal *Portal) convertDiscordMentions(msg *discordgo.Message, syncGhosts bool) *event.Mentions {
|
||||
var matrixMentions event.Mentions
|
||||
for _, mention := range msg.Mentions {
|
||||
puppet := portal.bridge.GetPuppetByID(mention.ID)
|
||||
if syncGhosts {
|
||||
puppet.UpdateInfo(nil, mention, "")
|
||||
}
|
||||
user := portal.bridge.GetUserByID(mention.ID)
|
||||
if user != nil {
|
||||
matrixMentions.UserIDs = append(matrixMentions.UserIDs, user.MXID)
|
||||
} else {
|
||||
matrixMentions.UserIDs = append(matrixMentions.UserIDs, puppet.MXID)
|
||||
}
|
||||
}
|
||||
slices.Sort(matrixMentions.UserIDs)
|
||||
matrixMentions.UserIDs = slices.Compact(matrixMentions.UserIDs)
|
||||
if msg.MentionEveryone {
|
||||
matrixMentions.Room = true
|
||||
}
|
||||
return &matrixMentions
|
||||
}
|
||||
|
||||
func (portal *Portal) convertDiscordTextMessage(ctx context.Context, intent *appservice.IntentAPI, msg *discordgo.Message) *ConvertedMessage {
|
||||
@@ -534,7 +659,7 @@ func (portal *Portal) convertDiscordTextMessage(ctx context.Context, intent *app
|
||||
var htmlParts []string
|
||||
if msg.Interaction != nil {
|
||||
puppet := portal.bridge.GetPuppetByID(msg.Interaction.User.ID)
|
||||
puppet.UpdateInfo(nil, msg.Interaction.User)
|
||||
puppet.UpdateInfo(nil, msg.Interaction.User, "")
|
||||
htmlParts = append(htmlParts, fmt.Sprintf(msgInteractionTemplateHTML, puppet.MXID, puppet.Name, msg.Interaction.Name))
|
||||
}
|
||||
if msg.Content != "" && !isPlainGifMessage(msg) {
|
||||
@@ -548,7 +673,7 @@ func (portal *Portal) convertDiscordTextMessage(ctx context.Context, intent *app
|
||||
with := log.With().
|
||||
Str("embed_type", string(embed.Type)).
|
||||
Int("embed_index", i)
|
||||
switch getEmbedType(embed) {
|
||||
switch getEmbedType(msg, embed) {
|
||||
case EmbedRich:
|
||||
log := with.Str("computed_embed_type", "rich").Logger()
|
||||
htmlParts = append(htmlParts, portal.convertDiscordRichEmbed(log.WithContext(ctx), intent, embed, msg.ID, i))
|
||||
@@ -581,5 +706,11 @@ func (portal *Portal) convertDiscordTextMessage(ctx context.Context, intent *app
|
||||
"com.beeper.linkpreviews": previews,
|
||||
}
|
||||
|
||||
if msg.WebhookID != "" && portal.bridge.Config.Bridge.PrefixWebhookMessages {
|
||||
content.EnsureHasHTML()
|
||||
content.Body = fmt.Sprintf("%s: %s", msg.Author.Username, content.Body)
|
||||
content.FormattedBody = fmt.Sprintf("<strong>%s</strong>: %s", html.EscapeString(msg.Author.Username), content.FormattedBody)
|
||||
}
|
||||
|
||||
return &ConvertedMessage{Type: event.EventMessage, Content: &content, Extra: extraContent}
|
||||
}
|
||||
|
||||
87
puppet.go
87
puppet.go
@@ -158,18 +158,6 @@ func (br *DiscordBridge) FormatPuppetMXID(did string) id.UserID {
|
||||
)
|
||||
}
|
||||
|
||||
func (br *DiscordBridge) updatePuppetsContactInfo() {
|
||||
if br.Config.Homeserver.Software != bridgeconfig.SoftwareHungry {
|
||||
return
|
||||
}
|
||||
for _, puppet := range br.GetAllPuppets() {
|
||||
if !puppet.ContactInfoSet && puppet.NameSet {
|
||||
puppet.ResendContactInfo()
|
||||
puppet.Update()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (puppet *Puppet) GetDisplayname() string {
|
||||
return puppet.Name
|
||||
}
|
||||
@@ -207,7 +195,7 @@ func (puppet *Puppet) updatePortalMeta(meta func(portal *Portal)) {
|
||||
}
|
||||
|
||||
func (puppet *Puppet) UpdateName(info *discordgo.User) bool {
|
||||
newName := puppet.bridge.Config.Bridge.FormatDisplayname(info)
|
||||
newName := puppet.bridge.Config.Bridge.FormatDisplayname(info, puppet.IsWebhook)
|
||||
if puppet.Name == newName && puppet.NameSet {
|
||||
return false
|
||||
}
|
||||
@@ -228,30 +216,54 @@ func (puppet *Puppet) UpdateName(info *discordgo.User) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (br *DiscordBridge) reuploadUserAvatar(intent *appservice.IntentAPI, guildID, userID, avatarID string) (id.ContentURI, error) {
|
||||
var downloadURL, ext string
|
||||
if guildID == "" {
|
||||
downloadURL = discordgo.EndpointUserAvatar(userID, avatarID)
|
||||
ext = "png"
|
||||
if strings.HasPrefix(avatarID, "a_") {
|
||||
downloadURL = discordgo.EndpointUserAvatarAnimated(userID, avatarID)
|
||||
ext = "gif"
|
||||
}
|
||||
} else {
|
||||
downloadURL = discordgo.EndpointGuildMemberAvatar(guildID, userID, avatarID)
|
||||
ext = "png"
|
||||
if strings.HasPrefix(avatarID, "a_") {
|
||||
downloadURL = discordgo.EndpointGuildMemberAvatarAnimated(guildID, userID, avatarID)
|
||||
ext = "gif"
|
||||
}
|
||||
}
|
||||
url := br.Config.Bridge.MediaPatterns.Avatar(userID, avatarID, ext)
|
||||
if !url.IsEmpty() {
|
||||
return url, nil
|
||||
}
|
||||
copied, err := br.copyAttachmentToMatrix(intent, downloadURL, false, AttachmentMeta{
|
||||
AttachmentID: fmt.Sprintf("avatar/%s/%s/%s", guildID, userID, avatarID),
|
||||
})
|
||||
if err != nil {
|
||||
return url, err
|
||||
}
|
||||
return copied.MXC, nil
|
||||
}
|
||||
|
||||
func (puppet *Puppet) UpdateAvatar(info *discordgo.User) bool {
|
||||
if puppet.Avatar == info.Avatar && puppet.AvatarSet {
|
||||
avatarID := info.Avatar
|
||||
if puppet.IsWebhook && !puppet.bridge.Config.Bridge.EnableWebhookAvatars {
|
||||
avatarID = ""
|
||||
}
|
||||
if puppet.Avatar == avatarID && puppet.AvatarSet {
|
||||
return false
|
||||
}
|
||||
avatarChanged := info.Avatar != puppet.Avatar
|
||||
puppet.Avatar = info.Avatar
|
||||
avatarChanged := avatarID != puppet.Avatar
|
||||
puppet.Avatar = avatarID
|
||||
puppet.AvatarSet = false
|
||||
puppet.AvatarURL = id.ContentURI{}
|
||||
|
||||
if puppet.Avatar != "" && (puppet.AvatarURL.IsEmpty() || avatarChanged) {
|
||||
downloadURL := discordgo.EndpointUserAvatar(info.ID, info.Avatar)
|
||||
ext := "png"
|
||||
if strings.HasPrefix(info.Avatar, "a_") {
|
||||
downloadURL = discordgo.EndpointUserAvatarAnimated(info.ID, info.Avatar)
|
||||
ext = "gif"
|
||||
}
|
||||
url := puppet.bridge.Config.Bridge.MediaPatterns.Avatar(info.ID, info.Avatar, ext)
|
||||
if url.IsEmpty() {
|
||||
var err error
|
||||
url, err = uploadAvatar(puppet.DefaultIntent(), downloadURL)
|
||||
if err != nil {
|
||||
puppet.log.Warn().Err(err).Str("avatar_id", puppet.Avatar).Msg("Failed to reupload user avatar")
|
||||
return true
|
||||
}
|
||||
url, err := puppet.bridge.reuploadUserAvatar(puppet.DefaultIntent(), "", info.ID, puppet.Avatar)
|
||||
if err != nil {
|
||||
puppet.log.Warn().Err(err).Str("avatar_id", puppet.Avatar).Msg("Failed to reupload user avatar")
|
||||
return true
|
||||
}
|
||||
puppet.AvatarURL = url
|
||||
}
|
||||
@@ -271,7 +283,7 @@ func (puppet *Puppet) UpdateAvatar(info *discordgo.User) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (puppet *Puppet) UpdateInfo(source *User, info *discordgo.User) {
|
||||
func (puppet *Puppet) UpdateInfo(source *User, info *discordgo.User, webhookID string) {
|
||||
puppet.syncLock.Lock()
|
||||
defer puppet.syncLock.Unlock()
|
||||
|
||||
@@ -294,6 +306,10 @@ func (puppet *Puppet) UpdateInfo(source *User, info *discordgo.User) {
|
||||
}
|
||||
|
||||
changed := false
|
||||
if webhookID != "" && webhookID == info.ID && !puppet.IsWebhook {
|
||||
puppet.IsWebhook = true
|
||||
changed = true
|
||||
}
|
||||
changed = puppet.UpdateContactInfo(info) || changed
|
||||
changed = puppet.UpdateName(info) || changed
|
||||
changed = puppet.UpdateAvatar(info) || changed
|
||||
@@ -308,6 +324,10 @@ func (puppet *Puppet) UpdateContactInfo(info *discordgo.User) bool {
|
||||
puppet.Username = info.Username
|
||||
changed = true
|
||||
}
|
||||
if puppet.GlobalName != info.GlobalName {
|
||||
puppet.GlobalName = info.GlobalName
|
||||
changed = true
|
||||
}
|
||||
if puppet.Discriminator != info.Discriminator {
|
||||
puppet.Discriminator = info.Discriminator
|
||||
changed = true
|
||||
@@ -316,7 +336,7 @@ func (puppet *Puppet) UpdateContactInfo(info *discordgo.User) bool {
|
||||
puppet.IsBot = info.Bot
|
||||
changed = true
|
||||
}
|
||||
if changed {
|
||||
if (changed && !puppet.IsWebhook) || !puppet.ContactInfoSet {
|
||||
puppet.ContactInfoSet = false
|
||||
puppet.ResendContactInfo()
|
||||
return true
|
||||
@@ -337,6 +357,9 @@ func (puppet *Puppet) ResendContactInfo() {
|
||||
"com.beeper.bridge.network": puppet.bridge.BeeperNetworkName,
|
||||
"com.beeper.bridge.is_network_bot": puppet.IsBot,
|
||||
}
|
||||
if puppet.IsWebhook {
|
||||
contactInfo["com.beeper.bridge.identifiers"] = []string{}
|
||||
}
|
||||
err := puppet.DefaultIntent().BeeperUpdateProfile(contactInfo)
|
||||
if err != nil {
|
||||
puppet.log.Warn().Err(err).Msg("Failed to store custom contact info in profile")
|
||||
|
||||
15
user.go
15
user.go
@@ -186,6 +186,12 @@ func (br *DiscordBridge) GetUserByID(id string) *User {
|
||||
return user
|
||||
}
|
||||
|
||||
func (br *DiscordBridge) GetCachedUserByID(id string) *User {
|
||||
br.usersLock.Lock()
|
||||
defer br.usersLock.Unlock()
|
||||
return br.usersByID[id]
|
||||
}
|
||||
|
||||
func (br *DiscordBridge) NewUser(dbUser *database.User) *User {
|
||||
user := &User{
|
||||
User: dbUser,
|
||||
@@ -630,6 +636,8 @@ func (user *User) eventHandler(rawEvt any) {
|
||||
user.pushPortalMessage(evt, "message create", evt.ChannelID, evt.GuildID)
|
||||
case *discordgo.MessageDelete:
|
||||
user.pushPortalMessage(evt, "message delete", evt.ChannelID, evt.GuildID)
|
||||
case *discordgo.MessageDeleteBulk:
|
||||
user.pushPortalMessage(evt, "bulk message delete", evt.ChannelID, evt.GuildID)
|
||||
case *discordgo.MessageUpdate:
|
||||
user.pushPortalMessage(evt, "message update", evt.ChannelID, evt.GuildID)
|
||||
case *discordgo.MessageReactionAdd:
|
||||
@@ -1226,10 +1234,17 @@ func (user *User) messageAckHandler(m *discordgo.MessageAck) {
|
||||
}
|
||||
|
||||
func (user *User) typingStartHandler(t *discordgo.TypingStart) {
|
||||
if t.UserID == user.DiscordID {
|
||||
return
|
||||
}
|
||||
portal := user.GetExistingPortalByID(t.ChannelID)
|
||||
if portal == nil || portal.MXID == "" {
|
||||
return
|
||||
}
|
||||
targetUser := user.bridge.GetCachedUserByID(t.UserID)
|
||||
if targetUser != nil {
|
||||
return
|
||||
}
|
||||
portal.handleDiscordTyping(t)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user