Golang实现Cobra命令闭环应用与oh-my-posh提示插件

下面我将介绍如何使用Golang的Cobra库创建命令行应用，并结合oh-my-posh实现美观的命令行提示。

1. Cobra基础应用实现

首先创建一个基本的Cobra命令行应用：

package main

import (
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
	Use:   "myapp",
	Short: "MyApp is a demo CLI application",
	Long:  `A longer description of my demo CLI application built with Cobra.`,
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("Hello from MyApp!")
	},
}

var versionCmd = &cobra.Command{
	Use:   "version",
	Short: "Print the version number of MyApp",
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("MyApp v1.0.0")
	},
}

func init() {
	rootCmd.AddCommand(versionCmd)
	
	// 添加全局flag
	rootCmd.PersistentFlags().StringVarP(&cfgFile, "config", "c", "", "config file (default is $HOME/.myapp.yaml)")
}

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func main() {
	Execute()
}

2. 添加oh-my-posh集成

oh-my-posh是一个强大的命令行提示工具，我们可以通过以下方式在Go应用中集成它的主题：

import (
	"os/exec"
	"runtime"
)

func setupPrompt() {
	// 检测oh-my-posh是否安装
	_, err := exec.LookPath("oh-my-posh")
	if err != nil {
		fmt.Println("oh-my-posh not found. Please install it first.")
		return
	}

	// 根据操作系统设置不同的提示主题
	var theme string
	switch runtime.GOOS {
	case "windows":
		theme = "jandedobbeleer.omp.json"
	case "darwin", "linux":
		theme = "agnoster.omp.json"
	default:
		theme = "default.omp.json"
	}

	// 设置提示
	cmd := exec.Command("oh-my-posh", "--init", "--shell", "bash", "--config", theme)
	err = cmd.Run()
	if err != nil {
		fmt.Printf("Failed to setup oh-my-posh: %v\n", err)
	}
}

然后在main函数中调用：

func main() {
	setupPrompt()
	Execute()
}

3. 完整示例：带配置的Cobra应用

下面是一个更完整的示例，包含配置读取和oh-my-posh集成：

package main

import (
	"fmt"
	"log"
	"os"
	"os/exec"
	"runtime"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var cfgFile string

type Config struct {
	Theme  string `mapstructure:"theme"`
	Debug  bool   `mapstructure:"debug"`
	Server struct {
		Host string `mapstructure:"host"`
		Port int    `mapstructure:"port"`
	} `mapstructure:"server"`
}

var appConfig Config

var rootCmd = &cobra.Command{
	Use:   "myapp",
	Short: "MyApp with Cobra and oh-my-posh",
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		initConfig()
		setupPrompt()
	},
	Run: func(cmd *cobra.Command, args []string) {
		if appConfig.Debug {
			log.Println("Debug mode enabled")
		}
		fmt.Printf("Server: %s:%d\n", appConfig.Server.Host, appConfig.Server.Port)
	},
}

func initConfig() {
	if cfgFile != "" {
		viper.SetConfigFile(cfgFile)
	} else {
		home, err := os.UserHomeDir()
		cobra.CheckErr(err)

		viper.AddConfigPath(home)
		viper.AddConfigPath(".")
		viper.SetConfigType("yaml")
		viper.SetConfigName(".myapp")
	}

	viper.AutomaticEnv()

	if err := viper.ReadInConfig(); err == nil {
		fmt.Println("Using config file:", viper.ConfigFileUsed())
	}

	if err := viper.Unmarshal(&appConfig); err != nil {
		log.Fatalf("Unable to decode config, %v", err)
	}
}

func setupPrompt() {
	if _, err := exec.LookPath("oh-my-posh"); err != nil {
		log.Println("oh-my-posh not found. Skipping prompt setup.")
		return
	}

	theme := appConfig.Theme
	if theme == "" {
		switch runtime.GOOS {
		case "windows":
			theme = "jandedobbeleer.omp.json"
		default:
			theme = "agnoster.omp.json"
		}
	}

	cmd := exec.Command("oh-my-posh", "--init", "--shell", "bash", "--config", theme)
	if err := cmd.Run(); err != nil {
		log.Printf("Failed to setup oh-my-posh: %v\n", err)
	}
}

func main() {
	rootCmd.PersistentFlags().StringVarP(&cfgFile, "config", "c", "", "config file (default is $HOME/.myapp.yaml)")
	
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

4. 配置示例

创建配置文件 ~/.myapp.yaml：

theme: "powerlevel10k_rainbow.omp.json"
debug: true
server:
  host: "localhost"
  port: 8080

5. 构建与使用

1. 安装依赖：

go get github.com/spf13/cobra
go get github.com/spf13/viper

1. 构建应用：

go build -o myapp

1. 运行应用：

./myapp --config ~/.myapp.yaml

总结

通过Cobra我们可以创建功能强大的命令行应用，而oh-my-posh则为终端提供了美观的提示。两者的结合可以提升开发者的命令行体验。在实际项目中，你可以根据需要添加更多命令和功能，以及定制更适合自己工作流的oh-my-posh主题。