service and interactor generation

service.go

@@ -1,34 +1,158 @@
 package tuugen
 
 import (
+	"embed"
+	"fmt"
 	"go/ast"
 	"go/parser"
 	"go/token"
+	"html/template"
+	"os"
+	"path/filepath"
+	"strings"
 )
 
-type serviceEndpoint struct {
-	name string
-	args []string
-}
-
-func GenerateService(filepath string) error {
-
-	//  load service template
-	// t, err := template.ParseFiles("templates/service.go.tmpl")
-	// if err != nil {
-	// 	return err
-	// }
-
+func GenerateService(FS embed.FS, serviceName, importPath, protoFilepath, outputFile string) error {
 	//	read interface from project built protos.
-	fset := token.NewFileSet()
-	f, err := parser.ParseFile(fset, filepath, nil, 0)
-	if err != nil {
-		return err
-	}
-	// generate the service template and store it to a file.
-	err = ast.Print(fset, f)
-	if err != nil {
+	if err := createFileFromProto(FS, serviceName, "templates/service.go.tmpl", importPath, protoFilepath, outputFile); err != nil {
 		return err
 	}
 	return nil
 }
+
+func GenerateInteractor(FS embed.FS, serviceName, importPath, protoFilepath, outputFile string) error {
+	//	read interface from project built protos.
+	if err := createFileFromProto(FS, serviceName, "templates/interactors.go.tmpl", importPath, protoFilepath, outputFile); err != nil {
+		return err
+	}
+	return nil
+}
+
+type serviceDef struct {
+	Imports []string
+	Funcs   []serviceFuncDef
+}
+
+type serviceFuncDef struct {
+	Name    string
+	Args    []string
+	Returns []string
+}
+
+func (s serviceFuncDef) ListToStr(list []string) string {
+	return strings.Join(list, ", ")
+}
+
+func (s serviceFuncDef) ArgsToStr() string {
+	args := []string{}
+	for _, a := range s.Args {
+		switch a {
+		case "context.Context":
+			args = append(args, fmt.Sprintf("ctx %s", a))
+		default:
+			args = append(args, fmt.Sprintf("req %s", a))
+		}
+	}
+	return s.ListToStr(args)
+}
+
+// generate the service/interactor template and store it to a file.
+func createFileFromProto(FS embed.FS, serviceName, templateFile, importPath, protoFilePath, outputFile string) error {
+	//  load service template
+	t, err := template.ParseFS(FS, templateFile)
+	if err != nil {
+		return err
+	}
+
+	// check/create file path
+	fp := filepath.Dir(outputFile)
+
+	if err := os.MkdirAll(fp, 0777); err != nil {
+		return err
+	}
+
+	f, err := os.Create(outputFile)
+	if err != nil {
+		return nil
+	}
+	funcs, err := methodsFromProto(protoFilePath, importPath, serviceName)
+	if err != nil {
+		return err
+	}
+
+	if err := t.Execute(f, funcs); err != nil {
+		return err
+	}
+
+	return f.Close()
+}
+
+func methodsFromProto(filepath, importPath, serviceName string) (serviceDef, error) {
+	funcs := []serviceFuncDef{}
+	fset := token.NewFileSet()
+	f, err := parser.ParseFile(fset, filepath, nil, 0)
+	if err != nil {
+		return serviceDef{}, fmt.Errorf("failed to parse proto go file: %w", err)
+	}
+
+	imports := f.Imports
+	importStrs := []string{}
+	// ast.Print(fset, imports)
+	// return serviceDef{}, nil
+	for _, i := range imports {
+		importStrs = append(importStrs, i.Name.Name)
+	}
+	importStrs = append(importStrs, importPath+"/internal/pb/service", importPath+"/internal/interactors")
+
+	for _, m := range f.Scope.Objects {
+		if m.Name != serviceName+"Server" {
+			continue
+		}
+		decl := m.Decl.(*ast.TypeSpec)
+		t := decl.Type.(*ast.InterfaceType)
+
+		for _, m := range t.Methods.List {
+			name := m.Names[0].Name
+			if strings.Contains(name, "mustEmbed") {
+				continue
+			}
+			f := serviceFuncDef{Name: name}
+			ft := m.Type.(*ast.FuncType)
+			if ft.Params != nil {
+				f.Args = parseFieldList(ft.Params)
+			}
+
+			if ft.Results != nil {
+				f.Returns = parseFieldList(ft.Results)
+			}
+			funcs = append(funcs, f)
+			// err = ast.Print(fset, ft.Params)
+			// if err != nil {
+			// 	return serviceDef{}, err
+			// }
+		}
+	}
+	return serviceDef{Funcs: funcs, Imports: importStrs}, nil
+}
+
+func parseFieldList(fl *ast.FieldList) []string {
+	args := []string{}
+	if fl.List == nil {
+		return args
+	}
+	for _, a := range fl.List {
+		argStr := ""
+		switch a.Type.(type) {
+		case *ast.SelectorExpr:
+			as := a.Type.(*ast.SelectorExpr)
+			argStr = as.X.(*ast.Ident).Name + "." + as.Sel.Name
+		case *ast.StarExpr:
+			as := a.Type.(*ast.StarExpr)
+			argStr = "*service." + as.X.(*ast.Ident).Name
+		case *ast.Ident:
+			argStr = a.Type.(*ast.Ident).Name
+		}
+		args = append(args, argStr)
+	}
+	return args
+}