补充完成逻辑

cmd/singlefile.go

@@ -0,0 +1,92 @@
+package cmd
+
+import (
+	"complie-erlang/config"
+	tm "complie-erlang/template"
+	"complie-erlang/worker"
+	"fmt"
+	"github.com/spf13/cobra"
+	"gopkg.in/yaml.v2"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+type SingleFile struct {
+	CheckFile string
+	OutFile   string
+	Cfg       string
+}
+
+func (s *SingleFile) run(_ *cobra.Command, _ []string) {
+
+	cfg := config.DefaultErlConfig()
+	templates := tm.NewTemplate()
+	worker1 := worker.NewSingleFile(templates)
+
+	xmlPath, err := worker1.FindConfigXMLPath(s.Cfg)
+	if err != nil {
+		fmt.Printf("Err 获取可执行文件路径失败: %v", err)
+		return
+	}
+
+	bytes, err := os.ReadFile(xmlPath)
+	if err != nil {
+		fmt.Printf("Err 读取配置文件失败: %v", err)
+		return
+	}
+
+	if err = yaml.Unmarshal(bytes, cfg); err != nil {
+		fmt.Printf("Err 解析配置文件 %s 失败: %v", xmlPath, err)
+		return
+	}
+
+	// 获取可执行文件所在目录
+	exePath, err := os.Executable()
+	if err != nil {
+		log.Fatalf("获取可执行文件路径失败: %v", err)
+		return
+	}
+
+	// 加载模版报错
+	if err := templates.ParseGlob(filepath.Join(filepath.Dir(exePath), cfg.TemplateDirPattern)); err != nil {
+		log.Fatalf("Err 加载模版报错: %v", err)
+		return
+	}
+
+	var parseTemKeys []string
+	for _, cfgPublic := range cfg.Keywords {
+		// 查询目标文件
+		parseTemKey, err := worker1.ParseTemKey(s.CheckFile, cfgPublic)
+		if err != nil {
+			log.Fatalf("loading templates: %v", err)
+			return
+		}
+
+		if len(parseTemKey) == 0 {
+			continue
+		}
+		parseTemKeys = append(parseTemKeys, parseTemKey...)
+	}
+	if err = os.WriteFile(s.OutFile, []byte(strings.Join(parseTemKeys, "\n\n")), os.ModePerm); err != nil {
+		fmt.Println("Err WriteFile:", err)
+		return
+	}
+	fmt.Println("ok")
+}
+
+func init() {
+	var singleSet = new(SingleFile)
+	var logsCmd = &cobra.Command{
+		Use:   "single",
+		Short: "single 单文件生成",
+		Long:  `这是一个文件监视器工具集合`,
+		Run:   singleSet.run,
+	}
+	logsCmd.PersistentFlags().StringVar(&singleSet.CheckFile, "file", "", "读取文件")
+	logsCmd.PersistentFlags().StringVar(&singleSet.OutFile, "out", "", "输出文件")
+	logsCmd.PersistentFlags().StringVar(&singleSet.Cfg, "cfg", "config.yaml", "配置地址")
+
+	rootCmd.AddCommand(logsCmd)
+}