8c65b35c57
* update image api *os.File to io.Reader * update code style * add reader test * supplementary reader test * update the reader in the form builder test * add commnet * update comment * update code style
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package openai
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/textproto"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type FormBuilder interface {
|
|
CreateFormFile(fieldname string, file *os.File) error
|
|
CreateFormFileReader(fieldname string, r io.Reader, filename string) error
|
|
WriteField(fieldname, value string) error
|
|
Close() error
|
|
FormDataContentType() string
|
|
}
|
|
|
|
type DefaultFormBuilder struct {
|
|
writer *multipart.Writer
|
|
}
|
|
|
|
func NewFormBuilder(body io.Writer) *DefaultFormBuilder {
|
|
return &DefaultFormBuilder{
|
|
writer: multipart.NewWriter(body),
|
|
}
|
|
}
|
|
|
|
func (fb *DefaultFormBuilder) CreateFormFile(fieldname string, file *os.File) error {
|
|
return fb.createFormFile(fieldname, file, file.Name())
|
|
}
|
|
|
|
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
|
|
|
|
func escapeQuotes(s string) string {
|
|
return quoteEscaper.Replace(s)
|
|
}
|
|
|
|
// CreateFormFileReader creates a form field with a file reader.
|
|
// The filename in parameters can be an empty string.
|
|
// The filename in Content-Disposition is required, But it can be an empty string.
|
|
func (fb *DefaultFormBuilder) CreateFormFileReader(fieldname string, r io.Reader, filename string) error {
|
|
h := make(textproto.MIMEHeader)
|
|
h.Set(
|
|
"Content-Disposition",
|
|
fmt.Sprintf(
|
|
`form-data; name="%s"; filename="%s"`,
|
|
escapeQuotes(fieldname),
|
|
escapeQuotes(filepath.Base(filename)),
|
|
),
|
|
)
|
|
|
|
fieldWriter, err := fb.writer.CreatePart(h)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = io.Copy(fieldWriter, r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (fb *DefaultFormBuilder) createFormFile(fieldname string, r io.Reader, filename string) error {
|
|
if filename == "" {
|
|
return fmt.Errorf("filename cannot be empty")
|
|
}
|
|
|
|
fieldWriter, err := fb.writer.CreateFormFile(fieldname, filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = io.Copy(fieldWriter, r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (fb *DefaultFormBuilder) WriteField(fieldname, value string) error {
|
|
return fb.writer.WriteField(fieldname, value)
|
|
}
|
|
|
|
func (fb *DefaultFormBuilder) Close() error {
|
|
return fb.writer.Close()
|
|
}
|
|
|
|
func (fb *DefaultFormBuilder) FormDataContentType() string {
|
|
return fb.writer.FormDataContentType()
|
|
}
|