NTO/crudgen
Archived
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: search

Browse Source
This commit is contained in:
Goga Coder
2025-03-15 18:43:43 +07:00
parent b26de27165
commit 8e7b0ab472
5 changed files with 140 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

129
internal/writer.go
View File

@@ -16,11 +16,13 @@ import (
"github.com/dave/dst/decorator"
)
func ImplementServiceStruct(modelName string, file *dst.File, reimplement bool) {
func ImplementServiceStruct(modelName string, file *dst.File, reimplement bool) (wasModified bool) {
serviceName := modelName + "Service"
isServiceStructDefined := false
var insertPos int
var decls []dst.Decl
var (
isServiceStructDefined bool
insertPos int
decls []dst.Decl
)
for i, decl := range file.Decls {
genDecl, ok := decl.(*dst.GenDecl)
@@ -53,7 +55,7 @@ func ImplementServiceStruct(modelName string, file *dst.File, reimplement bool)
}
if isServiceStructDefined && !reimplement {
return
return false
}
serviceStruct := &dst.GenDecl{
@@ -69,13 +71,16 @@ func ImplementServiceStruct(modelName string, file *dst.File, reimplement bool)
}
file.Decls = append(decls[:insertPos], append([]dst.Decl{serviceStruct}, decls[insertPos:]...)...)
return true
}
func ImplementModelAlias(modelName string, file *dst.File) {
isAliasDefined := false
func ImplementModelAlias(modelName string, file *dst.File) (wasModified bool) {
aliasTypeStandard := fmt.Sprintf("models.%s", CapitalizeFirst(modelName))
var insertPos int
var decls []dst.Decl
var (
insertPos int
decls []dst.Decl
isAliasDefined bool
)
for i, decl := range file.Decls {
genDecl, ok := decl.(*dst.GenDecl)
@@ -147,7 +152,7 @@ func ImplementModelAlias(modelName string, file *dst.File) {
Name: &dst.Ident{
Name: modelName,
},
Assign: true, // quick and dirty
Assign: true,
Type: &dst.Ident{
Name: aliasTypeStandard,
},
@@ -157,6 +162,9 @@ func ImplementModelAlias(modelName string, file *dst.File) {
if !isAliasDefined {
file.Decls = append(decls[:insertPos], append([]dst.Decl{&typeAlias}, decls[insertPos:]...)...)
return true
} else {
return false
}
}
@@ -169,8 +177,12 @@ func importExists(file *dst.File, importPath string) bool {
return false
}
func MaintainImports(file *dst.File) error {
var importDecl *dst.GenDecl
func MaintainImports(file *dst.File) (wasModified bool, e error) {
var (
modified bool
importDecl *dst.GenDecl
)
for _, decl := range file.Decls {
if genDecl, ok := decl.(*dst.GenDecl); ok && genDecl.Tok == token.IMPORT {
importDecl = genDecl
@@ -179,6 +191,7 @@ func MaintainImports(file *dst.File) error {
}
if importDecl == nil {
modified = true
importDecl = &dst.GenDecl{
Tok: token.IMPORT,
Specs: []dst.Spec{},
@@ -188,6 +201,7 @@ func MaintainImports(file *dst.File) error {
for _, importPath := range ServiceImports {
if !importExists(file, importPath) {
modified = true
importDecl.Specs = append(importDecl.Specs, &dst.ImportSpec{
Path: &dst.BasicLit{
Kind: token.STRING,
@@ -197,7 +211,7 @@ func MaintainImports(file *dst.File) error {
}
}
return nil
return modified, nil
}
func GenerateCrudMethodCode(methodName string, context CrudTemplatesContext) string {
@@ -237,12 +251,13 @@ func MethodCodeToDeclaration(methodCode string) (*dst.FuncDecl, error) {
return methodDecl, nil
}
func ImplementCrudMethods(modelName string, serviceName string, file *dst.File, reimplement bool) error {
func ImplementCrudMethods(modelName string, serviceName string, file *dst.File, reimplement bool) (wasModified bool, e error) {
templateContext := CrudTemplatesContext{
ServiceName: serviceName,
EntityType: modelName,
EntityPlural: ToPlural(modelName),
}
modified := reimplement
for _, methodName := range implementedMethods {
methodCode := GenerateCrudMethodCode(methodName, templateContext)
@@ -252,23 +267,27 @@ func ImplementCrudMethods(modelName string, serviceName string, file *dst.File,
fmt.Println(methodDecl)
panic(err)
}
err = ImplementMethod(file, methodDecl, reimplement)
methodModified, err := ImplementMethod(file, methodDecl, reimplement)
if err != nil {
return err
return false, err
}
if methodModified {
modified = true
}
}
return nil
return modified, nil
}
func ImplementMethod(file *dst.File, methodDecl *dst.FuncDecl, reimplement bool) error {
var decls []dst.Decl
methodImplemented := false
func ImplementMethod(file *dst.File, methodDecl *dst.FuncDecl, reimplement bool) (wasModified bool, e error) {
var (
decls []dst.Decl
methodImplemented bool
)
modified := reimplement
methodStructure := methodDecl.Recv.List[0].Names[0].Name
methodName := methodDecl.Name.Name
log.Printf("Standard method structure: %s\n", methodStructure)
log.Printf("Standard method name: %s\n", methodName)
for _, decl := range file.Decls {
decls = append(decls, decl)
@@ -277,14 +296,14 @@ func ImplementMethod(file *dst.File, methodDecl *dst.FuncDecl, reimplement bool)
continue
}
if len(funcDecl.Recv.List) > 0 && len(funcDecl.Recv.List[0].Names) > 0 {
fmt.Printf("Method structure: %s\n", funcDecl.Recv.List[0].Names[0].Name)
fmt.Printf("Method name: %s\n", funcDecl.Name.Name)
log.Printf("Method structure: %s\n", funcDecl.Recv.List[0].Names[0].Name)
log.Printf("Method name: %s\n", funcDecl.Name.Name)
if funcDecl.Recv.List[0].Names[0].Name == methodStructure {
if funcDecl.Name != nil && funcDecl.Name.Name == methodName {
if methodImplemented {
err := fmt.Sprintf("`%s` method redeclarated for struct `%s`", methodName, methodStructure)
log.Println(err)
return errors.New(err)
return false, errors.New(err)
} else {
methodImplemented = true
}
@@ -297,6 +316,7 @@ func ImplementMethod(file *dst.File, methodDecl *dst.FuncDecl, reimplement bool)
}
if reimplement || !methodImplemented {
modified = true
file.Decls = append(
decls,
&dst.FuncDecl{
@@ -310,72 +330,93 @@ func ImplementMethod(file *dst.File, methodDecl *dst.FuncDecl, reimplement bool)
)
}
return nil
return modified, nil
}
func CreateServiceFileIfNotExists(filePath string) error {
func CreateServiceFileIfNotExists(filePath string) (wasModified bool, e error) {
var modified bool
if _, err := os.Stat(filePath); err != nil {
// file wasn't created
f, err := os.Create(filePath)
if err != nil {
log.Fatalf("Failed to create file: %s", filePath)
return err
return false, err
}
_, err = f.Write([]byte("package services\n"))
if err != nil {
log.Fatalf("Failed to write file: %s", filePath)
return err
return false, err
}
modified = true
defer f.Close()
}
return nil
return modified, nil
}
func ImplementService(mainPkgPath string, modelName string, reimplement bool) error {
func ImplementService(mainPkgPath string, modelName string, reimplement bool) (wasModified bool, e error) {
serviceRelativePath := fmt.Sprintf("services/%s.go", strings.ToLower(modelName))
filePath := filepath.Join(mainPkgPath, serviceRelativePath)
serviceName := modelName + "Service"
modified := reimplement
err := CreateServiceFileIfNotExists(filePath)
fileModified, err := CreateServiceFileIfNotExists(filePath)
if err != nil {
return err
return false, err
}
if fileModified {
modified = true
}
fset := token.NewFileSet()
serviceFile, err := decorator.ParseFile(fset, filePath, nil, parser.ParseComments)
if err != nil {
log.Fatalf("Parsing error: %s: %v", mainPkgPath, err)
return err
return false, err
}
err = MaintainImports(serviceFile)
importsModified, err := MaintainImports(serviceFile)
if err != nil {
return err
return false, err
}
if importsModified {
modified = true
}
ImplementModelAlias(modelName, serviceFile)
ImplementServiceStruct(modelName, serviceFile, reimplement)
err = ImplementCrudMethods(modelName, serviceName, serviceFile, reimplement)
aliasAdded := ImplementModelAlias(modelName, serviceFile)
if aliasAdded {
modified = true
}
serviceStructModified := ImplementServiceStruct(modelName, serviceFile, reimplement)
if serviceStructModified {
modified = true
}
methodsModified, err := ImplementCrudMethods(modelName, serviceName, serviceFile, reimplement)
if err != nil {
return err
return false, err
}
if methodsModified {
modified = true
}
file, err := os.Create(filePath)
if err != nil {
return errors.New(fmt.Sprintf("Error occured to open `%s` service file: %v", modelName, err))
errMessage := errors.New(fmt.Sprintf("Error occured to open `%s` service file: %v", modelName, err))
return false, errMessage
}
err = decorator.Fprint(file, serviceFile)
if err != nil {
return errors.New(
errMessage := errors.New(
fmt.Sprintf("Error occurred to writing changes in `%s` service file: %v", modelName, err),
)
return false, errMessage
}
defer file.Close()
return nil
return modified, nil
}