#!/usr/bin/env bash

# bash install.sh --stage 0 --stop_stage 2

verbose=true;
stage=-1
stop_stage=1


# 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


work_dir="$(pwd)"
thirdparty_dir="$(pwd)/thirdparty"
pretrained_models_dir="$(pwd)/pretrained_models"
silero_vad_model_dir="${pretrained_models_dir}/silero_vad"

mkdir -p "${thirdparty_dir}"
mkdir -p "${pretrained_models_dir}"
mkdir -p "${silero_vad_model_dir}"


if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
  $verbose && echo "stage 0: install onnxruntime"

  cd "${thirdparty_dir}"

  # install onnxruntime
  # https://onnxruntime.ai/docs/install/
  # https://github.com/microsoft/onnxruntime/releases
  wget https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-1.20.1.tgz
  tar -zxvf onnxruntime-linux-x64-1.20.1.tgz

fi


if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
  $verbose && echo "stage 1: download silero_vad.onnx"

  cd "${silero_vad_model_dir}"

  # download model
  # https://github.com/snakers4/silero-vad/tree/master/src/silero_vad/data
  wget https://github.com/snakers4/silero-vad/raw/refs/heads/master/src/silero_vad/data/silero_vad.onnx

fi