#!/bin/sh
# script/run: build bin/sww with script/build and run discriminator and
# rproxy on the host, with the settings in the environment, until rproxy
# exits or the script is told to stop. rproxy listens on port 8080 unless
# SWRP_LISTEN_ADDR says otherwise, in front of the app at SWRP_UPSTREAM_URL.
# discriminator listens on port 8090 unless SWW_LISTEN_ADDR says otherwise, and
# rproxy asks it there unless SWRP_WAF_BACKEND says otherwise. Unless
# SWW_STATE_DIR is set, the state files go in bin/state, beside the binary,
# and unless SWW_RULES_DIR is set, the rule files are those of share/rules.d,
# which the image ships.
set -eu

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd -P)"
ROOT="$(cd "$SCRIPT_DIR/.." && pwd -P)"

DISCRIMINATOR=""
RPROXY=""

# stop_both: stop what runs, and wait for it, so that discriminator writes its state
# files.
stop_both() {
    trap '' HUP INT TERM
    kill $DISCRIMINATOR $RPROXY 2>/dev/null || true
    wait || true
}

main() {
    "$SCRIPT_DIR/build"
    if [ -z "${SWW_STATE_DIR+set}" ]; then
        SWW_STATE_DIR="$ROOT/bin/state"
        export SWW_STATE_DIR
        mkdir -p "$SWW_STATE_DIR"
    fi
    if [ -z "${SWW_RULES_DIR+set}" ]; then
        SWW_RULES_DIR="$ROOT/share/rules.d"
        export SWW_RULES_DIR
    fi
    SWW_LISTEN_ADDR="${SWW_LISTEN_ADDR:-:8090}"
    SWRP_WAF_BACKEND="${SWRP_WAF_BACKEND:-http://127.0.0.1:${SWW_LISTEN_ADDR##*:}}"
    export SWW_LISTEN_ADDR SWRP_WAF_BACKEND

    trap stop_both EXIT
    trap 'exit 1' HUP INT TERM

    "$ROOT/bin/sww" discriminator &
    DISCRIMINATOR=$!
    "$ROOT/bin/sww" rproxy &
    RPROXY=$!
    wait "$RPROXY"
}

main "$@"
