#!/usr/bin/env bash

export GIT_EDITOR=true

startContainers() {
    pushd ../payment-processor > /dev/null
        docker compose up --build -d 1> /dev/null 2>&1
    popd > /dev/null
    pushd ../participantes/$1 > /dev/null
        services=$(docker compose config --services | wc -l)
        echo "" > docker-compose.logs
        nohup docker compose up --build >> docker-compose.logs &
    popd > /dev/null
    #expectedServicesUp=$(( services + 4 ))
    #servicesUp=$(docker ps | grep ' Up ' | wc -l)
}

stopContainers() {
    pushd ../participantes/$1
        docker compose down -v --remove-orphans
        docker compose rm -s -v -f
        find * ! -group $(whoami) | xargs sudo rm -rf
    popd > /dev/null
    pushd ../payment-processor > /dev/null
        docker compose down --volumes > /dev/null
    popd > /dev/null
}

MAX_REQUESTS=550

while true; do

    # docker system prune -a -f --volumes

    for directory in ../participantes/*; do
    (
        git pull
        participant=$(echo $directory | sed -e 's/..\/participantes\///g' -e 's/\///g')
        echo "========================================"
        echo "  Participant $participant starting..."
        echo "========================================"

        testedFile="$directory/partial-results.json"

        if ! test -f $testedFile; then
            touch $testedFile
            echo "executing test for $participant..."
            stopContainers $participant
            startContainers $participant
            
            success=1
            max_attempts=15
            attempt=1
            while [ $success -ne 0 ] && [ $max_attempts -ge $attempt ]; do
                curl -f -s --max-time 3 localhost:9999/payments-summary
                success=$?
                echo "tried $attempt out of $max_attempts..."
                sleep 5
                ((attempt++))
            done

            if [ $success -eq 0 ]; then
                echo "" > $directory/k6.logs
                k6 run -e MAX_REQUESTS=$MAX_REQUESTS -e PARTICIPANT=$participant -e TOKEN=$(uuidgen) --log-output=file=$directory/k6.logs rinha.js
                stopContainers $participant
                echo "======================================="
                echo "working on $participant"
                sed -i '1001,$d' $directory/docker-compose.logs
                sed -i '1001,$d' $directory/k6.logs
                echo "log truncated at line 1000" >> $directory/docker-compose.logs
                echo "log truncated at line 1000" >> $directory/k6.logs
            else
                stopContainers $participant
                echo "[$(date)] Seu backend não respondeu nenhuma das $max_attempts tentativas de GET para http://localhost:9999/payments-summary. Teste abortado." > $directory/error.logs
                echo "[$(date)] Inspecione o arquivo docker-compose.logs para mais informações." >> $directory/error.logs
                echo "Could not get a successful response from backend... aborting test for $participant"
            fi

            git add $directory
            git commit -m "add $participant's partial result"
            git push

            echo "================================="
            echo "  Finished testing $participant!"
            echo "================================="

            sleep 5

        else
            echo "================================="
            echo "  Skipping $participant"
            echo "================================="
        fi
    )
    done

    date
    echo "generating results preview..."

    PREVIA_RESULTADOS=../PREVIA_RESULTADOS.md
    
    results=$(find ../participantes/*/partial-results.json -size +1b | wc -l)
    errors=$(find ../participantes/*/partial-results.json -size 0 | wc -l)
    total=$(find ../participantes/*/partial-results.json | wc -l)

    echo -e "# Prévia do Resultados da Rinha de Backend 2025" > $PREVIA_RESULTADOS
    echo -e "Atualizado em **$(date)**" >> $PREVIA_RESULTADOS
    echo -e "$total submissões / $results resultados / $errors submissões com erro" >> $PREVIA_RESULTADOS
    echo -e "*Testes executados com MAX_REQUESTS=$MAX_REQUESTS*."
    echo -e "\n" >> $PREVIA_RESULTADOS
    echo -e "| participante | p99 | bônus por desempenho (%) | multa ($) | lucro | submissão |" >> $PREVIA_RESULTADOS
    echo -e "| -- | -- | -- | -- | -- | -- |" >> $PREVIA_RESULTADOS

    for partialResult in ../participantes/*/partial-results.json; do
    (
        participant=$(echo $partialResult | sed -e 's/..\/participantes\///g' -e 's/\///g' -e 's/partial\-results\.json//g')
        link="https://github.com/zanfranceschi/rinha-de-backend-2025/tree/main/participantes/$participant"
        
        if [ -s $partialResult ]; then
            cat $partialResult | jq -r '(["|", .participante, "|", .p99.valor, "|", .p99.bonus, "|", .multa.total, "|", .total_liquido, "|", "['$participant']('$link')"]) | @tsv' >> $PREVIA_RESULTADOS
        fi
    )
    done

    echo -e "### Submissões com Erro" >> $PREVIA_RESULTADOS
    echo -e "\n" >> $PREVIA_RESULTADOS
    echo -e "| participante | submissão |" >> $PREVIA_RESULTADOS
    echo -e "| -- | -- |" >> $PREVIA_RESULTADOS
    for errorLog in ../participantes/*/error.logs; do
    (
        participant=$(echo $errorLog | sed -e 's/..\/participantes\///g' -e 's/\///g' -e 's/error\.logs//g')
        link="https://github.com/zanfranceschi/rinha-de-backend-2025/tree/main/participantes/$participant"
        echo "| $participant | [logs]($link) |" >> $PREVIA_RESULTADOS
    )
    done

    PREVIA_RESULTADOS_JSON=../previa-resultados+participantes-info.json
    python3 previa_resultados_json.py $PREVIA_RESULTADOS_JSON
    
    git pull
    git add $PREVIA_RESULTADOS_JSON
    git add $PREVIA_RESULTADOS
    git commit -m "previa resultados @ $(date)"
    git push
    echo "$(date) - waiting some time until next round..."
    sleep 300
done