helpers.go

package core

import (
	"fmt"
	"strconv"
)

// return the position of the first none space, or -1 if no white space exists
func SkipSpaces(data []byte) int {
	for index, b := range data {
		if b != ' ' {
			return index
		}
	}
	return -1
}

// Since these templates are possibly long-lived, let's free up any space
// which was accumulated while we grew these arrays
func TrimStrings(values []string) []string {
	if len(values) == cap(values) {
		return values
	}
	trimmed := make([]string, len(values))
	copy(trimmed, values)
	return trimmed
}

// Since these templates are possibly long-lived, let's free up any space
// which was accumulated while we grew these arrays
func TrimValues(values []Value) []Value {
	if len(values) == cap(values) {
		return values
	}
	trimmed := make([]Value, len(values))
	copy(trimmed, values)
	return trimmed
}

// Convert arbitrary data to []byte
func ToBytes(data interface{}) []byte {
	switch typed := data.(type) {
	case []byte:
		return typed
	case string:
		return []byte(typed)
	case bool:
		return []byte(strconv.FormatBool(typed))
	case float64:
		return []byte(strconv.FormatFloat(typed, 'g', -1, 64))
	case uint64:
		return []byte(strconv.FormatUint(typed, 10))
	case uint:
		return []byte(strconv.FormatUint(uint64(typed), 10))
	case int:
		return []byte(strconv.Itoa(typed))
	case fmt.Stringer:
		return []byte(typed.String())
	}
	return []byte(fmt.Sprintf("%v", data))
}

// Convert arbitrary data to string
func ToString(data interface{}) string {
	switch typed := data.(type) {
	case string:
		return typed
	case fmt.Stringer:
		return typed.String()
	default:
		return string(ToBytes(data))
	}
}

// Convert arbitrary data to string
func ToInt(data interface{}) (int, bool) {
	switch typed := data.(type) {
	case int:
		return typed, true
	case int32:
		return int(typed), true
	case int64:
		return int(typed), true
	case uint:
		return int(typed), true
	case float64:
		return int(typed), true
	case string:
		return stringToInt(typed)
	case []byte:
		return stringToInt(string(typed))
	default:
		return 0, false
	}
}

func stringToInt(s string) (int, bool) {
	if n, err := strconv.Atoi(s); err == nil {
		return n, true
	}
	return 0, false
}