#!/usr/bin/env bash

echo "First arg:"
echo $1

os='linux'
if [ "X$1" = "X-w" ]; then
    os='windows'
fi
echo "OS:" $os

src_dir=$(pwd)
test_root=build/lib/tests
if [ "X$os" = "Xwindows" ]; then
    test_root=$test_root/Debug
fi

if [ "X$os" = "Xlinux" ]; then
    drogon_ctl_exec=$(pwd)/build/drogon_ctl/drogon_ctl
else
    drogon_ctl_exec=$(pwd)/build/drogon_ctl/Debug/drogon_ctl.exe
    export PATH=$PATH:$src_dir/install/bin
fi
echo "drogon_ctl_exec: " ${drogon_ctl_exec}

if [ "X$os" = "Xwindows" ]; then
    integration_test_client_exec=./integration_test_client.exe
    integration_test_server_exec=./integration_test_server.exe
else
    integration_test_client_exec=./integration_test_client
    integration_test_server_exec=./integration_test_server
fi

function update_config_line()
{
    local key="$1"
    local value="$2"
    local file="$3"
    sed -i.bak -e "s/\"${key}\".*$/\"${key}\": ${value},/" "$file"
    rm -f "$file.bak"
}

function cleanup_integration_test_server()
{
    if [ "X$os" = "Xwindows" ]; then
        taskkill //F //IM integration_test_server.exe > /dev/null 2>&1 || true
    else
        killall integration_test_server > /dev/null 2>&1 || true
        pkill -f '/integration_test_server$' > /dev/null 2>&1 || true
    fi
}

function wait_for_url()
{
    local url="$1"
    local timeout_seconds="$2"
    local curl_args=(--silent --show-error --output /dev/null --max-time 2)

    if [ "$url" != "${url#https://}" ]; then
        curl_args+=(--insecure)
    fi

    local attempt=0
    while [ $attempt -lt $timeout_seconds ]; do
        if curl "${curl_args[@]}" "$url"; then
            return 0
        fi

        attempt=$((attempt + 1))
        sleep 1
    done

    return 1
}

function wait_for_integration_test_server()
{
    local server_pid="$1"
    local server_log="$2"

    if ! wait_for_url "http://127.0.0.1:8848/" 30; then
        echo "Timed out waiting for integration_test_server to accept HTTP requests"
        if kill -0 "$server_pid" > /dev/null 2>&1; then
            echo "integration_test_server is still running, recent log output:"
        else
            echo "integration_test_server exited before becoming ready, recent log output:"
        fi
        if [ -f "$server_log" ]; then
            tail -n 50 "$server_log"
        fi
        return 1
    fi

    wait_for_url "https://127.0.0.1:8849/" 5 > /dev/null 2>&1 || true
    return 0
}

function get_cpu_count()
{
    if command -v nproc > /dev/null 2>&1; then
        nproc
        return
    fi

    if command -v getconf > /dev/null 2>&1; then
        getconf _NPROCESSORS_ONLN
        return
    fi

    if command -v sysctl > /dev/null 2>&1; then
        sysctl -n hw.logicalcpu
        return
    fi

    echo 1
}

trap cleanup_integration_test_server EXIT

# Run the integration test server in the background and wait until it is ready.
function do_integration_test()
{
    pushd "$test_root"
    update_config_line "run_as_daemon" "false" config.example.json
    update_config_line "relaunch_on_error" "false" config.example.json
    update_config_line "number_of_threads" "1" config.example.json
    update_config_line "use_brotli" "true" config.example.json

    if [ "$1" = "stream_mode" ]; then
        update_config_line "enable_request_stream" "true" config.example.json
    else
        update_config_line "enable_request_stream" "false" config.example.json
    fi

    if [ ! -f "$integration_test_client_exec" ]; then
        echo "Build failed"
        exit -1
    fi
    if [ ! -f "$integration_test_server_exec" ]; then
        echo "Build failed"
        exit -1
    fi

    cleanup_integration_test_server

    local server_log=integration_test_server.log
    rm -f "$server_log"
    "$integration_test_server_exec" > "$server_log" 2>&1 &
    local server_pid=$!

    if ! wait_for_integration_test_server "$server_pid" "$server_log"; then
        exit -1
    fi

    echo "Running the integration test $1"
    "$integration_test_client_exec" -s

    if [ $? -ne 0 ]; then
        echo "Integration test failed $1"
        if [ -f "$server_log" ]; then
            tail -n 50 "$server_log"
        fi
        exit -1
    fi

    cleanup_integration_test_server
    popd
}

#Test drogon_ctl
function do_drogon_ctl_test()
{
    echo "Testing drogon_ctl"
    pushd "$test_root"
    rm -rf drogon_test

    ${drogon_ctl_exec} create project drogon_test

    ls -la
    cd drogon_test/controllers

    ${drogon_ctl_exec} create controller Test::SimpleCtrl
    ${drogon_ctl_exec} create controller -h Test::HttpCtrl
    ${drogon_ctl_exec} create controller -w Test::WebsockCtrl
    ${drogon_ctl_exec} create controller SimpleCtrl
    ${drogon_ctl_exec} create controller -h HttpCtrl
    ${drogon_ctl_exec} create controller -w WebsockCtrl

    if [ ! -f "Test_SimpleCtrl.h" -o ! -f "Test_SimpleCtrl.cc" -o ! -f "Test_HttpCtrl.h" -o ! -f "Test_HttpCtrl.cc" -o ! -f "Test_WebsockCtrl.h" -o ! -f "Test_WebsockCtrl.cc" ]; then
        echo "Failed to create controllers"
        exit -1
    fi

    if [ ! -f "SimpleCtrl.h" -o ! -f "SimpleCtrl.cc" -o ! -f "HttpCtrl.h" -o ! -f "HttpCtrl.cc" -o ! -f "WebsockCtrl.h" -o ! -f "WebsockCtrl.cc" ]; then
        echo "Failed to create controllers"
        exit -1
    fi

    cd ../filters

    ${drogon_ctl_exec} create filter Test::TestFilter

    if [ ! -f "Test_TestFilter.h" -o ! -f "Test_TestFilter.cc" ]; then
        echo "Failed to create filters"
        exit -1
    fi

    cd ../plugins

    ${drogon_ctl_exec} create plugin Test::TestPlugin

    if [ ! -f "Test_TestPlugin.h" -o ! -f "Test_TestPlugin.cc" ]; then
        echo "Failed to create plugins"
        exit -1
    fi

    cd ../views

    echo "Hello, world!" >>hello.csp

    cd ../build

    make_flags=''
    cmake_gen=''
    parallel=1
    cpu_count=$(get_cpu_count)

    # simulate ninja's parallelism
    case $cpu_count in
    1)
        parallel=$((cpu_count + 1))
        ;;
    2)
        parallel=$((cpu_count + 1))
        ;;
    *)
        parallel=$((cpu_count + 2))
        ;;
    esac

    if [ "X$os" = "Xlinux" ]; then
        if command -v ninja > /dev/null 2>&1; then
            cmake_gen='-G Ninja'
        else
            make_flags="$make_flags -j$parallel"
        fi
    else
        cmake_gen="$cmake_gen -DCMAKE_TOOLCHAIN_FILE=$src_dir/conan_toolchain.cmake \
                              -DCMAKE_PREFIX_PATH=$src_dir/install \
                              -DCMAKE_POLICY_DEFAULT_CMP0091=NEW \
                              -DCMAKE_CXX_STANDARD=17"
    fi
    cmake .. $cmake_gen

    if [ $? -ne 0 ]; then
        echo "Failed to run CMake for example project"
        exit -1
    fi

    cmake --build . -- $make_flags

    if [ $? -ne 0 ]; then
        echo "Error in testing"
        exit -1
    fi

        if [ "X$os" = "Xlinux" ]; then
            if [ ! -f "drogon_test" ]; then
                    echo "Failed to build drogon_test"
                    exit -1
            fi
    else
      if [ ! -f "Debug\drogon_test.exe" ]; then
          echo "Failed to build drogon_test"
          exit -1
      fi
    fi

    cd ../../
    rm -rf drogon_test
    popd
}

#unit testing
function do_unittest()
{
    echo "Unit testing"
    pushd "$src_dir/build"

    ctest . --output-on-failure
    if [ $? -ne 0 ]; then
        echo "Error in unit testing"
        exit -1
    fi
    popd
}

function do_db_test()
{
    pushd "$src_dir/build"
    if [ -f "./orm_lib/tests/db_test" ]; then
        echo "Test database"
        ./orm_lib/tests/db_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
    if [ -f "./orm_lib/tests/pipeline_test" ]; then
        echo "Test pipeline mode"
        ./orm_lib/tests/pipeline_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
    if [ -f "./orm_lib/tests/db_listener_test" ]; then
        echo "Test DbListener"
        ./orm_lib/tests/db_listener_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
    if [ -f "./orm_lib/tests/conn_options_test" ]; then
        echo "Test connection options"
        ./orm_lib/tests/conn_options_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
    if [ -f "./orm_lib/tests/db_api_test" ]; then
        echo "Test getDbClient() api"
        ./orm_lib/tests/db_api_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
    if [ -f "./nosql_lib/redis/tests/redis_test" ]; then
        echo "Test redis"
        ./nosql_lib/redis/tests/redis_test -s
        if [ $? -ne 0 ]; then
            echo "Error in testing"
            exit -1
        fi
    fi
}

if [ ! -f "$drogon_ctl_exec" ]
then
    echo "Warning: No built drogon_ctl, skip integration test and drogon_ctl test"
else
    do_integration_test
    do_integration_test stream_mode
    do_drogon_ctl_test
fi

if [ "X$1" = "X-t" ]; then
    do_unittest
    do_db_test
fi

echo "Everything is ok!"
exit 0