Add more functions to strings.tsh

Author: monstermichl

std/strings.tsh

@@ -59,6 +59,26 @@ func HasSuffix(s string, suffix string) bool {
 	return false
 }
 
+func Count(s string, substr string) int {
+	lenS := len(s)
+	lenSub := len(substr)
+
+	if lenSub == 0 {
+		return lenS + 1
+	}
+	c := 0
+
+	for i := 0; i < lenS; {
+		if HasPrefix(s[i:], substr) {
+			c++
+			i += lenSub
+		} else {
+			i++
+		}
+	}
+	return c
+}
+
 func Split(s string, sep string) []string {
 	sLen := len(s)
 	sepLen := len(sep)
@@ -101,3 +121,140 @@ func Split(s string, sep string) []string {
 	}
 	return elems
 }
+
+func Repeat(s string, count int) string {
+	new := s
+
+	for i := 1; i < count; i++ {
+		new += s
+	}
+	return new
+}
+
+func Replace(s string, old string, new string, n int) string {
+	res := ""
+	rep := 0
+	i := 0
+	lenOld := len(old)
+	lenNew := len(new)
+
+	if lenOld == 0 {
+		res = new
+		rep++
+	}
+
+	for i < len(s) && (rep < n || n < 0) {
+		c := s[i]
+
+		if lenOld == 0 {
+			res += c + new
+			i++
+		} else if HasPrefix(s[i:], old) {
+			res += new
+			i += lenOld
+		} else {
+			res += c
+			i++
+			continue
+		}
+		rep++
+	}
+	return res + s[i:]
+}
+
+func ReplaceAll(s string, old string, new string) string {
+	return Replace(s, old, new, -1)
+}
+
+func CutPrefix(s string, prefix string) (string, bool) {
+	if HasPrefix(s, prefix) {
+		return s[len(prefix):], true
+	}
+	return s, false
+}
+
+func CutSuffix(s string, suffix string) (string, bool) {
+	if HasSuffix(s, suffix) {
+		return s[0 : len(s)-len(suffix)], true
+	}
+	return s, false
+}
+
+func Cut(s string, sep string) (string, string, bool) {
+	lenS := len(s)
+
+	if lenS > 0 {
+		for i := 0; i < lenS; i++ {
+			if HasPrefix(s[i:], sep) {
+				return s[0:i], s[i+len(sep):], true
+			}
+		}
+	}
+	return s, "", false
+}
+
+func TrimPrefix(s string, prefix string) string {
+	s, c := CutPrefix(s, prefix)
+	return s
+}
+
+func TrimSuffix(s string, suffix string) string {
+	s, c := CutSuffix(s, suffix)
+	return s
+}
+
+func TrimLeft(s string, cutset string) string {
+	lenCS := len(cutset)
+
+	if len(s) > 0 && lenCS > 0 {
+		for {
+			trimmed := false
+
+			for i := 0; i < lenCS; i++ {
+				lenS := len(s)
+				s, cut := CutPrefix(s, cutset[i])
+
+				if cut {
+					trimmed = true
+				}
+			}
+
+			if !trimmed {
+				break
+			}
+		}
+	}
+	return s
+}
+
+func TrimRight(s string, cutset string) string {
+	lenCS := len(cutset)
+
+	if len(s) > 0 && lenCS > 0 {
+		for {
+			trimmed := false
+
+			for i := 0; i < lenCS; i++ {
+				lenS := len(s)
+				s, cut := CutSuffix(s, cutset[i])
+
+				if cut {
+					trimmed = true
+				}
+			}
+
+			if !trimmed {
+				break
+			}
+		}
+	}
+	return s
+}
+
+func Trim(s string, cutset string) string {
+	return TrimRight(TrimLeft(s, cutset), cutset)
+}
+
+func TrimSpace(s string) string {
+	return Trim(s, "\t\n\v\f\r ")
+}