VERSE
ebde9ba68f
- I think it would be better if one for commands only, one for the spec handling core logic so people won't be confused by the core logic inside the commands folder (cleaner I guess) - Regen docs
106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package ops
|
|
|
|
import (
|
|
"bufio"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/versenilvis/iris/spec"
|
|
)
|
|
|
|
func sshHostGenerator(tokens []string, _ string, _ string) []spec.Suggestion {
|
|
home, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
paths := []string{
|
|
filepath.Join(home, ".ssh", "config"),
|
|
"/etc/ssh/ssh_config",
|
|
}
|
|
|
|
seen := make(map[string]bool)
|
|
var results []spec.Suggestion
|
|
|
|
for _, path := range paths {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
scanner := bufio.NewScanner(f)
|
|
for scanner.Scan() {
|
|
parts := strings.Fields(scanner.Text())
|
|
if len(parts) < 2 || !strings.EqualFold(parts[0], "host") {
|
|
continue
|
|
}
|
|
|
|
for _, host := range parts[1:] {
|
|
// skip wildcards
|
|
if strings.ContainsAny(host, "*?!") {
|
|
continue
|
|
}
|
|
if seen[host] {
|
|
continue
|
|
}
|
|
seen[host] = true
|
|
results = append(results, spec.Suggestion{Cmd: host, Desc: "ssh host"})
|
|
}
|
|
}
|
|
_ = scanner.Err()
|
|
_ = f.Close()
|
|
}
|
|
|
|
return results
|
|
}
|
|
|
|
func init() {
|
|
sshOptions := []spec.Option{
|
|
{Name: "-p", Description: "port"},
|
|
{Name: "-i", Description: "identity file"},
|
|
{Name: "-L", Description: "local port forward"},
|
|
{Name: "-R", Description: "remote port forward"},
|
|
{Name: "-D", Description: "dynamic socks proxy"},
|
|
{Name: "-N", Description: "no remote command"},
|
|
{Name: "-f", Description: "background"},
|
|
{Name: "-v", Description: "verbose"},
|
|
{Name: "-A", Description: "agent forwarding"},
|
|
{Name: "-X", Description: "x11 forwarding"},
|
|
}
|
|
|
|
spec.Register(&spec.Spec{
|
|
Name: "ssh",
|
|
Description: "secure shell",
|
|
Generator: sshHostGenerator,
|
|
Options: sshOptions,
|
|
})
|
|
|
|
spec.Register(&spec.Spec{
|
|
Name: "scp",
|
|
Description: "secure copy",
|
|
Generator: sshHostGenerator,
|
|
Options: []spec.Option{
|
|
{Name: "-r", Description: "recursive"},
|
|
{Name: "-p", Description: "port"},
|
|
{Name: "-i", Description: "identity file"},
|
|
{Name: "-P", Description: "remote port"},
|
|
},
|
|
})
|
|
|
|
spec.Register(&spec.Spec{
|
|
Name: "rsync",
|
|
Description: "remote sync",
|
|
Generator: sshHostGenerator,
|
|
Options: []spec.Option{
|
|
{Name: "-av", Description: "archive + verbose"},
|
|
{Name: "-z", Description: "compress"},
|
|
{Name: "--delete", Description: "delete extraneous"},
|
|
{Name: "--progress", Description: "show progress"},
|
|
{Name: "-e", Description: "remote shell command"},
|
|
{Name: "--exclude", Description: "exclude pattern"},
|
|
{Name: "--dry-run", Description: "simulation only"},
|
|
},
|
|
})
|
|
}
|