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
}