filters.go

package filter

import (
	"net/url"
	"strings"

	"github.com/gookit/goutil/arrutil"
	"github.com/gookit/goutil/strutil"
)

var dontLimitType = map[string]uint8{
	"int":    1,
	"uint":   1,
	"int64":  1,
	"float":  1,
	"unique": 1,
	// list
	"trimStrings":   1,
	"stringsToInts": 1,
}

var filterAliases = map[string]string{
	"toInt":   "int",
	"toUint":  "uint",
	"toInt64": "int64",
	"toBool":  "bool",
	"camel":   "camelCase",
	"snake":   "snakeCase",
	"ltrim":   "trimLeft",
	"rtrim":   "trimRight",
	// --
	"lcFirst":      "lowerFirst",
	"ucFirst":      "upperFirst",
	"ucWord":       "upperWord",
	"distinct":     "unique",
	"trimList":     "trimStrings",
	"trim_list":    "trimStrings",
	"trim_strings": "trimStrings",
	"trimSpace":    "trim",
	"trim_space":   "trim",
	"uppercase":    "upper",
	"lowercase":    "lower",
	"escapeJs":     "escapeJS",
	"escape_js":    "escapeJS",
	"escapeHtml":   "escapeHTML",
	"escape_html":  "escapeHTML",
	"urlEncode":    "URLEncode",
	"url_encode":   "URLEncode",
	"encodeUrl":    "URLEncode",
	"encode_url":   "URLEncode",
	"urlDecode":    "URLDecode",
	"url_decode":   "URLDecode",
	"decodeUrl":    "URLDecode",
	"decode_url":   "URLDecode",
	// convert string
	"str2ints":  "strToInts",
	"str2arr":   "strToSlice",
	"str2list":  "strToSlice",
	"str2array": "strToSlice",
	"strToArr":  "strToSlice",
	"str2time":  "strToTime",
	// strings to ints
	"strings2ints": "stringsToInts",
}

// Name get real filter name.
func Name(name string) string {
	if rName, ok := filterAliases[name]; ok {
		return rName
	}
	return name
}

/*************************************************************
 * built in filters
 *************************************************************/

// Trim string
func Trim(s string, cutSet ...string) string {
	if len(cutSet) > 0 && cutSet[0] != "" {
		return strings.Trim(s, cutSet[0])
	}
	return strings.TrimSpace(s)
}

// TrimLeft char in the string.
func TrimLeft(s string, cutSet ...string) string {
	if len(cutSet) > 0 {
		return strings.TrimLeft(s, cutSet[0])
	}

	return strings.TrimLeft(s, " ")
}

// TrimRight char in the string.
func TrimRight(s string, cutSet ...string) string {
	if len(cutSet) > 0 {
		return strings.TrimRight(s, cutSet[0])
	}
	return strings.TrimRight(s, " ")
}

// TrimStrings trim string slice item.
func TrimStrings(ss []string, cutSet ...string) []string {
	return arrutil.TrimStrings(ss, cutSet...)
}

// URLEncode encode url string.
func URLEncode(s string) string {
	if pos := strings.IndexRune(s, '?'); pos > -1 { // escape query data
		return s[0:pos+1] + url.QueryEscape(s[pos+1:])
	}
	return s
}

// URLDecode decode url string.
func URLDecode(s string) string {
	if pos := strings.IndexRune(s, '?'); pos > -1 { // un-escape query data
		qy, err := url.QueryUnescape(s[pos+1:])
		if err == nil {
			return s[0:pos+1] + qy
		}
	}
	return s
}

// Unique value in the given array, slice.
func Unique(val any) any {
	switch tv := val.(type) {
	case []int:
		mp := make(map[int]int)
		for _, sVal := range tv {
			mp[sVal] = 1
		}

		// no repeat value
		if len(tv) == len(mp) {
			return tv
		}

		var ns []int
		for sVal := range mp {
			ns = append(ns, sVal)
		}
		return ns
	case []int64:
		mp := make(map[int64]int)
		for _, sVal := range tv {
			mp[sVal] = 1
		}

		// no repeat value
		if len(tv) == len(mp) {
			return tv
		}

		var ns []int64
		for sVal := range mp {
			ns = append(ns, sVal)
		}
		return ns
	case []string:
		mp := make(map[string]int)
		for _, sVal := range tv {
			mp[sVal] = 1
		}

		// no repeat value
		if len(tv) == len(mp) {
			return tv
		}

		var ns []string
		for sVal := range mp {
			ns = append(ns, sVal)
		}
		return ns
	}

	return val
}

// Substr cut string
func Substr(s string, pos, length int) string {
	return strutil.Substr(s, pos, length)
}

// Email filter, clear invalid chars.
func Email(s string) string {
	return strutil.FilterEmail(s)
}