144 lines
2.9 KiB
Go
144 lines
2.9 KiB
Go
package reqdata
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"git.xdrm.io/go/aicra/internal/cerr"
|
|
)
|
|
|
|
// ErrUnknownType is returned when encountering an unknown type
|
|
const ErrUnknownType = cerr.Error("unknown type")
|
|
|
|
// ErrInvalidJSON is returned when json parse failed
|
|
const ErrInvalidJSON = cerr.Error("invalid json")
|
|
|
|
// ErrInvalidRootType is returned when json is a map
|
|
const ErrInvalidRootType = cerr.Error("invalid json root type")
|
|
|
|
// Parameter represents an http request parameter
|
|
// that can be of type URL, GET, or FORM (multipart, json, urlencoded)
|
|
type Parameter struct {
|
|
// whether the value has been json-parsed
|
|
// for optimisation purpose, parameters are only parsed
|
|
// if they are required by the current service
|
|
Parsed bool
|
|
|
|
// whether the value is a file
|
|
File bool
|
|
|
|
// the actual parameter value
|
|
Value interface{}
|
|
}
|
|
|
|
// Parse parameter (json-like) if not already done
|
|
func (i *Parameter) Parse() error {
|
|
|
|
/* (1) Stop if already parsed or nil*/
|
|
if i.Parsed || i.Value == nil {
|
|
return nil
|
|
}
|
|
|
|
/* (2) Try to parse value */
|
|
parsed, err := parseParameter(i.Value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
i.Parsed = true
|
|
i.Value = parsed
|
|
|
|
return nil
|
|
}
|
|
|
|
// parseParameter parses http GET/POST data
|
|
// - []string
|
|
// - size = 1 : return json of first element
|
|
// - size > 1 : return array of json elements
|
|
// - string : return json if valid, else return raw string
|
|
func parseParameter(data interface{}) (interface{}, error) {
|
|
dtype := reflect.TypeOf(data)
|
|
dvalue := reflect.ValueOf(data)
|
|
|
|
switch dtype.Kind() {
|
|
|
|
/* (1) []string -> recursive */
|
|
case reflect.Slice:
|
|
|
|
// 1. Return nothing if empty
|
|
if dvalue.Len() == 0 {
|
|
return data, nil
|
|
}
|
|
|
|
// 2. only return first element if alone
|
|
if dvalue.Len() == 1 {
|
|
|
|
element := dvalue.Index(0)
|
|
|
|
// try to parse if a string (containing json)
|
|
if element.Kind() == reflect.String {
|
|
return parseParameter(element.String())
|
|
}
|
|
|
|
// already typed
|
|
return data, nil
|
|
|
|
}
|
|
|
|
// 3. Return all elements if more than 1
|
|
result := make([]interface{}, dvalue.Len())
|
|
|
|
for i, l := 0, dvalue.Len(); i < l; i++ {
|
|
element := dvalue.Index(i)
|
|
|
|
// ignore non-string
|
|
if element.Kind() != reflect.String {
|
|
result[i] = nil
|
|
continue
|
|
}
|
|
|
|
parsed, err := parseParameter(element.String())
|
|
if err != nil {
|
|
return data, err
|
|
}
|
|
result[i] = parsed
|
|
}
|
|
return result, nil
|
|
|
|
/* (2) string -> parse */
|
|
case reflect.String:
|
|
|
|
// build json wrapper
|
|
wrapper := fmt.Sprintf("{\"wrapped\":%s}", dvalue.String())
|
|
|
|
// try to parse as json
|
|
var result interface{}
|
|
err := json.Unmarshal([]byte(wrapper), &result)
|
|
|
|
// return if success
|
|
if err == nil {
|
|
|
|
mapval, ok := result.(map[string]interface{})
|
|
if !ok {
|
|
return dvalue.String(), ErrInvalidRootType
|
|
}
|
|
|
|
wrapped, ok := mapval["wrapped"]
|
|
if !ok {
|
|
return dvalue.String(), ErrInvalidJSON
|
|
}
|
|
|
|
return wrapped, nil
|
|
}
|
|
|
|
// else return as string
|
|
return dvalue.String(), nil
|
|
|
|
}
|
|
|
|
/* (3) NIL if unknown type */
|
|
return dvalue, ErrUnknownType
|
|
|
|
}
|