199 lines
6.5 KiB
Bash
Executable file
199 lines
6.5 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
set -o errtrace
|
|
|
|
# color schemes
|
|
# Ansi color code variables
|
|
red="\e[0;91m"
|
|
blue="\e[0;94m"
|
|
expand_bg="\e[K"
|
|
blue_bg="\e[0;104m${expand_bg}"
|
|
red_bg="\e[0;101m${expand_bg}"
|
|
green_bg="\e[0;102m${expand_bg}"
|
|
green="\e[0;92m"
|
|
white="\e[0;97m"
|
|
bold="\e[1m"
|
|
uline="\e[4m"
|
|
reset="\e[0m"
|
|
|
|
working_dir=$(pwd)
|
|
script_name=`basename "$0"`
|
|
|
|
echo -e ${reset}
|
|
|
|
## installing kubectl
|
|
which kubectl &> /dev/null || {
|
|
echo "kubectl not installed. Installing it..."
|
|
sudo curl -SsL https://dl.k8s.io/release/v1.20.0/bin/linux/amd64/kubectl -o /usr/local/bin/kubectl ; sudo chmod +x /usr/local/bin/kubectl
|
|
}
|
|
|
|
## installing stern
|
|
which stern &> /dev/null || {
|
|
echo "stern not installed. installing..."
|
|
sudo curl -SsL https://github.com/derdanne/stern/releases/download/2.1.16/stern_linux_amd64 -o /usr/local/bin/stern ; sudo chmod +x /usr/local/bin/stern
|
|
}
|
|
|
|
## installing k9s
|
|
which k9s &> /dev/null || {
|
|
echo "k9s not installed. Installing it..."
|
|
sudo curl -SsL https://github.com/derailed/k9s/releases/download/v0.24.2/k9s_Linux_x86_64.tar.gz -o /tmp/k9s.tar.gz
|
|
cd /tmp
|
|
tar -xf k9s.tar.gz
|
|
sudo mv k9s /usr/local/bin/k9s
|
|
sudo chmod +x /usr/local/bin/k9s
|
|
cd -
|
|
}
|
|
|
|
which ansible &> /dev/null || {
|
|
echo "ansible not installed. Installing it..."
|
|
which pip || (sudo apt update && sudo apt install python3-pip -y)
|
|
sudo pip3 install ansible==2.10.0
|
|
}
|
|
|
|
## installing helm
|
|
which helm &> /dev/null
|
|
if [[ $? -ne 0 ]]; then
|
|
echo "helm not installed. Installing it..."
|
|
curl -ssl https://get.helm.sh/helm-v3.4.2-linux-amd64.tar.gz -o /tmp/helm.tar.gz
|
|
tar -xf /tmp/helm.tar.gz
|
|
chmod +x linux-amd64/helm
|
|
sudo cp linux-amd64/helm /usr/local/bin/helm
|
|
rm -rf linux-amd64/helm /tmp/helm.tar.gz
|
|
fi
|
|
|
|
# ## checking kubernetes access
|
|
# kubectl cluster-info &> /dev/null
|
|
# if [[ $? -ne 0 ]]; then
|
|
# echo -e "${red}kubernetes cluster is not accessible.\nplease check ${bold}kubeconfig${reset}${red} env variable is set or ${bold}~/.kube/config exists.${reset}"
|
|
# exit 1
|
|
# fi
|
|
|
|
# make all stderr red
|
|
color()(set -o pipefail;"$@" 2>&1>&3|sed $'s,.*,\e[31m&\e[m,'>&2)3>&1
|
|
|
|
usage() {
|
|
echo -e ${bold}${yellow} '''
|
|
This script will install and configure OpenReplay apps and databases on the kubernetes cluster,
|
|
which is accesd with the ${HOME}/.kube/config or $KUBECONFIG env variable.
|
|
'''
|
|
cat <<"EOF"
|
|
___ ____ _
|
|
/ _ \ _ __ ___ _ __ | _ \ ___ _ __ | | __ _ _ _
|
|
| | | | '_ \ / _ \ '_ \| |_) / _ \ '_ \| |/ _` | | | |
|
|
| |_| | |_) | __/ | | | _ < __/ |_) | | (_| | |_| |
|
|
\___/| .__/ \___|_| |_|_| \_\___| .__/|_|\__,_|\__, |
|
|
|_| |_| |___/
|
|
|
|
EOF
|
|
echo -e "${green}Usage: openreplay-cli [ -h | --help ]
|
|
[ -v | --verbose ]
|
|
[ -m | --monitoring <Only for enterprise edition> ]
|
|
[ -e | --enterprise <enerprise_key> ]
|
|
[ -a | --app APP_NAME ] to install/reinstall specific application
|
|
[ -t | --type small|medium|ideal ]"
|
|
echo -e "${reset}${blue}type defines the resource limits applied for the installation:
|
|
small: 2core 8G machine
|
|
medium: 4core 16G machine
|
|
ideal: 8core 32G machine
|
|
|
|
apps can specifically be installed/reinstalled:
|
|
alerts assets chalice ender http integrations ios-proxy pg redis sink storage frontend postgresql redis clickhouse
|
|
${reset}"
|
|
echo type value: $installation_type
|
|
exit 0
|
|
}
|
|
|
|
# Defaults to minimum installation
|
|
installation_type=1
|
|
type() {
|
|
case "$1" in
|
|
small) installation_type=1 ;;
|
|
medium) installation_type=1.5 ;;
|
|
ideal) installation_type=2 ;;
|
|
*)
|
|
echo -e ${red}${bold}'ERROR!!!\nwrong value for `type`'${reset}
|
|
usage ;;
|
|
esac
|
|
}
|
|
|
|
function app(){
|
|
case $1 in
|
|
nginx)
|
|
# Resetting the redirection rule
|
|
sed -i 's/.* return 301 .*/ # return 301 https:\/\/$host$request_uri;/g' nginx-ingress/nginx-ingress/templates/configmap.yaml
|
|
[[ NGINX_REDIRECT_HTTPS -eq 1 ]] && {
|
|
sed -i "s/# return 301/return 301/g" nginx-ingress/nginx-ingress/templates/configmap.yaml
|
|
}
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type --tags nginx -v
|
|
exit 0
|
|
;;
|
|
postgresql|redis|clickhouse)
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type -e db_name=$1 --tags template --tags db -v
|
|
exit 0
|
|
;;
|
|
frontend)
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type --tags frontend -v
|
|
exit 0
|
|
;;
|
|
*)
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type -e app_name=$1 --tags app -v
|
|
exit 0
|
|
;;
|
|
esac
|
|
}
|
|
|
|
enterprise=0
|
|
function enterprise(){
|
|
enterprise=1
|
|
sed -i "s#enterprise_edition_license.*#enterprise_edition_license: \"${1}\"#g" vars.yaml
|
|
# Updating image version to be ee
|
|
sed -i "s/\(image_tag.*[0-9]\)\"$/\1-ee\"/" vars.yaml
|
|
echo "Importing enterprise code..."
|
|
cp -rf ../../ee/scripts/* ../
|
|
}
|
|
monitoring(){
|
|
if [[ enterprise -eq 0 ]]; then
|
|
echo -e "${red}Monitoring is supported only for enterprise edition.\n bash ./${script_name} -e <key> --monitoring ${reset}"
|
|
exit 1
|
|
fi
|
|
sed -i "s#enable_monitoring.*#enable_monitoring: \"true\"#g" vars.yaml
|
|
}
|
|
|
|
# Parsing command line args.
|
|
PARSED_ARGUMENTS=$(color getopt -a -n openreplay-cli -o vht:a:e:m --long verbose,help,type:,app:,enterprise:,monitoring -- "$@")
|
|
VALID_ARGUMENTS=$?
|
|
if [[ "$VALID_ARGUMENTS" != "0" ]]; then
|
|
usage
|
|
fi
|
|
|
|
eval set -- "$PARSED_ARGUMENTS"
|
|
while :
|
|
do
|
|
case "$1" in
|
|
-v | --verbose) VERBOSE=1 ; shift ;;
|
|
-h | --help) usage ; shift ;;
|
|
-t | --type) type $2 ; shift 2 ;;
|
|
-a | --app) app $2 ; shift 2 ;;
|
|
-e | --enterprise) enterprise $2 ; shift 2 ;;
|
|
-m | --monitoring) monitoring ; shift ;;
|
|
# -- means the end of the arguments; drop this, and break out of the while loop
|
|
--) shift; break ;;
|
|
# If invalid options were passed, then getopt should have reported an error,
|
|
# which we checked as VALID_ARGUMENTS when getopt was called...
|
|
*) echo "Unexpected option: $1 - this should not happen."
|
|
usage ;;
|
|
esac
|
|
done
|
|
|
|
[[ $VERBOSE -eq 1 ]] && set -x
|
|
|
|
{
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type --tags pre-check -v
|
|
} || exit $?
|
|
{
|
|
ansible-playbook -c local setup.yaml -e @vars.yaml -e scale=$installation_type --skip-tags pre-check -v
|
|
} || exit $?
|
|
|
|
|
|
|
|
|