mirror of
https://github.com/mautrix/discord.git
synced 2025-03-14 14:15:37 +00:00
client: set referers properly
This commit is contained in:
parent
8219516ede
commit
b330c5836e
9 changed files with 48 additions and 20 deletions
|
@ -73,6 +73,12 @@ func uploadDiscordAttachment(url string, data []byte) error {
|
||||||
for key, value := range discordgo.DroidFetchHeaders {
|
for key, value := range discordgo.DroidFetchHeaders {
|
||||||
req.Header.Set(key, value)
|
req.Header.Set(key, value)
|
||||||
}
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/octet-stream")
|
||||||
|
req.Header.Set("Referer", "https://discord.com/")
|
||||||
|
req.Header.Del("X-Debug-Options")
|
||||||
|
req.Header.Del("X-Discord-Locale")
|
||||||
|
req.Header.Del("X-Discord-Timezone")
|
||||||
|
req.Header.Del("X-Super-Properties")
|
||||||
|
|
||||||
resp, err := http.DefaultClient.Do(req)
|
resp, err := http.DefaultClient.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
@ -117,7 +117,7 @@ func (portal *Portal) collectBackfillMessages(log zerolog.Logger, source *User,
|
||||||
}
|
}
|
||||||
for {
|
for {
|
||||||
log.Debug().Str("before_id", before).Msg("Fetching messages for backfill")
|
log.Debug().Str("before_id", before).Msg("Fetching messages for backfill")
|
||||||
newMessages, err := source.Session.ChannelMessages(protoChannelID, messageFetchChunkSize, before, "", "")
|
newMessages, err := source.Session.ChannelMessages(protoChannelID, messageFetchChunkSize, before, "", "", portal.RefererOptIfUser(source.Session, protoChannelID)...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, false, err
|
return nil, false, err
|
||||||
}
|
}
|
||||||
|
@ -183,7 +183,7 @@ func (portal *Portal) backfillUnlimitedMissed(log zerolog.Logger, source *User,
|
||||||
}
|
}
|
||||||
for {
|
for {
|
||||||
log.Debug().Str("after_id", after).Msg("Fetching chunk of messages to backfill")
|
log.Debug().Str("after_id", after).Msg("Fetching chunk of messages to backfill")
|
||||||
messages, err := source.Session.ChannelMessages(protoChannelID, messageFetchChunkSize, "", after, "")
|
messages, err := source.Session.ChannelMessages(protoChannelID, messageFetchChunkSize, "", after, "", portal.RefererOptIfUser(source.Session, protoChannelID)...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Err(err).Msg("Error fetching chunk of messages to forward backfill")
|
log.Err(err).Msg("Error fetching chunk of messages to forward backfill")
|
||||||
return
|
return
|
||||||
|
|
|
@ -468,7 +468,7 @@ func fnSetRelay(ce *WrappedCommandEvent) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
case "create":
|
case "create":
|
||||||
perms, err := ce.User.Session.UserChannelPermissions(ce.User.DiscordID, portal.Key.ChannelID)
|
perms, err := ce.User.Session.UserChannelPermissions(ce.User.DiscordID, portal.Key.ChannelID, portal.RefererOptIfUser(ce.User.Session, "")...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Warn().Err(err).Msg("Failed to check user permissions")
|
log.Warn().Err(err).Msg("Failed to check user permissions")
|
||||||
ce.Reply("Failed to check if you have permission to create webhooks")
|
ce.Reply("Failed to check if you have permission to create webhooks")
|
||||||
|
@ -483,7 +483,7 @@ func fnSetRelay(ce *WrappedCommandEvent) {
|
||||||
name = strings.Join(ce.Args[1:], " ")
|
name = strings.Join(ce.Args[1:], " ")
|
||||||
}
|
}
|
||||||
log.Debug().Str("webhook_name", name).Msg("Creating webhook")
|
log.Debug().Str("webhook_name", name).Msg("Creating webhook")
|
||||||
webhookMeta, err = ce.User.Session.WebhookCreate(portal.Key.ChannelID, name, "")
|
webhookMeta, err = ce.User.Session.WebhookCreate(portal.Key.ChannelID, name, "", portal.RefererOptIfUser(ce.User.Session, "")...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Warn().Err(err).Msg("Failed to create webhook")
|
log.Warn().Err(err).Msg("Failed to create webhook")
|
||||||
ce.Reply("Failed to create webhook: %v", err)
|
ce.Reply("Failed to create webhook: %v", err)
|
||||||
|
|
|
@ -61,7 +61,7 @@ func (portal *Portal) getCommand(user *User, command string) (*discordgo.Applica
|
||||||
defer portal.commandsLock.Unlock()
|
defer portal.commandsLock.Unlock()
|
||||||
cmd, ok := portal.commands[command]
|
cmd, ok := portal.commands[command]
|
||||||
if !ok {
|
if !ok {
|
||||||
results, err := user.Session.ApplicationCommandsSearch(portal.Key.ChannelID, command)
|
results, err := user.Session.ApplicationCommandsSearch(portal.Key.ChannelID, command, portal.RefererOpt(""))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -247,7 +247,7 @@ func fnCommands(ce *WrappedCommandEvent) {
|
||||||
}
|
}
|
||||||
subcmd := strings.ToLower(ce.Args[0])
|
subcmd := strings.ToLower(ce.Args[0])
|
||||||
if subcmd == "search" {
|
if subcmd == "search" {
|
||||||
results, err := ce.User.Session.ApplicationCommandsSearch(ce.Portal.Key.ChannelID, ce.Args[1])
|
results, err := ce.User.Session.ApplicationCommandsSearch(ce.Portal.Key.ChannelID, ce.Args[1], ce.Portal.RefererOpt(""))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ce.Reply("Error searching for commands: %v", err)
|
ce.Reply("Error searching for commands: %v", err)
|
||||||
return
|
return
|
||||||
|
@ -297,7 +297,7 @@ func fnExec(ce *WrappedCommandEvent) {
|
||||||
ce.User.pendingInteractionsLock.Lock()
|
ce.User.pendingInteractionsLock.Lock()
|
||||||
ce.User.pendingInteractions[nonce] = ce
|
ce.User.pendingInteractions[nonce] = ce
|
||||||
ce.User.pendingInteractionsLock.Unlock()
|
ce.User.pendingInteractionsLock.Unlock()
|
||||||
err = ce.User.Session.SendInteractions(ce.Portal.GuildID, ce.Portal.Key.ChannelID, cmd, options, nonce)
|
err = ce.User.Session.SendInteractions(ce.Portal.GuildID, ce.Portal.Key.ChannelID, cmd, options, nonce, ce.Portal.RefererOpt(""))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ce.Reply("Error sending interaction: %v", err)
|
ce.Reply("Error sending interaction: %v", err)
|
||||||
ce.User.pendingInteractionsLock.Lock()
|
ce.User.pendingInteractionsLock.Lock()
|
||||||
|
|
|
@ -357,7 +357,11 @@ func (dma *DirectMediaAPI) fetchNewAttachmentURL(ctx context.Context, meta *Atta
|
||||||
var err error
|
var err error
|
||||||
messageIDStr := strconv.FormatUint(meta.MessageID, 10)
|
messageIDStr := strconv.FormatUint(meta.MessageID, 10)
|
||||||
if client.IsUser {
|
if client.IsUser {
|
||||||
msgs, err = client.ChannelMessages(channelIDStr, 5, "", "", messageIDStr)
|
var refs []discordgo.RequestOption
|
||||||
|
if portal != nil {
|
||||||
|
refs = append(refs, discordgo.WithChannelReferer(portal.GuildID, channelIDStr))
|
||||||
|
}
|
||||||
|
msgs, err = client.ChannelMessages(channelIDStr, 5, "", "", messageIDStr, refs...)
|
||||||
} else {
|
} else {
|
||||||
var msg *discordgo.Message
|
var msg *discordgo.Message
|
||||||
msg, err = client.ChannelMessage(channelIDStr, messageIDStr)
|
msg, err = client.ChannelMessage(channelIDStr, messageIDStr)
|
||||||
|
|
2
go.mod
2
go.mod
|
@ -42,4 +42,4 @@ require (
|
||||||
maunium.net/go/mauflag v1.0.0 // indirect
|
maunium.net/go/mauflag v1.0.0 // indirect
|
||||||
)
|
)
|
||||||
|
|
||||||
replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20241121222213-424cfdb527f5
|
replace github.com/bwmarrin/discordgo => github.com/beeper/discordgo v0.0.0-20241129150404-0ddeff8635e8
|
||||||
|
|
4
go.sum
4
go.sum
|
@ -1,7 +1,7 @@
|
||||||
github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
|
github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60=
|
||||||
github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM=
|
github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM=
|
||||||
github.com/beeper/discordgo v0.0.0-20241121222213-424cfdb527f5 h1:3w2UYUgbHJ8HJhmOi4fNSYRPV25rmoCcOoV1T74bzRI=
|
github.com/beeper/discordgo v0.0.0-20241129150404-0ddeff8635e8 h1:pJeDjlzwk6z6XTC1N54QNdAplXlnQ+Er+tO6ogquj0Q=
|
||||||
github.com/beeper/discordgo v0.0.0-20241121222213-424cfdb527f5/go.mod h1:59+AOzzjmL6onAh62nuLXmn7dJCaC/owDLWbGtjTcFA=
|
github.com/beeper/discordgo v0.0.0-20241129150404-0ddeff8635e8/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 h1:RrqgGjYQKalulkV8NGVIfkXQf6YYmOyiJKk8iXXhfZs=
|
||||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
|
30
portal.go
30
portal.go
|
@ -1167,7 +1167,7 @@ func (portal *Portal) startThreadFromMatrix(sender *User, threadRoot id.EventID)
|
||||||
AutoArchiveDuration: 24 * 60,
|
AutoArchiveDuration: 24 * 60,
|
||||||
Type: discordgo.ChannelTypeGuildPublicThread,
|
Type: discordgo.ChannelTypeGuildPublicThread,
|
||||||
Location: "Message",
|
Location: "Message",
|
||||||
})
|
}, portal.RefererOptIfUser(sender.Session, "")...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("error starting thread: %v", err)
|
return "", fmt.Errorf("error starting thread: %v", err)
|
||||||
}
|
}
|
||||||
|
@ -1497,6 +1497,20 @@ func (portal *Portal) convertReplyMessageToEmbed(eventID id.EventID, url string)
|
||||||
return embed, nil
|
return embed, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (portal *Portal) RefererOpt(threadID string) discordgo.RequestOption {
|
||||||
|
if threadID != "" && threadID != portal.Key.ChannelID {
|
||||||
|
return discordgo.WithThreadReferer(portal.GuildID, portal.Key.ChannelID, threadID)
|
||||||
|
}
|
||||||
|
return discordgo.WithChannelReferer(portal.GuildID, portal.Key.ChannelID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (portal *Portal) RefererOptIfUser(sess *discordgo.Session, threadID string) []discordgo.RequestOption {
|
||||||
|
if sess == nil || !sess.IsUser {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return []discordgo.RequestOption{portal.RefererOpt(threadID)}
|
||||||
|
}
|
||||||
|
|
||||||
func (portal *Portal) handleMatrixMessage(sender *User, evt *event.Event) {
|
func (portal *Portal) handleMatrixMessage(sender *User, evt *event.Event) {
|
||||||
if portal.IsPrivateChat() && sender.DiscordID != portal.Key.Receiver {
|
if portal.IsPrivateChat() && sender.DiscordID != portal.Key.Receiver {
|
||||||
go portal.sendMessageMetrics(evt, errUserNotReceiver, "Ignoring")
|
go portal.sendMessageMetrics(evt, errUserNotReceiver, "Ignoring")
|
||||||
|
@ -1626,7 +1640,7 @@ func (portal *Portal) handleMatrixMessage(sender *User, evt *event.Event) {
|
||||||
Name: att.Filename,
|
Name: att.Filename,
|
||||||
ID: sender.NextDiscordUploadID(),
|
ID: sender.NextDiscordUploadID(),
|
||||||
}},
|
}},
|
||||||
})
|
}, portal.RefererOpt(threadID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
go portal.sendMessageMetrics(evt, err, "Error preparing to reupload media in")
|
go portal.sendMessageMetrics(evt, err, "Error preparing to reupload media in")
|
||||||
return
|
return
|
||||||
|
@ -1667,7 +1681,7 @@ func (portal *Portal) handleMatrixMessage(sender *User, evt *event.Event) {
|
||||||
var msg *discordgo.Message
|
var msg *discordgo.Message
|
||||||
var err error
|
var err error
|
||||||
if !isWebhookSend {
|
if !isWebhookSend {
|
||||||
msg, err = sess.ChannelMessageSendComplex(channelID, &sendReq)
|
msg, err = sess.ChannelMessageSendComplex(channelID, &sendReq, portal.RefererOptIfUser(sess, threadID)...)
|
||||||
} else {
|
} else {
|
||||||
username, avatarURL := portal.getRelayUserMeta(sender)
|
username, avatarURL := portal.getRelayUserMeta(sender)
|
||||||
msg, err = relayClient.WebhookThreadExecute(portal.RelayWebhookID, portal.RelayWebhookSecret, true, threadID, &discordgo.WebhookParams{
|
msg, err = relayClient.WebhookThreadExecute(portal.RelayWebhookID, portal.RelayWebhookSecret, true, threadID, &discordgo.WebhookParams{
|
||||||
|
@ -1915,7 +1929,7 @@ func (portal *Portal) handleMatrixReaction(sender *User, evt *event.Event) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := sender.Session.MessageReactionAdd(msg.DiscordProtoChannelID(), msg.DiscordID, emojiID)
|
err := sender.Session.MessageReactionAddUser(portal.GuildID, msg.DiscordProtoChannelID(), msg.DiscordID, emojiID)
|
||||||
go portal.sendMessageMetrics(evt, err, "Error sending")
|
go portal.sendMessageMetrics(evt, err, "Error sending")
|
||||||
if err == nil {
|
if err == nil {
|
||||||
dbReaction := portal.bridge.DB.Reaction.New()
|
dbReaction := portal.bridge.DB.Reaction.New()
|
||||||
|
@ -2051,7 +2065,7 @@ func (portal *Portal) handleMatrixRedaction(sender *User, evt *event.Event) {
|
||||||
var err error
|
var err error
|
||||||
// TODO add support for deleting individual attachments from messages
|
// TODO add support for deleting individual attachments from messages
|
||||||
if sess != nil {
|
if sess != nil {
|
||||||
err = sess.ChannelMessageDelete(message.DiscordProtoChannelID(), message.DiscordID)
|
err = sess.ChannelMessageDelete(message.DiscordProtoChannelID(), message.DiscordID, portal.RefererOptIfUser(sess, message.ThreadID)...)
|
||||||
} else {
|
} else {
|
||||||
// TODO pre-validate that the message was sent by the webhook?
|
// TODO pre-validate that the message was sent by the webhook?
|
||||||
err = relayClient.WebhookMessageDelete(portal.RelayWebhookID, portal.RelayWebhookSecret, message.DiscordID)
|
err = relayClient.WebhookMessageDelete(portal.RelayWebhookID, portal.RelayWebhookSecret, message.DiscordID)
|
||||||
|
@ -2066,7 +2080,7 @@ func (portal *Portal) handleMatrixRedaction(sender *User, evt *event.Event) {
|
||||||
if sess != nil {
|
if sess != nil {
|
||||||
reaction := portal.bridge.DB.Reaction.GetByMXID(evt.Redacts)
|
reaction := portal.bridge.DB.Reaction.GetByMXID(evt.Redacts)
|
||||||
if reaction != nil && reaction.Channel == portal.Key {
|
if reaction != nil && reaction.Channel == portal.Key {
|
||||||
err := sess.MessageReactionRemove(reaction.DiscordProtoChannelID(), reaction.MessageID, reaction.EmojiName, reaction.Sender)
|
err := sess.MessageReactionRemoveUser(portal.GuildID, reaction.DiscordProtoChannelID(), reaction.MessageID, reaction.EmojiName, reaction.Sender)
|
||||||
go portal.sendMessageMetrics(evt, err, "Error sending")
|
go portal.sendMessageMetrics(evt, err, "Error sending")
|
||||||
if err == nil {
|
if err == nil {
|
||||||
reaction.Delete()
|
reaction.Delete()
|
||||||
|
@ -2135,7 +2149,7 @@ func (portal *Portal) HandleMatrixReadReceipt(brUser bridge.User, eventID id.Eve
|
||||||
Msg("Dropping read receipt: thread ID mismatch")
|
Msg("Dropping read receipt: thread ID mismatch")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
resp, err := sender.Session.ChannelMessageAckNoToken(msg.DiscordProtoChannelID(), msg.DiscordID)
|
resp, err := sender.Session.ChannelMessageAckNoToken(msg.DiscordProtoChannelID(), msg.DiscordID, portal.RefererOpt(msg.DiscordProtoChannelID()))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Err(err).Msg("Failed to send read receipt to Discord")
|
log.Err(err).Msg("Failed to send read receipt to Discord")
|
||||||
} else if resp.Token != nil {
|
} else if resp.Token != nil {
|
||||||
|
@ -2169,7 +2183,7 @@ func (portal *Portal) HandleMatrixTyping(newTyping []id.UserID) {
|
||||||
user := portal.bridge.GetUserByMXID(userID)
|
user := portal.bridge.GetUserByMXID(userID)
|
||||||
if user != nil && user.Session != nil {
|
if user != nil && user.Session != nil {
|
||||||
user.ViewingChannel(portal)
|
user.ViewingChannel(portal)
|
||||||
err := user.Session.ChannelTyping(portal.Key.ChannelID)
|
err := user.Session.ChannelTyping(portal.Key.ChannelID, portal.RefererOptIfUser(user.Session, "")...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
portal.log.Warn().Err(err).
|
portal.log.Warn().Err(err).
|
||||||
Str("user_id", user.MXID.String()).
|
Str("user_id", user.MXID.String()).
|
||||||
|
|
|
@ -112,6 +112,10 @@ func (thread *Thread) maybeInitialBackfill(source *User) {
|
||||||
thread.Parent.forwardBackfillInitial(source, thread)
|
thread.Parent.forwardBackfillInitial(source, thread)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (thread *Thread) RefererOpt() discordgo.RequestOption {
|
||||||
|
return discordgo.WithThreadReferer(thread.Parent.GuildID, thread.ParentID, thread.ID)
|
||||||
|
}
|
||||||
|
|
||||||
func (thread *Thread) Join(user *User) {
|
func (thread *Thread) Join(user *User) {
|
||||||
if user.IsInPortal(thread.ID) {
|
if user.IsInPortal(thread.ID) {
|
||||||
return
|
return
|
||||||
|
@ -137,7 +141,7 @@ func (thread *Thread) Join(user *User) {
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
if user.Session.IsUser {
|
if user.Session.IsUser {
|
||||||
err = user.Session.ThreadJoinWithLocation(thread.ID, discordgo.ThreadJoinLocationContextMenu)
|
err = user.Session.ThreadJoin(thread.ID, discordgo.WithLocationParam(discordgo.ThreadJoinLocationContextMenu), thread.RefererOpt())
|
||||||
} else {
|
} else {
|
||||||
err = user.Session.ThreadJoin(thread.ID)
|
err = user.Session.ThreadJoin(thread.ID)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue