编写 kubectl 插件

➜  ~ tree .kube
.kube
├── config
└── plugins
    └── swns
        ├── plugin.yaml
        └── swns

name: "targaryen"                 # 必填项: 用于 kuebctl 调用的插件名称
shortDesc: "Dragonized plugin"    # 必填项: 用于 help 该插件时的简短描述
longDesc: ""                      # 非必填: 插件的长描述
example: ""                       # 非必填: 插件的使用样例
command: "./dracarys"             # 必填项: 插件实际执行的文件位置，可以相对路径 or 绝对路径，或者在 PATH 里也行
flags:                            # 非必填: 插件支持的 flag
  - name: "heat"                  # 必填项: 如果你写了支持的 flag，那么此项必填
    shorthand: "h"                # 非必填: 该选项的缩短形式
    desc: "Fire heat"             # 必填项: 同样每个 flag 都必须书写描述
    defValue: "extreme"           # 非必填: 默认值
tree:                             # 允许定义一些子命令
  - ...                           # 子命令支持同样的设置属性(我想知道子命令的子命令的子命令支不支持...我还没去试过)

name: "swns"
shortDesc: "Switch NameSpace"
longDesc: "Switch Kubernetes current context namespace."
example: "kubectl plugin swns [NAMESPACE]"
command: "./swns"

// 注意: 下面的模板语法大括号中间没有空格，此处空格是为了防止博客渲染出错

package main

import (
	"fmt"
	"os"
	"os/exec"
	"strings"

	"github.com/mritd/promptx"
)

func main() {

	// 先拿到当前的 context
	cmd := exec.Command("kubectl", "config", "current-context")
	cmd.Stdin = os.Stdin
	cmd.Stderr = os.Stderr

	b, err := cmd.Output()
	checkAndExit(err)
	currentContext := strings.TrimSpace(string(b))

	// 如果提供了 NameSpace 字符串，我直接改就行了
	if len(os.Args) > 1 {
		cmd = exec.Command("kubectl", "config", "set-context", currentContext, "--namespace="+os.Args[1])
		cmd.Stdout = os.Stdout
		checkAndExit(cmd.Run())
		fmt.Printf("Kubernetes namespace switch to %s.\n", os.Args[1])
	} else {
		// 没提供我就得先把所有的 NameSpace 弄出来
		cmd = exec.Command("kubectl", "get", "ns", "-o", "template", "--template", "{ { range .items } }{ { .metadata.name } } { { end } }")
		b, err = cmd.Output()
		checkAndExit(err)
		allNameSpace := strings.Fields(string(b))

		// 弄到所有的 NameSpace 后，我在弄一个下拉列表(这是我自己造的一个下拉列表库)
		cfg := &promptx.SelectConfig{
			ActiveTpl:    "»  { { . | cyan } }",
			InactiveTpl:  "  { { . | white } }",
			SelectPrompt: "NameSpace",
			SelectedTpl:  "{ { \"» \" | green } }{ {\"NameSpace:\" | cyan } } { { . } }",
			DisPlaySize:  9,
			DetailsTpl:   ` `,
		}
		s := &promptx.Select{
			Items:  allNameSpace,
			Config: cfg,
		}

		// 用户选中一个 NameSpace 后我就拿到了想要设置的 NameSpace 字符串
		selectNameSpace := allNameSpace[s.Run()]

		// 跟上面套路一样，写进去就行了
		cmd = exec.Command("kubectl", "config", "set-context", currentContext, "--namespace="+selectNameSpace)
		cmd.Stdout = os.Stdout
		checkAndExit(cmd.Run())
		fmt.Printf("Kubernetes namespace switch to %s.\n", selectNameSpace)
	}
}

func checkErr(err error) bool {
	if err != nil {
		fmt.Println(err)
		return false
	}
	return true
}

func checkAndExit(err error) {
	if !checkErr(err) {
		os.Exit(1)
	}
}