generated from TBD54566975/tbd-project-template
-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor: switch to TBD54566975/scaffolder
- Loading branch information
1 parent
0648583
commit c1c8b4c
Showing
3 changed files
with
19 additions
and
113 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,128 +1,31 @@ | ||
| package internal | ||
|
|
||
| import ( | ||
| "io/fs" | ||
| "os" | ||
| "path/filepath" | ||
| "reflect" | ||
| "strings" | ||
| "text/template" | ||
|
|
||
| "github.com/alecthomas/errors" | ||
| "github.com/iancoleman/strcase" | ||
|
|
||
| "github.com/TBD54566975/scaffolder" | ||
| ) | ||
|
|
||
| // Scaffold evaluates the scaffolding files at the given destination against | ||
| // ctx. | ||
| // | ||
| // Both paths and file contents are evaluated. | ||
| // | ||
| // If a file name ends with ".tmpl", the ".tmpl" suffix is removed. | ||
| // | ||
| // The functions "snake", "camel", "lowerCamel", "kebab", "upper", and "lower" | ||
| // are available. | ||
| // | ||
| // This is inspired by [cookiecutter]. | ||
| // | ||
| // [cookiecutter]: https://github.com/cookiecutter/cookiecutter | ||
| func Scaffold(destination string, ctx any) error { | ||
| return errors.WithStack(walkDir(destination, func(path string, d fs.DirEntry) error { | ||
| info, err := d.Info() | ||
| if err != nil { | ||
| return errors.WithStack(err) | ||
| } | ||
|
|
||
| if strings.HasSuffix(path, ".tmpl") { | ||
| newPath := strings.TrimSuffix(path, ".tmpl") | ||
| if err = os.Rename(path, newPath); err != nil { | ||
| return errors.Wrap(err, "failed to rename file") | ||
| } | ||
| path = newPath | ||
| } | ||
|
|
||
| // Evaluate the last component of path name templates. | ||
| dir := filepath.Dir(path) | ||
| base := filepath.Base(path) | ||
| newName, err := evaluate(base, ctx) | ||
| if err != nil { | ||
| return errors.Wrapf(err, "%s", path) | ||
| } | ||
| // Rename if necessary. | ||
| if newName != base { | ||
| newName = filepath.Join(dir, newName) | ||
| err = os.Rename(path, newName) | ||
| if err != nil { | ||
| return errors.Wrap(err, "failed to rename file") | ||
| } | ||
| path = newName | ||
| } | ||
|
|
||
| if !info.Mode().IsRegular() { | ||
| return nil | ||
| } | ||
|
|
||
| // Evaluate file content. | ||
| template, err := os.ReadFile(path) | ||
| if err != nil { | ||
| return errors.Wrapf(err, "%s", path) | ||
| } | ||
| content, err := evaluate(string(template), ctx) | ||
| if err != nil { | ||
| return errors.Wrapf(err, "%s", path) | ||
| } | ||
| err = os.WriteFile(path, []byte(content), info.Mode()) | ||
| if err != nil { | ||
| return errors.Wrapf(err, "%s", path) | ||
| } | ||
| return nil | ||
| })) | ||
| } | ||
|
|
||
| // Walk dir executing fn after each entry. | ||
| func walkDir(dir string, fn func(path string, d fs.DirEntry) error) error { | ||
| entries, err := os.ReadDir(dir) | ||
| if err != nil { | ||
| return errors.WithStack(err) | ||
| } | ||
| for _, entry := range entries { | ||
| if entry.IsDir() { | ||
| err = walkDir(filepath.Join(dir, entry.Name()), fn) | ||
| if err != nil { | ||
| return errors.WithStack(err) | ||
| } | ||
| } | ||
| err = fn(filepath.Join(dir, entry.Name()), entry) | ||
| if err != nil { | ||
| return errors.WithStack(err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func evaluate(tmpl string, ctx any) (string, error) { | ||
| t, err := template.New("scaffolding").Funcs( | ||
| template.FuncMap{ | ||
| "snake": strcase.ToSnake, | ||
| "screamingSnake": strcase.ToScreamingSnake, | ||
| "camel": strcase.ToCamel, | ||
| "lowerCamel": strcase.ToLowerCamel, | ||
| "kebab": strcase.ToKebab, | ||
| "screamingKebab": strcase.ToScreamingKebab, | ||
| "upper": strings.ToUpper, | ||
| "lower": strings.ToLower, | ||
| "title": strings.Title, | ||
| "typename": func(v any) string { | ||
| return reflect.Indirect(reflect.ValueOf(v)).Type().Name() | ||
| }, | ||
| return errors.WithStack(scaffolder.Scaffold(destination, ctx, scaffolder.Functions(template.FuncMap{ | ||
| "snake": strcase.ToSnake, | ||
| "screamingSnake": strcase.ToScreamingSnake, | ||
| "camel": strcase.ToCamel, | ||
| "lowerCamel": strcase.ToLowerCamel, | ||
| "kebab": strcase.ToKebab, | ||
| "screamingKebab": strcase.ToScreamingKebab, | ||
| "upper": strings.ToUpper, | ||
| "lower": strings.ToLower, | ||
| "title": strings.Title, | ||
| "typename": func(v any) string { | ||
| return reflect.Indirect(reflect.ValueOf(v)).Type().Name() | ||
| }, | ||
| ).Parse(tmpl) | ||
| if err != nil { | ||
| return "", errors.Wrap(err, "failed to parse template") | ||
| } | ||
| newName := &strings.Builder{} | ||
| err = t.Execute(newName, ctx) | ||
| if err != nil { | ||
| return "", errors.Wrap(err, "failed to execute template") | ||
| } | ||
| return newName.String(), nil | ||
| }))) | ||
| } |