kt-connect/pkg/kt/command/check.go

package command

import (
	osexec "os/exec"

	"github.com/alibaba/kt-connect/pkg/kt"

	"runtime"

	"github.com/alibaba/kt-connect/pkg/kt/options"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	urfave "github.com/urfave/cli"

	"github.com/alibaba/kt-connect/pkg/kt/exec"
)

// NewCheckCommand return new check command
func NewCheckCommand(cli kt.CliInterface, options *options.DaemonOptions, action ActionInterface) urfave.Command {
	return urfave.Command{
		Name:  "check",
		Usage: "check local dependency for ktctl",
		Action: func(c *urfave.Context) error {
			if options.Debug {
				zerolog.SetGlobalLevel(zerolog.DebugLevel)
			}
			return action.Check(cli)
		},
	}
}

// Check check local denpendency for kt connect
func (action *Action) Check(cli kt.CliInterface) (err error) {
	log.Info().Msgf("system info %s-%s", runtime.GOOS, runtime.GOARCH)

	err = runCommandWithMsg(
		cli.Exec().SSH().Version(),
		"checking ssh version", "ssh is missing, please make sure command ssh is work right at your local first",
	)

	if err != nil {
		return
	}

	err = runCommandWithMsg(
		cli.Exec().Kubectl().Version(),
		"checking kubectl version", "kubectl is missing, please make sure kubectl is working right at your local first",
	)

	if err != nil {
		return
	}

	err = runCommandWithMsg(
		cli.Exec().SSHUttle().Version(),
		"checking sshuttle version", "sshuttle is missing, you can only use 'ktctl connect --method socks5' with Socks5 proxy mode",
	)

	if err != nil {
		return
	}

	log.Info().Msg("KT Connect is ready, enjoy it!")
	return nil
}

func runCommandWithMsg(cmd *osexec.Cmd, title string, msg string) (err error) {
	log.Info().Msg(title)
	err = exec.RunAndWait(cmd, title, true)
	if err != nil {
		log.Warn().Msg(msg)
	}
	return
}