wp-go/helper/strings/strings.go
2023-05-04 19:46:06 +08:00

121 lines
2.2 KiB
Go

package strings
import (
"crypto/md5"
"fmt"
"golang.org/x/exp/constraints"
"io"
"strconv"
"strings"
"unicode"
)
func Join(s ...string) (str string) {
if len(s) == 1 {
return s[0]
} else if len(s) > 1 {
b := strings.Builder{}
for _, s2 := range s {
b.WriteString(s2)
}
str = b.String()
}
return
}
func FirstUpper(s string) string {
if len(s) < 1 {
return s
}
return Join(strings.ToUpper(s[:1]), s[1:])
}
func ToInteger[T constraints.Integer](s string, defaults T) T {
if s == "" {
return defaults
}
i, err := strconv.ParseInt(s, 10, 64)
if err != nil {
return defaults
}
return T(i)
}
func ToInt[T constraints.Integer](s string) T {
defaults := T(0)
if s == "" {
return defaults
}
i, err := strconv.ParseInt(s, 10, 64)
if err != nil {
return defaults
}
return T(i)
}
func Md5(str string) string {
h := md5.New()
_, err := io.WriteString(h, str)
if err != nil {
return ""
}
return fmt.Sprintf("%x", h.Sum(nil))
}
func BuilderJoin(s *strings.Builder, str ...string) {
for _, ss := range str {
s.WriteString(ss)
}
}
func BuilderFormat(s *strings.Builder, format string, args ...any) {
s.WriteString(fmt.Sprintf(format, args...))
}
type Builder struct {
*strings.Builder
}
func NewBuilder() *Builder {
return &Builder{&strings.Builder{}}
}
func (b *Builder) WriteString(s ...string) (count int) {
for _, ss := range s {
i, _ := b.Builder.WriteString(ss)
count += i
}
return
}
func Replace(s string, replace map[string]string) string {
for k, v := range replace {
s = strings.ReplaceAll(s, k, v)
}
return s
}
func (b *Builder) Sprintf(format string, a ...any) int {
i, _ := fmt.Fprintf(b, format, a...)
return i
}
// CutSpecialDuplicate '\t', '\n', '\v', '\f', '\r', ' ', U+0085 (NEL), U+00A0 (NBSP)
func CutSpecialDuplicate(s, char string) string {
return strings.Join(strings.Fields(s), char)
}
// CamelCaseTo 驼峰单词转下划线或横杠单词 //分隔符
func CamelCaseTo(s string, delimiter rune) string {
var output []rune
for i, r := range s {
if i == 0 {
output = append(output, unicode.ToLower(r))
} else {
if unicode.IsUpper(r) {
output = append(output, delimiter)
}
output = append(output, unicode.ToLower(r))
}
}
return string(output)
}