forked from mirror/mautrix-discord
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
|
|
"maunium.net/go/mautrix/appservice"
|
|
"maunium.net/go/mautrix/id"
|
|
)
|
|
|
|
func (portal *Portal) downloadDiscordEmoji(id string, animated bool) ([]byte, string, error) {
|
|
var url string
|
|
var mimeType string
|
|
|
|
if animated {
|
|
// This url requests a gif, so that's what we set the mimetype to.
|
|
url = discordgo.EndpointEmojiAnimated(id)
|
|
mimeType = "image/gif"
|
|
} else {
|
|
// This url requests a png, so that's what we set the mimetype to.
|
|
url = discordgo.EndpointEmoji(id)
|
|
mimeType = "image/png"
|
|
}
|
|
|
|
req, err := http.NewRequest(http.MethodGet, url, nil)
|
|
if err != nil {
|
|
return nil, mimeType, err
|
|
}
|
|
|
|
req.Header.Set("User-Agent", discordgo.DroidBrowserUserAgent)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return nil, mimeType, err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
data, err := ioutil.ReadAll(resp.Body)
|
|
|
|
return data, mimeType, err
|
|
}
|
|
|
|
func (portal *Portal) uploadMatrixEmoji(intent *appservice.IntentAPI, data []byte, mimeType string) (id.ContentURI, error) {
|
|
uploaded, err := intent.UploadBytes(data, mimeType)
|
|
if err != nil {
|
|
return id.ContentURI{}, err
|
|
}
|
|
|
|
return uploaded.ContentURI, nil
|
|
}
|