gsh.go

package main

import (
	"fmt"
	"log"
	"os/user"
	"path"

	"github.com/alexflint/go-arg"
	"github.com/fatih/color"
	"github.com/ktr0731/go-fuzzyfinder"
)

const (
	defaultConfigPath = ".config/gsh.yml"
)

// args - application args
type args struct {
	Config    string   `arg:"-c,help:path to config"`
	PrintOnly bool     `arg:"-p,help:only print ssh command"`
	Patterns  []string `arg:"positional,help:patterns to match server name hostname and user"`
}

func main() {
	usr, err := user.Current()
	if err != nil {
		log.Fatal("Unable to get current user: ", err)
	}

	// default arguments
	var args args
	args.Config = path.Join(usr.HomeDir, defaultConfigPath)
	args.PrintOnly = false

	arg.MustParse(&args)

	// get server list from config file
	var servers []Server
	if servers, err = readConfig(args.Config, usr.Username, 22); err != nil {
		log.Fatalf("Failed to read config file '%s': %s\n", args.Config, err)
	}

	// match servers against patterns
	var matched []Server
	for _, s := range servers {
		if s.matchAll(args.Patterns) == true {
			matched = append(matched, s)
		}
	}

	if len(matched) == 0 {
		color.Red("no server matched")
		return
	}

	if len(matched) == 1 {
		connect(matched[0], args.PrintOnly)
		return
	}

	idx, err := fuzzyfinder.Find(matched, func(i int) string {
		return fmt.Sprintf("%s", matched[i].String())
	})
	if err != nil {
		if err != fuzzyfinder.ErrAbort {
			log.Fatalf("Failed to find server: %s", err)
		}
		return
	}

	connect(matched[idx], args.PrintOnly)
}