-
Notifications
You must be signed in to change notification settings - Fork 48
/
util.go
76 lines (65 loc) · 2.34 KB
/
util.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
// Copyright (c) 2020 Mercari, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of
// this software and associated documentation files (the "Software"), to deal in
// the Software without restriction, including without limitation the rights to
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
// the Software, and to permit persons to whom the Software is furnished to do so,
// subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
package internal
import (
"fmt"
"strings"
"github.com/kenshaw/snaker"
)
// reverseIndexRune finds the last rune r in s, returning -1 if not present.
func reverseIndexRune(s string, r rune) int {
if s == "" {
return -1
}
rs := []rune(s)
for i := len(rs) - 1; i >= 0; i-- {
if rs[i] == r {
return i
}
}
return -1
}
// SinguralizeIdentifier will singularize a identifier, returning it in
// CamelCase.
func SingularizeIdentifier(in Inflector, s string) string {
if i := reverseIndexRune(s, '_'); i != -1 {
s = s[:i] + "_" + in.Singularize(s[i+1:])
} else {
s = in.Singularize(s)
}
return snaker.ForceCamelIdentifier(s)
}
// SnakeToCamel converts the string to CamelCase
func SnakeToCamel(s string) string {
return snaker.ForceCamelIdentifier(s)
}
// CamelToSnake converts the string to snake_case
func CamelToScake(s string) string {
return snaker.CamelToSnakeIdentifier(s)
}
// EscapeColumnName will escape a column name if using reserved keyword as column name, returning it in
// surrounded back quotes.
func EscapeColumnName(s string) string {
if _, ok := reservedKeywords[strings.ToUpper(s)]; ok {
// return surrounded s with back quotes if reserved keyword
return fmt.Sprintf("`%s`", s)
}
// return s if not reserved keyword
return s
}