#!/usr/bin/env bash
set -euo pipefail

repository_root=$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)
workspace_venv="$repository_root/../.venv"
test_tmp=$(mktemp -d)
chmod 0700 "$test_tmp"

export ANSIBLE_COLLECTIONS_PATH="$test_tmp/collections"
export ANSIBLE_ROLES_PATH="$(dirname "$repository_root")"

inventory="$repository_root/tests/inventory"
converge="$repository_root/tests/converge.yml"
verify_config="$repository_root/tests/verify-config.yml"
ansible_playbook="$workspace_venv/bin/ansible-playbook"
binary_path=/usr/local/bin/scriptorium
config_path=/usr/local/etc/scriptorium/config.yml
profile_dir="$test_tmp/profiles"

cleanup() {
  exit_status=$?
  trap - EXIT
  "$ansible_playbook" -i "$inventory" "$converge" >/dev/null 2>&1 || true
  rm -rf -- "$test_tmp"
  exit "$exit_status"
}
trap cleanup EXIT

require_clean_recap() {
  grep -Eq 'changed=0 .*failed=0' "$1"
}

run_converge() {
  output_file=$1
  shift
  "$ansible_playbook" -i "$inventory" "$@" "$converge" | tee "$output_file"
}

mkdir -p "$profile_dir"
printf '%s\n' \
  'id: endpoint-profile' \
  'endpoint: http://127.0.0.1:8000/v1' \
  'model: offline-fixture' > "$profile_dir/endpoint-profile.yml"
printf '%s\n' \
  'id: backend-profile' \
  'backend: fixture-backend' \
  'model: offline-fixture' > "$profile_dir/backend-profile.yml"

cd "$repository_root"
"$workspace_venv/bin/ansible-galaxy" collection install \
  -r collections/requirements.yml \
  -p "$ANSIBLE_COLLECTIONS_PATH"
"$ansible_playbook" -i "$inventory" --syntax-check "$converge"
"$workspace_venv/bin/ansible-lint" "$converge" "$verify_config"

baseline_log="$test_tmp/baseline.log"
run_converge "$baseline_log"
"$binary_path" inspect profile --config "$config_path" \
  --profile-dir "$profile_dir" --profile endpoint-profile
baseline_repeat_log="$test_tmp/baseline-repeat.log"
run_converge "$baseline_repeat_log"
require_clean_recap "$baseline_repeat_log"

custom_go_binary="$test_tmp/go"
custom_go_invocations="$test_tmp/go-invocations.log"
printf '%s\n' \
  '#!/usr/bin/env bash' \
  'set -euo pipefail' \
  "printf '%s\\n' \"\$*\" >> $custom_go_invocations" \
  'exec /snap/bin/go "$@"' > "$custom_go_binary"
chmod 0755 "$custom_go_binary"
sudo rm -f -- "$binary_path"
recovery_log="$test_tmp/recovery.log"
run_converge "$recovery_log" \
  -e '{"scriptorium_manage_build_dependencies": false}' \
  -e "scriptorium_go_binary=$custom_go_binary"
grep -q . "$custom_go_invocations"
"$binary_path" inspect profile --config "$config_path" \
  --profile-dir "$profile_dir" --profile endpoint-profile
recovery_repeat_log="$test_tmp/recovery-repeat.log"
run_converge "$recovery_repeat_log" \
  -e '{"scriptorium_manage_build_dependencies": false}' \
  -e "scriptorium_go_binary=$custom_go_binary"
require_clean_recap "$recovery_repeat_log"

representative_args=(
  -e "scriptorium_profile_dir=$profile_dir"
  -e 'scriptorium_server_addr=127.0.0.1:18080'
  -e '{"scriptorium_server_max_request_bytes": 0, "scriptorium_server_artifact_root": null, "scriptorium_schema_dir": null, "scriptorium_defaults_render_format": "json", "scriptorium_backends": {"fixture-backend": {"endpoint": "http://127.0.0.1:8000/v1", "concurrency_limit": 1, "queue_capacity": 0, "extra_params": {"stream": false}}}}'
)
representative_log="$test_tmp/representative.log"
run_converge "$representative_log" "${representative_args[@]}"
"$binary_path" inspect profile --config "$config_path" --profile backend-profile
"$ansible_playbook" -i "$inventory" \
  -e "scriptorium_test_profile_dir=$profile_dir" "$verify_config"
representative_repeat_log="$test_tmp/representative-repeat.log"
run_converge "$representative_repeat_log" "${representative_args[@]}"
require_clean_recap "$representative_repeat_log"
