feat: support custom template replacement

Author: zhufuyi

pkg/replacer/replacer.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"text/template"
 	"time"
 
 	"github.com/zhufuyi/sponge/pkg/gofile"
@@ -28,6 +29,7 @@ type Replacer interface {
 	SaveFiles() error
 	ReadFile(filename string) ([]byte, error)
 	GetFiles() []string
+	SaveTemplateFiles(m map[string]interface{}, parentDir ...string) error
 }
 
 // replacerInfo replacer information
@@ -287,6 +289,86 @@ func (r *replacerInfo) SaveFiles() error {
 	return nil
 }
 
+// SaveTemplateFiles save file with setting
+func (r *replacerInfo) SaveTemplateFiles(m map[string]interface{}, parentDir ...string) error {
+	refDir := ""
+	if len(parentDir) > 0 {
+		refDir = strings.Join(parentDir, gofile.GetPathDelimiter())
+	}
+
+	writeData := make(map[string][]byte, len(r.files))
+	for _, file := range r.files {
+		data, err := replaceTemplateData(file, m)
+		if err != nil {
+			return err
+		}
+		newFilePath := r.getNewFilePath2(file, refDir)
+		newFilePath = trimExt(newFilePath)
+		if gofile.IsExists(newFilePath) {
+			return fmt.Errorf("file %s already exists, cancel code generation", newFilePath)
+		}
+		newFilePath, err = replaceTemplateFilePath(newFilePath, m)
+		if err != nil {
+			return err
+		}
+		writeData[newFilePath] = data
+	}
+
+	for file, data := range writeData {
+		err := saveToNewFile(file, data)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func replaceTemplateData(file string, m map[string]interface{}) ([]byte, error) {
+	data, err := os.ReadFile(file)
+	if err != nil {
+		return nil, fmt.Errorf("read file failed, err=%s", err)
+	}
+	if !bytes.Contains(data, []byte("{{")) {
+		return data, nil
+	}
+
+	builder := bytes.Buffer{}
+	tmpl, err := template.New(file).Parse(string(data))
+	if err != nil {
+		return nil, fmt.Errorf("parse data failed, err=%s", err)
+	}
+	err = tmpl.Execute(&builder, m)
+	if err != nil {
+		return nil, fmt.Errorf("execute data failed, err=%s", err)
+	}
+	return builder.Bytes(), nil
+}
+
+func replaceTemplateFilePath(file string, m map[string]interface{}) (string, error) {
+	if !strings.Contains(file, "{{") {
+		return file, nil
+	}
+
+	builder := strings.Builder{}
+	tmpl, err := template.New("file: " + file).Parse(file)
+	if err != nil {
+		return file, fmt.Errorf("parse file failed, err=%s", err)
+	}
+	err = tmpl.Execute(&builder, m)
+	if err != nil {
+		return file, fmt.Errorf("execute file failed, err=%s", err)
+	}
+	return builder.String(), nil
+}
+
+func trimExt(file string) string {
+	file = strings.TrimSuffix(file, ".tmpl")
+	file = strings.TrimSuffix(file, ".tpl")
+	file = strings.TrimSuffix(file, ".template")
+	return file
+}
+
 func (r *replacerInfo) isIgnoreFile(file string) bool {
 	isIgnore := false
 	for _, v := range r.ignoreFiles {
@@ -334,6 +416,18 @@ func (r *replacerInfo) getNewFilePath(file string) string {
 	return newFilePath
 }
 
+func (r *replacerInfo) getNewFilePath2(file string, refDir string) string {
+	if refDir == "" {
+		return r.getNewFilePath(file)
+	}
+
+	newFilePath := r.outPath + gofile.GetPathDelimiter() + refDir + gofile.GetPathDelimiter() + strings.Replace(file, r.path, "", 1)
+	if gofile.IsWindows() {
+		newFilePath = strings.ReplaceAll(newFilePath, "/", "\\")
+	}
+	return newFilePath
+}
+
 // if windows, convert the path splitter
 func (r *replacerInfo) convertPathDelimiter(filePath string) string {
 	if r.isActual && gofile.IsWindows() {

pkg/replacer/replacer_test.go

@@ -88,6 +88,22 @@ func TestNewWithFS(t *testing.T) {
 	}
 }
 
+func TestSaveTemplateFiles(t *testing.T) {
+	out := fmt.Sprintf("%s/replacer_test/template_%s", os.TempDir(), time.Now().Format("150405"))
+	m := map[string]interface{}{
+		"service":  map[string]interface{}{"name": "user", "version": 1.0},
+		"port":     8080,
+		"isSecure": true,
+	}
+
+	r, err := New("testDir")
+	assert.NoError(t, err)
+	_ = r.SetOutputDir(out)
+	err = r.SaveTemplateFiles(m)
+	assert.NoError(t, err)
+	t.Log(out)
+}
+
 func TestReplacerError(t *testing.T) {
 	_, err := New("/notfound")
 	assert.Error(t, err)

pkg/replacer/testDir/foo.txt

@@ -1 +1,7 @@
-456
+456
+
+{{.service}}
+{{.service.name}}
+{{.service.version}}
+{{.port}}
+{{.isSecure}}

pkg/replacer/testDir/replace/test.txt

@@ -1,3 +1,7 @@
 test1
 
 to do replace
+
+{{.service}}
+{{.port}}
+{{.isSecure}}