Blame view
filtering.go
2.32 KB
564ef6971 added simple sql ... |
1 2 3 4 5 |
package webutility import ( "fmt" "net/http" |
d869bd3f2 filtering fix |
6 |
"net/url" |
564ef6971 added simple sql ... |
7 8 |
"strings" ) |
dcfc40497 improved filtering |
9 |
type Filter map[string][]string |
564ef6971 added simple sql ... |
10 |
|
dcfc40497 improved filtering |
11 12 13 |
func (f Filter) Get(key string) (values []string, ok bool) { values, ok = f[key] return values, ok |
ab548c502 added Valid() to ... |
14 |
} |
dcfc40497 improved filtering |
15 16 17 18 19 20 |
func (f Filter) Size() int { return len(f) } func (f Filter) IsNotEmpty() bool { return len(f) > 0 |
ab548c502 added Valid() to ... |
21 |
} |
d06571d28 filter.ValueAt() |
22 23 24 25 26 27 28 29 30 |
func (f Filter) ValueAt(val string, index int) string { if filter, ok := f[val]; ok { if len(filter) > index { return filter[index] } } return "" } |
564ef6971 added simple sql ... |
31 |
func (fs Filter) validate(validFilters []string) (Filter, bool) { |
dcfc40497 improved filtering |
32 33 |
goodFilters := make(Filter) cnt, len := 0, 0 |
564ef6971 added simple sql ... |
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
for f, _ := range fs { len++ for _, v := range validFilters { if f == v { cnt++ goodFilters[f] = fs[f] } } } result := true if len > 0 && cnt == 0 { // if no valid filters are found declare filtering request as invalid result = false } return goodFilters, result } // requires input in format: "param1::value1|param2::value2..." func ParseFilters(req *http.Request, header string) (filters Filter) { q := req.FormValue(header) q = strings.Trim(q, "\"") kvp := strings.Split(q, "|") |
dcfc40497 improved filtering |
58 |
filters = make(Filter, len(kvp)) |
564ef6971 added simple sql ... |
59 60 61 62 |
for i, _ := range kvp { kv := strings.Split(kvp[i], "::") if len(kv) == 2 { |
d869bd3f2 filtering fix |
63 |
key, _ := url.QueryUnescape(kv[0]) |
dcfc40497 improved filtering |
64 65 66 67 68 69 70 |
// get values (if more than 1) vals := strings.Split(kv[1], ",") for _, v := range vals { u, _ := url.QueryUnescape(v) filters[key] = append(filters[key], u) } |
564ef6971 added simple sql ... |
71 72 73 74 75 |
} } return filters } |
d3cc6a289 < > filtering |
76 |
// TODO(marko): very dodgy, needs more robustness |
564ef6971 added simple sql ... |
77 78 79 80 |
func MakeFilterString(prefix string, filters Filter, validFilters []string) (res string, ok bool) { if prefix != "" { prefix += "." } |
dcfc40497 improved filtering |
81 82 |
if !filters.IsNotEmpty() { |
564ef6971 added simple sql ... |
83 84 85 86 87 |
return "", true } filters, ok = filters.validate(validFilters) if !ok { |
dcfc40497 improved filtering |
88 |
return "", false |
564ef6971 added simple sql ... |
89 |
} |
ad6d508b9 minor fix |
90 |
first := true |
dcfc40497 improved filtering |
91 |
for k, filter := range filters { |
dcfc40497 improved filtering |
92 93 94 |
symbol := "=" if first { |
4f226ef30 fixed filterd ord... |
95 |
res += " where " |
dcfc40497 improved filtering |
96 |
first = false |
564ef6971 added simple sql ... |
97 |
} else { |
4f226ef30 fixed filterd ord... |
98 |
res += " and " |
564ef6971 added simple sql ... |
99 |
} |
d3cc6a289 < > filtering |
100 |
|
dcfc40497 improved filtering |
101 102 |
res += "(" for i, f := range filter { |
66643e090 added support for != |
103 |
if strings.HasPrefix(f, "<") || strings.HasPrefix(f, ">") || strings.HasPrefix(f, "!") { |
dcfc40497 improved filtering |
104 |
symbol = string(f[0]) |
66643e090 added support for != |
105 |
f = strings.TrimLeft(f, "<>!") |
dcfc40497 improved filtering |
106 107 108 109 110 111 112 113 114 115 116 117 118 |
if strings.HasPrefix(f, "=") { f = strings.TrimLeft(f, "=") symbol += "=" } } res += fmt.Sprintf("%s%s %s '%s'", prefix, k, symbol, f) if i < len(filter)-1 { res += " or " } } res += ")" |
564ef6971 added simple sql ... |
119 120 121 122 |
} return res, ok } |