#!/bin/bash

# Copyright (c) The Diem Core Contributors.

# Copyright 2020-2021 The Datafuse Authors.

# SPDX-License-Identifier: Apache-2.0.

SCRIPT_PATH="$( cd "$( dirname "$0" )" >/dev/null 2>&1 && pwd )"

cd "$SCRIPT_PATH/../.." || exit

function add_to_profile {

eval "$1"

FOUND=$(grep -c "$1" < "${HOME}/.profile")

if [ "$FOUND" == "0" ]; then

echo "$1" >> "${HOME}"/.profile

fi

}

function update_path_and_profile {

touch "${HOME}"/.profile

mkdir -p "${HOME}"/bin

if [ -n "$CARGO_HOME" ]; then

add_to_profile "export CARGO_HOME=\"${CARGO_HOME}\""

add_to_profile "export PATH=\"${HOME}/bin:${CARGO_HOME}/bin:\$PATH\""

else

add_to_profile "export PATH=\"${HOME}/bin:${HOME}/.cargo/bin:\$PATH\""

fi

}

function install_build_essentials {

PACKAGE_MANAGER=$1

#Differently named packages for pkg-configs

if [[ "$PACKAGE_MANAGER" == "apt-get" ]]; then

install_pkg build-essential "$PACKAGE_MANAGER"

fi

if [[ "$PACKAGE_MANAGER" == "pacman" ]]; then

install_pkg base-devel "$PACKAGE_MANAGER"

fi

if [[ "$PACKAGE_MANAGER" == "apk" ]]; then

install_pkg alpine-sdk "$PACKAGE_MANAGER"

install_pkg coreutils "$PACKAGE_MANAGER"

fi

if [[ "$PACKAGE_MANAGER" == "yum" ]] || [[ "$PACKAGE_MANAGER" == "dnf" ]]; then

install_pkg gcc "$PACKAGE_MANAGER"

install_pkg gcc-c++ "$PACKAGE_MANAGER"

install_pkg make "$PACKAGE_MANAGER"

fi

#if [[ "$PACKAGE_MANAGER" == "brew" ]]; then

# install_pkg pkgconfig "$PACKAGE_MANAGER"

#fi

}

function install_rustup {

BATCH_MODE=$1

# Install Rust

[[ "${BATCH_MODE}" == "false" ]] && echo "Installing Rust......"

if rustup --version &>/dev/null; then

[[ "${BATCH_MODE}" == "false" ]] && echo "Rust is already installed"

else

curl https://sh.rustup.rs -sSf | sh -s -- -y --default-toolchain stable

PATH="${HOME}/.cargo/bin:${PATH}"

source $HOME/.cargo/env

fi

}

function install_pkg {

package=$1

PACKAGE_MANAGER=$2

PRE_COMMAND=()

if [ "$(whoami)" != 'root' ]; then

PRE_COMMAND=(sudo)

fi

if which "$package" &>/dev/null; then

echo "$package is already installed"

else

echo "Installing ${package}."

if [[ "$PACKAGE_MANAGER" == "yum" ]]; then

"${PRE_COMMAND[@]}" yum install "${package}" -y

elif [[ "$PACKAGE_MANAGER" == "apt-get" ]]; then

"${PRE_COMMAND[@]}" apt-get install "${package}" --no-install-recommends -y

elif [[ "$PACKAGE_MANAGER" == "pacman" ]]; then

"${PRE_COMMAND[@]}" pacman -Syu "$package" --noconfirm

elif [[ "$PACKAGE_MANAGER" == "apk" ]]; then

apk --update add --no-cache "${package}"

elif [[ "$PACKAGE_MANAGER" == "dnf" ]]; then

dnf install "$package"

elif [[ "$PACKAGE_MANAGER" == "brew" ]]; then

brew install "$package"

fi

fi

}

function install_pkg_config {

PACKAGE_MANAGER=$1

#Differently named packages for pkg-configs

if [[ "$PACKAGE_MANAGER" == "apt-get" ]] || [[ "$PACKAGE_MANAGER" == "dnf" ]]; then

install_pkg pkg-config "$PACKAGE_MANAGER"

fi

if [[ "$PACKAGE_MANAGER" == "pacman" ]]; then

install_pkg pkgconf "$PACKAGE_MANAGER"

fi

if [[ "$PACKAGE_MANAGER" == "brew" ]] || [[ "$PACKAGE_MANAGER" == "apk" ]] || [[ "$PACKAGE_MANAGER" == "yum" ]]; then

install_pkg pkgconfig "$PACKAGE_MANAGER"

fi

}

function install_toolchain {

version=$1

echo "Installing ${version} of rust toolchain"

rustup install "$version"

rustup set profile minimal

rustup component add rustfmt --toolchain "$version"

rustup component add clippy --toolchain "$version"

rustup component add miri --toolchain "$version"

rustup default "$version"

}

function usage {

cat <<EOF

usage: $0 [options]"

options:

-b Enable BATCH_MODE for installation

-t Install build tools

-o Install some operation tools

-p Install profile

-v Verbose mode

-y Install prover

-s Install codegen tools

EOF

}

function welcome_message {

cat <<EOF

Welcome to FuseQuery!

This script will download and install the necessary dependencies needed to

build, test and inspect FuseQuery.

Based on your selection, these tools will be included:

EOF

if [[ "$INSTALL_BUILD_TOOLS" == "true" ]]; then

cat <<EOF

Build tools (since -t or no option was provided):

* Rust (and the necessary components, e.g. rust-fmt, clippy)

* lcov

* pkg-config

* libssl-dev

* if linux, gcc-powerpc-linux-gnu

EOF

fi

if [[ "$OPERATIONS" == "true" ]]; then

cat <<EOF

Operation tools (since -o was provided):

* python3

* docker

EOF

fi

if [[ "$INSTALL_CODEGEN" == "true" ]]; then

cat <<EOF

Codegen tools (since -s was provided):

* Python3 (numpy, pyre-check)

EOF

fi

if [[ "$INSTALL_PROFILE" == "true" ]]; then

cat <<EOF

Moreover, ~/.profile will be updated (since -p was provided).

EOF

fi

cat <<EOF

If you'd prefer to install these dependencies yourself, please exit this script

now with Ctrl-C.

EOF

}

BATCH_MODE=true;

VERBOSE=false;

INSTALL_BUILD_TOOLS=false;

OPERATIONS=false;

INSTALL_PROFILE=false;

INSTALL_PROVER=false;

INSTALL_CODEGEN=false;

#parse args

while getopts "bmtopvysh" arg; do

case "$arg" in

b)

BATCH_MODE="true"

;;

t)

INSTALL_BUILD_TOOLS="true"

;;

o)

OPERATIONS="true"

;;

p)

INSTALL_PROFILE="true"

;;

v)

VERBOSE=true

;;

y)

INSTALL_PROVER="true"

;;

s)

INSTALL_CODEGEN="true"

;;

*)

usage;

exit 0;

;;

esac

done

if [[ "$VERBOSE" == "true" ]]; then

set -x

fi

if [[ "$INSTALL_BUILD_TOOLS" == "false" ]] && \

[[ "$OPERATIONS" == "false" ]] && \

[[ "$INSTALL_PROFILE" == "false" ]] && \

[[ "$INSTALL_PROVER" == "false" ]] && \

[[ "$INSTALL_CODEGEN" == "false" ]]; then

INSTALL_BUILD_TOOLS="true"

fi

if [ ! -f rust-toolchain ]; then

echo "Unknown location. Please run this from the datafuse repository. Abort."

exit 1

fi

PRE_COMMAND=()

if [ "$(whoami)" != 'root' ]; then

PRE_COMMAND=(sudo)

fi

PACKAGE_MANAGER=

if [[ "$(uname)" == "Linux" ]]; then

if command -v yum &> /dev/null; then

PACKAGE_MANAGER="yum"

elif command -v apt-get &> /dev/null; then

PACKAGE_MANAGER="apt-get"

elif command -v pacman &> /dev/null; then

PACKAGE_MANAGER="pacman"

elif command -v apk &>/dev/null; then

PACKAGE_MANAGER="apk"

elif command -v dnf &>/dev/null; then

echo "WARNING: dnf package manager support is experimental"

PACKAGE_MANAGER="dnf"

else

echo "Unable to find supported package manager (yum, apt-get, dnf, or pacman). Abort"

exit 1

fi

elif [[ "$(uname)" == "Darwin" ]]; then

if which brew &>/dev/null; then

PACKAGE_MANAGER="brew"

else

echo "Missing package manager Homebrew (https://brew.sh/). Abort"

exit 1

fi

else

echo "Unknown OS. Abort."

exit 1

fi

if [[ "$BATCH_MODE" == "false" ]]; then

welcome_message

printf "Proceed with installing necessary dependencies? (y/N) > "

read -e -r input

if [[ "$input" != "y"* ]]; then

echo "Exiting..."

exit 0

fi

fi

if [[ "$PACKAGE_MANAGER" == "apt-get" ]]; then

[[ "$BATCH_MODE" == "false" ]] && echo "Updating apt-get......"

"${PRE_COMMAND[@]}" apt-get update

[[ "$BATCH_MODE" == "false" ]] && echo "Installing ca-certificates......"

"${PRE_COMMAND[@]}" install_pkg ca-certificates "$PACKAGE_MANAGER"

fi

[[ "$INSTALL_PROFILE" == "true" ]] && update_path_and_profile

install_pkg curl "$PACKAGE_MANAGER"

if [[ "$INSTALL_BUILD_TOOLS" == "true" ]]; then

install_build_essentials "$PACKAGE_MANAGER"

install_pkg cmake "$PACKAGE_MANAGER"

install_pkg clang "$PACKAGE_MANAGER"

install_pkg llvm "$PACKAGE_MANAGER"

install_pkg_config "$PACKAGE_MANAGER"

install_rustup "$BATCH_MODE"

install_toolchain "$(cat ./rust-toolchain)"

install_pkg lcov "$PACKAGE_MANAGER"

fi

if [[ "$OPERATIONS" == "true" ]]; then

install_pkg python3 "$PACKAGE_MANAGER"

install_pkg git "$PACKAGE_MANAGER"

#for timeout

if [[ "$PACKAGE_MANAGER" == "apt-get" ]]; then

install_pkg coreutils "$PACKAGE_MANAGER"

fi

fi

if [[ "$INSTALL_CODEGEN" == "true" ]]; then

install_pkg clang "$PACKAGE_MANAGER"

install_pkg llvm "$PACKAGE_MANAGER"

if [[ "$PACKAGE_MANAGER" == "apt-get" ]]; then

install_pkg python3-all-dev "$PACKAGE_MANAGER"

install_pkg python3-setuptools "$PACKAGE_MANAGER"

install_pkg python3-pip "$PACKAGE_MANAGER"

else

install_pkg python3 "$PACKAGE_MANAGER"

fi

"${PRE_COMMAND[@]}" python3 -m pip install coscmd PyYAML

fi

[[ "${BATCH_MODE}" == "false" ]] && cat <<EOF

Finished installing all dependencies.

You should now be able to build the project by running:

cargo build

EOF

exit 0