@@ -14,9 +14,6 @@ import (
14
14
"text/template"
15
15
)
16
16
17
- const processorsFolder = "processors"
18
- const modelsFolder = "models"
19
-
20
17
var ValidCaseNormalized = []string {"snakecase" , "camelcase" , "pascalcase" }
21
18
22
19
func Generate (routines []Routine , config * Config ) error {
@@ -64,19 +61,20 @@ func Generate(routines []Routine, config *Config) error {
64
61
}
65
62
66
63
func generateDbContext (routines []Routine , hashMap * map [string ]string , config * Config ) error {
67
- dbcontextTemplate , err := parseTemplates (config .DbContextTemplate )
64
+ dbContextTemplate , err := parseTemplate (config .DbContextTemplate )
68
65
if err != nil {
69
66
return fmt .Errorf ("loading dbContext template: %s" , err )
70
67
}
71
68
72
69
data := & DbContextData {
70
+ Config : config ,
73
71
Functions : routines ,
74
72
}
75
73
76
74
filename := changeCase ("DbContext" + config .GeneratedFileExtension , config .GeneratedFileCase )
77
75
fp := filepath .Join (config .OutputFolder , filename )
78
76
79
- changed , err := generateFile (data , dbcontextTemplate , fp , hashMap )
77
+ changed , err := generateFile (data , dbContextTemplate , fp , hashMap )
80
78
if err != nil {
81
79
return err
82
80
}
@@ -92,24 +90,28 @@ func generateDbContext(routines []Routine, hashMap *map[string]string, config *C
92
90
}
93
91
94
92
func generateModels (routines []Routine , hashMap * map [string ]string , config * Config ) error {
95
-
96
- moduleTemplate , err := parseTemplates (config .ModelTemplate )
93
+ moduleTemplate , err := parseTemplate (config .ModelTemplate )
97
94
if err != nil {
98
95
return fmt .Errorf ("loading module template: %s" , err )
99
96
}
100
97
101
- err = os .MkdirAll (filepath .Join (config .OutputFolder , modelsFolder ), 0777 )
98
+ err = os .MkdirAll (filepath .Join (config .OutputFolder , config . ModelsFolderName ), 0777 )
102
99
103
100
for _ , routine := range routines {
104
101
if ! routine .HasReturn {
105
102
continue
106
103
}
107
104
108
105
filename := changeCase (routine .ModelName + config .GeneratedFileExtension , config .GeneratedFileCase )
109
- relPath := filepath .Join (modelsFolder , filename )
106
+ relPath := filepath .Join (config . ModelsFolderName , filename )
110
107
filePath := filepath .Join (config .OutputFolder , relPath )
111
108
112
- changed , err := generateFile (routine , moduleTemplate , filePath , hashMap )
109
+ data := & ModelTemplateData {
110
+ Config : config ,
111
+ Routine : routine ,
112
+ }
113
+
114
+ changed , err := generateFile (data , moduleTemplate , filePath , hashMap )
113
115
if err != nil {
114
116
return fmt .Errorf ("generating models: %s" , err )
115
117
}
@@ -125,12 +127,12 @@ func generateModels(routines []Routine, hashMap *map[string]string, config *Conf
125
127
}
126
128
127
129
func generateProcessors (routines []Routine , hashMap * map [string ]string , config * Config ) error {
128
- processorTemplate , err := parseTemplates (config .ProcessorTemplate )
130
+ processorTemplate , err := parseTemplate (config .ProcessorTemplate )
129
131
if err != nil {
130
132
return fmt .Errorf ("loading processor template: %s" , err )
131
133
}
132
134
133
- err = os .MkdirAll (filepath .Join (config .OutputFolder , processorsFolder ), 0777 )
135
+ err = os .MkdirAll (filepath .Join (config .OutputFolder , config . ProcessorsFolderName ), 0777 )
134
136
if err != nil {
135
137
return fmt .Errorf ("creating processor output folder: %s" , err )
136
138
}
@@ -143,10 +145,15 @@ func generateProcessors(routines []Routine, hashMap *map[string]string, config *
143
145
}
144
146
145
147
filename := changeCase (routine .ProcessorName + config .GeneratedFileExtension , config .GeneratedFileCase )
146
- relPath := filepath .Join (processorsFolder , filename )
148
+ relPath := filepath .Join (config . ProcessorsFolderName , filename )
147
149
filePath := filepath .Join (config .OutputFolder , relPath )
148
150
149
- changed , err := generateFile (routine , processorTemplate , filePath , hashMap )
151
+ data := & ProcessorTemplateData {
152
+ Config : config ,
153
+ Routine : routine ,
154
+ }
155
+
156
+ changed , err := generateFile (data , processorTemplate , filePath , hashMap )
150
157
if err != nil {
151
158
return fmt .Errorf ("generating processor %s: %s" , routine .ProcessorName , err )
152
159
}
@@ -161,7 +168,7 @@ func generateProcessors(routines []Routine, hashMap *map[string]string, config *
161
168
return nil
162
169
}
163
170
164
- func parseTemplates (templatePath string ) (* template.Template , error ) {
171
+ func parseTemplate (templatePath string ) (* template.Template , error ) {
165
172
if ! common .PathExists (templatePath ) {
166
173
return nil , fmt .Errorf ("template file %s does not exist" , templatePath )
167
174
0 commit comments