105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"log"
|
|
"io/ioutil"
|
|
"fmt"
|
|
"strings"
|
|
"os/user"
|
|
"github.com/manifoldco/promptui"
|
|
)
|
|
|
|
type host struct {
|
|
Name string
|
|
HostName string
|
|
User string
|
|
}
|
|
|
|
func readFile(file string) string {
|
|
contents, err := ioutil.ReadFile(file)
|
|
if err != nil {
|
|
log.Fatalf("%q does not exist", file)
|
|
}
|
|
return fmt.Sprintf("%s", contents)
|
|
}
|
|
|
|
func readHosts(lines []string) []host {
|
|
hosts := make([]host,0)
|
|
current := host{}
|
|
|
|
for _, line := range lines {
|
|
fields := strings.Fields(line)
|
|
if len(fields) <= 1 {
|
|
continue
|
|
}
|
|
nameLower := strings.ToLower(fields[0])
|
|
if nameLower == "host" {
|
|
if current.Name != "" {
|
|
hosts = append(hosts, current)
|
|
}
|
|
currentUser, err := user.Current()
|
|
currentUserName := ""
|
|
if err == nil {
|
|
currentUserName = currentUser.Username
|
|
}
|
|
current = host{Name:fields[1], HostName: fields[1], User: currentUserName}
|
|
}
|
|
if nameLower == "hostname" {
|
|
current.HostName = fields[1]
|
|
}
|
|
if nameLower == "user" {
|
|
current.User = fields[1]
|
|
}
|
|
}
|
|
if current.Name != "" {
|
|
hosts = append(hosts, current)
|
|
}
|
|
|
|
return hosts
|
|
}
|
|
|
|
func main() {
|
|
args := os.Args[1:]
|
|
if len(args) == 0 {
|
|
log.Fatalf("Usage: %s [configfile]", os.Args[0])
|
|
}
|
|
fileName := args[0]
|
|
lines := strings.Split(readFile(fileName), "\n")
|
|
for i, line := range lines {
|
|
lines[i] = strings.Trim(line, "\r\t ")
|
|
}
|
|
hosts := readHosts(lines)
|
|
|
|
searcher := func(input string, index int) bool {
|
|
host := hosts[index]
|
|
name := strings.Replace(strings.ToLower(host.Name+host.User+host.HostName), " ", "", -1)
|
|
input = strings.Replace(strings.ToLower(input), " ", "", -1)
|
|
|
|
return strings.Contains(name, input)
|
|
}
|
|
|
|
templates := &promptui.SelectTemplates {
|
|
Label: "{{ . }}",
|
|
Active: ">{{ .User | cyan }}@({{ .HostName | red }})",
|
|
Inactive: " {{ .User | cyan }}@({{ .HostName | red }})",
|
|
Selected: "{{ .User | cyan }}@({{ .HostName | red }})",
|
|
}
|
|
|
|
prompt := promptui.Select{
|
|
Label: "Select Host/Option",
|
|
Items: hosts,
|
|
Templates: templates,
|
|
Size: 4,
|
|
Searcher: searcher,
|
|
}
|
|
|
|
i, _, err := prompt.Run()
|
|
|
|
if err != nil {
|
|
fmt.Printf("Prompt failed %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("You choose number %d: %s\n", i+1, hosts[i].Name)
|
|
} |