-
Notifications
You must be signed in to change notification settings - Fork 93
/
stringutil.go
65 lines (50 loc) · 1.01 KB
/
stringutil.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package stringutil
import (
"strings"
"unicode/utf8"
)
// NeedsQuotes reports whether s should be wrapped with quote
// before being used as a document key.
// If it returns true, s must be used with strconv.Quote or
// using the fmt '%q' formatter.
func NeedsQuotes(s string) bool {
if s == "" {
return true
}
for len(s) > 0 {
r, wid := utf8.DecodeRuneInString(s)
if wid > 1 {
return true
}
if r == utf8.RuneError {
return true
}
if ('0' <= r && r <= '9') || ('a' <= r && r <= 'z') || ('A' <= r && r <= 'Z') || r == '_' {
s = s[wid:]
continue
}
return true
}
return false
}
// NormalizeIdentifier wraps s around the given quotes, if needed.
func NormalizeIdentifier(s string, with rune) string {
if s == "" {
return s
}
if !NeedsQuotes(s) {
return s
}
var sb strings.Builder
sb.WriteRune(with)
for len(s) > 0 {
r, wid := utf8.DecodeRuneInString(s)
if r == with {
sb.WriteRune('\\')
}
sb.WriteRune(r)
s = s[wid:]
}
sb.WriteRune(with)
return sb.String()
}