utils pkg unify

pkg/util/stringutils2/stringutils.go

@@ -0,0 +1,100 @@
+package stringutils2
+
+import (
+	"fmt"
+	"strings"
+)
+
+func EscapeString(str string, pairs [][]string) string {
+	if len(pairs) == 0 {
+		pairs = [][]string{
+			{"\\", `\\`},
+			{"\n", `\n`},
+			{"\r", `\r`},
+			{"\t", `\t`},
+			{`"`, `\"`},
+			{"'", `\'`},
+			{"$", `\$`},
+		}
+	}
+	for _, pair := range pairs {
+		k, v := pair[0], pair[1]
+		str = strings.Replace(str, k, v, -1)
+	}
+	return str
+}
+
+func EscapeEchoString(str string) (string, error) {
+	pairs := [][]string{
+		{"\\", `\\`},
+		{"\n", `\n`},
+		{"\r", `\r`},
+		{"\t", `\t`},
+		{`"`, `\"`},
+	}
+	innerPairs := [][]string{
+		{"\\", `\\`},
+		{"\n", `\n`},
+		{"\r", `\r`},
+		{"\t", `\t`},
+		{`"`, `\"`},
+		{"$", `\$`},
+	}
+	segs, err := SplitByQuotation(str)
+	if err != nil {
+		return "", err
+	}
+	ret := ""
+	for idx := 0; idx < len(segs); idx++ {
+		s := EscapeString(segs[idx], innerPairs)
+		if idx%2 == 1 {
+			s = EscapeString(segs[idx], pairs)
+			s = `\"` + EscapeString(s, innerPairs) + `\"`
+		}
+		ret += s
+	}
+	return ret, nil
+}
+
+func findQuotationPos(line string, offset int) int {
+	if offset > len(line) {
+		return -1
+	}
+	subStr := line[offset:]
+	pos := strings.Index(subStr, `"`)
+	if pos < 0 {
+		return pos
+	}
+	pos += offset
+	if pos > 0 && line[pos-1] == '\\' {
+		return findQuotationPos(line, pos+1)
+	}
+	return pos
+}
+
+func SplitByQuotation(line string) ([]string, error) {
+	segs := []string{}
+	offset := 0
+	for offset < len(line) {
+		pos := findQuotationPos(line, offset)
+		if pos < 0 {
+			segs = append(segs, line[offset:])
+			offset = len(line)
+		} else {
+			if pos == offset {
+				offset += 1
+			} else {
+				segs = append(segs, line[offset:pos])
+				offset = pos + 1
+			}
+			pos = findQuotationPos(line, offset)
+			if pos < 0 {
+				return nil, fmt.Errorf("Unpaired quotations: %s", line[offset:])
+			} else {
+				segs = append(segs, line[offset:pos])
+				offset = pos + 1
+			}
+		}
+	}
+	return segs, nil
+}