/
helper.go
112 lines (101 loc) · 2.06 KB
/
helper.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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package generator
import (
"encoding/json"
"gopkg.in/yaml.v3"
"os"
"strings"
"unicode"
)
// ParseStruct function accepts a string and tries to parse it into a Struct
func ParseStruct(s string) (*Config, error) {
var Data Config
err := yaml.Unmarshal([]byte(s), &Data)
if err != nil {
err := json.Unmarshal([]byte(s), &Data)
if err != nil {
return nil, err
}
return nil, err
}
return &Data, nil
}
func ParseStructFromFileName(fileName string) (*Config, error) {
// Read the file
dat, err := os.ReadFile(fileName)
if err != nil {
return nil, err
}
return ParseStruct(string(dat))
}
func toCamelCase(name string) string {
words := strings.Split(name, "_")
result := ""
if name == "" {
return name
}
for _, word := range words {
result += strings.ToUpper(string(word[0])) + strings.ToLower(word[1:])
}
return result
}
func toTitleCase(s string) string {
// Use a closure here to remember state.
// Hackish but effective. Depends on Map scanning in order and calling
// the closure once per rune.
prev := ' '
return strings.Map(
func(r rune) rune {
if isSeparator(prev) {
prev = r
return unicode.ToTitle(r)
}
prev = r
return r
},
s)
}
func isSeparator(r rune) bool {
// ASCII alphanumerics and underscore are not separators
if r <= 0x7F {
switch {
case '0' <= r && r <= '9':
return false
case 'a' <= r && r <= 'z':
return false
case 'A' <= r && r <= 'Z':
return false
case r == '_':
return false
}
return true
}
// Letters and digits are not separators
if unicode.IsLetter(r) || unicode.IsDigit(r) {
return false
}
// Otherwise, all we can do for now is treat spaces as separators.
return unicode.IsSpace(r)
}
func ToSnakeCase(name string) string {
name = strings.Replace(name, " ", "_", -1)
name = strings.ToLower(name)
return name
}
func isValidType(fieldType string) bool {
validTypes := []string{
"string",
"int64",
"int32",
"int16",
"float64",
"bool",
"time.Time",
"struct",
}
for _, t := range validTypes {
if fieldType == t {
return true
}
}
return false
}