85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
|
package multipart
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Read all until the next boundary is found
|
||
|
|
func (i *MultipartReader) readComponent() ([]string, error) {
|
||
|
|
|
||
|
|
component := make([]string, 0)
|
||
|
|
|
||
|
|
for { // Read until boundary or error
|
||
|
|
|
||
|
|
line, _, err := i.reader.ReadLine()
|
||
|
|
|
||
|
|
/* (1) Stop on error */
|
||
|
|
if err != nil {
|
||
|
|
return component, err
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (2) Stop at boundary */
|
||
|
|
if strings.HasPrefix(string(line), i.boundary) {
|
||
|
|
return component, err
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (3) Ignore empty lines */
|
||
|
|
if len(line) > 0 {
|
||
|
|
component = append(component, string(line))
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// Parses a single component from its raw lines
|
||
|
|
func (i *MultipartReader) parseComponent(line []string) error {
|
||
|
|
|
||
|
|
// next line index to use
|
||
|
|
cursor := 1
|
||
|
|
|
||
|
|
/* (1) Fail if invalid line count */
|
||
|
|
if len(line) < 2 {
|
||
|
|
return fmt.Errorf("Missing data to parse component")
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (2) Split meta data */
|
||
|
|
meta := strings.Split(line[0], "; ")
|
||
|
|
|
||
|
|
if len(meta) < 2 {
|
||
|
|
return fmt.Errorf("Missing component meta data")
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (3) Extract name */
|
||
|
|
if !strings.HasPrefix(meta[1], `name="`) {
|
||
|
|
return fmt.Errorf("Cannot extract component name")
|
||
|
|
}
|
||
|
|
name := meta[1][len(`name="`) : len(meta[1])-1]
|
||
|
|
|
||
|
|
/* (4) Check if it is a file */
|
||
|
|
isFile := len(meta) > 2 && strings.HasPrefix(meta[2], `filename="`)
|
||
|
|
|
||
|
|
// skip next line (Content-Type) if file
|
||
|
|
if isFile {
|
||
|
|
cursor++
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (5) Create index if name not already used */
|
||
|
|
already, isset := i.Components[name]
|
||
|
|
if !isset {
|
||
|
|
|
||
|
|
i.Components[name] = &MultipartComponent{
|
||
|
|
File: isFile,
|
||
|
|
Data: make([]string, 0),
|
||
|
|
}
|
||
|
|
already = i.Components[name]
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/* (6) Store new value */
|
||
|
|
already.Data = append(already.Data, strings.Join(line[cursor:], "\n"))
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|