#!/usr/bin/env bash
#
# The MIT License
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#

set -e

DEFAULT_INFLUXDB_VERSION="1.8"
INFLUXDB_VERSION="${INFLUXDB_VERSION:-$DEFAULT_INFLUXDB_VERSION}"
INFLUXDB_IMAGE=influxdb:${INFLUXDB_VERSION}

DEFAULT_INFLUXDB_V2_VERSION="latest"
INFLUXDB_V2_VERSION="${INFLUXDB_V2_VERSION:-$DEFAULT_INFLUXDB_V2_VERSION}"
INFLUXDB_V2_IMAGE=influxdb:${INFLUXDB_V2_VERSION}

DEFAULT_DOCKER_NET_DRIVER="bridge"
DOCKER_NET_DRIVER="${DOCKER_NET_DRIVER:-$DEFAULT_DOCKER_NET_DRIVER}"

SCRIPT_PATH="$( cd "$(dirname "$0")" ; pwd -P )"

docker kill influxdb || true
docker rm influxdb || true
docker kill influxdb_v2 || true
docker rm influxdb_v2 || true
docker kill influxdb_v2_onboarding || true
docker rm influxdb_v2_onboarding || true
docker network rm influx_network || true
docker network create -d ${DOCKER_NET_DRIVER} influx_network --subnet 192.168.0.0/24 --gateway 192.168.0.1

echo
echo "Restarting InfluxDB [${INFLUXDB_IMAGE}] ..."
echo

#
# InfluxDB
#

docker pull ${INFLUXDB_IMAGE} || true
docker run \
       --detach \
       --name influxdb \
       --network influx_network \
       --publish 8086:8086 \
       --publish 8089:8089/udp \
       --volume ${SCRIPT_PATH}/influxdb.conf:/etc/influxdb/influxdb.conf \
       ${INFLUXDB_IMAGE}

echo "Wait to start InfluxDB"
wget -S --spider --tries=20 --retry-connrefused --waitretry=5 http://localhost:8086/ping

#
# InfluxDB 2.x
#
echo
echo "Restarting InfluxDB 2.x [${INFLUXDB_V2_IMAGE}] ... "
echo

docker pull ${INFLUXDB_V2_IMAGE} || true
docker run \
       --env INFLUXD_HTTP_BIND_ADDRESS=:9999 \
       --detach \
       --name influxdb_v2 \
       --network influx_network \
       --publish 9999:9999 \
       ${INFLUXDB_V2_IMAGE}

echo "Wait to start InfluxDB 2.x"
wget -S --spider --tries=20 --retry-connrefused --waitretry=5 http://localhost:9999/metrics

echo
echo "Post onBoarding request, to setup initial user (my-user@my-password), org (my-org) and bucketSetup (my-bucket)"
echo
curl -i -X POST http://localhost:9999/api/v2/setup -H 'accept: application/json' \
    -d '{
            "username": "my-user",
            "password": "my-password",
            "org": "my-org",
            "bucket": "my-bucket",
            "token": "my-token"
        }'

#
# InfluxDB 2.x
#
echo
echo "Restarting InfluxDB 2.x for onboarding test... "
echo

docker run \
       --env INFLUXD_HTTP_BIND_ADDRESS=:9999 \
       --detach \
       --name influxdb_v2_onboarding \
       --network influx_network \
       --publish 9990:9999 \
       ${INFLUXDB_V2_IMAGE}