使用wmenu创建Golang命令行交互式菜单

wmenu是一个简单易用的Golang库，用于在命令行中创建交互式菜单。它支持单选、多选等功能，并且可以自定义菜单项的样式。

安装wmenu

go get github.com/dixonwille/wmenu/v5

基本使用示例

package main

import (
	"fmt"
	"os"

	"github.com/dixonwille/wmenu/v5"
)

func main() {
	menu := wmenu.NewMenu("请选择一个选项:")

	// 添加菜单选项
	menu.Option("选项 1", nil, false, nil)
	menu.Option("选项 2", nil, false, nil)
	menu.Option("选项 3", nil, false, nil)
	menu.Option("退出", nil, false, func(opt wmenu.Opt) error {
		os.Exit(0)
		return nil
	})

	// 设置菜单动作处理函数
	menu.Action(func(opts []wmenu.Opt) error {
		opt := opts[0] // 获取用户选择的第一个选项
		fmt.Printf("您选择了: %s\n", opt.Text)
		return nil
	})

	// 运行菜单
	err := menu.Run()
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

高级功能示例

1. 多选菜单

func multiSelectExample() {
	menu := wmenu.NewMenu("请选择多个选项(用逗号分隔):")
	menu.IsYesNo(false)
	menu.MultiSelect(true) // 启用多选

	menu.Option("苹果", "apple", false, nil)
	menu.Option("香蕉", "banana", false, nil)
	menu.Option("橙子", "orange", false, nil)
	menu.Option("葡萄", "grape", false, nil)

	menu.Action(func(opts []wmenu.Opt) error {
		fmt.Println("您选择了:")
		for _, opt := range opts {
			fmt.Printf("- %s (值: %v)\n", opt.Text, opt.Value)
		}
		return nil
	})

	_ = menu.Run()
}

2. 带默认值的菜单

func defaultOptionExample() {
	menu := wmenu.NewMenu("请选择一个颜色(默认红色):")
	
	// 第三个参数true表示这是默认选项
	menu.Option("红色", "red", true, nil)
	menu.Option("蓝色", "blue", false, nil)
	menu.Option("绿色", "green", false, nil)

	menu.Action(func(opts []wmenu.Opt) error {
		opt := opts[0]
		fmt.Printf("您选择的颜色是: %s (值: %v)\n", opt.Text, opt.Value)
		return nil
	})

	_ = menu.Run()
}

3. 自定义菜单样式

func customStyleExample() {
	menu := wmenu.NewMenu("自定义样式菜单:")
	
	// 自定义颜色
	menu.SetSeparator(" > ")
	menu.Color(wmenu.BrightBlue, wmenu.Red, wmenu.BrightYellow, wmenu.Green, wmenu.BrightGreen)

	menu.Option("第一项", 1, false, nil)
	menu.Option("第二项", 2, false, nil)
	menu.Option("第三项", 3, false, nil)

	menu.Action(func(opts []wmenu.Opt) error {
		fmt.Printf("选择的值: %v\n", opts[0].Value)
		return nil
	})

	_ = menu.Run()
}

错误处理

wmenu会返回可能出现的错误，应该妥善处理：

func errorHandlingExample() {
	menu := wmenu.NewMenu("带有错误处理的菜单:")
	
	menu.Option("继续", nil, false, nil)
	menu.Option("退出", nil, false, func(opt wmenu.Opt) error {
		return fmt.Errorf("用户选择退出")
	})

	menu.Action(func(opts []wmenu.Opt) error {
		fmt.Println("继续执行...")
		return nil
	})

	err := menu.Run()
	if err != nil {
		fmt.Printf("发生错误: %v\n", err)
		os.Exit(1)
	}
}

实际应用示例

func mainMenu() {
	for {
		menu := wmenu.NewMenu("主菜单:")
		
		menu.Option("用户管理", nil, false, func(opt wmenu.Opt) error {
			userManagementMenu()
			return nil
		})
		
		menu.Option("系统设置", nil, false, func(opt wmenu.Opt) error {
			systemSettingsMenu()
			return nil
		})
		
		menu.Option("退出系统", nil, false, func(opt wmenu.Opt) error {
			os.Exit(0)
			return nil
		})

		menu.Action(func(opts []wmenu.Opt) error {
			// 主菜单的动作由子菜单处理
			return nil
		})

		if err := menu.Run(); err != nil {
			fmt.Println("菜单错误:", err)
		}
	}
}

func userManagementMenu() {
	// 用户管理子菜单实现
}

func systemSettingsMenu() {
	// 系统设置子菜单实现
}

wmenu库提供了简单直观的API来创建命令行菜单，适合各种CLI工具的开发。通过组合不同的选项和动作，可以构建出功能丰富的交互式命令行应用。