134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package forms
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
"unicode"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/araddon/dateparse"
|
|
)
|
|
|
|
// Query parameters for GET /api/v1/photos
|
|
type PhotoSearchForm struct {
|
|
Query string `form:"q"`
|
|
|
|
Title string `form:"title"`
|
|
Description string `form:"description"`
|
|
Notes string `form:"notes"`
|
|
Artist string `form:"artist"`
|
|
Hash string `form:"hash"`
|
|
Duplicate bool `form:"duplicate"`
|
|
Lat float64 `form:"lat"`
|
|
Long float64 `form:"long"`
|
|
Dist uint `form:"dist"`
|
|
Fmin float64 `form:"fmin"`
|
|
Fmax float64 `form:"fmax"`
|
|
Chroma uint `form:"chroma"`
|
|
Mono bool `form:"mono"`
|
|
Portrait bool `form:"portrait"`
|
|
Location bool `form:"location"`
|
|
Tags string `form:"tags"`
|
|
Country string `form:"country"`
|
|
Color string `form:"color"`
|
|
Camera int `form:"camera"`
|
|
Before time.Time `form:"before" time_format:"2006-01-02"`
|
|
After time.Time `form:"after" time_format:"2006-01-02"`
|
|
Favorites bool `form:"favorites"`
|
|
|
|
Count int `form:"count" binding:"required"`
|
|
Offset int `form:"offset"`
|
|
Order string `form:"order"`
|
|
}
|
|
|
|
func (f *PhotoSearchForm) ParseQueryString() (result error) {
|
|
var key, value []byte
|
|
var escaped, isKeyValue bool
|
|
|
|
query := f.Query
|
|
|
|
f.Query = ""
|
|
|
|
formValues := reflect.ValueOf(f).Elem()
|
|
|
|
query = strings.TrimSpace(query) + "\n"
|
|
|
|
for _, char := range query {
|
|
if unicode.IsSpace(char) && !escaped {
|
|
if isKeyValue {
|
|
fieldName := string(bytes.Title(bytes.ToLower(key)))
|
|
field := formValues.FieldByName(fieldName)
|
|
stringValue := string(bytes.ToLower(value))
|
|
|
|
if field.CanSet() {
|
|
switch field.Interface().(type) {
|
|
case time.Time:
|
|
if timeValue, err := dateparse.ParseAny(stringValue); err != nil {
|
|
result = err
|
|
} else {
|
|
field.Set(reflect.ValueOf(timeValue))
|
|
}
|
|
case float64:
|
|
if floatValue, err := strconv.ParseFloat(stringValue, 64); err != nil {
|
|
result = err
|
|
} else {
|
|
field.SetFloat(floatValue)
|
|
}
|
|
case int, int64:
|
|
if intValue, err := strconv.Atoi(stringValue); err != nil {
|
|
result = err
|
|
} else {
|
|
field.SetInt(int64(intValue))
|
|
}
|
|
case uint, uint64:
|
|
if intValue, err := strconv.Atoi(stringValue); err != nil {
|
|
result = err
|
|
} else {
|
|
field.SetUint(uint64(intValue))
|
|
}
|
|
case string:
|
|
field.SetString(stringValue)
|
|
case bool:
|
|
if stringValue == "1" || stringValue == "true" || stringValue == "yes" {
|
|
field.SetBool(true)
|
|
} else if stringValue == "0" || stringValue == "false" || stringValue == "no" {
|
|
field.SetBool(false)
|
|
} else {
|
|
result = fmt.Errorf("not a bool value: %s", fieldName)
|
|
}
|
|
default:
|
|
result = fmt.Errorf("unsupported field type: %s", fieldName)
|
|
}
|
|
} else {
|
|
result = fmt.Errorf("unknown form field: %s", fieldName)
|
|
}
|
|
} else {
|
|
f.Query = string(bytes.ToLower(key))
|
|
}
|
|
|
|
escaped = false
|
|
isKeyValue = false
|
|
key = key[:0]
|
|
value = value[:0]
|
|
} else if char == ':' {
|
|
isKeyValue = true
|
|
} else if char == '"' {
|
|
escaped = !escaped
|
|
} else if isKeyValue {
|
|
value = append(value, byte(char))
|
|
} else {
|
|
key = append(key, byte(char))
|
|
}
|
|
}
|
|
|
|
if result != nil {
|
|
log.Errorf("error while parsing search form: %s", result)
|
|
}
|
|
|
|
return result
|
|
}
|