#!/usr/bin/env bash YW=$(echo "\033[33m") RD=$(echo "\033[01;31m") BL=$(echo "\033[36m") GN=$(echo "\033[1;92m") CL=$(echo "\033[m") RETRY_NUM=10 RETRY_EVERY=3 NUM=$RETRY_NUM CM="${GN}✓${CL}" CROSS="${RD}✗${CL}" BFR="\\r\\033[K" HOLD="-" set -o errexit set -o errtrace set -o nounset set -o pipefail shopt -s expand_aliases alias die='EXIT=$? LINE=$LINENO error_exit' trap die ERR function error_exit() { trap - ERR local reason="Unknown failure occurred." local msg="${1:-$reason}" local flag="${RD}‼ ERROR ${CL}$EXIT@$LINE" echo -e "$flag $msg" 1>&2 exit $EXIT } function msg_info() { local msg="$1" echo -ne " ${HOLD} ${YW}${msg}..." } function msg_ok() { local msg="$1" echo -e "${BFR} ${CM} ${GN}${msg}${CL}" } function msg_error() { local msg="$1" echo -e "${BFR} ${CROSS} ${RD}${msg}${CL}" } msg_info "Setting up Container OS " sed -i "/$LANG/ s/\(^# \)//" /etc/locale.gen locale-gen >/dev/null while [ "$(hostname -I)" = "" ]; do echo 1>&2 -en "${CROSS}${RD} No Network! " sleep $RETRY_EVERY ((NUM--)) if [ $NUM -eq 0 ]; then echo 1>&2 -e "${CROSS}${RD} No Network After $RETRY_NUM Tries${CL}" exit 1 fi done msg_ok "Set up Container OS" msg_ok "Network Connected: ${BL}$(hostname -I)" set +e alias die='' if nc -zw1 8.8.8.8 443; then msg_ok "Internet Connected"; else msg_error "Internet NOT Connected" read -r -p "Would you like to continue anyway? " prompt if [[ $prompt == "y" || $prompt == "Y" || $prompt == "yes" || $prompt == "Yes" ]]; then echo -e " ⚠️ ${RD}Expect Issues Without Internet${CL}" else echo -e " 🖧 Check Network Settings" exit 1 fi fi RESOLVEDIP=$(nslookup "github.com" | awk -F':' '/^Address: / { matched = 1 } matched { print $2}' | xargs) if [[ -z "$RESOLVEDIP" ]]; then msg_error "DNS Lookup Failure"; else msg_ok "DNS Resolved github.com to $RESOLVEDIP"; fi alias die='EXIT=$? LINE=$LINENO error_exit' set -e msg_info "Updating Container OS" apt-get update &>/dev/null apt-get -y upgrade &>/dev/null msg_ok "Updated Container OS" msg_info "Installing Dependencies" apt-get install -y curl &>/dev/null apt-get install -y sudo &>/dev/null msg_ok "Installed Dependencies" msg_info "Installing pip3" apt-get install -y python3-pip &>/dev/null msg_ok "Installed pip3" get_latest_release() { curl -sL https://api.github.com/repos/$1/releases/latest | grep '"tag_name":' | cut -d'"' -f4 } DOCKER_LATEST_VERSION=$(get_latest_release "moby/moby") CORE_LATEST_VERSION=$(get_latest_release "home-assistant/core") PORTAINER_LATEST_VERSION=$(get_latest_release "portainer/portainer") msg_info "Installing Docker $DOCKER_LATEST_VERSION" DOCKER_CONFIG_PATH='/etc/docker/daemon.json' mkdir -p $(dirname $DOCKER_CONFIG_PATH) cat >$DOCKER_CONFIG_PATH <<'EOF' { "log-driver": "journald" } EOF sh <(curl -sSL https://get.docker.com) &>/dev/null msg_ok "Installed Docker $DOCKER_LATEST_VERSION" msg_info "Pulling Portainer $PORTAINER_LATEST_VERSION Image" docker pull portainer/portainer-ce:latest &>/dev/null msg_ok "Pulled Portainer $PORTAINER_LATEST_VERSION Image" msg_info "Installing Portainer $PORTAINER_LATEST_VERSION" docker volume create portainer_data >/dev/null docker run -d \ -p 8000:8000 \ -p 9000:9000 \ --name=portainer \ --restart=always \ -v /var/run/docker.sock:/var/run/docker.sock \ -v portainer_data:/data \ portainer/portainer-ce:latest &>/dev/null msg_ok "Installed Portainer $PORTAINER_LATEST_VERSION" msg_info "Pulling Home Assistant $CORE_LATEST_VERSION Image" docker pull homeassistant/home-assistant:stable &>/dev/null msg_ok "Pulled Home Assistant $CORE_LATEST_VERSION Image" msg_info "Installing Home Assistant $CORE_LATEST_VERSION" docker volume create hass_config >/dev/null docker run -d \ --name homeassistant \ --privileged \ --restart unless-stopped \ -v /var/run/docker.sock:/var/run/docker.sock \ -v /dev:/dev \ -v hass_config:/config \ -v /etc/localtime:/etc/localtime:ro \ --net=host \ homeassistant/home-assistant:stable &>/dev/null msg_ok "Installed Home Assistant $CORE_LATEST_VERSION" msg_info "Creating Update Menu Script" pip3 install runlike &>/dev/null UPDATE_PATH='/root/update' UPDATE_CONTAINERS_PATH='/root/update-containers.sh' cat >$UPDATE_PATH <<'EOF' #!/bin/sh set -o errexit show_menu(){ normal=`echo "\033[m"` safe=`echo "\033[32m"` menu=`echo "\033[36m"` number=`echo "\033[33m"` bgred=`echo "\033[41m"` fgred=`echo "\033[31m"` hostname -I printf "\n${menu}*********************************************${normal}\n" printf "${menu}**${number} 1)${safe} Switch to Stable Branch ${normal}\n" printf "${menu}**${number} 2)${number} Switch to Beta Branch ${normal}\n" printf "${menu}**${number} 3)${fgred} Switch to Dev Branch ${normal}\n" printf "${menu}**${number} 4)${safe} Backup Home Assistant Data (to root) ${normal}\n" printf "${menu}**${number} 5)${number} Restore Home Assistant Data ${normal}\n" printf "${menu}**${number} 6)${fgred} Edit Home Assistant Configuration ${normal}\n" printf "${menu}**${number} 7)${safe} Restart Home Assistant ${normal}\n" printf "${menu}**${number} 8)${safe} Just Update Containers ${normal}\n" printf "${menu}**${number} 9)${number} Remove Unused Images ${normal}\n" printf "${menu}**${number} 10)${safe} Update Host OS ${normal}\n" printf "${menu}**${number} 11)${safe} Reboot Host OS ${normal}\n" printf "${menu}*********************************************${normal}\n" printf "Please choose an option from the menu and enter or ${fgred}x to exit. ${normal}" read opt } option_picked(){ msgcolor=`echo "\033[01;31m"` normal=`echo "\033[00;00m"` message=${@:-"${normal}Error: No message passed"} printf "${msgcolor}${message}${normal}\n" } clear show_menu while [ $opt != '' ] do if [ $opt = '' ]; then exit; else case $opt in 1) clear; option_picked "Switching to Stable Branch"; TAG=stable break; ;; 2) clear; option_picked "Switching to Beta Branch"; TAG=beta break; ;; 3) while true; do read -p "Are you sure you want to Switch to Dev Branch? Proceed(y/n)?" yn case $yn in [Yy]* ) break;; [Nn]* ) exit;; * ) echo "Please answer yes or no.";; esac done clear; option_picked "Switching to Dev Branch"; TAG=dev break; ;; 4) clear; option_picked "Backing up Home Assistant Data to root (hass_config)"; rm -r hass_config; cp -pR /var/lib/docker/volumes/hass_config/ /root/; sleep 2; clear; show_menu; ;; 5) while true; do read -p "Are you sure you want to Restore Home Assistant Data? Proceed(y/n)?" yn case $yn in [Yy]* ) break;; [Nn]* ) exit;; * ) echo "Please answer yes or no.";; esac done clear; option_picked "Restoring Home Assistant Data from root (hass_config)"; rm -r /var/lib/docker/volumes/hass_config/_data; cp -pR /root/hass_config/_data /var/lib/docker/volumes/hass_config/; sleep 2; clear; show_menu; ;; 6) while true; do read -p "Are you sure you want to Edit Home Assistant Configuration? Proceed(y/n)?" yn case $yn in [Yy]* ) break;; [Nn]* ) exit;; * ) echo "Please answer yes or no.";; esac done clear; option_picked "Editing Home Assistant Configuration"; nano /var/lib/docker/volumes/hass_config/_data/configuration.yaml; clear; show_menu; ;; 7) clear; option_picked "Restarting Home Assistant"; docker restart homeassistant; exit; ;; 8) clear; option_picked "Just Updating Containers"; ./update-containers.sh; sleep 2; clear; show_menu; ;; 9) clear; option_picked "Removing Unused Images"; docker image prune -af; sleep 2; clear; show_menu; ;; 10) clear; option_picked "Updating Host OS"; apt update && apt upgrade -y; sleep 2; clear; show_menu; ;; 11) clear; option_picked "Reboot Host OS"; reboot; exit; ;; x)exit; ;; \n)exit; ;; *)clear; option_picked "Please choose an option from the menu"; show_menu; ;; esac fi done docker pull homeassistant/home-assistant:$TAG docker rm --force homeassistant docker run -d \ --name homeassistant \ --privileged \ --restart unless-stopped \ -v /var/run/docker.sock:/var/run/docker.sock \ -v /dev:/dev \ -v hass_config:/config \ -v /etc/localtime:/etc/localtime:ro \ -v /etc/timezone:/etc/timezone:ro \ --net=host \ homeassistant/home-assistant:$TAG EOF sudo chmod +x /root/update cat >$UPDATE_CONTAINERS_PATH <<'EOF' #!/bin/bash set -o errexit CONTAINER_LIST="${1:-$(docker ps -q)}" for container in ${CONTAINER_LIST}; do CONTAINER_IMAGE="$(docker inspect --format "{{.Config.Image}}" --type container ${container})" RUNNING_IMAGE="$(docker inspect --format "{{.Image}}" --type container "${container}")" docker pull "${CONTAINER_IMAGE}" LATEST_IMAGE="$(docker inspect --format "{{.Id}}" --type image "${CONTAINER_IMAGE}")" if [[ "${RUNNING_IMAGE}" != "${LATEST_IMAGE}" ]]; then echo "Updating ${container} image ${CONTAINER_IMAGE}" DOCKER_COMMAND="$(runlike "${container}")" docker rm --force "${container}" eval ${DOCKER_COMMAND} fi done EOF sudo chmod +x /root/update-containers.sh msg_ok "Created Update Menu Script" mkdir /root/hass_config PASS=$(grep -w "root" /etc/shadow | cut -b6) if [[ $PASS != $ ]]; then msg_info "Customizing Container" rm /etc/motd rm /etc/update-motd.d/10-uname touch ~/.hushlogin GETTY_OVERRIDE="/etc/systemd/system/container-getty@1.service.d/override.conf" mkdir -p $(dirname $GETTY_OVERRIDE) cat <$GETTY_OVERRIDE [Service] ExecStart= ExecStart=-/sbin/agetty --autologin root --noclear --keep-baud tty%I 115200,38400,9600 \$TERM EOF systemctl daemon-reload systemctl restart $(basename $(dirname $GETTY_OVERRIDE) | sed 's/\.d//') msg_ok "Customized Container" fi if [[ "${SSH_ROOT}" == "yes" ]]; then sed -i "s/#PermitRootLogin prohibit-password/PermitRootLogin yes/g" /etc/ssh/sshd_config systemctl restart sshd fi msg_info "Cleaning up" apt-get autoremove >/dev/null apt-get autoclean >/dev/null msg_ok "Cleaned"