50 lines
1 KiB
Text
50 lines
1 KiB
Text
|
#!/usr/bin/env nu
|
||
|
|
# TODO: add tunnel show subcmd to quickly view open tunnels
|
||
|
|
|
||
|
|
def start-ssh-controller [host: string] {
|
||
|
|
print "initializing connection to $host"
|
||
|
|
ssh -M -f -N $host
|
||
|
|
}
|
||
|
|
|
||
|
|
def is-null [ name: string, arg: any] {
|
||
|
|
if $arg == null {
|
||
|
|
print $"expected argument for --($name)"
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# script `--help` will improve :)
|
||
|
|
# https://github.com/nushell/nushell/pull/13445
|
||
|
|
|
||
|
|
# control background tunnels
|
||
|
|
# usage:
|
||
|
|
# tunnel up -p 4321 -h computer
|
||
|
|
# tunnel down -h computer
|
||
|
|
def main [] {
|
||
|
|
print "use subcommands up/down to control ssh tunnels"
|
||
|
|
help main
|
||
|
|
}
|
||
|
|
|
||
|
|
# activate ssh tunnel
|
||
|
|
def "main up" [
|
||
|
|
--port: int # port number
|
||
|
|
--host: string # host name
|
||
|
|
] {
|
||
|
|
is-null "port" $port
|
||
|
|
is-null "host" $host
|
||
|
|
|
||
|
|
if (ssh -O check $host | complete).exit_code != 0 {
|
||
|
|
start-ssh-controller $host
|
||
|
|
}
|
||
|
|
ssh -fNL $"($port):localhost:($port)" $host
|
||
|
|
}
|
||
|
|
|
||
|
|
# shutdown ssh tunnels on host
|
||
|
|
def "main down" [
|
||
|
|
--host: string # host name
|
||
|
|
] {
|
||
|
|
is-null "host" $host
|
||
|
|
print "exiting connection to $host"
|
||
|
|
ssh -O exit $host
|
||
|
|
}
|