install.sh

#!/bin/bash
VERSION=0.0.0

if [[ $EUID -ne 0 ]]; then
   sudo bash `realpath "$0"`
   exit 
fi

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
NC='\033[0m'
REBOOT=false
SYSD_LOC=/usr/lib/systemd/system
ENV_FILE=/etc/environment
REBOOT=false 

limits_conf() {
  local LINE1='pi hard rtprio 99'
  local LINE2='pi soft rtprio 99'
  local FILE='/etc/security/limits.conf'
  grep -qF -- "$LINE1" $FILE || { echo "$LINE1" >> $FILE; REBOOT=true; }
  grep -qF -- "$LINE2" $FILE || { echo "$LINE2" >> $FILE; REBOOT=true; }
}

echo -e "${GREEN}Installing Vizy version ${VERSION}...${NC}"

if [[ -z "${VIZY_HOME}" ]]; then
    eval `cat /etc/environment` 
    if [[ -z "${VIZY_HOME}" ]]; then
        PREV_INSTALL=false
        REBOOT=true
        DEFAULT_HOME="${HOME}/vizy"
        echo -en "${YELLOW}Where would you like to install Vizy? (Press ENTER to choose ${DEFAULT_HOME}):${NC}"
        read VIZY_HOME
        VIZY_HOME=${VIZY_HOME:-"${DEFAULT_HOME}"}
        # Clean ENV_FILE of any previous lines
        sed -i '/^VIZY_HOME/d' "${ENV_FILE}"
        echo "VIZY_HOME=${VIZY_HOME}" >> "${ENV_FILE}"
        DEST_DIR="${VIZY_HOME}"
    fi
fi
if [[ -n "${VIZY_HOME}" ]]; then
    PREV_INSTALL=true
    DEST_DIR="${VIZY_HOME}.new"
fi

# Update firmware if necessary
echo -e "\n${GREEN}Checking power firmware...${NC}\n"
scripts/update_power_firmware
# Check/install services
if ! scripts/install_services; then
    REBOOT=true
fi
# Change limits.conf file if necessary
limits_conf

# Install any wheels if included 
WHLS="*.whl"
echo "${PWD}"
for f in ${WHLS}; do
    echo -e "\n${GREEN}Installing ${f}...${NC}\n"
    pip3 install --force-reinstall ${f} 
done

# Uninstall vizy
echo -e "\n${GREEN}Uninstalling previous Vizy version...${NC}\n"
pip3 uninstall -y vizy
# Install vizy
echo -e "\n${GREEN}Installing Vizy...${NC}\n"
python3 setup.py install --force
# Copy to final destination
echo -e "\n${GREEN}Copying...${NC}"
mkdir -p "${DEST_DIR}"
if [ -d apps ]; then
    cp -r apps "${DEST_DIR}"
fi
if [ -d examples ]; then
    cp -r examples "${DEST_DIR}"
fi
if [ -d scripts ]; then
    cp -r scripts "${DEST_DIR}"
fi
if [ -d sys ]; then
    cp -r sys "${DEST_DIR}"
fi

if ${PREV_INSTALL}; then
    # Copy settings in etc directory
    cp -r "${VIZY_HOME}/etc" "${DEST_DIR}"
    # Remove previous backup (we only keep one)
    rm -rf "${VIZY_HOME}.bak"
    # Rename direcories
    mv "${VIZY_HOME}" "${VIZY_HOME}.bak"
    mv "${DEST_DIR}" "${VIZY_HOME}"
fi

# Change ownership to pi
chown -R pi "${VIZY_HOME}"

if ${REBOOT}; then
    echo -en "\n${YELLOW}Reboot required.  Would you like to reboot now? (y or n):${NC}" 
    read -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        reboot now
    fi
elif ${PREV_INSTALL}; then
    # Restart vizy software
    # If we're installing via Vizyvisor, this will kill ourselves before 
    # we print a reassuring "success" message (which is important), so we 
    # have an option to skip/defer. 
    if [[ -z "${VIZY_NO_RESTART}" ]]; then
        echo -e "\n${GREEN}Restarting...${NC}"
        service vizy-power-monitor restart
        service vizy-server restart
    fi
fi