#!/usr/bin/env bash

# usage
usage() {
USAGE=$(cat <<EOF
Usage: smartctl_tests_long

Schedules smartd long tests. Please see man smartctl_tests_long.
EOF
)
	echo "$USAGE";
}

set -e;

# check for config file
apply_config() {
  local config=$1;

  if [[ ! -f "$config" ]]; then
      echo "No config file specified";
      echo "";
      usage;
      exit 1;
  fi

  set -a;
  # shellcheck disable=SC1090
  source "$config";
  set +a;
}

source_config() {
  local configFallback=$1;
  local configGlobalFallback=$2;

  if [[ -f "$configFallback" ]]; then
    apply_config "$configFallback";
    return;
  fi

  if [[ -f "$configGlobalFallback" ]]; then
    apply_config "$configGlobalFallback";
    return;
  fi
}

check_required() {
  type hostname &> /dev/null || { echo "Requiring 'hostname' but it's not installed"; exit 1; }
  type smartctl &> /dev/null || { echo "Requiring 'smartctl' but it's not installed"; exit 1; }
}

SMARTCTL_TESTS_LONG_DEVICES=(sda)

source_config "$HOME/.smartctl_tests_long.conf" "/etc/smartctl_tests_long.conf"
check_required

HOSTNAME=$(hostname)

echo "Starting long tests for ...";

for d in "${SMARTCTL_TESTS_LONG_DEVICES[@]}"; do
    DEVICE="/dev/$d";
    echo "$DEVICE";
    smartctl -t long "$DEVICE";
    SUBJECT="[smartd $HOSTNAME] queued long test"
    MESSAGE="Queued a smartd long test for $DEVICE."
    notifier "${SUBJECT}" "${MESSAGE}"
done