mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
f73d049392
## What changed? Refactor code generators: 1. Extracted templates to separate files. 2. Extracted common helpers to `codegen` package. 3. Simplified code generators. All generated files are almost unchanged (removed leading empty line). ## Why? Better maintainability. ## How did you test it? - [ ] built - [x] run locally and tested manually - [ ] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
100 lines
2.1 KiB
Go
100 lines
2.1 KiB
Go
package codegen
|
|
|
|
import (
|
|
"go/parser"
|
|
"go/token"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"golang.org/x/text/cases"
|
|
"golang.org/x/text/language"
|
|
)
|
|
|
|
func GenerateToFile[T any](
|
|
generator func(io.Writer, T) error,
|
|
generatorArg T,
|
|
outPath string,
|
|
outFileName string,
|
|
) {
|
|
filename := filepath.Join(outPath, outFileName+"_gen.go")
|
|
w, err := os.Create(filename)
|
|
FatalIfErr(err)
|
|
defer func() {
|
|
FatalIfErr(w.Close())
|
|
_, err = parser.ParseFile(token.NewFileSet(), filename, nil, parser.SkipObjectResolution)
|
|
FatalIfErr(err)
|
|
}()
|
|
err = generator(w, generatorArg)
|
|
FatalIfErr(err)
|
|
}
|
|
|
|
func GenerateTemplateToWriter(tmpl string, data any, w io.Writer) error {
|
|
t, err := template.New("code").Parse(tmpl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return t.Execute(w, data)
|
|
}
|
|
|
|
func GenerateTemplateToFile(
|
|
tmpl string,
|
|
data any,
|
|
outPath string,
|
|
outFileName string,
|
|
) {
|
|
GenerateToFile(func(w io.Writer, data any) error {
|
|
return GenerateTemplateToWriter(tmpl, data, w)
|
|
}, data, outPath, outFileName)
|
|
}
|
|
|
|
func FatalIfErr(err error) {
|
|
if err != nil {
|
|
//nolint:revive // okay to call Fatal here since this is part of the build process, not the server.
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Fatalf(format string, v ...any) {
|
|
//nolint:revive // okay to call Fatal here since this is part of the build process, not the server.
|
|
log.Fatalf(format, v...)
|
|
}
|
|
|
|
func CamelCaseToSnakeCase(s string) string {
|
|
if s == "" {
|
|
return ""
|
|
}
|
|
t := make([]rune, 0, len(s)+5)
|
|
for i, c := range s {
|
|
if IsASCIIUpper(c) {
|
|
if i != 0 {
|
|
t = append(t, '_')
|
|
}
|
|
c ^= ' ' // Make it a lower letter.
|
|
}
|
|
t = append(t, c)
|
|
}
|
|
return string(t)
|
|
}
|
|
|
|
func SnakeCaseToPascalCase(s string) string {
|
|
// Split the string by underscores
|
|
words := strings.Split(s, "_")
|
|
|
|
// Capitalize the first letter of each word
|
|
for i, word := range words {
|
|
// Convert first rune to upper and the rest to lower case
|
|
words[i] = cases.Title(language.AmericanEnglish).String(strings.ToLower(word))
|
|
}
|
|
|
|
// Join them back into a single string
|
|
return strings.Join(words, "")
|
|
}
|
|
|
|
func IsASCIIUpper(c rune) bool {
|
|
return 'A' <= c && c <= 'Z'
|
|
}
|