#!/usr/bin/env bash

## Test two nodes-cluster discover each other using DNS A records lookup result.

set -euo pipefail

cd -P -- "$(dirname -- "$0")/.."

IMAGE="${1}"

NET='test_node_discovery_dns'
NODE1='emqx1'
NODE2='emqx2'
COOKIE='this-is-a-secret'

# cleanup
docker rm -f dnsmasq >/dev/null 2>&1 || true
docker rm -f "$NODE1" >/dev/null 2>&1 || true
docker rm -f "$NODE2" >/dev/null 2>&1 || true
docker network rm "$NET" >/dev/null 2>&1 || true

docker network create --subnet=172.18.0.0/16 $NET

IP0="172.18.0.100"
IP1="172.18.0.101"
IP2="172.18.0.102"
DOMAIN="dnstest.mynet"

# create configs for dnsmasq
cat <<-EOF > "/tmp/dnsmasq.conf"
conf-dir=/etc/dnsmasq,*.conf
addn-hosts=/etc/hosts.$DOMAIN
EOF

cat <<-EOF > "/tmp/hosts.$DOMAIN"
$IP1 $DOMAIN
$IP2 $DOMAIN
EOF

cat <<-EOF > /tmp/dnsmasq.base.conf
domain-needed
bogus-priv
no-hosts
keep-in-foreground
no-resolv
expand-hosts
server=8.8.8.8
EOF

docker run -d -t --name dnsmasq \
    --net "$NET" \
    --ip "$IP0" \
    -v /tmp/dnsmasq.conf:/etc/dnsmasq.conf \
    -v "/tmp/hosts.$DOMAIN:/etc/hosts.$DOMAIN" \
    -v "/tmp/dnsmasq.base.conf:/etc/dnsmasq/0.base.conf" \
    --cap-add=NET_ADMIN \
    storytel/dnsmasq dnsmasq --no-daemon --log-queries

start_emqx() {
    NAME="$1"
    IP="$2"
    DASHBOARD_PORT="$3"
    docker run -d -t \
        --name "$NAME" \
        --net "$NET" \
        --ip "$IP" \
        --dns "$IP0" \
        -p "$DASHBOARD_PORT:18083" \
        -e EMQX_LOG__CONSOLE_HANDLER__LEVEL=debug \
        -e EMQX_NODE_COOKIE="$COOKIE" \
        -e EMQX_cluster__discovery_strategy='dns' \
        -e EMQX_cluster__dns__name="$DOMAIN" \
        -e EMQX_cluster__dns__record_type="a" \
        "$IMAGE"
}

start_emqx "$NODE1" "$IP1" 18083
start_emqx "$NODE2" "$IP2" 18084