package cmd

import (
	"agent-wdd/config"
	"fmt"
	"io"
	"os"
	"strings"

	"github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
	Use:   "wdd",
	Short: "wdd应用程序是wdd封装的NB工具",
	Long: `使用golang的强大特性，加上 WDD 的高超技术，打造一个方便、高效、适用于各种LINUX系统的瑞士军刀工具!
尽情享用！ 尽情享用！ 尽情享用！`,
	DisableAutoGenTag: true,
	CompletionOptions: cobra.CompletionOptions{
		DisableDefaultCmd:   true,
		DisableNoDescFlag:   true,
		DisableDescriptions: true,
		HiddenDefaultCmd:    true,
	},
}

func init() {
	// 初始化配置
	cobra.OnInitialize(config.InitConfig)
}

func Execute() {

	// 1. base命令
	baseCmd := &cobra.Command{
		Use:   "base",
		Short: "服务器基础操作",
	}
	addBaseSubcommands(baseCmd)

	// 2. zsh命令
	zshCmd := &cobra.Command{
		Use:   "zsh",
		Short: "zsh相关的内容",
	}
	addZshSubcommands(zshCmd)

	// 3. proxy命令
	proxyCmd := &cobra.Command{
		Use:   "proxy",
		Short: "主机代理相关的内容",
	}
	addProxySubcommands(proxyCmd)

	// 4. acme命令
	acmeCmd := &cobra.Command{
		Use:   "acme",
		Short: "acme相关的内容",
	}
	addAcmeSubcommands(acmeCmd)

	// 5. wdd命令
	wddCmd := &cobra.Command{
		Use:   "wdd",
		Short: "WDD相关操作",
	}
	addWddSubcommands(wddCmd)

	// 6. security命令
	securityCmd := &cobra.Command{
		Use:   "security",
		Short: "安全相关操作",
	}

	// 7. info命令
	infoCmd := &cobra.Command{
		Use:   "info",
		Short: "主机信息",
	}
	addInfoSubcommands(infoCmd)

	// 8. version命令
	versionCmd := &cobra.Command{
		Use:   "version",
		Short: "打印版本信息",
		Run: func(cmd *cobra.Command, args []string) {
			// 实现version逻辑
			fmt.Println("版本信息: 1.0.0")
		},
	}

	// 9. config命令
	configCmd := &cobra.Command{
		Use:   "config",
		Short: "配置文件管理",
	}
	addConfigSubcommands(configCmd)

	// 10. download命令
	downloadCmd := &cobra.Command{
		Use:   "download",
		Short: "文件下载，直接下载 [url] [dest_path]",
	}

	addDownloadSubcommands(downloadCmd)

	// 11. cert命令
	certCmd := &cobra.Command{
		Use:   "cert",
		Short: "SSL证书管理",
	}
	addCertManagerSubcommands(certCmd)

	helpCmd := &cobra.Command{
		Use:   "help",
		Short: "帮助信息",
		Run: func(cmd *cobra.Command, args []string) {
			// 在你的命令执行函数中
			// cmd.Root().SetHelpFunc(func(c *cobra.Command, s []string) {
			// 	// 直接使用默认帮助模板
			// 	c.Usage()
			// })

			// 自定义实现 打印全部的命令
			printAllCommands(cmd.Root(), os.Stdout)
		},
	}

	// 添加所有根命令
	rootCmd.AddCommand(
		baseCmd,
		zshCmd,
		proxyCmd,
		acmeCmd,
		wddCmd,
		securityCmd,
		infoCmd,
		versionCmd,
		configCmd,
		downloadCmd,
		certCmd,
		helpCmd,
	)

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

}

func printAllCommands(cmd *cobra.Command, writer io.Writer) {
	// 打印命令标题
	fmt.Fprintf(writer, "All available commands:\n\n")

	// 递归打印命令树
	var traverse func(*cobra.Command, int)
	traverse = func(c *cobra.Command, depth int) {
		if c.Hidden || c.Deprecated != "" {
			return
		}

		// 生成命令前缀缩进
		indent := strings.Repeat("  ", depth)

		// 打印当前命令信息
		fmt.Fprintf(writer, "%s%-20s%s\n",
			indent,
			c.Name(),
			strings.TrimSpace(c.Short))

		// 递归子命令
		for _, subCmd := range c.Commands() {
			traverse(subCmd, depth+1)
		}
	}

	// 从根命令开始遍历
	for _, c := range cmd.Commands() {
		traverse(c, 0)
	}
}