#!/bin/bash
# vim:ft=sh ts=4 sw=4 sts=4
###############################################################################
#
# 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.
#
###############################################################################
#
# YUM repo install script.

set -e
logfile="crate-install.log"

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


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

function prf_err() {
    echo -e "\\n$RED$1$END"
}

function check_java8_debian() {
    # Check if Java 8 is installed on OS that does not ship with Java 8 in official repository.
    if type -p java > /dev/null; then
        # Java is already installed. Check if it's >= Java8
        JAVA_VERSION=$(java -version 2>&1 | awk '/version/ {gsub(/"/, "", $3); split($3, parts, ".")} END {print (parts[1] == 1 ? parts[2] : parts[1])}')
        if [ "$JAVA_VERSION" -lt 8 ]; then
            install_java
        fi
    else
        # No Java installed. Install via third party repo.
        install_java
    fi
}

function install_java() {
    prf "* Installing Java 11 from openjdk-r/pparepository"
    $OPTSUDO sudo add-apt-repository ppa:openjdk-r/ppa
    $OPTSUDO apt-get update
    $OPTSUDO apt-get install -y openjdk-11-jre-headless
}

# OS/Distro Detection
if [ -f /etc/os-release ]; then
    # Arch Linux, Debian, Ubuntu, Raspbian
    # shellcheck disable=SC1091
    . /etc/os-release
    OS=$ID
elif [ -f /etc/debian_version ]; then
    # Old Debian releases
    OS=Debian
elif [ -f /etc/redhat-release ]; then
    # Just mark as RedHat and we'll use Python version detection
    # to know what to install
    OS=RedHat
    # check for systemd
    rpm -q systemd >> /dev/null && SYSTEMD_AVAILABLE=0
elif [ -f /etc/lsb-release ]; then
    # shellcheck disable=SC1091
    . /etc/lsb-release
    OS=$DISTRIB_ID
elif [ -f /etc/arch-release ]; then
    # Archlinux Docker image has an arch-release file instead of os-release
    OS="arch"
elif [ -f /etc/system-release ]; then
    # Amazon Linux
    OS=Amazon
else
    OS=$(uname -s)
fi

if [ "$OS" = "Darwin" ]; then
    prf_err "This script does not support installing on the Mac.
Please use the 1-step trial script available at https://crate.io/docs/."
    exit 1;
fi

# Make lower case
OS=${OS,,}

if [ "$(id -u)" -eq 0 ]; then
    ASROOT=" --asroot "
    OPTSUDO=""
else
    ASROOT=""
    OPTSUDO="sudo "
fi

# Set up a named pipe for logging
npipe=/tmp/$$.tmp
mknod $npipe p

# Log all output to a log for error checking
tee <$npipe $logfile &
exec 1>&-
exec 1>$npipe 2>&1
trap 'rm -f $npipe' EXIT

function on_error() {
    prf_err "It looks like you hit an issue when trying to install Crate.

Troubleshooting and basic usage information for Crate are available at:

    https://crate.io/docs/

If you're still having problems, please send an email to support@crate.io
with the contents of crate-install.log and we'll do our very best to help you
solve your problem.\\n"
}
trap on_error ERR


# Install the necessary package sources
if [ "$OS" = "redhat" ] || [ "$OS" = "amazon" ] || [ "$OS" = "amzn" ]  || [ "$OS" = "centos" ]; then

    rpm -q crate-release >> /dev/null && CRATE_RELEASE_AVAILABLE=0
    if [ "$CRATE_RELEASE_AVAILABLE" == "0" ]; then
        prf "* The crate repository is already installed"
    else
        prf "* Installing YUM sources for Crate\\n"
        $OPTSUDO rpm --import https://cdn.crate.io/downloads/yum/RPM-GPG-KEY-crate
        if [ "$SYSTEMD_AVAILABLE" == "0" ]; then
            CRATE_RELEASE_RPM="https://cdn.crate.io/downloads/yum/7/noarch/crate-release-7.0-1.noarch.rpm"
        else
            CRATE_RELEASE_RPM="https://cdn.crate.io/downloads/yum/6/x86_64/crate-release-6.5-1.noarch.rpm"
        fi
        $OPTSUDO rpm -Uvh $CRATE_RELEASE_RPM
    fi

    rpm -q crate >> /dev/null && CRATE_AVAILABLE=0
    if [ "$CRATE_AVAILABLE" == "0" ]; then
        prf "* The Crate package is already installed"
    else
        prf "\\n* Installing the Crate package\\n\\n"
        $OPTSUDO yum -y install crate
    fi

    prf "* Starting the Service...\\n\\n"
    if [ "$SYSTEMD_AVAILABLE" == "0" ]; then
        $OPTSUDO mkdir -p /opt/crate/data/crate
        $OPTSUDO chown crate:crate /opt/crate/data/crate
        $OPTSUDO systemctl daemon-reload
        $OPTSUDO systemctl enable crate.service
        $OPTSUDO systemctl start crate.service
    else
        $OPTSUDO mkdir -p /opt/crate/data/crate
        $OPTSUDO chown crate:crate /opt/crate/data/crate
        $OPTSUDO service crate start
    fi
elif [ "$OS" == "debian" ] || [ "$OS" == "raspbian" ]; then
    prf "\\n* Installing APT repository for Crate\\n"
    $OPTSUDO apt-get install -y apt-transport-https
    $OPTSUDO wget https://cdn.crate.io/downloads/apt/DEB-GPG-KEY-crate
    $OPTSUDO apt-key add DEB-GPG-KEY-crate -y
    VERSION=$(cut -d "." -f1 /etc/debian_version)
    if [ "$VERSION" == "7" ]; then
        DISTRO="wheezy"
        check_java8_debian
    elif [ "$VERSION" == "8" ]; then
        DISTRO="jessie"
        if [ "$($OPTSUDO grep -e backports /etc/apt/sources.list | wc -l)" -lt 1 ]; then
          echo "deb http://http.debian.net/debian ${DISTRO}-backports main" | $OPTSUDO tee -a /etc/apt/sources.list
        fi
    else
        prf_err "Your version $VERSION is not supported by Crate."
        exit 1
    fi
    if [ "$($OPTSUDO grep -e crate /etc/apt/sources.list | wc -l)" -lt 2 ]; then
        echo "deb https://cdn.crate.io/downloads/apt/stable/ ${DISTRO} main" | $OPTSUDO tee -a /etc/apt/sources.list
        echo "deb-src https://cdn.crate.io/downloads/apt/stable/ ${DISTRO} main" | $OPTSUDO tee -a /etc/apt/sources.list
    fi
    $OPTSUDO apt-get update

    if dpkg -s "crate" | grep "installed" ; then
        prf "* The Crate package is already installed"
    else
        prf "\\n* Installing the Crate package\\n\\n"
        $OPTSUDO apt-get install -y crate
    fi

    prf "* Starting the Service...\\n\\n"
    $OPTSUDO mkdir -p /opt/crate/data/crate
    $OPTSUDO chown crate:crate /opt/crate/data/crate
    if type -p systemd > /dev/null; then
        $OPTSUDO systemctl enable crate
        if $OPTSUDO systemctl status crate | grep "running" ; then
            prf "* Crate is already running"
            $OPTSUDO systemctl restart crate
        else
            prf "* Starting the Service...\\n\\n"
            $OPTSUDO systemctl start crate
        fi
    else
        if $OPTSUDO service crate status | grep "running" ; then
            prf "* Crate is already running"
            $OPTSUDO service crate restart
        else
            prf "* Starting the Service...\\n\\n"
            $OPTSUDO service crate start
        fi
    fi
elif [ "$OS" = "ubuntu" ]; then
    MAJOR_VERSION=$(echo "$VERSION_ID" | cut -d "." -f 1)
    if [ "$MAJOR_VERSION" -lt "16" ]; then
        check_java8_debian
    fi

    if [ "${UBUNTU_CODENAME}x" = "x" ]; then
        source /etc/lsb-release
        UBUNTU_CODENAME=$DISTRIB_CODENAME
    fi

    prf "\\n* Installing APT repository for Crate\\n"
    $OPTSUDO wget https://cdn.crate.io/downloads/apt/DEB-GPG-KEY-crate
    $OPTSUDO apt-key add DEB-GPG-KEY-crate
    if [ ! -f /etc/apt/sources.list.d/crate.list ]; then
        echo "deb https://cdn.crate.io/downloads/deb/stable/ ${UBUNTU_CODENAME} main" | $OPTSUDO tee -a /etc/apt/sources.list.d/crate.list
        echo "deb-src https://cdn.crate.io/downloads/deb/stable/ ${UBUNTU_CODENAME} main" | $OPTSUDO tee -a /etc/apt/sources.list.d/crate.list
    fi
    if [ "$MAJOR_VERSION" -lt "18" ]; then
        # The last version was in "artful" (17.10)
        $OPTSUDO apt-get install -y python-software-properties software-properties-common
    fi
    $OPTSUDO apt-get update

    if dpkg -s "crate" | grep "installed" ; then
        prf "* The Crate package is already installed"
    else
        prf "\\n* Installing the Crate package\\n\\n"
        $OPTSUDO apt-get install -y crate
    fi

    if $OPTSUDO service crate status | grep "running" ; then
        prf "* Crate is already running"
    else
        prf "* Starting the Service...\\n\\n"
        $OPTSUDO mkdir -p /opt/crate/data/crate
        $OPTSUDO chown crate:crate /opt/crate/data/crate
        $OPTSUDO service crate start

    fi
elif [ "$OS" = "arch" ]; then
    prf "\\n* Installing Crate from Arch Linux AUR\\n"
    prf "\\n* Ensuring binutils is installed\\n"
    $OPTSUDO pacman -S --noconfirm --asdeps --needed binutils
    mkdir -p ~/builds
    if [ -d "$HOME/builds/crate" ]; then
        prf "\\n* Deleting old builds\\n"
        rm -rf ~/builds/crate
    fi
    prf "\\n* Getting build files\\n"
    cd ~/builds && curl -O https://aur.archlinux.org/cgit/aur.git/snapshot/crate.tar.gz
    cd ~/builds && tar xzvf crate.tar.gz
    prf "\\n* building and installing crate package\\n"
    cd ~/builds/crate && makepkg $ASROOT -sfi
    prf "\\n* starting daemon\\n"
    $OPTSUDO systemctl start crate
else
    prf_err "Your OS or distribution $OS is not supported by this install script.
Please visit

    https://crate.io/docs/

For help.\\n"
    exit 1;
fi


prf "Your Crate Service has started up for the first time.

To checkout the admin UI open http://$(hostname -f):4200/admin in your browser
"