string.go

package yiigo

import "strings"

// AddSlashes 在字符串的每个引号前添加反斜杠
func AddSlashes(s string) string {
	var builder strings.Builder
	for _, ch := range s {
		if ch == '\'' || ch == '"' || ch == '\\' {
			builder.WriteRune('\\')
		}
		builder.WriteRune(ch)
	}
	return builder.String()
}

// StripSlashes 删除字符串中的反斜杠
func StripSlashes(s string) string {
	l, skip := len(s), false

	var builder strings.Builder
	for i, ch := range s {
		if skip {
			builder.WriteRune(ch)
			skip = false
			continue
		}
		if ch == '\\' {
			if i+1 < l && s[i+1] == '\\' {
				skip = true
			}
			continue
		}
		builder.WriteRune(ch)
	}
	return builder.String()
}

// QuoteMeta 在字符串预定义的字符前添加反斜杠
func QuoteMeta(s string) string {
	var builder strings.Builder
	for _, ch := range s {
		switch ch {
		case '.', '+', '\\', '(', '$', ')', '[', '^', ']', '*', '?':
			builder.WriteRune('\\')
		}
		builder.WriteRune(ch)
	}
	return builder.String()
}