#!/bin/bash

# Copyright 2024 Versity Software
# This file is licensed under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http:#www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

# Function to display help information
show_help() {
    echo "Usage: $0 [option...]"
    echo "   -h, --help          Display this help message and exit"
    echo "                       Separate the below by comma"
    echo "   s3api               Run all tests with s3api cli"
    echo "   s3api-multipart     Run multipart tests with s3api cli"
    echo "   s3api-bucket        Run bucket tests with s3api cli"
    echo "   s3api-object        Run object tests with s3api cli"
    echo "   s3api-policy        Run policy tests with s3api cli"
    echo "   s3api-user          Run user tests with s3api cli"
    echo "   s3                  Run tests with s3 cli"
    echo "   s3cmd               Run tests with s3cmd utility"
    echo "   s3cmd-user          Run user tests with s3cmd utility"
    echo "   s3cmd-non-user      Run non-user tests with s3cmd utility"
    echo "   s3cmd-file-count    Run file count test with s3cmd utility"
    echo "   mc                  Run tests with mc utility"
    echo "   mc-non-file-count   Run non-file count tests with mc utility"
    echo "   mc-file-count       Run file count test with mc utility"
    echo "   rest                Run tests with rest cli"
}

handle_param() {
  case $1 in
      -h|--help)
          show_help
          exit 0
          ;;
      s3|s3-file-count|s3-non-file-count|s3api|s3cmd|s3cmd-user|s3cmd-non-user|s3cmd-file-count|mc|mc-non-file-count|mc-file-count|s3api-user|rest|s3api-policy|s3api-bucket|s3api-object|s3api-multipart)
          run_suite "$1"
          ;;
      *) # Handle unrecognized options or positional arguments
          echo "Unrecognized option: $1" >&2
          exit 1
          ;;
  esac
}

run_suite() {
  exit_code=0
  case $1 in
    s3api)
      echo "Running all s3api tests ..."
      "$HOME"/bin/bats ./tests/test_s3api_bucket.sh || exit_code=$?
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_s3api_object.sh || exit_code=$?
      fi
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_s3api_policy.sh || exit_code=$?
      fi
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_s3api_multipart.sh || exit_code=$?
      fi
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_user_aws.sh || exit_code=$?
      fi
      ;;
    s3api-multipart)
      echo "Running s3api multipart tests ..."
      "$HOME"/bin/bats ./tests/test_s3api_multipart.sh || exit_code=$?
      ;;
    s3api-policy)
      echo "Running s3api policy tests ..."
      "$HOME"/bin/bats ./tests/test_s3api_policy.sh || exit_code=$?
      ;;
    s3api-bucket)
      echo "Running s3api bucket tests ..."
      "$HOME"/bin/bats ./tests/test_s3api_bucket.sh || exit_code=$?
      ;;
    s3api-object)
      echo "Running s3api object tests ..."
      "$HOME"/bin/bats ./tests/test_s3api_object.sh || exit_code=$?
      ;;
    s3)
      echo "Running s3 tests ..."
      "$HOME"/bin/bats ./tests/test_s3.sh || exit_code=$?
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_s3_file_count.sh || exit_code=$?
      fi
      ;;
    s3-non-file-count)
      echo "Running s3 non-file count tests ..."
      "$HOME"/bin/bats ./tests/test_s3.sh || exit_code=$?
      ;;
    s3-file-count)
      echo "Running s3 file count test ..."
      "$HOME"/bin/bats ./tests/test_s3_file_count.sh || exit_code=$?
      ;;
    s3cmd)
      echo "Running s3cmd tests ..."
      "$HOME"/bin/bats ./tests/test_s3cmd.sh || exit_code=$?
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_user_s3cmd.sh || exit_code=$?
      fi
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_s3cmd_file_count.sh || exit_code=$?
      fi
      ;;
    s3cmd-user)
      echo "Running s3cmd user tests ..."
      "$HOME"/bin/bats ./tests/test_user_s3cmd.sh || exit_code=$?
      ;;
    s3cmd-non-user)
      echo "Running s3cmd non-user tests ..."
      "$HOME"/bin/bats ./tests/test_s3cmd.sh || exit_code=$?
      ;;
    s3cmd-file-count)
      echo "Running s3cmd file count test ..."
      "$HOME"/bin/bats ./tests/test_s3cmd_file_count.sh || exit_code=$?
      ;;
    mc)
      echo "Running mc tests ..."
      "$HOME"/bin/bats ./tests/test_mc.sh || exit_code=$?
      if [[ $exit_code -eq 0 ]]; then
        "$HOME"/bin/bats ./tests/test_mc_file_count.sh || exit_code=$?
      fi
      ;;
    mc-non-file-count)
      echo "Running mc non-file count tests ..."
      "$HOME"/bin/bats ./tests/test_mc.sh || exit_code=$?
      ;;
    mc-file-count)
      echo "Running mc file count test ..."
      "$HOME"/bin/bats ./tests/test_mc_file_count.sh || exit_code=$?
      ;;
    rest)
      echo "Running rest tests ..."
      if ! "$HOME"/bin/bats ./tests/test_rest.sh; then
        exit_code=1
      elif ! "$HOME"/bin/bats ./tests/test_rest_acl.sh; then
        exit_code=1
      elif ! "$HOME"/bin/bats ./tests/test_rest_chunked.sh; then
        exit_code=1
      elif ! "$HOME"/bin/bats ./tests/test_rest_checksum.sh; then
        exit_code=1
      elif ! "$HOME"/bin/bats ./tests/test_rest_versioning.sh; then
        exit_code=1
      elif ! "$HOME"/bin/bats ./tests/test_rest_bucket.sh; then
        exit_code=1
      fi
      ;;
    s3api-user)
      echo "Running s3api user tests ..."
      "$HOME"/bin/bats ./tests/test_user_aws.sh || exit_code=$?
  esac
  if [ $exit_code -ne 0 ]; then
    exit 1
  fi
}

if [ $# -le 0 ]; then
  show_help
  exit 0
fi

IFS=',' read -ra options <<< "$1"
for option in "${options[@]}"; do
  handle_param "$option"
done

# shellcheck disable=SC2086
exit 0