107 lines
2.1 KiB
Plaintext
107 lines
2.1 KiB
Plaintext
use re
|
|
use str
|
|
use moi/util/condition
|
|
use moi/util/file
|
|
use moi/util/ip
|
|
use moi/util/list
|
|
use moi/util/map
|
|
use moi/util/option
|
|
|
|
var options-ssh = [ 1 2 4 6 A D f g I i L l m o v a b C c e F k N n p q R s T t X x ]
|
|
var options-scp = [ 3 4 6 B C p q r v c F i l o P S ]
|
|
|
|
var names = [
|
|
benjamin
|
|
luence
|
|
]
|
|
|
|
fn -kh {
|
|
cat ~/.ssh/known_hosts | peach {|l|
|
|
put [(str:split ' ' $l)]
|
|
} | peach {|e|
|
|
var domains @_ = $@e
|
|
str:split ',' $domains
|
|
}
|
|
}
|
|
|
|
fn -port {|cmd @argv|
|
|
var o = (condition:cset (eq $cmd 'ssh') '-p' '-P')
|
|
var margs = (option:map $argv)
|
|
var p = (map:value-of $margs $o &default=[])
|
|
if (list:empty $p) {
|
|
put 22
|
|
} else {
|
|
put $p[-1]
|
|
}
|
|
}
|
|
|
|
fn -complete-names {|| each {|n| put $n'@' } $names }
|
|
|
|
fn -complete-domains {|name hosts|
|
|
each {|h|
|
|
put $name@$h
|
|
} $hosts
|
|
}
|
|
|
|
fn -complete-remote-dir {|port address dir|
|
|
try {
|
|
ssh -p $port $address 'ls '$dir | each {|f|
|
|
put $address':'$dir$f
|
|
}
|
|
} except e {
|
|
}
|
|
}
|
|
|
|
fn -complete-args {|hosts cmd @argv|
|
|
var arg = $argv[-1]
|
|
var i = (str:index $arg @)
|
|
if (< $i 0) {
|
|
-complete-names
|
|
all $hosts
|
|
if (eq $cmd scp) {
|
|
edit:complete-filename $cmd $@argv
|
|
}
|
|
return
|
|
}
|
|
var n h = $arg[..$i] $arg[(+ $i 1)..]
|
|
if (eq $cmd scp) {
|
|
set i = (str:index $h :)
|
|
if (>= $i 0) {
|
|
var d = $h[(+ $i 1)..]
|
|
set h = $h[..$i]
|
|
if (list:includes $h $hosts) {
|
|
var p = (-port $cmd @argv)
|
|
var e = [ (str:split / $d) ]
|
|
if (not-eq $e[-1] '') {
|
|
set d = (str:join / $e[..-1])
|
|
if (eq $d '') {
|
|
set d = /
|
|
}
|
|
}
|
|
-complete-remote-dir $p $n@$h $d
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
-complete-domains $n $hosts
|
|
}
|
|
|
|
fn complete {|@argv|
|
|
var @hosts = (-kh)
|
|
var cmd = $argv[0]
|
|
var is-ssh = (eq $cmd ssh)
|
|
var po = (condition:cset $is-ssh -p -P)
|
|
if (<= (count $argv) 2) {
|
|
all (condition:cset $is-ssh $options-ssh $options-scp)
|
|
-complete-args $hosts $@argv
|
|
} elif (eq $argv[-2] $po) {
|
|
put 22
|
|
} else {
|
|
-complete-args $hosts $@argv
|
|
}
|
|
}
|
|
|
|
set edit:completion:arg-completer[scp] = $complete~
|
|
set edit:completion:arg-completer[ssh] = $complete~
|