summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/nelsonken/gomf/form_builder.go
diff options
context:
space:
mode:
authorWim <wim@42.be>2019-02-15 18:20:32 +0100
committerGitHub <noreply@github.com>2019-02-15 18:20:32 +0100
commit716751cf7685471bbc969e19fe26f23e66f3c0b4 (patch)
tree6db2350f28a76ae6e1cf7c3cea7263503057d82f /vendor/github.com/nelsonken/gomf/form_builder.go
parent077b818d82a98855d5577f4f9de5d2b2d2cc6a50 (diff)
parent6ebd5cbbd8a941e0bc5f99f0d8e99cfd1d8ac0d7 (diff)
downloadmatterbridge-msglm-716751cf7685471bbc969e19fe26f23e66f3c0b4.tar.gz
matterbridge-msglm-716751cf7685471bbc969e19fe26f23e66f3c0b4.tar.bz2
matterbridge-msglm-716751cf7685471bbc969e19fe26f23e66f3c0b4.zip
Refactor and update RocketChat bridge (#707)
* Add support for editing/deleting messages * Add support for uploading files * Add support for avatars * Use the Rocket.Chat.Go.SDK * Use the rest and streaming api
Diffstat (limited to 'vendor/github.com/nelsonken/gomf/form_builder.go')
-rw-r--r--vendor/github.com/nelsonken/gomf/form_builder.go89
1 files changed, 89 insertions, 0 deletions
diff --git a/vendor/github.com/nelsonken/gomf/form_builder.go b/vendor/github.com/nelsonken/gomf/form_builder.go
new file mode 100644
index 00000000..9b0d2294
--- /dev/null
+++ b/vendor/github.com/nelsonken/gomf/form_builder.go
@@ -0,0 +1,89 @@
+package gomf
+
+import (
+ "bytes"
+ "context"
+ "fmt"
+ "mime/multipart"
+ "net/http"
+ "net/textproto"
+)
+
+type FormBuilder struct {
+ w *multipart.Writer
+ b *bytes.Buffer
+}
+
+func New() *FormBuilder {
+ buf := new(bytes.Buffer)
+ writer := multipart.NewWriter(buf)
+ return &FormBuilder{
+ w: writer,
+ b: buf,
+ }
+}
+
+func (ufw *FormBuilder) WriteField(name, value string) error {
+ w, err := ufw.w.CreateFormField(name)
+ if err != nil {
+ return err
+ }
+
+ _, err = w.Write([]byte(value))
+ if err != nil {
+ return err
+ }
+
+ return nil
+}
+
+// WriteFile if contentType is empty-string, will auto convert to application/octet-stream
+func (ufw *FormBuilder) WriteFile(fieldName, fileName, contentType string, content []byte) error {
+ if contentType == "" {
+ contentType = "application/octet-stream"
+ }
+
+ wx, err := ufw.w.CreatePart(textproto.MIMEHeader{
+ "Content-Type": []string{
+ contentType,
+ },
+ "Content-Disposition": []string{
+ fmt.Sprintf(`form-data; name="%s"; filename="%s"`, fieldName, fileName),
+ },
+ })
+ if err != nil {
+ return err
+ }
+
+ _, err = wx.Write(content)
+ if err != nil {
+ return err
+ }
+
+ return nil
+}
+
+func (fb *FormBuilder) Close() error {
+ return fb.w.Close()
+}
+
+func (fb *FormBuilder) GetBuffer() *bytes.Buffer {
+ return fb.b
+}
+
+func (fb *FormBuilder) GetHTTPRequest(ctx context.Context, reqURL string) (*http.Request, error) {
+ err := fb.Close()
+ if err != nil {
+ return nil, err
+ }
+
+ req, err := http.NewRequest("POST", reqURL, fb.b)
+ if err != nil {
+ return nil, err
+ }
+
+ req.Header.Set("Content-Type", fb.w.FormDataContentType())
+ req = req.WithContext(ctx)
+
+ return req, nil
+}