source "${BASH_SOURCE[0]%/*}/windows-compat.bash"
source "${BASH_SOURCE[0]%/*}/local-remote.bash"

if [ -z "$BATS_TMPDIR" ]; then
    export BATS_TMPDIR=$HOME/batstmp/
    mkdir $BATS_TMPDIR
fi

export DOLT_CONTEXT_VALIDATION_ENABLED=true
export DOLT_ENABLE_DYNAMIC_ASSERTS=true

nativebatsdir() { echo `nativepath $BATS_TEST_DIRNAME/$1`; }
batshelper() { echo `nativebatsdir helper/$1`; }

stash_current_dolt_user() {
    export STASHED_DOLT_USER_NAME=`dolt config --global --get user.name`
    export STASHED_DOLT_USER_EMAIL=`dolt config --global --get user.email`
}

restore_stashed_dolt_user() {
    dolt config --global --add user.name "$STASHED_DOLT_USER_NAME"
    dolt config --global --add user.email "$STASHED_DOLT_USER_EMAIL"
    unset STASHED_DOLT_USER_NAME STASHED_DOLT_USER_EMAIL
}

set_dolt_user() {
    dolt config --global --add user.name "$1" > /dev/null 2>&1
    dolt config --global --add user.email "$2" > /dev/null 2>&1
}

unset_dolt_user() {
  dolt config --global --unset user.name
  dolt config --global --unset user.email
}

current_dolt_user_name() {
    dolt config --global --get user.name
}

current_dolt_user_email() {
    dolt config --global --get user.email
}

# get_head_commit returns the commit hash for the current HEAD
# optionally takes a branch name as an argument to get the head commit of that branch
get_head_commit() {
  branch=$1
  if [ -z "$branch" ]; then
    dolt log -n 1 | grep -m 1 commit | cut -c 13-44
  else
    dolt --branch "$branch" log -n 1 | grep -m 1 commit | cut -c 13-44
  fi
}

dolt_log_in_PST() {
    TZ=PST+8 dolt log -n1
}

setup_no_dolt_init() {
    # Ensure locally installed dolt (via `go install`) takes precedence
    export PATH=~/go/bin:$PATH
    cd $BATS_TMPDIR
    
    # remove directory if exists
    # reruns recycle pids
    rm -rf "dolt-repo-$$"

    # Append the directory name with the pid of the calling process so
    # multiple tests can be run in parallel on the same machine
    mkdir "dolt-repo-$$"
    cd "dolt-repo-$$"

    if [ -z "$DOLT_TEST_RETRIES" ]; then
        export BATS_TEST_RETRIES="$DOLT_TEST_RETRIES"
    fi

    # Our tests use a mix of root and dolt users, and the CLI
    # commands which authenticate will block for user input if we don't
    # set DOLT_CLI_PASSWORD - so we set it to the empty string by default.
    # The DOLT_SILENCE_USER_REQ_FOR_TESTING environment variable is to skip
    # a check which errors out if a password is presented but no user is specified.
    # The combination of these two flags allows us to avoid altering hundreds
    # of existing tests which predate the authentication restrictions added
    # during the cli -> sql migration.
    export DOLT_CLI_PASSWORD=""
    export DOLT_SILENCE_USER_REQ_FOR_TESTING="Y"
}

assert_feature_version() {
    run dolt version --feature
    # Tests that don't end in a valid dolt dir will fail the above
    # command, don't check its output in that case
    if [ "$status" -eq 0 ]; then
        [[ "$output" =~ "feature version: 7" ]] || exit 1
    else
      # Clear status to avoid BATS failing if this is the last run command
      status=0
    fi
}

setup_common() {
    setup_no_dolt_init
    dolt init
    setup_remote_server
}

teardown_common() {
    # rm -rf can fail with a "directory not empty" error in some cases. This seems to be a misleading
    # error message; the real error is that a file is still in use. Instead of waiting longer for
    # any processes to finish, we just ignore any error removing temp files and use 'true' as the last
    # command in this function to ensure that teardown_common doesn't fail a test just because we
    # couldn't delete any temporary test files.
    teardown_remote_server
    rm -rf "$BATS_TMPDIR/dolt-repo-$$"
    true
}

log_status_eq() {
    if ! [ "$status" -eq $1 ]; then
        echo "status: expected $1, received $status"
        printf "output:\n$output"
        exit 1
    fi
}

log_output_has() {
    if ! [[ "$output" =~ $1 ]]; then
        echo "output did not have $1"
        printf "output:\n$output"
        exit 1
    fi
}

parquet() {
  if [ -z "$PARQUET_RUNTIME_JAR" ]; then
    echo "must supply PARQUET_RUNTIME_JAR to use parquet cli"
    exit 1
  else
    out=$(hadoop jar "$PARQUET_RUNTIME_JAR" org.apache.parquet.cli.Main "$@")
    echo "$out"
  fi
}

nativevar DOLT_ROOT_PATH $BATS_TMPDIR/config-$$ /p
dolt config --global --add metrics.disabled true > /dev/null 2>&1
set_dolt_user "Bats Tests" "bats@email.fake"