-
Notifications
You must be signed in to change notification settings - Fork 4
/
utils.go
78 lines (64 loc) · 1.32 KB
/
utils.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
66
67
68
69
70
71
72
73
74
75
76
77
78
package bulk
import (
"fmt"
"strings"
)
func toSnakeCase(input string) string {
var sb strings.Builder
var upperCount int
for i, c := range input {
switch {
case isUppercase(c):
if i > 0 && (upperCount == 0 || nextIsLower(input, i)) {
sb.WriteByte('_')
}
sb.WriteByte(byte(c - 'A' + 'a'))
upperCount++
case isLowercase(c):
sb.WriteByte(byte(c))
upperCount = 0
case isDigit(c):
if i == 0 {
panic("Identifier must start with a character: `" + input + "`")
}
sb.WriteByte(byte(c))
default:
panic("Invalid identifier: `" + input + "`")
}
}
return sb.String()
}
// nextIsLower The next character is lower case, but not the last 's'.
// nextIsLower("HTMLFile", 1) expected: "html_file"
// => true
// nextIsLower("URLs", -1) expected: "urls"
// => false
func nextIsLower(input string, i int) bool {
i++
if i >= len(input) {
return false
}
c := input[i]
if c == 's' && i == len(input)-1 {
return false
}
return isLowercase(rune(c))
}
func isDigit(r rune) bool {
return r >= '0' && r <= '9'
}
func isLowercase(r rune) bool {
return r >= 'a' && r <= 'z'
}
func isUppercase(r rune) bool {
return r >= 'A' && r <= 'Z'
}
func escapeSqlName(name string) string {
return fmt.Sprintf("`%s`", strings.Trim(name, "`"))
}
func min(x, y int) int {
if x < y {
return x
}
return y
}