#!/bin/bash

# This script is used to run tests for the InfluxDB documentation.
# The script is designed to be run in a Docker container. It is used to substitute placeholder values.

# Function to check if an option is present in the arguments
has_option() {
    local target="$1"
    shift
    for arg in "$@"; do
        if [ "$arg" == "$target" ]; then
            return 0
        fi
    done
    return 1
}

verbose=0
# Check if "--option" is present in the CMD arguments
if has_option "-v" "$@"; then
    verbose=1
    echo "Using verbose mode..."
fi

BASE_DIR=$(pwd)
cd $TEMP_DIR

for file in `find . -type f \( -iname '*.md' \)` ; do
  if [ -f "$file" ]; then
    echo "PRETEST: substituting values in $file"

    # Replaces placeholder values with environment variable references.

    # Non-language-specific replacements.
    sed -i 's|https:\/\/{{< influxdb/host >}}|$INFLUX_HOST|g;
    ' $file

    # Python-specific replacements.
    # Use f-strings to identify placeholders in Python while also keeping valid syntax if
    # the user replaces the value.
    # Remember to import os for your example code.
    sed -i 's/f"DATABASE_TOKEN"/os.getenv("INFLUX_TOKEN")/g;
    s/f"API_TOKEN"/os.getenv("INFLUX_TOKEN")/g;
    s/f"BUCKET_NAME"/os.getenv("INFLUX_DATABASE")/g;
    s/f"DATABASE_NAME"/os.getenv("INFLUX_DATABASE")/g;
    s|f"{{< influxdb/host >}}"|os.getenv("INFLUX_HOSTNAME")|g;
    s|f"RETENTION_POLICY_NAME\|RETENTION_POLICY"|"autogen"|g;
    ' $file

    # Shell-specific replacements.
    ## In JSON Heredoc
    sed -i 's|"orgID": "ORG_ID"|"orgID": "$INFLUX_ORG"|g;
    s|"name": "BUCKET_NAME"|"name": "$INFLUX_DATABASE"|g;' \
    $file

    sed -i 's/API_TOKEN/$INFLUX_TOKEN/g;
    s/ORG_ID/$INFLUX_ORG/g;
    s/DATABASE_TOKEN/$INFLUX_TOKEN/g;
    s/--bucket-id BUCKET_ID/--bucket-id $INFLUX_BUCKET_ID/g;
    s/BUCKET_NAME/$INFLUX_DATABASE/g;
    s/DATABASE_NAME/$INFLUX_DATABASE/g;
    s/--id DBRP_ID/--id $INFLUX_DBRP_ID/g;
    s/get-started/$INFLUX_DATABASE/g;
    s/RETENTION_POLICY_NAME\|RETENTION_POLICY/$INFLUX_RETENTION_POLICY/g;
    s/CONFIG_NAME/CONFIG_$(shuf -i 0-100 -n1)/g;' \
    $file

    # v2-specific replacements.
    sed -i 's|https:\/\/us-west-2-1.aws.cloud2.influxdata.com|$INFLUX_HOST|g;
    s|{{< latest-patch >}}|${influxdb_latest_patches_v2}|g;
    s|{{< latest-patch cli=true >}}|${influxdb_latest_cli_v2}|g;' \
    $file

    # Skip package manager commands.
    sed -i 's|sudo dpkg.*$||g;
    s|sudo yum.*$||g;' \
    $file

    # Environment-specific replacements.
    sed -i 's|sudo ||g;' \
    $file
  fi
  if [ $verbose -eq 1 ]; then
    echo "FILE CONTENTS:"
    cat $file
  fi
done

# Miscellaneous test setup.
# For macOS samples.
mkdir -p ~/Downloads && rm -rf ~/Downloads/*
# Clean up installed files from previous runs.
gpg -q --batch --yes --delete-key D8FF8E1F7DF8B07E > /dev/null 2>&1

# Activate the Python virtual environment configured in the Dockerfile.
. /opt/venv/bin/activate

# List installed Python dependencies.
pip list

# Run test commands with options provided in the CMD of the Dockerfile.
# pytest rootdir is the directory where pytest.ini is located (/test).
if [ -d ./content/influxdb/cloud-dedicated/ ]; then
echo "Running content/influxdb/cloud-dedicated tests..."
pytest --codeblocks --envfile $BASE_DIR/.env.dedicated ./content/influxdb/cloud-dedicated/ $@
fi

if [ -d ./content/influxdb/cloud-serverless/ ]; then
echo "Running content/influxdb/cloud-serverless tests..."
pytest --codeblocks --envfile $BASE_DIR/.env.serverless ./content/influxdb/cloud-serverless/ $@
fi

if [ -d ./content/telegraf/ ]; then
echo "Running content/telegraf tests..."
pytest --codeblocks --envfile $BASE_DIR/.env.telegraf ./content/telegraf/ $@
fi