#!/bin/bash
# SPDX-License-Identifier: BSD-3-Clause OR Apache-2.0

set -euo pipefail

# uD3TN + HDTN Interoperability test
# It is recommended to run this script in a Docker image based on the
# `hdtn_interop` Dockerfile provided with uD3TN as follows:
# $ docker run -it -v "$(pwd):/ud3tn" ud3tn-hdtn-interop:<tag> \
#   bash -c '/ud3tn/test/dockerfiles/prepare_for_test.sh /ud3tn /ud3tn_build && cd /ud3tn_build && source /ud3tn_venv/bin/activate && test/run_hdtn_test.sh'


BP_VERSION=7

# Config variables
HDTN_EID=ipn:10.0
UD3TN1_EID=ipn:1.0
UD3TN2_EID=ipn:2.0
SOURCE_AGENTID=3
SINK_AGENTID=5
SINK_EID=ipn:2.5
BPGEN_EID=ipn:1.2
HDTN_CONFIG_FILE=$HDTN_SOURCE_ROOT/config_files/hdtn/hdtn_ingress1tcpcl_port4556_egress1tcpcl_port4558flowid2.json
HDTN_CONTACT_PLAN=contactPlanCutThroughMode_unlimitedRate.json
HDTN_BPGEN_CONFIG=$HDTN_SOURCE_ROOT/config_files/outducts/bpgen_one_tcpcl_port4556.json
HDTN_BPSINK_CONFIG=$HDTN_SOURCE_ROOT/config_files/inducts/bpsink_one_tcpcl_port4558.json


UD3TN_DIR="$(pwd)"
cd "$UD3TN_DIR"

exit_handler() {
    cd "$UD3TN_DIR"

    kill -TERM $UD3TN1_PID || true
    kill -TERM $UD3TN2_PID || true
    kill -2 $HDTN_PID || true

    echo
    echo ">>> HDTN LOGFILE"
    cat "/tmp/hdtn.log" || true
    echo
    echo ">>> uD3TN1 LOGFILE"
    cat "/tmp/ud3tn1.log" || true
    echo
    echo ">>> uD3TN2 LOGFILE"
    cat "/tmp/ud3tn2.log" || true
    echo

    rm -f -r $TMPDIR
}

rm -f hdtn_payload.txt received_payload.txt
rm -f /tmp/hdtn*log /tmp/ud3tn*.log

# uD3TN1 -> HDTN -> uD3TN2 Test
# Start first uD3TN instance (uD3TN1)
"$UD3TN_DIR/build/posix/ud3tn" -s $UD3TN_DIR/ud3tn1.socket -S $UD3TN_DIR/ud3tn1.aap2.socket -c "sqlite:ud3tn1.sqlite;tcpclv3:127.0.0.1,4554" -b $BP_VERSION -e "$UD3TN1_EID" -L 4 > /tmp/ud3tn1.log 2>&1 &
UD3TN1_PID=$!

# Start second uD3TN instance (uD3TN2)
"$UD3TN_DIR/build/posix/ud3tn" -s $UD3TN_DIR/ud3tn2.socket -S $UD3TN_DIR/ud3tn2.aap2.socket -c "sqlite:ud3tn2.sqlite;tcpclv3:127.0.0.1,4558" -b $BP_VERSION -e "$UD3TN2_EID" -L 4 > /tmp/ud3tn2.log 2>&1 &
UD3TN2_PID=$!

# Start HDTN
"$HDTN_SOURCE_ROOT/build/module/hdtn_one_process/hdtn-one-process" --hdtn-config-file=$HDTN_CONFIG_FILE --contact-plan-file=$HDTN_CONTACT_PLAN > /tmp/hdtn.log 2>&1 &
HDTN_PID=$!

# Create temporary directory
TMPDIR="$(mktemp -d)"

trap exit_handler EXIT

# Configure contact to HDTN in uD3TN1 which allows to reach uD3TN2
sleep 1
aap2-config --socket $UD3TN_DIR/ud3tn1.aap2.socket --schedule 1 3600 10000 --reaches "$UD3TN2_EID" "$HDTN_EID" tcpclv3:127.0.0.1:4556

# Send a bundle to uD3TN1, addressed to uD3TN2
PAYLOAD="THISISTHEBUNDLEPAYLOAD"
(sleep 0.5 && aap-send --socket $UD3TN_DIR/ud3tn1.socket --agentid "$SOURCE_AGENTID" "$SINK_EID" "$PAYLOAD") &

timeout -v 10 stdbuf -oL aap-receive --socket $UD3TN_DIR/ud3tn2.socket --agentid "$SINK_AGENTID" --count 1 --verify-pl "$PAYLOAD" --newline -vv

# Send `hdtn_payload.txt` to uD3TN aap_receiver so we could send it back
echo $PAYLOAD > hdtn_payload.txt
(sleep 1 && $HDTN_SOURCE_ROOT/build/common/bpcodec/apps/bpsendfile --file-or-folder-path=hdtn_payload.txt --my-uri-eid="$BPGEN_EID" --dest-uri-eid="$SINK_EID" --use-bp-version-7 --outducts-config-file=$HDTN_BPGEN_CONFIG)&
timeout -v 10 stdbuf -oL aap-receive --socket $UD3TN_DIR/ud3tn2.socket --agentid "$SINK_AGENTID" --count 1 -vv -o received_payload.txt

# Terminate uD3TN2 so that HDTN_ could start
echo "Terminating uD3TN_2..." && kill -TERM $UD3TN2_PID
echo "Waiting for uD3TN to exit gracefully - if it doesn't, check for sanitizer warnings."
wait $UD3TN2_PID

# uD3TN1 -> HDTN -> HDTN_bpreceivefile Test
# Start HDTN_bpreceivefile
$HDTN_SOURCE_ROOT/build/common/bpcodec/apps/bpreceivefile --save-directory=$TMPDIR --inducts-config-file=$HDTN_BPSINK_CONFIG --my-uri-eid="$SINK_EID" &
HDTNSINK_PID=$!

# Configure contact to HDTN in uD3TN1 which allows to reach HDTN_bpreceivefile
aap2-config --socket $UD3TN_DIR/ud3tn1.aap2.socket --schedule 1 3600 10000 --reaches "$UD3TN2_EID" "$HDTN_EID" tcpclv3:127.0.0.1:4556

# Send bundle which contains `hdtn_payload.txt` to HDTN_bpreceivefile
(sleep 2 && (cat $UD3TN_DIR/received_payload.txt | aap-send --socket $UD3TN_DIR/ud3tn1.socket --agentid "$SOURCE_AGENTID" "$SINK_EID"))&

inotifywait $TMPDIR -t 10 -e create |
        while read -r directory action file; do
            if [[ "$file" == "hdtn_payload.txt" ]]; then
                RECEIVED_PAYLOAD="$(cat $TMPDIR/hdtn_payload.txt)"
                echo "Payload was received successfully."
                if [[ "$RECEIVED_PAYLOAD" != "$PAYLOAD" ]]; then
                    echo "Received payload does not match: $RECEIVED_PAYLOAD"
                    exit 1
                fi
                exit;
            fi
        done

echo "Terminating uD3TN_1..." && kill -TERM $UD3TN1_PID
echo "Waiting for uD3TN to exit gracefully - if it doesn't, check for sanitizer warnings."
wait $UD3TN1_PID