mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-01 03:03:48 +00:00
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/textproto"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/diamondburned/arikawa/discord"
|
|
"github.com/diamondburned/arikawa/internal/json"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type SendMessageData struct {
|
|
Content string `json:"content"`
|
|
Nonce string `json:"nonce"`
|
|
TTS bool `json:"tts"`
|
|
|
|
Embed *discord.Embed `json:"embed"`
|
|
|
|
Files []SendMessageFile `json:"-"`
|
|
}
|
|
|
|
type SendMessageFile struct {
|
|
Name string
|
|
ContentType string // auto-detect if empty
|
|
Reader io.Reader
|
|
}
|
|
|
|
var quoteEscaper = strings.NewReplacer(`\`, `\\`, `"`, `\"`)
|
|
|
|
func (data *SendMessageData) WriteMultipart(c json.Driver, w io.Writer) error {
|
|
body := multipart.NewWriter(w)
|
|
|
|
// Encode the JSON body first
|
|
h := textproto.MIMEHeader{}
|
|
h.Set("Content-Disposition", `form-data; name="payload_json"`)
|
|
h.Set("Content-Type", "application/json")
|
|
|
|
w, err := body.CreatePart(h)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to create bodypart for JSON")
|
|
}
|
|
|
|
if err := c.EncodeStream(w, data); err != nil {
|
|
return errors.Wrap(err, "Failed to encode JSON")
|
|
}
|
|
|
|
// Content-Type buffer
|
|
var buf []byte
|
|
|
|
for i, file := range data.Files {
|
|
h := textproto.MIMEHeader{}
|
|
h.Set("Content-Disposition", fmt.Sprintf(
|
|
`form-data; name="file%d"; filename="%s"`,
|
|
i, quoteEscaper.Replace(file.Name),
|
|
))
|
|
|
|
w, err := body.CreatePart(h)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to create bodypart for "+
|
|
strconv.Itoa(i))
|
|
}
|
|
|
|
if file.ContentType == "" {
|
|
if buf == nil {
|
|
buf = make([]byte, 512)
|
|
}
|
|
|
|
n, err := file.Reader.Read(buf)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to read first 512 bytes for "+
|
|
strconv.Itoa(i))
|
|
}
|
|
|
|
file.ContentType = http.DetectContentType(buf[:n])
|
|
data.Files[i] = file
|
|
|
|
h.Set("Content-Type", file.ContentType)
|
|
|
|
// Prematurely write
|
|
if _, err := w.Write(buf[:n]); err != nil {
|
|
return errors.Wrap(err, "Failed to write buffer for "+
|
|
strconv.Itoa(i))
|
|
}
|
|
|
|
} else {
|
|
h.Set("Content-Type", file.ContentType)
|
|
}
|
|
|
|
if _, err := io.Copy(w, file.Reader); err != nil {
|
|
return errors.Wrap(err, "Failed to write file for "+
|
|
strconv.Itoa(i))
|
|
}
|
|
}
|
|
|
|
if err := body.Close(); err != nil {
|
|
return errors.Wrap(err, "Failed to close body writer")
|
|
}
|
|
|
|
return nil
|
|
}
|