run.sh

#!/usr/bin/env bash
################################################################
#                                                              #
#              Production runner script for:                   #
#                                                              #
#                  Postfix Log Parser                          #
#            (C) 2019 Privex Inc.   GNU AGPL v3                #
#                                                              #
#      Privex Site: https://www.privex.io/                     #
#                                                              #
#      Github Repo: https://github.com/Privex/postfix-parser   #
#                                                              #
################################################################

if [ -t 1 ]; then
    BOLD="$(tput bold)" RED="$(tput setaf 1)" GREEN="$(tput setaf 2)" YELLOW="$(tput setaf 3)" BLUE="$(tput setaf 4)"
    MAGENTA="$(tput setaf 5)" CYAN="$(tput setaf 6)" WHITE="$(tput setaf 7)" RESET="$(tput sgr0)"
else
    BOLD="" RED="" GREEN="" YELLOW="" BLUE=""
    MAGENTA="" CYAN="" WHITE="" RESET=""
fi

export PATH="${HOME}/.local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:${PATH}"

# easy coloured messages function
# written by @someguy123
function msg () {
    # usage: msg [color] message
    if [[ "$#" -eq 0 ]]; then echo ""; return; fi;
    if [[ "$#" -eq 1 ]]; then
        echo -e "$1"
        return
    fi

    ts="no"
    if [[ "$#" -gt 2 ]] && [[ "$1" == "ts" ]]; then
        ts="yes"
        shift
    fi
    if [[ "$#" -gt 2 ]] && [[ "$1" == "bold" ]]; then
        echo -n "${BOLD}"
        shift
    fi
    [[ "$ts" == "yes" ]] && _msg="[$(date +'%Y-%m-%d %H:%M:%S %Z')] ${@:2}" || _msg="${@:2}"

    case "$1" in
        bold) echo -e "${BOLD}${_msg}${RESET}";;
        [Bb]*) echo -e "${BLUE}${_msg}${RESET}";;
        [Yy]*) echo -e "${YELLOW}${_msg}${RESET}";;
        [Rr]*) echo -e "${RED}${_msg}${RESET}";;
        [Gg]*) echo -e "${GREEN}${_msg}${RESET}";;
        * ) echo -e "${_msg}";;
    esac
}



######
# Directory where the script is located, so we can source files regardless of where PWD is
######

DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

cd "$DIR"

[[ -f .env ]] && source .env || msg yellow " >> Warning: No .env file found."

# Override these defaults inside of `.env`
: ${HOST='127.0.0.1'}
: ${PORT='8487'}
: ${GU_WORKERS='10'}    # Number of Gunicorn worker processes

case "$1" in
    dev*)
        QUART_APP=wsgi QUART_ENV=development ./wsgi.py
        ;;
    prod*)
        pipenv run hypercorn -b "${HOST}:${PORT}" -w "$GU_WORKERS" wsgi
        ;;
    cron|import|parse*)
        pipenv run ./manage.py parse
        ;;
    *)
        echo "Runner script for Privex's Postfix Log Parser"
        echo ""
        msg bold red "Unknown command.\n"
        msg bold green "Postfix Log Parser - (C) 2019 Privex Inc."
        msg bold green "    Website: https://www.privex.io/ \n    Source: https://github.com/Privex/postfix-parser\n"
        msg green "Available run.sh commands:\n"
        msg yellow "\t dev - Start the Flask development server - UNSAFE FOR PRODUCTION"
        msg yellow "\t prod - Start the production Hypercorn server"
        msg yellow "\t parse - Parse and import Postfix logs"
        msg
        ;;
esac