summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-kv/util.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/codeberg.org/gruf/go-kv/util.go')
-rw-r--r--vendor/codeberg.org/gruf/go-kv/util.go71
1 files changed, 71 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-kv/util.go b/vendor/codeberg.org/gruf/go-kv/util.go
new file mode 100644
index 000000000..a9526bf3d
--- /dev/null
+++ b/vendor/codeberg.org/gruf/go-kv/util.go
@@ -0,0 +1,71 @@
+package kv
+
+import (
+ "strconv"
+
+ "codeberg.org/gruf/go-byteutil"
+ "codeberg.org/gruf/go-kv/format"
+)
+
+// appendQuoteKey will append and escape/quote a formatted key string.
+func appendQuoteKey(buf *byteutil.Buffer, str string) {
+ switch {
+ case !strconv.CanBackquote(str):
+ // Append quoted and escaped string
+ buf.B = strconv.AppendQuote(buf.B, str)
+ case format.ContainsDoubleQuote(str):
+ // Double quote and escape string
+ buf.B = append(buf.B, '"')
+ buf.B = format.AppendEscape(buf.B, str)
+ buf.B = append(buf.B, '"')
+ case len(str) < 1 || format.ContainsSpaceOrTab(str):
+ // Double quote this string as-is
+ buf.WriteString(`"` + str + `"`)
+ default:
+ // Append string as-is
+ buf.WriteString(str)
+ }
+}
+
+// appendQuoteValue will append and escape/quote a formatted value string.
+func appendQuoteValue(buf *byteutil.Buffer, str string) {
+ switch {
+ case !strconv.CanBackquote(str):
+ // Append quoted and escaped string
+ buf.B = strconv.AppendQuote(buf.B, str)
+ return
+ case !doubleQuoted(str):
+ if format.ContainsDoubleQuote(str) {
+ // Double quote and escape string
+ buf.B = append(buf.B, '"')
+ buf.B = format.AppendEscape(buf.B, str)
+ buf.B = append(buf.B, '"')
+ return
+ } else if format.ContainsSpaceOrTab(str) {
+ // Double quote this string as-is
+ buf.WriteString(`"` + str + `"`)
+ return
+ }
+ }
+
+ // Append string as-is
+ buf.WriteString(str)
+}
+
+// doubleQuoted will return whether 'str' is double quoted.
+func doubleQuoted(str string) bool {
+ if len(str) < 2 ||
+ str[0] != '"' || str[len(str)-1] != '"' {
+ return false
+ }
+ var delim bool
+ for i := len(str) - 2; i >= 0; i-- {
+ switch str[i] {
+ case '\\':
+ delim = !delim
+ default:
+ return !delim
+ }
+ }
+ return !delim
+}