feat(jzero): remove embedx

Author: jaronnie

cmd/jzero/.template/frame/gateway/app/cmd/server.go.tpl

@@ -6,7 +6,6 @@ import (
 	configurator "github.com/zeromicro/go-zero/core/configcenter"
 	"github.com/jzero-io/jzero/core/configcenter/subscriber"
 	"github.com/common-nighthawk/go-figure"
-	"github.com/jzero-io/jzero/core/embedx"
 	"github.com/spf13/cobra"
 	"github.com/zeromicro/go-zero/core/logx"
 	"github.com/zeromicro/go-zero/core/service"
@@ -35,7 +34,7 @@ var serverCmd = &cobra.Command{
     	logx.Must(logx.SetUp(c.Log.LogConf))
 
     	// write pb to local
-        c.Gateway.Upstreams[0].ProtoSets, err = embedx.WriteToLocal(pb.Embed, embedx.WithFileMatchFunc(func(path string) bool {
+        c.Gateway.Upstreams[0].ProtoSets, err = pb.WriteToLocal(pb.Embed, pb.WithFileMatchFunc(func(path string) bool {
 			return filepath.Ext(path) == ".pb"
 		}))
         logx.Must(err)

cmd/jzero/.template/frame/gateway/app/desc/pb/embed.go.tpl

@@ -1,10 +1,105 @@
 package pb
 
 import (
-    "embed"
+	"embed"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+
+	"github.com/pkg/errors"
 )
 
 var (
 	//go:embed *
 	Embed embed.FS
-)
+)
+
+type Opts func(config *embedxConfig)
+
+type embedxConfig struct {
+	Dir           string
+	FileMatchFunc func(path string) bool
+}
+
+func WithDir(dir string) Opts {
+	return func(config *embedxConfig) {
+		config.Dir = dir
+	}
+}
+
+func WithFileMatchFunc(fileFilter func(path string) bool) Opts {
+	return func(config *embedxConfig) {
+		config.FileMatchFunc = fileFilter
+	}
+}
+
+func WriteToLocal(ef embed.FS, opts ...Opts) ([]string, error) {
+	config := &embedxConfig{}
+
+	for _, opt := range opts {
+		opt(config)
+	}
+
+	var fileList []string
+
+	err := fs.WalkDir(ef, ".", func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+		if !d.IsDir() {
+			data, err := ef.ReadFile(path)
+			if err != nil {
+				return err
+			}
+			if config.Dir != "" {
+				if stat, err := os.Stat(config.Dir); err != nil {
+					if !os.IsExist(err) {
+						err = os.MkdirAll(config.Dir, 0o755)
+						if err != nil {
+							return err
+						}
+					}
+				} else {
+					if !stat.IsDir() {
+						return errors.Errorf("%s: not a directory", config.Dir)
+					}
+				}
+			}
+
+			var tmpFile *os.File
+			if config.FileMatchFunc != nil {
+				if config.FileMatchFunc(path) {
+					if tmpFile, err = createTemp(config.Dir, path, data); err != nil {
+						return err
+					}
+				}
+			} else {
+				if tmpFile, err = createTemp(config.Dir, path, data); err != nil {
+					return err
+				}
+			}
+			if tmpFile != nil {
+				fileList = append(fileList, tmpFile.Name())
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	return fileList, nil
+}
+
+func createTemp(dir, path string, data []byte) (*os.File, error) {
+	tmpFile, err := os.CreateTemp(dir, fmt.Sprintf("*%s", filepath.Ext(path)))
+	if err != nil {
+		return nil, err
+	}
+	defer tmpFile.Close()
+	if _, err := tmpFile.Write(data); err != nil {
+		return nil, err
+	}
+
+	return tmpFile, nil
+}