dotfiles/overlays/itp/itp.sh
Magic_RB 9b371b8662
Add InfluxDB provisioning script
Signed-off-by: Magic_RB <magic_rb@redalder.org>
2023-09-03 18:03:00 +02:00

201 lines
4.5 KiB
Bash

# -*- sh-basic-offset: 2 indent-tabs-mode: nil -*-
declare -a _positional_args
declare _dry_run=no
declare _state_file
declare _task_file
declare _org
while [[ $# -gt 0 ]]; do
case $1 in
-n|--dry-run)
_dry_run="yes"
shift # past argument
;;
-s|--state)
if [[ -z "${2:-}" ]] ; then
echo "$1 takes one argument"
exit 2
fi
_state_file="$2"
shift # past argument
shift # past value
;;
-f|--file)
if [[ -z "${2:-}" ]] ; then
echo "$1 takes one argument"
exit 2
fi
_task_file="$2"
shift # past argument
shift # past value
;;
-o|--org)
if [[ -z "${2:-}" ]] ; then
echo "$1 takes one argument"
exit 2
fi
_org="$2"
shift # past argument
shift # past value
;;
--*|-*)
echo "Unknown option $1"
exit 1
;;
*)
_positional_args+=("$1") # save positional arg
shift # past argument
;;
esac
done
set -- "${_positional_args[@]}"
if [[ -z "${_state_file:-}" ]] ; then
echo "-s|--state must be specified"
exit 2
fi
if [[ -z "${_task_file:-}" ]] ; then
echo "-f|--file must be specified"
exit 2
fi
if ! [[ -f "$_task_file" ]] ; then
echo "$_task_file must exist and be writable"
exit 2
fi
if ! [[ -w "$_state_file" ]] ; then
touch "$_state_file" || ( echo "Couldn't create state file" ; exit 2 )
echo "[]" > "$_state_file"
fi
if [[ -z "${_org:-}" ]] ; then
echo "-o|--org must be specified"
exit 2
fi
if [[ -z "${INFLUX_HOST:-}" ]] ; then
echo "INFLUX_HOST must be set"
exit 2
fi
if [[ -z "${INFLUX_TOKEN:-}" ]] ; then
echo "INFLUX_TOKEN must be set"
exit 2
fi
if ! influx task list --org "$_org" >/dev/null 2>&1 ; then
echo "Failed to establish connection to InfuxDB"
exit 2
fi
declare -a _tasks_to_remove
declare -a _tasks_to_add
declare -a _tasks_to_update
mapfile -t _tasks_to_remove < <(comm -23 <(jq -r '.[].name' < "$_state_file" | sort) <(jq -r '.[].name' < "$_task_file" | sort))
mapfile -t _tasks_to_add < <(comm -13 <(jq -r '.[].name' < "$_state_file" | sort) <(jq -r '.[].name' < "$_task_file" | sort))
mapfile -t _tasks_to_update < <(comm -12 <(jq -r '.[].name' < "$_state_file" | sort) <(jq -r '.[].name' < "$_task_file" | sort))
function task_key_by_name() {
_file="$1"
_task="$2"
_key="$3"
# shellcheck disable=SC2086 # Intended splitting of JQ_OPTS
jq ${JQ_OPTS:-} '.[] | select(.name == "'"$_task"'") | .'"$_key" < "$_file"
}
function prepend_if_not_null() {
_prepend=$1
_input=$(cat)
if [[ "$_input" != "null" ]] ; then
echo "$_prepend $_input"
fi
}
function generate_flux_file() {
_task="$1"
cat <<EOF
option task = {
name: $(task_key_by_name "$_task_file" "$task" "name")
$(task_key_by_name "$_task_file" "$task" "cron" | prepend_if_not_null ", cron: ")
$(JQ_OPTS="-r" task_key_by_name "$_task_file" "$task" "every" | prepend_if_not_null ", every: ")
}
$(cat "$(JQ_OPTS="-r" task_key_by_name "$_task_file" "$task" "flux_file")")
EOF
}
function task_delete() {
_id="$1"
influx task delete --id "$_id"
}
function task_create() {
_flux_file="$1"
influx task create --org "$_org" --file <(cat "$_flux_file")
}
function task_update() {
_task="$1"
_flux_file="$2"
influx task update --id "$(JQ_OPTS="-r" task_key_by_name "$_state_file" "$_task" "id")" --file "$_flux_file"
}
for task in "${_tasks_to_remove[@]}" ; do
if [[ "$_dry_run" == "no" ]] ; then
echo "Removing $task"
task_delete "$(JQ_OPTS="-r" task_key_by_name "$_state_file" "$task" "id")"
else
echo "Would remove $task"
fi
done
for task in "${_tasks_to_add[@]}" ; do
if [[ "$_dry_run" == "no" ]] ; then
echo "Adding $task"
_flux_file=$(generate_flux_file "$task")
task_create <(cat <<<"$_flux_file")
else
echo "Would add $task"
fi
done
for task in "${_tasks_to_update[@]}" ; do
if [[ "$_dry_run" == "no" ]] ; then
echo "Updating $task"
_flux_file=$(generate_flux_file "$task")
_output="$(task_update "$task" <(cat <<<"$_flux_file") 2>&1)"
if [[ "$?" == "1" ]] && [[ "$_output" == *"404"* ]] ; then
echo "Failed to update, creating: $_output"
task_create <(cat <<<"$_flux_file")
fi
else
echo "Would update $task"
fi
done
_current_state=$(influx task list --org "$_org" --json)
if [[ "$_dry_run" == "no" ]] ; then
echo "Saving new state"
cat "$_task_file" <(cat <<<"$_current_state") | jq -s '[.[0] as $new | .[1] as $old | $old[] | select(.name | IN($new[].name))]' > "$_state_file"
fi