#!/usr/bin/env bash

# bash install.sh --stage 1 --stop_stage 1 --system_version centos

verbose=true;
stage=-1
stop_stage=2

work_dir="$(pwd)"


# parse options
while true; do
  [ -z "${1:-}" ] && break;  # break if there are no arguments
  case "$1" in
    --*) name=$(echo "$1" | sed s/^--// | sed s/-/_/g);
      eval '[ -z "${'"$name"'+xxx}" ]' && echo "$0: invalid option $1" 1>&2 && exit 1;
      old_value="(eval echo \\$$name)";
      if [ "${old_value}" == "true" ] || [ "${old_value}" == "false" ]; then
        was_bool=true;
      else
        was_bool=false;
      fi

      # Set the variable to the right value-- the escaped quotes make it work if
      # the option had spaces, like --cmd "queue.pl -sync y"
      eval "${name}=\"$2\"";

      # Check that Boolean-valued arguments are really Boolean.
      if $was_bool && [[ "$2" != "true" && "$2" != "false" ]]; then
        echo "$0: expected \"true\" or \"false\": $1 $2" 1>&2
        exit 1;
      fi
      shift 2;
      ;;

    *) break;
  esac
done


$verbose && echo "system_version: ${system_version}"

pretrained_models_dir="$(pwd)/pretrained_models"

mkdir -p "${pretrained_models_dir}"


if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
  $verbose && echo "stage 1: download fasttext models"
  cd "${pretrained_models_dir}" || exit 1;

  wget https://dl.fbaipublicfiles.com/fasttext/supervised-models/lid.176.bin
  wget https://dl.fbaipublicfiles.com/fasttext/supervised-models/lid.176.ftz

fi