feat: add buf cmd

cmd/buf.go

@@ -0,0 +1,43 @@
+package cmd
+
+import (
+	"fmt"
+	"os/exec"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func CommandBuf(root *cobra.Command) {
+	cmd := &cobra.Command{
+		Use:   "buf",
+		Short: "run buf commands",
+		RunE:  commandBufE,
+	}
+
+	root.AddCommand(cmd)
+}
+
+func commandBufE(cmd *cobra.Command, args []string) error {
+	if _, err := exec.LookPath("buf"); err != nil {
+		log.Warn("buf 命令不存在，正在安装 buf...")
+		log.Info("go install github.com/bufbuild/buf/cmd/buf@v1.48.0")
+		installCmd := exec.Command("go", "install", "github.com/bufbuild/buf/cmd/buf@v1.48.0")
+		if err := installCmd.Run(); err != nil {
+			return fmt.Errorf("安装 buf 失败: %v", err)
+		}
+		log.Info("buf 安装成功")
+
+		if _, err := exec.LookPath("buf"); err != nil {
+			return fmt.Errorf("buf 命令不存在，请检查 $PATH")
+		}
+	}
+
+	log.Info("buf 命令已存在，正在运行 buf generate...")
+	generateCmd := exec.Command("buf", "generate")
+	if err := generateCmd.Run(); err != nil {
+		return fmt.Errorf("运行 buf generate 失败: %v", err)
+	}
+	log.Info("buf generate 运行成功")
+	return nil
+}