45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
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...")
|
|
log.Info("PROTOBUF GUIDE: https://buf.build/docs/best-practices/style-guide/")
|
|
generateCmd := exec.Command("buf", "generate")
|
|
if err := generateCmd.Run(); err != nil {
|
|
return fmt.Errorf("运行 buf generate 失败: %v", err)
|
|
}
|
|
log.Info("buf generate 运行成功")
|
|
return nil
|
|
}
|