-
Notifications
You must be signed in to change notification settings - Fork 551
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add parameters placeholders and modifier
- Loading branch information
Pantani
authored and
Pantani
committed
Nov 22, 2023
1 parent
9fe1f4c
commit 5cd468e
Showing
6 changed files
with
169 additions
and
13 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
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
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
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
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 |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| package modulecreate | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "path/filepath" | ||
|
|
||
| "github.com/gobuffalo/genny/v2" | ||
|
|
||
| "github.com/ignite/cli/ignite/pkg/placeholder" | ||
| "github.com/ignite/cli/ignite/templates/module" | ||
| ) | ||
|
|
||
| func paramsTypesModify(replacer placeholder.Replacer, opts ParamsOptions) genny.RunFn { | ||
| return func(r *genny.Runner) error { | ||
| path := filepath.Join(opts.AppPath, "x", opts.ModuleName, "types/params.go") | ||
| f, err := r.Disk.Find(path) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| content := f.String() | ||
| for _, param := range opts.Params { | ||
| // param key and default value. | ||
| templateVars := `var ( | ||
| // Key%[2]v represents the %[2]v parameter. | ||
| Key%[2]v = []byte("%[2]v") | ||
| // TODO: Determine the default value | ||
| // Default%[2]v represents the %[2]v default value. | ||
| Default%[2]v %[3]v = %[4]v | ||
| ) | ||
| %[1]v` | ||
| replacementVars := fmt.Sprintf( | ||
| templateVars, | ||
| module.PlaceholderParamsVars, | ||
| param.Name.UpperCamel, | ||
| param.DataType(), | ||
| param.Value(), | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsVars, replacementVars) | ||
|
|
||
| // add parameter to the new method. | ||
| templateNewParam := "%[2]v %[3]v,\n%[1]v" | ||
| replacementNewParam := fmt.Sprintf( | ||
| templateNewParam, | ||
| module.PlaceholderParamsNewParam, | ||
| param.Name.LowerCamel, | ||
| param.DataType(), | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsNewParam, replacementNewParam) | ||
|
|
||
| // add parameter to the struct into the new method. | ||
| templateNewStruct := "%[2]v,\n%[1]v" | ||
| replacementNewStruct := fmt.Sprintf( | ||
| templateNewStruct, | ||
| module.PlaceholderParamsNewStruct, | ||
| param.Name.LowerCamel, | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsNewStruct, replacementNewStruct) | ||
|
|
||
| // add default parameter. | ||
| templateDefault := `Default%[2]v, | ||
| %[1]v` | ||
| replacementDefault := fmt.Sprintf( | ||
| templateDefault, | ||
| module.PlaceholderParamsDefault, | ||
| param.Name.UpperCamel, | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsDefault, replacementDefault) | ||
|
|
||
| // add new param set pair. | ||
| templateSetPairs := `paramtypes.NewParamSetPair(Key%[2]v, &p.%[2]v, validate%[2]v), | ||
| %[1]v` | ||
| replacementSetPairs := fmt.Sprintf( | ||
| templateSetPairs, | ||
| module.PlaceholderParamsSetPairs, | ||
| param.Name.UpperCamel, | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsSetPairs, replacementSetPairs) | ||
|
|
||
| // add param field to the validate method. | ||
| templateValidate := `if err := validate%[2]v(p.%[2]v); err != nil { | ||
| return err | ||
| } | ||
| %[1]v` | ||
| replacementValidate := fmt.Sprintf( | ||
| templateValidate, | ||
| module.PlaceholderParamsValidate, | ||
| param.Name.UpperCamel, | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsValidate, replacementValidate) | ||
|
|
||
| // add param field to the validate method. | ||
| templateValidation := `// validate%[2]v validates the %[2]v parameter. | ||
| func validate%[2]v(v interface{}) error { | ||
| %[3]v, ok := v.(%[4]v) | ||
| if !ok { | ||
| return fmt.Errorf("invalid parameter type: %T", v) | ||
| } | ||
| // TODO implement validation | ||
| _ = %[3]v | ||
| return nil | ||
| } | ||
| %[1]v` | ||
| replacementValidation := fmt.Sprintf( | ||
| templateValidation, | ||
| module.PlaceholderParamsValidation, | ||
| param.Name.UpperCamel, | ||
| param.Name.LowerCamel, | ||
| param.DataType(), | ||
| ) | ||
| content = replacer.Replace(content, module.PlaceholderParamsValidation, replacementValidation) | ||
|
|
||
| } | ||
|
|
||
| newFile := genny.NewFileS(path, content) | ||
| return r.File(newFile) | ||
| } | ||
| } |
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