jshiffer/matterbridge
1
0
Fork 1
mirror of https://github.com/42wim/matterbridge.git synced 2024-12-17 22:42:01 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use native upload when relaying from discord.

Browse Source 
Resolves #1965
This commit is contained in:
Yousef Mansy 2023-02-18 16:41:19 -08:00
parent 24f6747516
commit 3ffd96a350
1 changed files with 28 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
bridge/discord/handlers.go
View File

@ -1,7 +1,10 @@
package bdiscord
import (
"path"
"github.com/42wim/matterbridge/bridge/config"
"github.com/42wim/matterbridge/bridge/helper"
"github.com/bwmarrin/discordgo"
"github.com/davecgh/go-spew/spew"
)
@ -98,15 +101,14 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
return
}
// add the url of the attachments to content
if len(m.Attachments) > 0 {
for _, attach := range m.Attachments {
m.Content = m.Content + "\n" + attach.URL
}
rmsg := config.Message{
Account: b.Account,
Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg",
UserID: m.Author.ID,
ID: m.ID,
Extra: make(map[string][]interface{}),
}
rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", UserID: m.Author.ID, ID: m.ID}
b.Log.Debugf("== Receiving event %#v", m.Message)
if m.Content != "" {
@ -138,8 +140,14 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
}
}
// no empty messages
if rmsg.Text == "" {
if len(m.Attachments) > 0 {
b.handleDownloadFile(&rmsg, m)
}
hasAttachment := len(rmsg.Extra["file"]) > 0
// no empty messages unless has attachment
if rmsg.Text == "" && !hasAttachment {
return
}
@ -279,3 +287,14 @@ func handleEmbed(embed *discordgo.MessageEmbed) string {
return result
}
func (b *Bdiscord) handleDownloadFile(rmsg *config.Message, m *discordgo.MessageCreate) {
for _, attach := range m.Attachments {
data, err := helper.DownloadFile(attach.URL)
if err != nil {
b.Log.Errorf("download %s failed %#v", attach.URL, err)
}
helper.HandleDownloadData(b.Log, rmsg, path.Base(attach.URL), rmsg.Text, attach.URL, data, b.General)
}
}