99 lines
1.8 KiB
Go
99 lines
1.8 KiB
Go
package multipart
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
// Reader is a multipart reader.
|
|
type Reader struct {
|
|
// io.Reader used for http.Request.Body reading
|
|
reader *bufio.Reader
|
|
|
|
// boundary used to separate multipart MultipartDatas
|
|
boundary string
|
|
|
|
// result will be inside this field
|
|
Data map[string]*Component
|
|
}
|
|
|
|
// NewReader creates a new reader from a reader and a boundary.
|
|
func NewReader(r io.Reader, boundary string) (*Reader, error) {
|
|
reader := &Reader{
|
|
reader: nil,
|
|
boundary: fmt.Sprintf("--%s", boundary),
|
|
Data: make(map[string]*Component),
|
|
}
|
|
|
|
// 1. Create reader
|
|
dst, ok := r.(*bufio.Reader)
|
|
if !ok {
|
|
dst = bufio.NewReader(r)
|
|
}
|
|
reader.reader = dst
|
|
|
|
// 2. "move" reader right after the first boundary
|
|
var err error
|
|
line := make([]byte, 0)
|
|
|
|
for err == nil && string(line) != reader.boundary {
|
|
line, _, err = dst.ReadLine()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 3. return reader
|
|
return reader, nil
|
|
|
|
}
|
|
|
|
// Parse parses the multipart components from the request
|
|
func (reader *Reader) Parse() error {
|
|
|
|
// for each component (until boundary)
|
|
for {
|
|
|
|
comp := &Component{
|
|
ContentType: "raw",
|
|
Data: make([]byte, 0),
|
|
Headers: make(map[string]string),
|
|
}
|
|
|
|
// 1. Read and parse data
|
|
err := comp.read(reader.reader, reader.boundary)
|
|
|
|
// 3. Dispatch error
|
|
if err != nil && err != io.EOF {
|
|
return err
|
|
}
|
|
|
|
name := comp.GetHeader("name")
|
|
if len(name) < 1 {
|
|
return ErrMissingDataName
|
|
}
|
|
|
|
if _, nameUsed := reader.Data[name]; nameUsed {
|
|
return ErrDataNameConflict
|
|
}
|
|
|
|
reader.Data[name] = comp
|
|
|
|
if err == io.EOF {
|
|
return nil
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Get returns a multipart data by name, nil if not found
|
|
func (reader *Reader) Get(_key string) *Component {
|
|
data, ok := reader.Data[_key]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return data
|
|
}
|