-
Notifications
You must be signed in to change notification settings - Fork 0
/
multipart.go
104 lines (87 loc) · 2.45 KB
/
multipart.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package internal
import (
"encoding/json"
"fmt"
"io"
"mime/multipart"
"net/http"
"net/textproto"
"strings"
"github.com/hasura/ndc-rest-schema/schema"
"github.com/hasura/ndc-sdk-go/utils"
)
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
func escapeQuotes(s string) string {
return quoteEscaper.Replace(s)
}
// MultipartWriter extends multipart.Writer with helpers
type MultipartWriter struct {
*multipart.Writer
}
// NewMultipartWriter creates a MultipartWriter instance
func NewMultipartWriter(w io.Writer) *MultipartWriter {
return &MultipartWriter{multipart.NewWriter(w)}
}
// WriteDataURI write a file from data URI string
func (w *MultipartWriter) WriteDataURI(name string, value any, headers http.Header) error {
b64, err := utils.DecodeString(value)
if err != nil {
return fmt.Errorf("%s: %s", name, err)
}
dataURI, err := DecodeDataURI(b64)
if err != nil {
return fmt.Errorf("%s: %s", name, err)
}
h := make(textproto.MIMEHeader)
for key, header := range headers {
h[key] = header
}
h.Set("Content-Disposition",
fmt.Sprintf(`form-data; name="%s"; filename="%s"`,
escapeQuotes(name), escapeQuotes(name)))
if dataURI.MediaType == "" {
h.Set("Content-Type", "application/octet-stream")
} else {
h.Set("Content-Type", dataURI.MediaType)
}
p, err := w.CreatePart(h)
if err != nil {
return fmt.Errorf("%s: %s", name, err)
}
_, err = p.Write([]byte(dataURI.Data))
return err
}
// WriteField calls CreateFormField and then writes the given value with json encoding.
func (w *MultipartWriter) WriteJSON(fieldName string, value any, headers http.Header) error {
bs, err := json.Marshal(value)
if err != nil {
return err
}
h := createFieldMIMEHeader(fieldName, headers)
h.Set(schema.ContentTypeHeader, schema.ContentTypeJSON)
p, err := w.CreatePart(h)
if err != nil {
return err
}
_, err = p.Write(bs)
return err
}
// WriteField calls CreateFormField and then writes the given value.
func (w *MultipartWriter) WriteField(fieldName, value string, headers http.Header) error {
h := createFieldMIMEHeader(fieldName, headers)
p, err := w.CreatePart(h)
if err != nil {
return err
}
_, err = p.Write([]byte(value))
return err
}
func createFieldMIMEHeader(fieldName string, headers http.Header) textproto.MIMEHeader {
h := make(textproto.MIMEHeader)
for key, header := range headers {
h[key] = header
}
h.Set("Content-Disposition",
fmt.Sprintf(`form-data; name="%s"`, escapeQuotes(fieldName)))
return h
}