nto-cli/generation/scheme.go

package generation

import (
	_ "embed"
	"fmt"
	"log"
	"nto_cli/model"
	"nto_cli/utils"
	"os"
	"path/filepath"
	"strings"
	"text/template"
)

//go:embed templates/scheme.tmpl
var SchemeTemplate string

const GolangServicesPath = "../../bindings/app/internal/services"

type Dependency struct {
	ImportName  string
	ServiceName string
	LowerName   string
	FieldName   string
}

type TemplateData struct {
	StructName         string
	LowerName          string
	GolangServicesPath string
	Fields             []model.Field
	Dependencies       []Dependency
}

func GenerateScheme(model *model.Model, mkPath string) {
	data := TemplateData{
		StructName:         model.Name,
		LowerName:          strings.ToLower(model.Name),
		GolangServicesPath: GolangServicesPath,
		Fields:             model.Fields,
		Dependencies:       processDependencies(model.Fields),
	}

	schemeFilename := strings.ToUpper(model.Name[:1]) + strings.ToLower(model.Name[1:]) + "Scheme.vue"
	schemeFilePath := filepath.Join(mkPath, schemeFilename)
	schemeFile, err := os.Create(schemeFilePath)
	if err != nil {
		log.Fatalf("Failed to create file: %s", err)
	}
	defer func(schemeFile *os.File) {
		err := schemeFile.Close()
		if err != nil {
			log.Fatalf("Failed to close file: %s", err)
		}
	}(schemeFile)

	tmpl, err := template.New("scheme").Parse(SchemeTemplate)
	if err != nil {
		panic(fmt.Sprintf("Failed to parse template: %s", err))
	}

	err = tmpl.Execute(schemeFile, data)
	if err != nil {
		log.Fatalf("Failed to execute template: %s", err)
	}
	log.Printf("Scheme for `%s` model is written: %s", model.Name, schemeFilePath)
	_ = utils.FormatFilesWithPrettier([]string{schemeFilePath})
}

func processDependencies(fields []model.Field) []Dependency {
	var dependencies []Dependency

	for _, field := range fields {
		dependency := field.Metadata.RelatedModel
		if dependency == "" {
			continue
		}
		dependencies = append(dependencies, Dependency{
			ImportName:  strings.ToUpper(dependency[:1]) + strings.ToLower(dependency[1:]) + "Service",
			ServiceName: strings.ToLower(dependency) + "Service",
			LowerName:   strings.ToLower(dependency),
			FieldName:   field.Name,
		})

	}

	return dependencies
}