#!/bin/bash
###############################################################################
#
# CRATE Utilities: https://github.com/crate/crate-utils
#
# Licensed to CRATE Technology GmbH ("Crate") under one or more contributor
# license agreements.  See the NOTICE file distributed with this work for
# additional information regarding copyright ownership.  Crate licenses
# this file to you under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.  You may
# obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
# License for the specific language governing permissions and limitations
# under the License.
#
# However, if you have executed another commercial license agreement
# with Crate these terms will supersede the license and you may use the
# software solely pursuant to the terms of the relevant commercial agreement.
#
###############################################################################
#
# CrateDB quick setup program

set -e

INV="\\033[7m"
BRN="\\033[33m"
RED="\\033[31m"
END="\\033[0m\\033[27m"

function wait_for_user() {
    read -p "Press RETURN to continue or any other key to abort" -r -n1 -s x
    if [[ "$x" != '' ]]; then
        exit 1
    fi
}

function prf() {
    echo -e "$INV$BRN$1$END"
}

function on_error() {
    echo -e "$RED
It looks like you hit an issue when trying CrateDB.

Troubleshooting and basic usage information for CrateDB are available at:

    https://crate.io/docs/
$END"
}
trap on_error ERR

function on_exit() {
    # kill crate on exit
    kill "$(jobs -p)"
}

function pre_start_cmd() {
    # display info about crate admin on non gui systems
    if [[ ! $OS = "Darwin" && ! -n $DISPLAY ]]; then
        [ "$(hostname -d)" ] && HOST=$(hostname -f) || HOST=$(hostname)
        prf "- CrateDB will be started in the foreground\\n"
        prf "To open the CrateDB Admin UI, go to http://$HOST:4200/\\n"
    fi
}

function post_start_cmd() {
    # open crate admin if system has gui
    if [[ $OS = "Darwin" || -n $DISPLAY ]]; then
        open 'http://localhost:4200/#/help'
    fi
}

function wait_until_running() {
    # wait until crate is listening on port 4200
    while ! nc -vz localhost 4200 > /dev/null 2>&1 /dev/null; do
        sleep 0.1
    done
}

# Architecture and OS detection
ARCH=$(uname -m)
OS=$(uname -s)

if [ "$ARCH" = "x86_64" ]; then
  if [ "$OS" = "Linux" ]; then
    PLATFORM="x64_linux"
  elif [ "$OS" = "Darwin" ]; then
    PLATFORM="x64_mac"
  else
    echo -e "\\n$RED \\n The platform ${ARCH}:${OS} is not supported.$END\\n\\n"
    exit 1
  fi
elif [ "$ARCH" = "aarch64" ]; then
  if [ "$OS" = "arch" ]; then
    PLATFORM="aarch64_linux"
  else
    echo -e "\\n$RED \\n The platform ${ARCH}:${OS} is not supported.$END\\n\\n"
    exit 1
  fi
else
  echo -e "\\n$RED \\n The architecture: ${ARCH} is not supported.$END\\n\\n"
  exit 1
fi

STABLE_RELEASE_VERSION=$(curl -s https://crate.io/versions.json \
  | grep -o '"crate": *"[^"]*"' \
  | tr -d '" ' \
  | cut -d ":" -f2)
STABLE_RELEASE_FILENAME="crate-${STABLE_RELEASE_VERSION}.tar.gz"
STABLE_RELEASE_DIR="crate-${STABLE_RELEASE_VERSION}"
STABLE_RELEASE_URL=$(curl -Ls -I -w "%{url_effective}" \
  https://cdn.crate.io/downloads/releases/cratedb/${PLATFORM}/${STABLE_RELEASE_FILENAME} | tail -n1)

trap on_exit EXIT

prf "Thank you for trying out CrateDB $STABLE_RELEASE_VERSION\\n"

if [ ! -f "$STABLE_RELEASE_FILENAME" ]; then
    prf "* Downloading CrateDB"
    curl -L --max-redirs 1 -f ${STABLE_RELEASE_URL} > "$STABLE_RELEASE_FILENAME"
else
    prf "- CrateDB $STABLE_RELEASE_VERSION has already been downloaded"
fi

if [ ! -d "$STABLE_RELEASE_DIR" ]; then
    mkdir "$STABLE_RELEASE_DIR" && tar -xzf "$STABLE_RELEASE_FILENAME" -C "$STABLE_RELEASE_DIR" --strip-components 1
else
    prf "- Runtime directory $STABLE_RELEASE_DIR has already been created"
fi

prf "* Starting CrateDB"
pre_start_cmd
"$STABLE_RELEASE_DIR/bin/crate" &
wait_until_running
post_start_cmd
wait