mirror of https://github.com/gobeam/stringy.git
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package stringy
|
|
|
|
import (
|
|
"errors"
|
|
"regexp"
|
|
"strings"
|
|
"unicode"
|
|
)
|
|
|
|
var selectCapitalRegexp = regexp.MustCompile(SelectCapital)
|
|
|
|
func caseHelper(input string, isCamel bool, rule ...string) []string {
|
|
if !isCamel {
|
|
input = selectCapitalRegexp.ReplaceAllString(input, ReplaceCapital)
|
|
}
|
|
input = strings.Join(strings.Fields(strings.TrimSpace(input)), " ")
|
|
if len(rule) > 0 && len(rule)%2 != 0 {
|
|
panic(errors.New(OddError))
|
|
}
|
|
rule = append(rule, ".", " ", "_", " ", "-", " ")
|
|
|
|
replacer := strings.NewReplacer(rule...)
|
|
input = replacer.Replace(input)
|
|
words := strings.Fields(input)
|
|
return words
|
|
}
|
|
|
|
func contains(slice []string, item string) bool {
|
|
set := make(map[string]struct{}, len(slice))
|
|
for _, s := range slice {
|
|
set[s] = struct{}{}
|
|
}
|
|
_, ok := set[item]
|
|
return ok
|
|
}
|
|
|
|
func getInput(i input) (input string) {
|
|
if i.Result != "" {
|
|
input = i.Result
|
|
} else {
|
|
input = i.Input
|
|
}
|
|
return
|
|
}
|
|
|
|
func replaceStr(input, search, replace, types string) string {
|
|
lcInput := strings.ToLower(input)
|
|
lcSearch := strings.ToLower(search)
|
|
if input == "" || !strings.Contains(lcInput, lcSearch) {
|
|
return input
|
|
}
|
|
var start int
|
|
if types == "last" {
|
|
start = strings.LastIndex(lcInput, lcSearch)
|
|
} else {
|
|
start = strings.Index(lcInput, lcSearch)
|
|
}
|
|
end := start + len(search)
|
|
return input[:start] + replace + input[end:]
|
|
}
|
|
|
|
func ucfirst(val string) string {
|
|
for _, v := range val {
|
|
return string(unicode.ToUpper(v)) + val[len(string(v)):]
|
|
}
|
|
return ""
|
|
}
|