110 lines
2.1 KiB
Bash
110 lines
2.1 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
print_long_help() {
|
||
|
|
cat <<EOF
|
||
|
|
Start a Pluto Notebook server through an ssh connection
|
||
|
|
|
||
|
|
USAGE:
|
||
|
|
rpluto [--help] [-p <host-port>] <local-port> <ssh-addr>
|
||
|
|
|
||
|
|
FLAGS:
|
||
|
|
-h, --help show this help message and exit
|
||
|
|
|
||
|
|
OPTIONS:
|
||
|
|
-p <host-port> Port use for on host for the Pluto Server
|
||
|
|
|
||
|
|
ARGS:
|
||
|
|
<local-port> Port on client to connect Pluto Server
|
||
|
|
<ssh-addr> Either an ~/.ssh/config alias or the host address
|
||
|
|
EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
print_short_help() {
|
||
|
|
cat <<EOF
|
||
|
|
USAGE:
|
||
|
|
rpluto [--help] [-p <host-port>] <local-port> <ssh-addr>
|
||
|
|
|
||
|
|
For more information try --help
|
||
|
|
EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
print_start_msg() {
|
||
|
|
cat <<EOF
|
||
|
|
Starting remote Pluto Notebook. This may take a few seconds...
|
||
|
|
Wait for a popup or connect here using a browser:
|
||
|
|
|
||
|
|
http://localhost:$1
|
||
|
|
|
||
|
|
When you're finished type ^C in this terminal
|
||
|
|
EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
# 1: Client port
|
||
|
|
# 2: SSH alias or address
|
||
|
|
# [3]: Host port
|
||
|
|
ssh_pluto_tunnel() {
|
||
|
|
if [[ "$(uname -s)" == "Darwin" ]]; then
|
||
|
|
sleep 4 && open -n -a "Google Chrome" --args --incognito "http://localhost:$1" &
|
||
|
|
elif [[ "$(uname -s)" == "Linux" ]]; then
|
||
|
|
sleep 4 && chromium --incognito "http://localhost:$1" &>/dev/null
|
||
|
|
fi
|
||
|
|
|
||
|
|
ssh -L "$1":localhost:"${3:-7000}" "$2"\
|
||
|
|
"julia -e\
|
||
|
|
\"import Pluto;
|
||
|
|
Pluto.run(
|
||
|
|
port=${3:-7000},
|
||
|
|
require_secret_for_open_links=false,
|
||
|
|
require_secret_for_access=false
|
||
|
|
)\""
|
||
|
|
}
|
||
|
|
|
||
|
|
exit_msg(){
|
||
|
|
printf "\nNotebook has been shutdown\n"
|
||
|
|
exit 0
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
# Print help and exit ====
|
||
|
|
if [[ $1 == '--help' ]]; then
|
||
|
|
print_long_help
|
||
|
|
exit 0
|
||
|
|
elif [[ "$1" =~ '-h' ]]; then
|
||
|
|
print_short_help
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
# Parse arguments ====
|
||
|
|
positional_args=()
|
||
|
|
|
||
|
|
while [[ "$#" -gt 0 ]]; do
|
||
|
|
arg="$1"
|
||
|
|
|
||
|
|
case $arg in
|
||
|
|
-p)
|
||
|
|
host_port="$2"
|
||
|
|
shift
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
positional_args+=("$1")
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
done
|
||
|
|
|
||
|
|
set -- "${positional_args[@]}"
|
||
|
|
|
||
|
|
# Set graceful exit
|
||
|
|
if [[ "$#" -eq 2 ]]; then
|
||
|
|
trap exit_msg SIGINT
|
||
|
|
|
||
|
|
print_start_msg "$1"
|
||
|
|
|
||
|
|
ssh_pluto_tunnel $1 $2 $host_port
|
||
|
|
else
|
||
|
|
print_short_help
|
||
|
|
exit 2
|
||
|
|
fi
|