abra-capsul/abra-capsul

160 lines
7.3 KiB
Bash
Executable File

#!/usr/bin/env bash
DOC="
Capsul provider plugin for abra
Usage:
abra-capsul <name> <size> <os> <ssh-key>
ENVIRONMENT
Read configuration from these variables, or prompt if unset:
- \$CAPSUL_API_TOKEN
- \$CAPSUL_SERVER
"
# docopt parser below, refresh this parser with `docopt.sh abra-capsul`
# shellcheck disable=2016,1075
docopt() { parse() { if ${DOCOPT_DOC_CHECK:-true}; then local doc_hash
if doc_hash=$(printf "%s" "$DOC" | (sha256sum 2>/dev/null || shasum -a 256)); then
if [[ ${doc_hash:0:5} != "$digest" ]]; then
stderr "The current usage doc (${doc_hash:0:5}) does not match \
what the parser was generated with (${digest})
Run \`docopt.sh\` to refresh the parser."; _return 70; fi; fi; fi
local root_idx=$1; shift; argv=("$@"); parsed_params=(); parsed_values=()
left=(); testdepth=0; local arg; while [[ ${#argv[@]} -gt 0 ]]; do
if [[ ${argv[0]} = "--" ]]; then for arg in "${argv[@]}"; do
parsed_params+=('a'); parsed_values+=("$arg"); done; break
elif [[ ${argv[0]} = --* ]]; then parse_long
elif [[ ${argv[0]} = -* && ${argv[0]} != "-" ]]; then parse_shorts
elif ${DOCOPT_OPTIONS_FIRST:-false}; then for arg in "${argv[@]}"; do
parsed_params+=('a'); parsed_values+=("$arg"); done; break; else
parsed_params+=('a'); parsed_values+=("${argv[0]}"); argv=("${argv[@]:1}"); fi
done; local idx; if ${DOCOPT_ADD_HELP:-true}; then
for idx in "${parsed_params[@]}"; do [[ $idx = 'a' ]] && continue
if [[ ${shorts[$idx]} = "-h" || ${longs[$idx]} = "--help" ]]; then
stdout "$trimmed_doc"; _return 0; fi; done; fi
if [[ ${DOCOPT_PROGRAM_VERSION:-false} != 'false' ]]; then
for idx in "${parsed_params[@]}"; do [[ $idx = 'a' ]] && continue
if [[ ${longs[$idx]} = "--version" ]]; then stdout "$DOCOPT_PROGRAM_VERSION"
_return 0; fi; done; fi; local i=0; while [[ $i -lt ${#parsed_params[@]} ]]; do
left+=("$i"); ((i++)) || true; done
if ! required "$root_idx" || [ ${#left[@]} -gt 0 ]; then error; fi; return 0; }
parse_shorts() { local token=${argv[0]}; local value; argv=("${argv[@]:1}")
[[ $token = -* && $token != --* ]] || _return 88; local remaining=${token#-}
while [[ -n $remaining ]]; do local short="-${remaining:0:1}"
remaining="${remaining:1}"; local i=0; local similar=(); local match=false
for o in "${shorts[@]}"; do if [[ $o = "$short" ]]; then similar+=("$short")
[[ $match = false ]] && match=$i; fi; ((i++)) || true; done
if [[ ${#similar[@]} -gt 1 ]]; then
error "${short} is specified ambiguously ${#similar[@]} times"
elif [[ ${#similar[@]} -lt 1 ]]; then match=${#shorts[@]}; value=true
shorts+=("$short"); longs+=(''); argcounts+=(0); else value=false
if [[ ${argcounts[$match]} -ne 0 ]]; then if [[ $remaining = '' ]]; then
if [[ ${#argv[@]} -eq 0 || ${argv[0]} = '--' ]]; then
error "${short} requires argument"; fi; value=${argv[0]}; argv=("${argv[@]:1}")
else value=$remaining; remaining=''; fi; fi; if [[ $value = false ]]; then
value=true; fi; fi; parsed_params+=("$match"); parsed_values+=("$value"); done
}; parse_long() { local token=${argv[0]}; local long=${token%%=*}
local value=${token#*=}; local argcount; argv=("${argv[@]:1}")
[[ $token = --* ]] || _return 88; if [[ $token = *=* ]]; then eq='='; else eq=''
value=false; fi; local i=0; local similar=(); local match=false
for o in "${longs[@]}"; do if [[ $o = "$long" ]]; then similar+=("$long")
[[ $match = false ]] && match=$i; fi; ((i++)) || true; done
if [[ $match = false ]]; then i=0; for o in "${longs[@]}"; do
if [[ $o = $long* ]]; then similar+=("$long"); [[ $match = false ]] && match=$i
fi; ((i++)) || true; done; fi; if [[ ${#similar[@]} -gt 1 ]]; then
error "${long} is not a unique prefix: ${similar[*]}?"
elif [[ ${#similar[@]} -lt 1 ]]; then
[[ $eq = '=' ]] && argcount=1 || argcount=0; match=${#shorts[@]}
[[ $argcount -eq 0 ]] && value=true; shorts+=(''); longs+=("$long")
argcounts+=("$argcount"); else if [[ ${argcounts[$match]} -eq 0 ]]; then
if [[ $value != false ]]; then
error "${longs[$match]} must not have an argument"; fi
elif [[ $value = false ]]; then
if [[ ${#argv[@]} -eq 0 || ${argv[0]} = '--' ]]; then
error "${long} requires argument"; fi; value=${argv[0]}; argv=("${argv[@]:1}")
fi; if [[ $value = false ]]; then value=true; fi; fi; parsed_params+=("$match")
parsed_values+=("$value"); }; required() { local initial_left=("${left[@]}")
local node_idx; ((testdepth++)) || true; for node_idx in "$@"; do
if ! "node_$node_idx"; then left=("${initial_left[@]}"); ((testdepth--)) || true
return 1; fi; done; if [[ $((--testdepth)) -eq 0 ]]; then
left=("${initial_left[@]}"); for node_idx in "$@"; do "node_$node_idx"; done; fi
return 0; }; value() { local i; for i in "${!left[@]}"; do local l=${left[$i]}
if [[ ${parsed_params[$l]} = "$2" ]]; then
left=("${left[@]:0:$i}" "${left[@]:((i+1))}")
[[ $testdepth -gt 0 ]] && return 0; local value
value=$(printf -- "%q" "${parsed_values[$l]}"); if [[ $3 = true ]]; then
eval "var_$1+=($value)"; else eval "var_$1=$value"; fi; return 0; fi; done
return 1; }; stdout() { printf -- "cat <<'EOM'\n%s\nEOM\n" "$1"; }; stderr() {
printf -- "cat <<'EOM' >&2\n%s\nEOM\n" "$1"; }; error() {
[[ -n $1 ]] && stderr "$1"; stderr "$usage"; _return 1; }; _return() {
printf -- "exit %d\n" "$1"; exit "$1"; }; set -e; trimmed_doc=${DOC:1:193}
usage=${DOC:34:49}; digest=9b8b6; shorts=(); longs=(); argcounts=(); node_0(){
value _name_ a; }; node_1(){ value _size_ a; }; node_2(){ value _os_ a; }
node_3(){ value _ssh_key_ a; }; node_4(){ required 0 1 2 3; }; node_5(){
required 4; }; cat <<<' docopt_exit() { [[ -n $1 ]] && printf "%s\n" "$1" >&2
printf "%s\n" "${DOC:34:49}" >&2; exit 1; }'; unset var__name_ var__size_ \
var__os_ var__ssh_key_; parse 5 "$@"; local prefix=${DOCOPT_PREFIX:-''}
unset "${prefix}_name_" "${prefix}_size_" "${prefix}_os_" "${prefix}_ssh_key_"
eval "${prefix}"'_name_=${var__name_:-}'
eval "${prefix}"'_size_=${var__size_:-}'; eval "${prefix}"'_os_=${var__os_:-}'
eval "${prefix}"'_ssh_key_=${var__ssh_key_:-}'; local docopt_i=1
[[ $BASH_VERSION =~ ^4.3 ]] && docopt_i=2; for ((;docopt_i>0;docopt_i--)); do
declare -p "${prefix}_name_" "${prefix}_size_" "${prefix}_os_" \
"${prefix}_ssh_key_"; done; }
# docopt parser above, complete command for generating this parser is `docopt.sh abra-capsul`
ABRA_DIR="${ABRA_DIR:-$HOME/.abra}"
ABRA_VENDOR_DIR="$ABRA_DIR/vendor"
JQ="$ABRA_VENDOR_DIR/jq"
abra_capsul() {
declare abra_capsul__name_ abra_capsul__size_ abra_capsul__os_ \
abra_capsul__ssh_key_
DOCOPT_PREFIX=abra_capsul_
eval "$(docopt "$@")"
if ! type curl > /dev/null 2>&1; then
error "curl program is not installed"
fi
if [[ -z "$CAPSUL_API_TOKEN" ]]; then
read -rp "Capsul API token: " CAPSUL_API_TOKEN
fi
if [[ -z "$CAPSUL_SERVER" ]]; then
read -rp "Capsul server address: " CAPSUL_SERVER
fi
NAME="$abra_capsul__name_"
SIZE="$abra_capsul__size_"
OS="$abra_capsul__os_"
SSH_KEY="$abra_capsul__ssh_key_"
response=$(curl --silent \
-X POST \
-H "Authorization: $CAPSUL_API_TOKEN" \
-H "Content-Type: application/json" \
-d "{\"name\":\"$NAME\", \
\"size\": \"$SIZE\", \
\"os\": \"$OS\", \
\"ssh_key_0\": \"$SSH_KEY\" \
}" \
"$CAPSUL_SERVER/api/capsul/create")
vm_id=$(echo "$response" | $JQ -r ".id")
if [ "$vm_id" == "null" ] || [ -z "$vm_id" ]; then
error_msg=$(echo "$response" | jq -r ".errors")
echo "Oops, something went wrong. Capsul API responded with: $error_msg"
exit 1
fi
echo "Your new Capsul VPS is up, with ID $vm_id. Enjoy!"
}
abra_capsul "$@"