diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-kv/util.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-kv/util.go | 71 |
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 +} |