| 1 | #!/usr/bin/env bash | 
| 2 |  | 
| 3 | # Script variables | 
| 4 | PYTHON_MAJOR=3 | 
| 5 | PYTHON_MINOR=7 | 
| 6 | ACCEPTABLE_PYTHON_COMMANDS="python3 python3.10 python3.9 python3.8 python3.7 python" | 
| 7 |  | 
| 8 | SCRIPT_SOURCE_DIR=$(dirname -- "${BASH_SOURCE[0]}") | 
| 9 |  | 
| 10 | while read -r requirement | 
| 11 | do | 
| 12 | REQUIRED_PIP_MODULES="$REQUIRED_PIP_MODULES $(sed -r "s/:.*//g" <<< $requirement)" | 
| 13 | REQUIRED_PYTHON_MODULES="$REQUIRED_PYTHON_MODULES $(sed -r "s/.*://g" <<< $requirement)" | 
| 14 | done < $SCRIPT_SOURCE_DIR/requirements.txt | 
| 15 |  | 
| 16 | # Plumbing | 
| 17 | set -o pipefail | 
| 18 |  | 
| 19 | RESTORE=$(echo -en '\033[0m') | 
| 20 | STANDOUT=$(echo -en '\033[7m') | 
| 21 | RED=$(echo -en '\033[00;31m') | 
| 22 | GREEN=$(echo -en '\033[00;32m') | 
| 23 | YELLOW=$(echo -en '\033[00;33m') | 
| 24 | PURPLE=$(echo -en '\033[00;35m') | 
| 25 | LIGHTGRAY=$(echo -en '\033[00;37m') | 
| 26 | LRED=$(echo -en '\033[01;31m') | 
| 27 | LGREEN=$(echo -en '\033[01;32m') | 
| 28 | LYELLOW=$(echo -en '\033[01;33m') | 
| 29 | LBLUE=$(echo -en '\033[01;34m') | 
| 30 | LCYAN=$(echo -en '\033[01;36m') | 
| 31 |  | 
| 32 |  | 
| 33 |  | 
| 34 | function usage { | 
| 35 | echo "Usage: $0 <prod|test>" | 
| 36 | exit 0 | 
| 37 | } | 
| 38 |  | 
| 39 | if [ "$EUID" -ne 0 ]; then | 
| 40 | echo "error: must run as root" | 
| 41 | usage | 
| 42 | fi | 
| 43 |  | 
| 44 | # Handle argument input | 
| 45 | case $1 in | 
| 46 | h|help|?|-h|-help|-?|--h|--help|--?) | 
| 47 | usage | 
| 48 | ;; | 
| 49 | prod) | 
| 50 | ;; | 
| 51 | test) | 
| 52 | ;; | 
| 53 | *) | 
| 54 | usage | 
| 55 | ;; | 
| 56 | esac | 
| 57 |  | 
| 58 | MODE=$1 | 
| 59 |  | 
| 60 | # Determine Python interpreter to use | 
| 61 | echo "${LIGHTGRAY}Determining Python interpreter${RESTORE}" | 
| 62 | if [ -z $PYTHON_COMMAND ]; then | 
| 63 | for COMMAND in $ACCEPTABLE_PYTHON_COMMANDS | 
| 64 | do | 
| 65 | if command -v $COMMAND &> /dev/null; then | 
| 66 | PYTHON_COMMAND=$COMMAND | 
| 67 | break | 
| 68 | fi | 
| 69 | done | 
| 70 | fi | 
| 71 |  | 
| 72 | # Expand interpreter command, verify with `import sys` test instruction | 
| 73 | PYTHON_COMMAND=$(command -v $PYTHON_COMMAND) | 
| 74 | if [ -z $PYTHON_COMMAND ]; then | 
| 75 | echo "  ${RED}Python interpreter not found${RESTORE}" | 
| 76 | exit 1 | 
| 77 | fi | 
| 78 | if [ -h "$PYTHON_COMMAND" ]; then | 
| 79 | PYTHON_COMMAND=$(readlink -f $PYTHON_COMMAND) | 
| 80 | fi | 
| 81 | echo "  Trying interpreter [ ${LYELLOW}$PYTHON_COMMAND${RESTORE} ]" | 
| 82 | if ! $PYTHON_COMMAND -c "import sys"; then | 
| 83 | echo "  ${RED}Executable is not Python${RESTORE}" | 
| 84 | exit 1 | 
| 85 | fi | 
| 86 |  | 
| 87 | # Verifying Python version | 
| 88 | echo "${LIGHTGRAY}Checking Python version${RESTORE} [ needs ${LIGHTGRAY}>=$PYTHON_MAJOR.$PYTHON_MINOR${RESTORE} ]" | 
| 89 | PYTHON_VERSION_STRING=$($PYTHON_COMMAND -c "print('.'.join([str(a) for a in __import__('sys').version_info[:3]]))") | 
| 90 | if ! $PYTHON_COMMAND -c "import sys;exit(not(sys.version_info.major==$PYTHON_MAJOR and sys.version_info.minor>=$PYTHON_MINOR))"; then | 
| 91 | echo "  ${RED}Python version must be ${RESTORE}[ ${LCYAN}>=$PYTHON_MAJOR.$PYTHON_MINOR${RESTORE} ]${RED}."\ | 
| 92 | "Installed is ${RESTORE}[ ${LCYAN}$PYTHON_VERSION_STRING${RESTORE} ]" | 
| 93 | exit 1 | 
| 94 | fi | 
| 95 | echo "  Version [ ${LCYAN}$PYTHON_VERSION_STRING${RESTORE} ] acceptable" | 
| 96 |  | 
| 97 | # Verifying required modules are installed | 
| 98 | echo "${LIGHTGRAY}Checking Python modules installed${RESTORE}" | 
| 99 | for MODULE in $(seq 1 $(wc -w <<< $REQUIRED_PIP_MODULES)) | 
| 100 | do | 
| 101 | PIP_MODULE=$(awk -v N=$MODULE '{print $N}' <<< "$REQUIRED_PIP_MODULES") | 
| 102 | PYTHON_MODULE=$(awk -v N=$MODULE '{print $N}' <<< "$REQUIRED_PYTHON_MODULES") | 
| 103 | if ! $PYTHON_COMMAND -c "import $PYTHON_MODULE" &> /dev/null; then | 
| 104 | echo "  ${RED}Required Python module ${RESTORE}[ ${LBLUE}$PYTHON_MODULE${RESTORE} ] ${RED}not found${RESTORE}" | 
| 105 | echo "  Install with ${PURPLE}$PYTHON_COMMAND -m pip install $PIP_MODULE${RESTORE}" | 
| 106 | exit 1 | 
| 107 | fi | 
| 108 | echo "  Module [ ${LBLUE}$PYTHON_MODULE${RESTORE} ] found" | 
| 109 | done | 
| 110 | echo "  ${GREEN}All required modules found${RESTORE}" | 
| 111 |  | 
| 112 | function runcmd { | 
| 113 | "$@" 2>&1 | sed 's/^/  /'; | 
| 114 | } | 
| 115 |  | 
| 116 |  | 
| 117 | echo "${LIGHTGRAY}Running HTML templating script${RESTORE}" | 
| 118 | runcmd $SCRIPT_SOURCE_DIR/root/compile.py $HTML_OUT | 
| 119 | if [ $? != 0 ]; then | 
| 120 | echo "  ${RED}Error running templating script${RESTORE}" | 
| 121 | else | 
| 122 | echo "  ${GREEN}Templating successful${RESTORE}" | 
| 123 | fi | 
| 124 |  | 
| 125 | echo "${LIGHTGRAY}Installing nginx configuration${RESTORE}" | 
| 126 |  | 
| 127 | if [ ! -f $SCRIPT_SOURCE_DIR/nginx/nginx.conf ]; then | 
| 128 | echo "  ${RED}Core configuration file ${RESTORE}[ ${LBLUE}$SCRIPT_SOURCE_DIR/nginx/nginx.conf${RESTORE} ] ${RED}not found${RESTORE}" | 
| 129 | exit 1 | 
| 130 | fi | 
| 131 | runcmd cp -v $SCRIPT_SOURCE_DIR/nginx/nginx.conf /etc/nginx/nginx.conf | 
| 132 |  | 
| 133 | runcmd rm -rv /etc/nginx/sites/joshstock.in | 
| 134 | runcmd mkdir -p /etc/nginx/sites/joshstock.in | 
| 135 | if [ ! -d $SCRIPT_SOURCE_DIR/nginx/$MODE ]; then | 
| 136 | echo "  ${RED}Directory ${RESTORE}[ ${LBLUE}$SCRIPT_SOURCE_DIR/nginx/$MODE${RESTORE} ] ${RED}not found${RESTORE}" | 
| 137 | exit 1 | 
| 138 | fi | 
| 139 | runcmd cp -rv $SCRIPT_SOURCE_DIR/nginx/$MODE/* /etc/nginx/sites/joshstock.in | 
| 140 | echo "  ${GREEN}nginx configuration successfully installed${RESTORE}" | 
| 141 |  | 
| 142 | echo "${LIGHTGRAY}Installing resty-gitweb configuration${RESTORE}" | 
| 143 | runcmd cp -v $SCRIPT_SOURCE_DIR/resty-gitweb.yaml /etc/resty-gitweb.yaml | 
| 144 | echo "  ${GREEN}resty-gitweb configuration successfully installed${RESTORE}" | 
| 145 |  | 
| 146 | if [[ ! $(lsof -i TCP:80) =~ "nginx" ]]; then | 
| 147 | echo "${LIGHTGRAY}(Re)starting nginx${RESTORE}" | 
| 148 | systemctl restart nginx | 
| 149 | else | 
| 150 | echo "${LIGHTGRAY}Reloading nginx${RESTORE}" | 
| 151 | systemctl reload nginx | 
| 152 | fi | 
| 153 | if [ $? != 0 ]; then | 
| 154 | echo "  ${RED}nginx configuration not accepted${RESTORE}" | 
| 155 | runcmd systemctl status nginx | 
| 156 | exit 1 | 
| 157 | else | 
| 158 | echo "  ${GREEN}nginx loaded new configuration successfully${RESTORE}" | 
| 159 | echo "${BOLD}${LGREEN}Successful deployment!${RESTORE}" | 
| 160 | exit 0 | 
| 161 | fi | 
| 162 |  |