2015-04-05 15:03:16 -04:00
|
|
|
// Copyright © 2013-14 Steve Francia <spf@spf13.com>.
|
|
|
|
//
|
|
|
|
// Licensed under the Simple Public License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
// http://opensource.org/licenses/Simple-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
|
|
|
package tpl
|
|
|
|
|
|
|
|
import (
|
2015-09-22 16:24:24 -04:00
|
|
|
"bitbucket.org/pkg/inflect"
|
2015-04-05 15:03:16 -04:00
|
|
|
"bytes"
|
2015-09-12 15:45:12 -04:00
|
|
|
"encoding/base64"
|
2015-04-05 15:03:16 -04:00
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"html"
|
|
|
|
"html/template"
|
|
|
|
"os"
|
|
|
|
"reflect"
|
|
|
|
"sort"
|
|
|
|
"strconv"
|
|
|
|
"strings"
|
2015-05-26 06:33:32 -04:00
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/spf13/cast"
|
|
|
|
"github.com/spf13/hugo/helpers"
|
|
|
|
jww "github.com/spf13/jwalterweatherman"
|
2015-04-05 15:03:16 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
var funcMap template.FuncMap
|
|
|
|
|
|
|
|
func Eq(x, y interface{}) bool {
|
|
|
|
normalize := func(v interface{}) interface{} {
|
|
|
|
vv := reflect.ValueOf(v)
|
|
|
|
switch vv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
return vv.Int()
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
return vv.Float()
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
return vv.Uint()
|
|
|
|
default:
|
|
|
|
return v
|
|
|
|
}
|
|
|
|
}
|
|
|
|
x = normalize(x)
|
|
|
|
y = normalize(y)
|
|
|
|
return reflect.DeepEqual(x, y)
|
|
|
|
}
|
|
|
|
|
|
|
|
func Ne(x, y interface{}) bool {
|
|
|
|
return !Eq(x, y)
|
|
|
|
}
|
|
|
|
|
|
|
|
func Ge(a, b interface{}) bool {
|
|
|
|
left, right := compareGetFloat(a, b)
|
|
|
|
return left >= right
|
|
|
|
}
|
|
|
|
|
|
|
|
func Gt(a, b interface{}) bool {
|
|
|
|
left, right := compareGetFloat(a, b)
|
|
|
|
return left > right
|
|
|
|
}
|
|
|
|
|
|
|
|
func Le(a, b interface{}) bool {
|
|
|
|
left, right := compareGetFloat(a, b)
|
|
|
|
return left <= right
|
|
|
|
}
|
|
|
|
|
|
|
|
func Lt(a, b interface{}) bool {
|
|
|
|
left, right := compareGetFloat(a, b)
|
|
|
|
return left < right
|
|
|
|
}
|
|
|
|
|
2015-10-02 11:30:21 -04:00
|
|
|
func Dictionary(values ...interface{}) (map[string]interface{}, error) {
|
|
|
|
if len(values)%2 != 0 {
|
|
|
|
return nil, errors.New("invalid dict call")
|
|
|
|
}
|
|
|
|
dict := make(map[string]interface{}, len(values)/2)
|
2015-10-12 14:47:06 -04:00
|
|
|
for i := 0; i < len(values); i += 2 {
|
2015-10-02 11:30:21 -04:00
|
|
|
key, ok := values[i].(string)
|
|
|
|
if !ok {
|
|
|
|
return nil, errors.New("dict keys must be strings")
|
|
|
|
}
|
2015-10-12 14:47:06 -04:00
|
|
|
dict[key] = values[i+1]
|
2015-10-02 11:30:21 -04:00
|
|
|
}
|
|
|
|
return dict, nil
|
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
func compareGetFloat(a interface{}, b interface{}) (float64, float64) {
|
|
|
|
var left, right float64
|
|
|
|
var leftStr, rightStr *string
|
|
|
|
var err error
|
|
|
|
av := reflect.ValueOf(a)
|
|
|
|
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.Array, reflect.Chan, reflect.Map, reflect.Slice:
|
|
|
|
left = float64(av.Len())
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
left = float64(av.Int())
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
left = av.Float()
|
|
|
|
case reflect.String:
|
|
|
|
left, err = strconv.ParseFloat(av.String(), 64)
|
|
|
|
if err != nil {
|
|
|
|
str := av.String()
|
|
|
|
leftStr = &str
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bv := reflect.ValueOf(b)
|
|
|
|
|
|
|
|
switch bv.Kind() {
|
|
|
|
case reflect.Array, reflect.Chan, reflect.Map, reflect.Slice:
|
|
|
|
right = float64(bv.Len())
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
right = float64(bv.Int())
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
right = bv.Float()
|
|
|
|
case reflect.String:
|
|
|
|
right, err = strconv.ParseFloat(bv.String(), 64)
|
|
|
|
if err != nil {
|
|
|
|
str := bv.String()
|
|
|
|
rightStr = &str
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case leftStr == nil || rightStr == nil:
|
|
|
|
case *leftStr < *rightStr:
|
|
|
|
return 0, 1
|
|
|
|
case *leftStr > *rightStr:
|
|
|
|
return 1, 0
|
|
|
|
default:
|
|
|
|
return 0, 0
|
|
|
|
}
|
|
|
|
|
|
|
|
return left, right
|
|
|
|
}
|
|
|
|
|
|
|
|
// Slicing in Slicestr is done by specifying a half-open range with
|
|
|
|
// two indices, start and end. 1 and 4 creates a slice including elements 1 through 3.
|
|
|
|
// The end index can be omitted, it defaults to the string's length.
|
2015-08-14 04:36:56 -04:00
|
|
|
func Slicestr(a interface{}, startEnd ...interface{}) (string, error) {
|
2015-04-05 15:03:16 -04:00
|
|
|
aStr, err := cast.ToStringE(a)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
2015-08-14 04:36:56 -04:00
|
|
|
var argStart, argEnd int
|
|
|
|
|
|
|
|
argNum := len(startEnd)
|
|
|
|
|
|
|
|
if argNum > 0 {
|
2015-08-15 09:47:16 -04:00
|
|
|
if argStart, err = cast.ToIntE(startEnd[0]); err != nil {
|
|
|
|
return "", errors.New("start argument must be integer")
|
2015-08-14 04:36:56 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if argNum > 1 {
|
2015-08-15 09:47:16 -04:00
|
|
|
if argEnd, err = cast.ToIntE(startEnd[1]); err != nil {
|
|
|
|
return "", errors.New("end argument must be integer")
|
2015-08-14 04:36:56 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if argNum > 2 {
|
2015-04-05 15:03:16 -04:00
|
|
|
return "", errors.New("too many arguments")
|
|
|
|
}
|
|
|
|
|
2015-08-07 02:52:22 -04:00
|
|
|
asRunes := []rune(aStr)
|
|
|
|
|
2015-08-14 04:36:56 -04:00
|
|
|
if argNum > 0 && (argStart < 0 || argStart >= len(asRunes)) {
|
2015-04-30 05:26:34 -04:00
|
|
|
return "", errors.New("slice bounds out of range")
|
|
|
|
}
|
|
|
|
|
2015-08-14 04:36:56 -04:00
|
|
|
if argNum == 2 {
|
|
|
|
if argEnd < 0 || argEnd > len(asRunes) {
|
2015-04-30 05:26:34 -04:00
|
|
|
return "", errors.New("slice bounds out of range")
|
|
|
|
}
|
2015-08-14 04:36:56 -04:00
|
|
|
return string(asRunes[argStart:argEnd]), nil
|
|
|
|
} else if argNum == 1 {
|
|
|
|
return string(asRunes[argStart:]), nil
|
2015-04-05 15:03:16 -04:00
|
|
|
} else {
|
2015-08-07 02:52:22 -04:00
|
|
|
return string(asRunes[:]), nil
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// Substr extracts parts of a string, beginning at the character at the specified
|
|
|
|
// position, and returns the specified number of characters.
|
|
|
|
//
|
|
|
|
// It normally takes two parameters: start and length.
|
|
|
|
// It can also take one parameter: start, i.e. length is omitted, in which case
|
|
|
|
// the substring starting from start until the end of the string will be returned.
|
|
|
|
//
|
|
|
|
// To extract characters from the end of the string, use a negative start number.
|
|
|
|
//
|
|
|
|
// In addition, borrowing from the extended behavior described at http://php.net/substr,
|
|
|
|
// if length is given and is negative, then that many characters will be omitted from
|
|
|
|
// the end of string.
|
2015-06-06 11:21:14 -04:00
|
|
|
func Substr(a interface{}, nums ...interface{}) (string, error) {
|
2015-04-05 15:03:16 -04:00
|
|
|
aStr, err := cast.ToStringE(a)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
var start, length int
|
2015-06-06 11:21:14 -04:00
|
|
|
|
2015-08-07 02:52:22 -04:00
|
|
|
asRunes := []rune(aStr)
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
switch len(nums) {
|
2015-06-06 11:21:14 -04:00
|
|
|
case 0:
|
|
|
|
return "", errors.New("too less arguments")
|
2015-04-05 15:03:16 -04:00
|
|
|
case 1:
|
2015-08-15 09:47:16 -04:00
|
|
|
if start, err = cast.ToIntE(nums[0]); err != nil {
|
|
|
|
return "", errors.New("start argument must be integer")
|
2015-06-06 11:21:14 -04:00
|
|
|
}
|
2015-08-07 02:52:22 -04:00
|
|
|
length = len(asRunes)
|
2015-04-05 15:03:16 -04:00
|
|
|
case 2:
|
2015-08-15 09:47:16 -04:00
|
|
|
if start, err = cast.ToIntE(nums[0]); err != nil {
|
|
|
|
return "", errors.New("start argument must be integer")
|
2015-06-06 11:21:14 -04:00
|
|
|
}
|
2015-08-15 09:47:16 -04:00
|
|
|
if length, err = cast.ToIntE(nums[1]); err != nil {
|
|
|
|
return "", errors.New("length argument must be integer")
|
2015-06-06 11:21:14 -04:00
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
default:
|
|
|
|
return "", errors.New("too many arguments")
|
|
|
|
}
|
|
|
|
|
2015-08-07 02:52:22 -04:00
|
|
|
if start < -len(asRunes) {
|
2015-04-05 15:03:16 -04:00
|
|
|
start = 0
|
|
|
|
}
|
2015-08-07 02:52:22 -04:00
|
|
|
if start > len(asRunes) {
|
2015-04-05 15:03:16 -04:00
|
|
|
return "", errors.New(fmt.Sprintf("start position out of bounds for %d-byte string", len(aStr)))
|
|
|
|
}
|
|
|
|
|
|
|
|
var s, e int
|
|
|
|
if start >= 0 && length >= 0 {
|
|
|
|
s = start
|
|
|
|
e = start + length
|
|
|
|
} else if start < 0 && length >= 0 {
|
2015-08-07 02:52:22 -04:00
|
|
|
s = len(asRunes) + start - length + 1
|
|
|
|
e = len(asRunes) + start + 1
|
2015-04-05 15:03:16 -04:00
|
|
|
} else if start >= 0 && length < 0 {
|
|
|
|
s = start
|
2015-08-07 02:52:22 -04:00
|
|
|
e = len(asRunes) + length
|
2015-04-05 15:03:16 -04:00
|
|
|
} else {
|
2015-08-07 02:52:22 -04:00
|
|
|
s = len(asRunes) + start
|
|
|
|
e = len(asRunes) + length
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
if s > e {
|
|
|
|
return "", errors.New(fmt.Sprintf("calculated start position greater than end position: %d > %d", s, e))
|
|
|
|
}
|
2015-08-07 02:52:22 -04:00
|
|
|
if e > len(asRunes) {
|
|
|
|
e = len(asRunes)
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
2015-08-07 02:52:22 -04:00
|
|
|
return string(asRunes[s:e]), nil
|
2015-04-05 15:03:16 -04:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func Split(a interface{}, delimiter string) ([]string, error) {
|
|
|
|
aStr, err := cast.ToStringE(a)
|
|
|
|
if err != nil {
|
|
|
|
return []string{}, err
|
|
|
|
}
|
|
|
|
return strings.Split(aStr, delimiter), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func Intersect(l1, l2 interface{}) (interface{}, error) {
|
|
|
|
if l1 == nil || l2 == nil {
|
|
|
|
return make([]interface{}, 0), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
l1v := reflect.ValueOf(l1)
|
|
|
|
l2v := reflect.ValueOf(l2)
|
|
|
|
|
|
|
|
switch l1v.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
switch l2v.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
r := reflect.MakeSlice(l1v.Type(), 0, 0)
|
|
|
|
for i := 0; i < l1v.Len(); i++ {
|
|
|
|
l1vv := l1v.Index(i)
|
|
|
|
for j := 0; j < l2v.Len(); j++ {
|
|
|
|
l2vv := l2v.Index(j)
|
|
|
|
switch l1vv.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
if l1vv.Type() == l2vv.Type() && l1vv.String() == l2vv.String() && !In(r, l2vv) {
|
|
|
|
r = reflect.Append(r, l2vv)
|
|
|
|
}
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
switch l2vv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
if l1vv.Int() == l2vv.Int() && !In(r, l2vv) {
|
|
|
|
r = reflect.Append(r, l2vv)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
switch l2vv.Kind() {
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
if l1vv.Float() == l2vv.Float() && !In(r, l2vv) {
|
|
|
|
r = reflect.Append(r, l2vv)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return r.Interface(), nil
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(l2).Type().String())
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(l1).Type().String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func In(l interface{}, v interface{}) bool {
|
|
|
|
lv := reflect.ValueOf(l)
|
|
|
|
vv := reflect.ValueOf(v)
|
|
|
|
|
|
|
|
switch lv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
for i := 0; i < lv.Len(); i++ {
|
|
|
|
lvv := lv.Index(i)
|
2015-10-12 11:58:40 -04:00
|
|
|
lvv, isNil := indirect(lvv)
|
|
|
|
if isNil {
|
|
|
|
continue
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
switch lvv.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
if vv.Type() == lvv.Type() && vv.String() == lvv.String() {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
switch vv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
if vv.Int() == lvv.Int() {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
switch vv.Kind() {
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
if vv.Float() == lvv.Float() {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case reflect.String:
|
|
|
|
if vv.Type() == lv.Type() && strings.Contains(lv.String(), vv.String()) {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// indirect is taken from 'text/template/exec.go'
|
|
|
|
func indirect(v reflect.Value) (rv reflect.Value, isNil bool) {
|
|
|
|
for ; v.Kind() == reflect.Ptr || v.Kind() == reflect.Interface; v = v.Elem() {
|
|
|
|
if v.IsNil() {
|
|
|
|
return v, true
|
|
|
|
}
|
|
|
|
if v.Kind() == reflect.Interface && v.NumMethod() > 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return v, false
|
|
|
|
}
|
|
|
|
|
|
|
|
// First is exposed to templates, to iterate over the first N items in a
|
|
|
|
// rangeable list.
|
|
|
|
func First(limit interface{}, seq interface{}) (interface{}, error) {
|
|
|
|
|
|
|
|
if limit == nil || seq == nil {
|
|
|
|
return nil, errors.New("both limit and seq must be provided")
|
|
|
|
}
|
|
|
|
|
|
|
|
limitv, err := cast.ToIntE(limit)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
if limitv < 1 {
|
|
|
|
return nil, errors.New("can't return negative/empty count of items from sequence")
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice, reflect.String:
|
|
|
|
// okay
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
if limitv > seqv.Len() {
|
|
|
|
limitv = seqv.Len()
|
|
|
|
}
|
|
|
|
return seqv.Slice(0, limitv).Interface(), nil
|
|
|
|
}
|
|
|
|
|
2015-06-10 18:11:47 -04:00
|
|
|
// Last is exposed to templates, to iterate over the last N items in a
|
|
|
|
// rangeable list.
|
|
|
|
func Last(limit interface{}, seq interface{}) (interface{}, error) {
|
|
|
|
|
|
|
|
if limit == nil || seq == nil {
|
|
|
|
return nil, errors.New("both limit and seq must be provided")
|
|
|
|
}
|
|
|
|
|
|
|
|
limitv, err := cast.ToIntE(limit)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
if limitv < 1 {
|
|
|
|
return nil, errors.New("can't return negative/empty count of items from sequence")
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice, reflect.String:
|
|
|
|
// okay
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
if limitv > seqv.Len() {
|
|
|
|
limitv = seqv.Len()
|
|
|
|
}
|
|
|
|
return seqv.Slice(seqv.Len()-limitv, seqv.Len()).Interface(), nil
|
|
|
|
}
|
|
|
|
|
2015-06-10 17:49:55 -04:00
|
|
|
// After is exposed to templates, to iterate over all the items after N in a
|
|
|
|
// rangeable list. It's meant to accompany First
|
2015-06-10 17:58:57 -04:00
|
|
|
func After(index interface{}, seq interface{}) (interface{}, error) {
|
2015-06-10 17:49:55 -04:00
|
|
|
|
2015-06-10 17:58:57 -04:00
|
|
|
if index == nil || seq == nil {
|
2015-06-10 17:49:55 -04:00
|
|
|
return nil, errors.New("both limit and seq must be provided")
|
|
|
|
}
|
|
|
|
|
2015-06-10 17:58:57 -04:00
|
|
|
indexv, err := cast.ToIntE(index)
|
2015-06-10 17:49:55 -04:00
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2015-06-10 17:58:57 -04:00
|
|
|
if indexv < 1 {
|
2015-06-10 17:49:55 -04:00
|
|
|
return nil, errors.New("can't return negative/empty count of items from sequence")
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice, reflect.String:
|
|
|
|
// okay
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
2015-06-10 17:58:57 -04:00
|
|
|
if indexv >= seqv.Len() {
|
2015-06-10 17:49:55 -04:00
|
|
|
return nil, errors.New("no items left")
|
|
|
|
}
|
2015-06-10 17:58:57 -04:00
|
|
|
return seqv.Slice(indexv, seqv.Len()).Interface(), nil
|
2015-06-10 17:49:55 -04:00
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
var (
|
|
|
|
zero reflect.Value
|
|
|
|
errorType = reflect.TypeOf((*error)(nil)).Elem()
|
2015-05-26 06:33:32 -04:00
|
|
|
timeType = reflect.TypeOf((*time.Time)(nil)).Elem()
|
2015-04-05 15:03:16 -04:00
|
|
|
)
|
|
|
|
|
2015-05-26 06:33:32 -04:00
|
|
|
func timeUnix(v reflect.Value) int64 {
|
|
|
|
if v.Type() != timeType {
|
|
|
|
panic("coding error: argument must be time.Time type reflect Value")
|
|
|
|
}
|
|
|
|
return v.MethodByName("Unix").Call([]reflect.Value{})[0].Int()
|
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
func evaluateSubElem(obj reflect.Value, elemName string) (reflect.Value, error) {
|
|
|
|
if !obj.IsValid() {
|
|
|
|
return zero, errors.New("can't evaluate an invalid value")
|
|
|
|
}
|
|
|
|
typ := obj.Type()
|
|
|
|
obj, isNil := indirect(obj)
|
|
|
|
|
|
|
|
// first, check whether obj has a method. In this case, obj is
|
|
|
|
// an interface, a struct or its pointer. If obj is a struct,
|
|
|
|
// to check all T and *T method, use obj pointer type Value
|
|
|
|
objPtr := obj
|
|
|
|
if objPtr.Kind() != reflect.Interface && objPtr.CanAddr() {
|
|
|
|
objPtr = objPtr.Addr()
|
|
|
|
}
|
|
|
|
mt, ok := objPtr.Type().MethodByName(elemName)
|
|
|
|
if ok {
|
|
|
|
if mt.PkgPath != "" {
|
|
|
|
return zero, fmt.Errorf("%s is an unexported method of type %s", elemName, typ)
|
|
|
|
}
|
|
|
|
// struct pointer has one receiver argument and interface doesn't have an argument
|
|
|
|
if mt.Type.NumIn() > 1 || mt.Type.NumOut() == 0 || mt.Type.NumOut() > 2 {
|
|
|
|
return zero, fmt.Errorf("%s is a method of type %s but doesn't satisfy requirements", elemName, typ)
|
|
|
|
}
|
|
|
|
if mt.Type.NumOut() == 1 && mt.Type.Out(0).Implements(errorType) {
|
|
|
|
return zero, fmt.Errorf("%s is a method of type %s but doesn't satisfy requirements", elemName, typ)
|
|
|
|
}
|
|
|
|
if mt.Type.NumOut() == 2 && !mt.Type.Out(1).Implements(errorType) {
|
|
|
|
return zero, fmt.Errorf("%s is a method of type %s but doesn't satisfy requirements", elemName, typ)
|
|
|
|
}
|
|
|
|
res := objPtr.Method(mt.Index).Call([]reflect.Value{})
|
|
|
|
if len(res) == 2 && !res[1].IsNil() {
|
|
|
|
return zero, fmt.Errorf("error at calling a method %s of type %s: %s", elemName, typ, res[1].Interface().(error))
|
|
|
|
}
|
|
|
|
return res[0], nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// elemName isn't a method so next start to check whether it is
|
|
|
|
// a struct field or a map value. In both cases, it mustn't be
|
|
|
|
// a nil value
|
|
|
|
if isNil {
|
|
|
|
return zero, fmt.Errorf("can't evaluate a nil pointer of type %s by a struct field or map key name %s", typ, elemName)
|
|
|
|
}
|
|
|
|
switch obj.Kind() {
|
|
|
|
case reflect.Struct:
|
|
|
|
ft, ok := obj.Type().FieldByName(elemName)
|
|
|
|
if ok {
|
|
|
|
if ft.PkgPath != "" {
|
|
|
|
return zero, fmt.Errorf("%s is an unexported field of struct type %s", elemName, typ)
|
|
|
|
}
|
|
|
|
return obj.FieldByIndex(ft.Index), nil
|
|
|
|
}
|
|
|
|
return zero, fmt.Errorf("%s isn't a field of struct type %s", elemName, typ)
|
|
|
|
case reflect.Map:
|
|
|
|
kv := reflect.ValueOf(elemName)
|
|
|
|
if kv.Type().AssignableTo(obj.Type().Key()) {
|
|
|
|
return obj.MapIndex(kv), nil
|
|
|
|
}
|
|
|
|
return zero, fmt.Errorf("%s isn't a key of map type %s", elemName, typ)
|
|
|
|
}
|
|
|
|
return zero, fmt.Errorf("%s is neither a struct field, a method nor a map element of type %s", elemName, typ)
|
|
|
|
}
|
|
|
|
|
|
|
|
func checkCondition(v, mv reflect.Value, op string) (bool, error) {
|
2015-06-27 13:15:42 -04:00
|
|
|
v, vIsNil := indirect(v)
|
|
|
|
if !v.IsValid() {
|
|
|
|
vIsNil = true
|
|
|
|
}
|
|
|
|
mv, mvIsNil := indirect(mv)
|
|
|
|
if !mv.IsValid() {
|
|
|
|
mvIsNil = true
|
|
|
|
}
|
|
|
|
if vIsNil || mvIsNil {
|
|
|
|
switch op {
|
|
|
|
case "", "=", "==", "eq":
|
|
|
|
return vIsNil == mvIsNil, nil
|
|
|
|
case "!=", "<>", "ne":
|
|
|
|
return vIsNil != mvIsNil, nil
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
var ivp, imvp *int64
|
|
|
|
var svp, smvp *string
|
|
|
|
var ima []int64
|
|
|
|
var sma []string
|
|
|
|
if mv.Type() == v.Type() {
|
|
|
|
switch v.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
iv := v.Int()
|
|
|
|
ivp = &iv
|
|
|
|
imv := mv.Int()
|
|
|
|
imvp = &imv
|
|
|
|
case reflect.String:
|
|
|
|
sv := v.String()
|
|
|
|
svp = &sv
|
|
|
|
smv := mv.String()
|
|
|
|
smvp = &smv
|
2015-05-26 06:33:32 -04:00
|
|
|
case reflect.Struct:
|
|
|
|
switch v.Type() {
|
|
|
|
case timeType:
|
|
|
|
iv := timeUnix(v)
|
|
|
|
ivp = &iv
|
|
|
|
imv := timeUnix(mv)
|
|
|
|
imvp = &imv
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if mv.Kind() != reflect.Array && mv.Kind() != reflect.Slice {
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
if mv.Type().Elem() != v.Type() {
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
switch v.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
iv := v.Int()
|
|
|
|
ivp = &iv
|
|
|
|
for i := 0; i < mv.Len(); i++ {
|
|
|
|
ima = append(ima, mv.Index(i).Int())
|
|
|
|
}
|
|
|
|
case reflect.String:
|
|
|
|
sv := v.String()
|
|
|
|
svp = &sv
|
|
|
|
for i := 0; i < mv.Len(); i++ {
|
|
|
|
sma = append(sma, mv.Index(i).String())
|
|
|
|
}
|
2015-05-26 06:33:32 -04:00
|
|
|
case reflect.Struct:
|
|
|
|
switch v.Type() {
|
|
|
|
case timeType:
|
|
|
|
iv := timeUnix(v)
|
|
|
|
ivp = &iv
|
|
|
|
for i := 0; i < mv.Len(); i++ {
|
|
|
|
ima = append(ima, timeUnix(mv.Index(i)))
|
|
|
|
}
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
switch op {
|
|
|
|
case "", "=", "==", "eq":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp == *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp == *smvp, nil
|
|
|
|
}
|
|
|
|
case "!=", "<>", "ne":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp != *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp != *smvp, nil
|
|
|
|
}
|
|
|
|
case ">=", "ge":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp >= *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp >= *smvp, nil
|
|
|
|
}
|
|
|
|
case ">", "gt":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp > *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp > *smvp, nil
|
|
|
|
}
|
|
|
|
case "<=", "le":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp <= *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp <= *smvp, nil
|
|
|
|
}
|
|
|
|
case "<", "lt":
|
|
|
|
if ivp != nil && imvp != nil {
|
|
|
|
return *ivp < *imvp, nil
|
|
|
|
} else if svp != nil && smvp != nil {
|
|
|
|
return *svp < *smvp, nil
|
|
|
|
}
|
|
|
|
case "in", "not in":
|
|
|
|
var r bool
|
|
|
|
if ivp != nil && len(ima) > 0 {
|
|
|
|
r = In(ima, *ivp)
|
|
|
|
} else if svp != nil {
|
|
|
|
if len(sma) > 0 {
|
|
|
|
r = In(sma, *svp)
|
|
|
|
} else if smvp != nil {
|
|
|
|
r = In(*smvp, *svp)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
if op == "not in" {
|
|
|
|
return !r, nil
|
|
|
|
} else {
|
|
|
|
return r, nil
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return false, errors.New("no such an operator")
|
|
|
|
}
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func Where(seq, key interface{}, args ...interface{}) (r interface{}, err error) {
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
kv := reflect.ValueOf(key)
|
|
|
|
|
|
|
|
var mv reflect.Value
|
|
|
|
var op string
|
|
|
|
switch len(args) {
|
|
|
|
case 1:
|
|
|
|
mv = reflect.ValueOf(args[0])
|
|
|
|
case 2:
|
|
|
|
var ok bool
|
|
|
|
if op, ok = args[0].(string); !ok {
|
|
|
|
return nil, errors.New("operator argument must be string type")
|
|
|
|
}
|
|
|
|
op = strings.TrimSpace(strings.ToLower(op))
|
|
|
|
mv = reflect.ValueOf(args[1])
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't evaluate the array by no match argument or more than or equal to two arguments")
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value of type " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
|
|
|
|
var path []string
|
|
|
|
if kv.Kind() == reflect.String {
|
|
|
|
path = strings.Split(strings.Trim(kv.String(), "."), ".")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
rv := reflect.MakeSlice(seqv.Type(), 0, 0)
|
|
|
|
for i := 0; i < seqv.Len(); i++ {
|
|
|
|
var vvv reflect.Value
|
|
|
|
rvv := seqv.Index(i)
|
|
|
|
if kv.Kind() == reflect.String {
|
|
|
|
vvv = rvv
|
|
|
|
for _, elemName := range path {
|
|
|
|
vvv, err = evaluateSubElem(vvv, elemName)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
vv, _ := indirect(rvv)
|
|
|
|
if vv.Kind() == reflect.Map && kv.Type().AssignableTo(vv.Type().Key()) {
|
|
|
|
vvv = vv.MapIndex(kv)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if ok, err := checkCondition(vvv, mv, op); ok {
|
|
|
|
rv = reflect.Append(rv, rvv)
|
|
|
|
} else if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return rv.Interface(), nil
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't iterate over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Apply, given a map, array, or slice, returns a new slice with the function fname applied over it.
|
|
|
|
func Apply(seq interface{}, fname string, args ...interface{}) (interface{}, error) {
|
|
|
|
if seq == nil {
|
|
|
|
return make([]interface{}, 0), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if fname == "apply" {
|
|
|
|
return nil, errors.New("can't apply myself (no turtles allowed)")
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
|
|
|
fn, found := funcMap[fname]
|
|
|
|
if !found {
|
|
|
|
return nil, errors.New("can't find function " + fname)
|
|
|
|
}
|
|
|
|
|
|
|
|
fnv := reflect.ValueOf(fn)
|
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
r := make([]interface{}, seqv.Len())
|
|
|
|
for i := 0; i < seqv.Len(); i++ {
|
|
|
|
vv := seqv.Index(i)
|
|
|
|
|
|
|
|
vvv, err := applyFnToThis(fnv, vv, args...)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
r[i] = vvv.Interface()
|
|
|
|
}
|
|
|
|
|
|
|
|
return r, nil
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't apply over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func applyFnToThis(fn, this reflect.Value, args ...interface{}) (reflect.Value, error) {
|
|
|
|
n := make([]reflect.Value, len(args))
|
|
|
|
for i, arg := range args {
|
|
|
|
if arg == "." {
|
|
|
|
n[i] = this
|
|
|
|
} else {
|
|
|
|
n[i] = reflect.ValueOf(arg)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-30 04:51:01 -04:00
|
|
|
num := fn.Type().NumIn()
|
|
|
|
|
|
|
|
if fn.Type().IsVariadic() {
|
|
|
|
num--
|
|
|
|
}
|
|
|
|
|
2015-05-02 05:32:38 -04:00
|
|
|
// TODO(bep) see #1098 - also see template_tests.go
|
|
|
|
/*if len(args) < num {
|
2015-04-30 04:51:01 -04:00
|
|
|
return reflect.ValueOf(nil), errors.New("Too few arguments")
|
2015-04-30 05:41:25 -04:00
|
|
|
} else if len(args) > num {
|
|
|
|
return reflect.ValueOf(nil), errors.New("Too many arguments")
|
2015-05-02 05:32:38 -04:00
|
|
|
}*/
|
2015-04-30 04:51:01 -04:00
|
|
|
|
2015-05-01 11:00:22 -04:00
|
|
|
for i := 0; i < num; i++ {
|
|
|
|
if xt, targ := n[i].Type(), fn.Type().In(i); !xt.AssignableTo(targ) {
|
|
|
|
return reflect.ValueOf(nil), errors.New("called apply using " + xt.String() + " as type " + targ.String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
res := fn.Call(n)
|
|
|
|
|
|
|
|
if len(res) == 1 || res[1].IsNil() {
|
|
|
|
return res[0], nil
|
|
|
|
} else {
|
|
|
|
return reflect.ValueOf(nil), res[1].Interface().(error)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Delimit(seq, delimiter interface{}, last ...interface{}) (template.HTML, error) {
|
|
|
|
d, err := cast.ToStringE(delimiter)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
var dLast *string
|
|
|
|
for _, l := range last {
|
|
|
|
dStr, err := cast.ToStringE(l)
|
|
|
|
if err != nil {
|
|
|
|
dLast = nil
|
|
|
|
}
|
|
|
|
dLast = &dStr
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return "", errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
|
|
|
var str string
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Map:
|
|
|
|
sortSeq, err := Sort(seq)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
seqv = reflect.ValueOf(sortSeq)
|
|
|
|
fallthrough
|
|
|
|
case reflect.Array, reflect.Slice, reflect.String:
|
|
|
|
for i := 0; i < seqv.Len(); i++ {
|
|
|
|
val := seqv.Index(i).Interface()
|
|
|
|
valStr, err := cast.ToStringE(val)
|
|
|
|
if err != nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
switch {
|
|
|
|
case i == seqv.Len()-2 && dLast != nil:
|
|
|
|
str += valStr + *dLast
|
|
|
|
case i == seqv.Len()-1:
|
|
|
|
str += valStr
|
|
|
|
default:
|
|
|
|
str += valStr + d
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
default:
|
|
|
|
return "", errors.New("can't iterate over " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
|
|
|
|
return template.HTML(str), nil
|
|
|
|
}
|
|
|
|
|
2015-08-06 13:27:15 -04:00
|
|
|
func Sort(seq interface{}, args ...interface{}) (interface{}, error) {
|
2015-04-05 15:03:16 -04:00
|
|
|
seqv := reflect.ValueOf(seq)
|
|
|
|
seqv, isNil := indirect(seqv)
|
|
|
|
if isNil {
|
|
|
|
return nil, errors.New("can't iterate over a nil value")
|
|
|
|
}
|
|
|
|
|
2015-08-07 07:05:18 -04:00
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice, reflect.Map:
|
|
|
|
// ok
|
|
|
|
default:
|
|
|
|
return nil, errors.New("can't sort " + reflect.ValueOf(seq).Type().String())
|
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
// Create a list of pairs that will be used to do the sort
|
2015-08-06 13:27:15 -04:00
|
|
|
p := pairList{SortAsc: true, SliceType: reflect.SliceOf(seqv.Type().Elem())}
|
2015-04-05 15:03:16 -04:00
|
|
|
p.Pairs = make([]pair, seqv.Len())
|
|
|
|
|
2015-08-07 07:05:18 -04:00
|
|
|
var sortByField string
|
2015-04-05 15:03:16 -04:00
|
|
|
for i, l := range args {
|
|
|
|
dStr, err := cast.ToStringE(l)
|
|
|
|
switch {
|
|
|
|
case i == 0 && err != nil:
|
2015-08-07 07:05:18 -04:00
|
|
|
sortByField = ""
|
2015-04-05 15:03:16 -04:00
|
|
|
case i == 0 && err == nil:
|
2015-08-07 07:05:18 -04:00
|
|
|
sortByField = dStr
|
2015-04-05 15:03:16 -04:00
|
|
|
case i == 1 && err == nil && dStr == "desc":
|
|
|
|
p.SortAsc = false
|
|
|
|
case i == 1:
|
|
|
|
p.SortAsc = true
|
|
|
|
}
|
|
|
|
}
|
2015-08-07 07:05:18 -04:00
|
|
|
path := strings.Split(strings.Trim(sortByField, "."), ".")
|
2015-04-05 15:03:16 -04:00
|
|
|
|
|
|
|
switch seqv.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
for i := 0; i < seqv.Len(); i++ {
|
|
|
|
p.Pairs[i].Key = reflect.ValueOf(i)
|
|
|
|
p.Pairs[i].Value = seqv.Index(i)
|
2015-08-07 07:05:18 -04:00
|
|
|
if sortByField == "" || sortByField == "value" {
|
|
|
|
p.Pairs[i].SortByValue = p.Pairs[i].Value
|
|
|
|
} else {
|
|
|
|
v := p.Pairs[i].Value
|
|
|
|
var err error
|
|
|
|
for _, elemName := range path {
|
|
|
|
v, err = evaluateSubElem(v, elemName)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
p.Pairs[i].SortByValue = v
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
case reflect.Map:
|
|
|
|
keys := seqv.MapKeys()
|
|
|
|
for i := 0; i < seqv.Len(); i++ {
|
|
|
|
p.Pairs[i].Key = keys[i]
|
|
|
|
p.Pairs[i].Value = seqv.MapIndex(keys[i])
|
2015-08-07 07:05:18 -04:00
|
|
|
if sortByField == "" {
|
|
|
|
p.Pairs[i].SortByValue = p.Pairs[i].Key
|
|
|
|
} else if sortByField == "value" {
|
|
|
|
p.Pairs[i].SortByValue = p.Pairs[i].Value
|
|
|
|
} else {
|
|
|
|
v := p.Pairs[i].Value
|
|
|
|
var err error
|
|
|
|
for _, elemName := range path {
|
|
|
|
v, err = evaluateSubElem(v, elemName)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
p.Pairs[i].SortByValue = v
|
|
|
|
}
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
}
|
2015-08-06 13:27:15 -04:00
|
|
|
return p.sort(), nil
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
// Credit for pair sorting method goes to Andrew Gerrand
|
|
|
|
// https://groups.google.com/forum/#!topic/golang-nuts/FT7cjmcL7gw
|
|
|
|
// A data structure to hold a key/value pair.
|
|
|
|
type pair struct {
|
2015-08-07 07:05:18 -04:00
|
|
|
Key reflect.Value
|
|
|
|
Value reflect.Value
|
|
|
|
SortByValue reflect.Value
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
// A slice of pairs that implements sort.Interface to sort by Value.
|
|
|
|
type pairList struct {
|
2015-08-07 07:05:18 -04:00
|
|
|
Pairs []pair
|
|
|
|
SortAsc bool
|
|
|
|
SliceType reflect.Type
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p pairList) Swap(i, j int) { p.Pairs[i], p.Pairs[j] = p.Pairs[j], p.Pairs[i] }
|
|
|
|
func (p pairList) Len() int { return len(p.Pairs) }
|
|
|
|
func (p pairList) Less(i, j int) bool {
|
2015-08-07 07:05:18 -04:00
|
|
|
return Lt(p.Pairs[i].SortByValue.Interface(), p.Pairs[j].SortByValue.Interface())
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
// sorts a pairList and returns a slice of sorted values
|
2015-08-06 13:27:15 -04:00
|
|
|
func (p pairList) sort() interface{} {
|
2015-04-05 15:03:16 -04:00
|
|
|
if p.SortAsc {
|
|
|
|
sort.Sort(p)
|
|
|
|
} else {
|
|
|
|
sort.Sort(sort.Reverse(p))
|
|
|
|
}
|
2015-08-06 13:27:15 -04:00
|
|
|
sorted := reflect.MakeSlice(p.SliceType, len(p.Pairs), len(p.Pairs))
|
2015-04-05 15:03:16 -04:00
|
|
|
for i, v := range p.Pairs {
|
2015-08-06 13:27:15 -04:00
|
|
|
sorted.Index(i).Set(v.Value)
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
2015-08-06 13:27:15 -04:00
|
|
|
return sorted.Interface()
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func IsSet(a interface{}, key interface{}) bool {
|
|
|
|
av := reflect.ValueOf(a)
|
|
|
|
kv := reflect.ValueOf(key)
|
|
|
|
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.Array, reflect.Chan, reflect.Slice:
|
|
|
|
if int64(av.Len()) > kv.Int() {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
case reflect.Map:
|
|
|
|
if kv.Type() == av.Type().Key() {
|
|
|
|
return av.MapIndex(kv).IsValid()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
func ReturnWhenSet(a, k interface{}) interface{} {
|
|
|
|
av, isNil := indirect(reflect.ValueOf(a))
|
|
|
|
if isNil {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
|
|
|
var avv reflect.Value
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.Array, reflect.Slice:
|
|
|
|
index, ok := k.(int)
|
|
|
|
if ok && av.Len() > index {
|
|
|
|
avv = av.Index(index)
|
|
|
|
}
|
|
|
|
case reflect.Map:
|
|
|
|
kv := reflect.ValueOf(k)
|
|
|
|
if kv.Type().AssignableTo(av.Type().Key()) {
|
|
|
|
avv = av.MapIndex(kv)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if avv.IsValid() {
|
|
|
|
switch avv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
return avv.Int()
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
return avv.Uint()
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
return avv.Float()
|
|
|
|
case reflect.String:
|
|
|
|
return avv.String()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2015-04-15 14:31:05 -04:00
|
|
|
func Highlight(in interface{}, lang, opts string) template.HTML {
|
2015-04-05 15:03:16 -04:00
|
|
|
var str string
|
|
|
|
av := reflect.ValueOf(in)
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
str = av.String()
|
|
|
|
}
|
|
|
|
|
2015-04-15 14:31:05 -04:00
|
|
|
return template.HTML(helpers.Highlight(html.UnescapeString(str), lang, opts))
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
var markdownTrimPrefix = []byte("<p>")
|
|
|
|
var markdownTrimSuffix = []byte("</p>\n")
|
|
|
|
|
|
|
|
func Markdownify(text string) template.HTML {
|
|
|
|
m := helpers.RenderBytes(&helpers.RenderingContext{Content: []byte(text), PageFmt: "markdown"})
|
|
|
|
m = bytes.TrimPrefix(m, markdownTrimPrefix)
|
|
|
|
m = bytes.TrimSuffix(m, markdownTrimSuffix)
|
|
|
|
return template.HTML(m)
|
|
|
|
}
|
|
|
|
|
|
|
|
func refPage(page interface{}, ref, methodName string) template.HTML {
|
|
|
|
value := reflect.ValueOf(page)
|
|
|
|
|
|
|
|
method := value.MethodByName(methodName)
|
|
|
|
|
|
|
|
if method.IsValid() && method.Type().NumIn() == 1 && method.Type().NumOut() == 2 {
|
|
|
|
result := method.Call([]reflect.Value{reflect.ValueOf(ref)})
|
|
|
|
|
|
|
|
url, err := result[0], result[1]
|
|
|
|
|
|
|
|
if !err.IsNil() {
|
|
|
|
jww.ERROR.Printf("%s", err.Interface())
|
|
|
|
return template.HTML(fmt.Sprintf("%s", err.Interface()))
|
|
|
|
}
|
|
|
|
|
|
|
|
if url.String() == "" {
|
|
|
|
jww.ERROR.Printf("ref %s could not be found\n", ref)
|
|
|
|
return template.HTML(ref)
|
|
|
|
}
|
|
|
|
|
|
|
|
return template.HTML(url.String())
|
|
|
|
}
|
|
|
|
|
|
|
|
jww.ERROR.Printf("Can only create references from Page and Node objects.")
|
|
|
|
return template.HTML(ref)
|
|
|
|
}
|
|
|
|
|
|
|
|
func Ref(page interface{}, ref string) template.HTML {
|
|
|
|
return refPage(page, ref, "Ref")
|
|
|
|
}
|
|
|
|
|
|
|
|
func RelRef(page interface{}, ref string) template.HTML {
|
|
|
|
return refPage(page, ref, "RelRef")
|
|
|
|
}
|
|
|
|
|
|
|
|
func Chomp(text interface{}) (string, error) {
|
|
|
|
s, err := cast.ToStringE(text)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return strings.TrimRight(s, "\r\n"), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Trim leading/trailing characters defined by b from a
|
|
|
|
func Trim(a interface{}, b string) (string, error) {
|
|
|
|
aStr, err := cast.ToStringE(a)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return strings.Trim(aStr, b), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Replace all occurences of b with c in a
|
|
|
|
func Replace(a, b, c interface{}) (string, error) {
|
|
|
|
aStr, err := cast.ToStringE(a)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
bStr, err := cast.ToStringE(b)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
cStr, err := cast.ToStringE(c)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return strings.Replace(aStr, bStr, cStr, -1), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// DateFormat converts the textual representation of the datetime string into
|
|
|
|
// the other form or returns it of the time.Time value. These are formatted
|
|
|
|
// with the layout string
|
|
|
|
func DateFormat(layout string, v interface{}) (string, error) {
|
|
|
|
t, err := cast.ToTimeE(v)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return t.Format(layout), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func SafeHTML(text string) template.HTML {
|
|
|
|
return template.HTML(text)
|
|
|
|
}
|
|
|
|
|
|
|
|
// "safeHTMLAttr" is currently disabled, pending further discussion
|
|
|
|
// on its use case. 2015-01-19
|
|
|
|
func SafeHTMLAttr(text string) template.HTMLAttr {
|
|
|
|
return template.HTMLAttr(text)
|
|
|
|
}
|
|
|
|
|
|
|
|
func SafeCSS(text string) template.CSS {
|
|
|
|
return template.CSS(text)
|
|
|
|
}
|
|
|
|
|
|
|
|
func SafeURL(text string) template.URL {
|
|
|
|
return template.URL(text)
|
|
|
|
}
|
|
|
|
|
|
|
|
func doArithmetic(a, b interface{}, op rune) (interface{}, error) {
|
|
|
|
av := reflect.ValueOf(a)
|
|
|
|
bv := reflect.ValueOf(b)
|
|
|
|
var ai, bi int64
|
|
|
|
var af, bf float64
|
|
|
|
var au, bu uint64
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
ai = av.Int()
|
|
|
|
switch bv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
bi = bv.Int()
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
af = float64(ai) // may overflow
|
|
|
|
ai = 0
|
|
|
|
bf = bv.Float()
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
bu = bv.Uint()
|
|
|
|
if ai >= 0 {
|
|
|
|
au = uint64(ai)
|
|
|
|
ai = 0
|
|
|
|
} else {
|
|
|
|
bi = int64(bu) // may overflow
|
|
|
|
bu = 0
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return nil, errors.New("Can't apply the operator to the values")
|
|
|
|
}
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
af = av.Float()
|
|
|
|
switch bv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
bf = float64(bv.Int()) // may overflow
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
bf = bv.Float()
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
bf = float64(bv.Uint()) // may overflow
|
|
|
|
default:
|
|
|
|
return nil, errors.New("Can't apply the operator to the values")
|
|
|
|
}
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
au = av.Uint()
|
|
|
|
switch bv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
bi = bv.Int()
|
|
|
|
if bi >= 0 {
|
|
|
|
bu = uint64(bi)
|
|
|
|
bi = 0
|
|
|
|
} else {
|
|
|
|
ai = int64(au) // may overflow
|
|
|
|
au = 0
|
|
|
|
}
|
|
|
|
case reflect.Float32, reflect.Float64:
|
|
|
|
af = float64(au) // may overflow
|
|
|
|
au = 0
|
|
|
|
bf = bv.Float()
|
|
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
|
bu = bv.Uint()
|
|
|
|
default:
|
|
|
|
return nil, errors.New("Can't apply the operator to the values")
|
|
|
|
}
|
|
|
|
case reflect.String:
|
|
|
|
as := av.String()
|
|
|
|
if bv.Kind() == reflect.String && op == '+' {
|
|
|
|
bs := bv.String()
|
|
|
|
return as + bs, nil
|
|
|
|
} else {
|
|
|
|
return nil, errors.New("Can't apply the operator to the values")
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return nil, errors.New("Can't apply the operator to the values")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch op {
|
|
|
|
case '+':
|
|
|
|
if ai != 0 || bi != 0 {
|
|
|
|
return ai + bi, nil
|
|
|
|
} else if af != 0 || bf != 0 {
|
|
|
|
return af + bf, nil
|
|
|
|
} else if au != 0 || bu != 0 {
|
|
|
|
return au + bu, nil
|
|
|
|
} else {
|
|
|
|
return 0, nil
|
|
|
|
}
|
|
|
|
case '-':
|
|
|
|
if ai != 0 || bi != 0 {
|
|
|
|
return ai - bi, nil
|
|
|
|
} else if af != 0 || bf != 0 {
|
|
|
|
return af - bf, nil
|
|
|
|
} else if au != 0 || bu != 0 {
|
|
|
|
return au - bu, nil
|
|
|
|
} else {
|
|
|
|
return 0, nil
|
|
|
|
}
|
|
|
|
case '*':
|
|
|
|
if ai != 0 || bi != 0 {
|
|
|
|
return ai * bi, nil
|
|
|
|
} else if af != 0 || bf != 0 {
|
|
|
|
return af * bf, nil
|
|
|
|
} else if au != 0 || bu != 0 {
|
|
|
|
return au * bu, nil
|
|
|
|
} else {
|
|
|
|
return 0, nil
|
|
|
|
}
|
|
|
|
case '/':
|
|
|
|
if bi != 0 {
|
|
|
|
return ai / bi, nil
|
|
|
|
} else if bf != 0 {
|
|
|
|
return af / bf, nil
|
|
|
|
} else if bu != 0 {
|
|
|
|
return au / bu, nil
|
|
|
|
} else {
|
|
|
|
return nil, errors.New("Can't divide the value by 0")
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return nil, errors.New("There is no such an operation")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func Mod(a, b interface{}) (int64, error) {
|
|
|
|
av := reflect.ValueOf(a)
|
|
|
|
bv := reflect.ValueOf(b)
|
|
|
|
var ai, bi int64
|
|
|
|
|
|
|
|
switch av.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
ai = av.Int()
|
|
|
|
default:
|
|
|
|
return 0, errors.New("Modulo operator can't be used with non integer value")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch bv.Kind() {
|
|
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
|
bi = bv.Int()
|
|
|
|
default:
|
|
|
|
return 0, errors.New("Modulo operator can't be used with non integer value")
|
|
|
|
}
|
|
|
|
|
|
|
|
if bi == 0 {
|
|
|
|
return 0, errors.New("The number can't be divided by zero at modulo operation")
|
|
|
|
}
|
|
|
|
|
|
|
|
return ai % bi, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func ModBool(a, b interface{}) (bool, error) {
|
|
|
|
res, err := Mod(a, b)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
return res == int64(0), nil
|
|
|
|
}
|
|
|
|
|
2015-09-12 15:45:12 -04:00
|
|
|
func Base64Decode(content interface{}) (string, error) {
|
|
|
|
conv, err := cast.ToStringE(content)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
dec, err := base64.StdEncoding.DecodeString(conv)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return string(dec), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func Base64Encode(content interface{}) (string, error) {
|
|
|
|
conv, err := cast.ToStringE(content)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return base64.StdEncoding.EncodeToString([]byte(conv)), nil
|
|
|
|
}
|
|
|
|
|
2015-04-05 15:03:16 -04:00
|
|
|
func init() {
|
|
|
|
funcMap = template.FuncMap{
|
2015-09-12 15:45:12 -04:00
|
|
|
"urlize": helpers.URLize,
|
|
|
|
"sanitizeURL": helpers.SanitizeURL,
|
|
|
|
"sanitizeurl": helpers.SanitizeURL,
|
|
|
|
"eq": Eq,
|
|
|
|
"ne": Ne,
|
|
|
|
"gt": Gt,
|
|
|
|
"ge": Ge,
|
|
|
|
"lt": Lt,
|
|
|
|
"le": Le,
|
2015-10-02 11:30:21 -04:00
|
|
|
"dict": Dictionary,
|
2015-09-12 15:45:12 -04:00
|
|
|
"in": In,
|
|
|
|
"slicestr": Slicestr,
|
|
|
|
"substr": Substr,
|
|
|
|
"split": Split,
|
|
|
|
"intersect": Intersect,
|
|
|
|
"isSet": IsSet,
|
|
|
|
"isset": IsSet,
|
|
|
|
"echoParam": ReturnWhenSet,
|
|
|
|
"safeHTML": SafeHTML,
|
|
|
|
"safeCSS": SafeCSS,
|
|
|
|
"safeURL": SafeURL,
|
|
|
|
"absURL": func(a string) template.HTML { return template.HTML(helpers.AbsURL(a)) },
|
|
|
|
"relURL": func(a string) template.HTML { return template.HTML(helpers.RelURL(a)) },
|
|
|
|
"markdownify": Markdownify,
|
|
|
|
"first": First,
|
|
|
|
"last": Last,
|
|
|
|
"after": After,
|
|
|
|
"where": Where,
|
|
|
|
"delimit": Delimit,
|
|
|
|
"sort": Sort,
|
|
|
|
"highlight": Highlight,
|
|
|
|
"add": func(a, b interface{}) (interface{}, error) { return doArithmetic(a, b, '+') },
|
|
|
|
"sub": func(a, b interface{}) (interface{}, error) { return doArithmetic(a, b, '-') },
|
|
|
|
"div": func(a, b interface{}) (interface{}, error) { return doArithmetic(a, b, '/') },
|
|
|
|
"mod": Mod,
|
|
|
|
"mul": func(a, b interface{}) (interface{}, error) { return doArithmetic(a, b, '*') },
|
|
|
|
"modBool": ModBool,
|
|
|
|
"lower": func(a string) string { return strings.ToLower(a) },
|
|
|
|
"upper": func(a string) string { return strings.ToUpper(a) },
|
|
|
|
"title": func(a string) string { return strings.Title(a) },
|
|
|
|
"partial": Partial,
|
|
|
|
"ref": Ref,
|
|
|
|
"relref": RelRef,
|
|
|
|
"apply": Apply,
|
|
|
|
"chomp": Chomp,
|
|
|
|
"replace": Replace,
|
|
|
|
"trim": Trim,
|
|
|
|
"dateFormat": DateFormat,
|
|
|
|
"getJSON": GetJSON,
|
|
|
|
"getCSV": GetCSV,
|
|
|
|
"readDir": ReadDir,
|
|
|
|
"seq": helpers.Seq,
|
|
|
|
"getenv": func(varName string) string { return os.Getenv(varName) },
|
|
|
|
"base64Decode": Base64Decode,
|
|
|
|
"base64Encode": Base64Encode,
|
2015-09-22 16:24:24 -04:00
|
|
|
"pluralize": func(in interface{}) (string, error) {
|
|
|
|
word, err := cast.ToStringE(in)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return inflect.Pluralize(word), nil
|
|
|
|
},
|
2015-09-22 16:31:02 -04:00
|
|
|
"singularize": func(in interface{}) (string, error) {
|
|
|
|
word, err := cast.ToStringE(in)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return inflect.Singularize(word), nil
|
|
|
|
},
|
2015-04-05 15:03:16 -04:00
|
|
|
}
|
|
|
|
}
|