single-node.sh

#!/bin/bash
# Copyright 2018 Evgeny Golyshev. All Rights Reserved.
#
# Licensed 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.

set -e

USAGE="USAGE: $0 build <target directory>|create-superuser|dbshell|list|loaddata|makemessages|makemigrations|migrate|rebuild|remove|restart|shell <service>|start|stop-all|stop <service>"

if [ -z "$1" ]; then
    >&2 echo "${USAGE}"
    exit 1
fi

if [ "$(id -u)" != "0" ]; then
    >&2 echo "This script must be run as root"
    exit 1
fi

. ./essentials.sh

. ./functions.sh

if [ -f localsettings ]; then
    # localsettings may not exist, so it's necessary to ignore SC1091.
    # shellcheck disable=SC1091
    . ./localsettings
fi

set -x

export ALLOWED_HOSTS=${ALLOWED_HOSTS:=localhost,127.0.0.1}

export CUSDEB_WEB_CLIENT_PORT=${CUSDEB_WEB_CLIENT_PORT:=8000}

export CUSDEB_API_PORT=${CUSDEB_API_PORT:=8001}

export BM_PORT=${BM_PORT:=8002}

export DOMINION_PORT=${DOMINION_PORT:=8003}

export ORION_PORT=${ORION_PORT:=8004}

export HELPIK_PORT=${HELPIK_PORT:=8005}

export TZ_PORT=${TZ_PORT:=8006}

export ANONYMOUS_PORT=${ANONYMOUS_PORT:=8007}

export BUILD_RESULT_PORT=${BUILD_RESULT_PORT:=8008}

export MONGO_DATABASE=${MONGO_DATABASE:=cusdeb}

export MONGO_HOST=${MONGO_HOST:=localhost}

export MONGO_PORT=${MONGO_PORT:=33018}

export MONGO_TAG=${MONGO_TAG:=4.2}

export PG_NAME=${PG_NAME:=cusdeb}

export PG_HOST=${PG_HOST:=localhost}

export PG_PASSWORD=${PG_PASSWORD:=secret}

export PG_PORT=${PG_PORT:=54321}

export PG_TAG=${PG_TAG:=12}

export PG_USER=${PG_USER:=postgres}

export RABBITMQ_PORT=${RABBITMQ_PORT:=5672}

export RABBITMQ_TAG=${RABBITMQ_TAG:=3.7}

export REDIS_HOST=${REDIS_HOST=localhost}

export REDIS_PORT=${REDIS_PORT:=63791}

export REDIS_TAG=${REDIS_TAG:=6}

export MAILHOG_PORT=${MAILHOG_PORT:=1025}

export MAILHOST_WEB_PORT=${MAILHOST_WEB_PORT:=8025}

export MAILHOG_TAG=${MAILHOG_TAG:=v1.0.0}

export SECRET_KEY=${SECRET_KEY:="secret"}

export EMAIL_HOST=${EMAIL_HOST:=localhost}

export EMAIL_PORT=${EMAIL_PORT:=1025}

export EMAIL_HOST_USER=${EMAIL_HOST_USER:=test}

export EMAIL_HOST_PASSWORD=${EMAIL_HOST_PASSWORD:=test}

export EMAIL_USE_TLS=${EMAIL_USE_TLS:=false}

export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}

export PKG_CONFIG_PATH=${PKG_CONFIG_PATH}

USER="$(get_owner .)"
export USER

set +x

info "checking dependencies"
check_dependencies

#
# Parse params
#

case $1 in
build)
    if [ -f cusdeb ]; then
        fatal "cusdeb-single-node is already installed. Run rebuild command."
        exit 1
    fi

    if [ -z "$2" ]; then
        fatal "target directory is not specified"
        exit 1
    fi

    if [ ! -d "$2" ]; then
        fatal "$2 does not exist"
        exit 1
    fi

    resume_state
    state="$(get_state)"

    TARGET="$(get_absolute_path "$2")"

    build_env "${state}" full

    ;;
rebuild)
    TARGET="$(<cusdeb)"

    if [ -z "$2" ]; then
        recipe=fast
    else
        recipe="$2"
    fi

    switch_state_to init
    state="$(get_state)"

    info "cleaning up"

    case $recipe in
    fast)
        pushd "$TARGET"
            find . -maxdepth 1 ! -name 'chroots' ! -name '.' ! -name '..' -exec rm -r {} \;
        popd
        ;;
    full)
        if [ "$(ls -A "$TARGET")" ]; then
            # shellcheck disable=SC2115
            rm -r "$TARGET"/*
        fi
    esac

    build_env "$state" "$recipe"

    ;;
create-superuser)
    run_manage_py createsuperuser

    ;;
dbshell)
    run_manage_py dbshell

    ;;
list)
    TARGET="$(<cusdeb)"
    export TARGET

    supervisorctl -c config/supervisord.conf status | awk '{ print $1 " (" $2 ")" }' | while IFS= read -r service; do
        >&2 echo "${service}"
    done

    ;;
loaddata)
    if [ -z "$2" ]; then
        fatal "fixture is not specified"
        exit 1
    fi

    run_manage_py loaddata "$2"

    ;;
makemessages)
    run_manage_py makemessages -l ru

    ;;
makemigrations)
    run_manage_py makemigrations

    ;;
migrate)
    run_manage_py migrate

    ;;
remove)
    check_if_cusdeb_single_node_is_installed

    stop_containers

    TARGET="$(<cusdeb)"

    if [[ -d "${TARGET}" ]]; then
        if [[ "$(ls -A "${TARGET}")" ]]; then
            if prompt "Do you want to delete all contents of the directory ${TARGET}?"; then
                for item in "${TARGET}"/*; do
                    info "removing ${item}"
                    rm -r "${item}"
                done
            fi
        else
            info "${TARGET} is empty"
        fi
    else
        info "${TARGET} does not exist"
    fi

    rm cusdeb

    ;;
restart)
    >&2 echo "stub"
    ;;
shell)
    run_manage_py shell

    ;;
start)
    check_if_cusdeb_single_node_is_installed

    TARGET="$(cat cusdeb)"
    export TARGET

    export_node_envs

    check_ports

    run_containers
    trap "stop_containers && exit 130" 2

    print_doc

    run_daemons

    ;;
stop-all)
    check_if_cusdeb_single_node_is_installed

    stop_containers

    TARGET="$(cat cusdeb)"
    export TARGET

    stop_daemons

    ;;
stop)
    check_if_cusdeb_single_node_is_installed

    TARGET="$(<cusdeb)"
    export TARGET

    if [[ -z $2 ]]; then
        fatal "<service> is not specified. To see all services run 'sudo $0 list'"
        exit 1
    fi

    supervisorctl -c config/supervisord.conf stop "$2"

    ;;
*)
    >&2 echo "${USAGE}"
    exit 1
    ;;
esac