Go CLI 应用开发

fmx check ./cnt-content/full          # 校验全部文档，问题以非零码退出
fmx check --fix ./cnt-content/full    # 可修复项（缺 title 时用文件名兜底）直接修
fmx stats ./cnt-content/full          # 子命令：统计各模块文档数
fmx --version

func main() {
    // os.Args[0] 是程序自身路径，用户输入从 [1] 开始
    if len(os.Args) < 2 {
        fmt.Fprintln(os.Stderr, "usage: fmx <dir>")
        os.Exit(2) // 2 = 用法错误，与 1（业务失败）区分
    }
    dir := os.Args[1]
    fmt.Println("checking", dir)
}

func main() {
    fix := flag.Bool("fix", false, "自动修复可修复项")
    minTitle := flag.Int("min-title", 4, "title 最短长度")
    flag.Parse() // 之后 os.Args 里只剩位置参数

    dirs := flag.Args()
    if len(dirs) == 0 {
        dirs = []string{"."}
    }
    for _, d := range dirs {
        run(d, *fix, *minTitle)
    }
}

var fix bool

var rootCmd = &cobra.Command{
    Use:   "fmx",
    Short: "frontmatter 校验工具",
    Long:  "fmx 校验 cnt-content 文档的 frontmatter 合法性与 title 存在性。",
    Version: "0.3.0",
}

var checkCmd = &cobra.Command{
    Use:   "check [dirs...]",
    Short: "校验文档",
    Args:  cobra.MinimumNArgs(1),
    RunE: func(cmd *cobra.Command, args []string) error {
        return runCheck(args, fix)
    },
}

func init() {
    checkCmd.Flags().BoolVarP(&fix, "fix", "f", false, "自动修复可修复项")
    rootCmd.AddCommand(checkCmd)
}

func main() {
    if err := rootCmd.Execute(); err != nil {
        os.Exit(1) // cobra 已把错误打到 stderr
    }
}

// 优先级：flag 显式传入 > FMX_MIN_TITLE 环境变量 > 默认值
func resolveMinTitle(cmdFlag int) int {
    if cmdFlag != 0 { // flag 显式传了非零值
        return cmdFlag
    }
    if v := os.Getenv("FMX_MIN_TITLE"); v != "" {
        if n, err := strconv.Atoi(v); err == nil {
            return n
        }
    }
    return 4
}

package main

import (
    "embed"
    "fmt"
)

//go:embed templates/*.tmpl NOTICE.txt
var assets embed.FS

func main() {
    // 路径相对于本 .go 源文件所在目录，必须以子目录开头（不能 . 或 ..）
    b, err := assets.ReadFile("templates/report.tmpl")
    if err != nil {
        panic(err)
    }
    fmt.Println(len(b))
}

func runCheck(dirs []string, fix bool) error {
    issues := 0
    for _, dir := range dirs {
        err := filepath.WalkDir(dir, func(path string, d fs.DirEntry, err error) error {
            if err != nil {
                return err // 目录不可达：向上抛，由 main 决定退出码
            }
            if d.IsDir() || (!strings.HasSuffix(path, ".md") && !strings.HasSuffix(path, ".mdx")) {
                return nil
            }
            n, err := checkFile(path, fix)
            if err != nil {
                return err
            }
            issues += n
            return nil
        })
        if err != nil {
            return err
        }
    }
    if issues > 0 {
        fmt.Fprintf(os.Stderr, "found %d issue(s)\n", issues)
        return errIssuesFound // 让 main 映射为退出码 1
    }
    return nil
}

func checkFile(path string, fix bool) (int, error) {
    raw, err := os.ReadFile(path)
    if err != nil {
        return 0, err
    }
    text, fm, ok := splitFrontmatter(string(raw))
    if !ok {
        fmt.Fprintf(os.Stderr, "%s: missing frontmatter\n", path)
        return 1, nil
    }
    if _, has := fm["title"]; has {
        return 0, nil
    }
    if !fix {
        fmt.Fprintf(os.Stderr, "%s: missing title\n", path)
        return 1, nil
    }
    // --fix：用文件名兜底 title，写回文件
    base := strings.TrimSuffix(filepath.Base(path), filepath.Ext(path))
    fixed := reTitle.ReplaceAllString(text, fmt.Sprintf("title: %s\n", base))
    if err := os.WriteFile(path, []byte(fixed), 0o644); err != nil {
        return 0, err
    }
    fmt.Fprintf(os.Stderr, "%s: title filled from filename\n", path)
    return 0, nil
}