#!/bin/sh
# The release build's own test loop: builds the test image and the release image from
# docker/Dockerfile.release for one PostgreSQL major, starts each on a throwaway server (port 5437, no volume) and runs
# the quick suites. The release image has no page hooks, so the hook-based tests skip there (conftest); the test image
# runs every suite. Works from the exported standalone tree as well as from this repository.
#
#   sh scripts/ci_release.sh            # PG_MAJOR=17
#   PG_MAJOR=18 sh scripts/ci_release.sh
#   WHICH=release sh scripts/ci_release.sh   # the release image only (test | release | both)
#   ARCHFLAGS=-march=x86-64-v3 sh scripts/ci_release.sh   # portable build (default: -march=native)
#   CI_NAME=..., PORT=..., TAG_SUFFIX=...   # container name, port and image tag suffix, so several jobs can share one
#                                           # Docker host; CI_CLEAN=1 removes this run's containers and images at the end
set -e
cd "$(dirname "$0")/.."
PY=${PY:-.venv/bin/python}
PG_MAJOR=${PG_MAJOR:-17}
PORT=${PORT:-5437}
WHICH=${WHICH:-both}
CI_NAME=${CI_NAME:-trigpoint-release-ci}
ARCHFLAGS=${ARCHFLAGS:-}
LOGDIR=${RUNNER_TEMP:-/tmp}
TEST_IMG=trigpoint-postgres-test:pg$PG_MAJOR${TAG_SUFFIX:-}
REL_IMG=trigpoint/postgres:0.5-pg$PG_MAJOR${TAG_SUFFIX:-}
cleanup() {
    docker rm -f $CI_NAME-test $CI_NAME-release >/dev/null 2>&1 || true
    [ "${CI_CLEAN:-0}" = 1 ] && docker rmi $TEST_IMG $REL_IMG >/dev/null 2>&1 || true
}
trap cleanup EXIT
SUITES="tests/test_pg_extension.py tests/test_pg_check.py tests/test_pg_edge.py tests/test_pg_compact.py tests/test_pg_cosine.py tests/test_pg_security.py tests/test_pg_errcodes.py tests/test_pg_filter.py tests/test_pg_upgrade.py tests/test_pg_fuzz.py tests/test_pg_cancel.py tests/test_pg_shared_cap.py"

run_suites() {  # $1 = image, $2 = label
    name=$CI_NAME-$2
    docker rm -f $name >/dev/null 2>&1 || true
    # A runner inside a container (Docker socket of the host) cannot reach a published port on its own localhost:
    # the server then joins the runner's network and is reached by its address. Otherwise it is published on
    # 127.0.0.1 at $PORT, or at a free port Docker picks (PORT=0), so parallel jobs never collide.
    SELF_NET=
    [ -f /.dockerenv ] && SELF_NET=$(docker inspect -f '{{range $k, $v := .NetworkSettings.Networks}}{{$k}} {{end}}' "$(hostname)" 2>/dev/null | awk '{print $1}')
    [ -f /.dockerenv ] && [ -z "$SELF_NET" ] && echo "warning: inside a container but $(hostname) is not a container on this Docker host; publishing a port instead"
    if [ -n "$SELF_NET" ]; then NETARGS="--network $SELF_NET"; elif [ "$PORT" = 0 ]; then NETARGS="-p 127.0.0.1::5432"; else NETARGS="-p 127.0.0.1:$PORT:5432"; fi
    docker run -d --rm --name $name $NETARGS --shm-size 2g -e POSTGRES_USER=vsroute -e POSTGRES_PASSWORD=vsroute -e POSTGRES_DB=vsroute \
        "$1" postgres -c shared_buffers=512MB -c maintenance_work_mem=512MB -c max_parallel_workers_per_gather=0 -c max_worker_processes=16 >/dev/null
    # pg_isready also answers for the image's temporary init server (socket only); wait for the real one over TCP
    for i in $(seq 1 60); do docker exec -e PGPASSWORD=vsroute $name psql -h 127.0.0.1 -U vsroute -d vsroute -tAc 'SELECT 1' >/dev/null 2>&1 && break; sleep 1; done
    if [ -n "$SELF_NET" ]; then
        DBHOST=$(docker inspect -f "{{(index .NetworkSettings.Networks \"$SELF_NET\").IPAddress}}" $name); DBPORT=5432
    else
        DBHOST=127.0.0.1; DBPORT=$(docker port $name 5432/tcp | head -1 | sed 's/.*://')
    fi
    echo "server $name at $DBHOST:$DBPORT${SELF_NET:+ (network $SELF_NET)}"
    docker exec $name psql -U vsroute -tAc "CREATE EXTENSION vector; CREATE EXTENSION trigpoint; SELECT version(); SELECT trigpoint_version();"
    PGHOST=$DBHOST PGPORT=$DBPORT PGUSER=vsroute PGPASSWORD=vsroute PGDATABASE=vsroute $PY -m pytest $SUITES -q -p no:cacheprovider 2>&1 | tee $LOGDIR/trigpoint_ci_${CI_NAME}_$2.log | tail -3
    [ -f docs/operator-guide.md ] && sh scripts/doc_test.sh $name
    docker stop $name >/dev/null
    LOG=$LOGDIR/trigpoint_ci_${CI_NAME}_$2.log
    tail -1 $LOG | grep -q " passed" && ! tail -1 $LOG | grep -q "failed\|error" || {
        # the failures with their tracebacks (pytest -q prints them before the summary), last 150 lines
        sed -n '/= FAILURES =\|= ERRORS =/,$p' $LOG | tail -150
        echo "$2 ($1): suites failed"; exit 1; }
}

if [ "$WHICH" = "test" ] || [ "$WHICH" = "both" ]; then
    echo "== test image (TEST_HOOKS=1), PG $PG_MAJOR"
    docker build -q -f docker/Dockerfile.release --build-arg PG_MAJOR=$PG_MAJOR --build-arg ARCHFLAGS="$ARCHFLAGS" --build-arg TEST_HOOKS=1 -t $TEST_IMG . >/dev/null
    run_suites $TEST_IMG test
fi
if [ "$WHICH" = "release" ] || [ "$WHICH" = "both" ]; then
    echo "== release image, PG $PG_MAJOR"
    docker build -q -f docker/Dockerfile.release --build-arg PG_MAJOR=$PG_MAJOR --build-arg ARCHFLAGS="$ARCHFLAGS" -t $REL_IMG . >/dev/null
    rm -f dist/postgresql-$PG_MAJOR-trigpoint_*.deb; mkdir -p dist && docker build -q -f docker/Dockerfile.release --build-arg PG_MAJOR=$PG_MAJOR --build-arg ARCHFLAGS="$ARCHFLAGS" --target deb -o dist . >/dev/null && ls -l dist/*.deb
    run_suites $REL_IMG release
fi
echo "ci_release: green"