#!/usr/bin/env bash

# usage
usage() {
USAGE=$(cat <<EOF
Usage: smartctl_tests_long [CONFIG_FILE (absolute path)]

If no CONFIG_FILE is given, HOME/.smartctl_tests_long.conf or /etc/smartctl_tests_long.conf is used. This fallback option
has to exist or the script will exit.

Configuration can be done in any file and any pre-defined variable can be overwritten.

The following are at least required for the script to work:
- SMARTCTL_TESTS_LONG_MAIL_ADDRESS="" // mail.rc has to be configured
- SMARTCTL_TESTS_LONG_DEVICES=(sda) // array of block devices (not partitions!) to check

You can copy this script to '/usr/local/bin' and use create a custom CONFIG_FILE as user. Examples can be found in '/usr/share/doc/<scriptname>'.
EOF
)
	echo "$USAGE";
}

set -e;

HOSTNAME=$(hostname)
SMARTCTL_TESTS_LONG_MAIL_ADDRESS="";
SMARTCTL_TESTS_LONG_DEVICES=(sda)

# 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;
  source "$config";
  set +a;
}

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

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

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

  if [[ -f "$configGlobalFallback" ]]; then
    apply_config "$configGlobalFallback";
    return;
  fi
}
source_config "$1" "$HOME/.smartctl_tests_long.conf" "/etc/smartctl_tests_long.conf"

echo "Starting long tests for ...";

for d in ${SMARTCTL_TESTS_LONG_DEVICES[@]}; do
    DEVICE="/dev/"$d;
    echo $DEVICE;
    smartctl -t long $DEVICE;
   mailx -s "[smartd $HOSTNAME] queued long test" $SMARTCTL_TESTS_LONG_MAIL_ADDRESS << EOF
Queued a smartd long test for $DEVICE.
EOF
done