#!/bin/bash
-LIBRARY_VERSION=`cat library/setup.py | grep version | awk -F"'" '{print $2}'`
-LIBRARY_NAME=`cat library/setup.py | grep name | awk -F"'" '{print $2}'`
CONFIG=/boot/config.txt
DATESTAMP=`date "+%Y-%M-%d-%H-%M-%S"`
+CONFIG_BACKUP=false
+APT_HAS_UPDATED=false
+USER_HOME=/home/$SUDO_USER
+RESOURCES_TOP_DIR=$USER_HOME/Pimoroni
+WD=`pwd`
-printf "$LIBRARY_NAME $LIBRARY_VERSION Python Library: Installer\n\n"
+user_check() {
+ if [ $(id -u) -ne 0 ]; then
+ printf "Script must be run as root. Try 'sudo ./install.sh'\n"
+ exit 1
+ fi
+}
+
+confirm() {
+ if [ "$FORCE" == '-y' ]; then
+ true
+ else
+ read -r -p "$1 [y/N] " response < /dev/tty
+ if [[ $response =~ ^(yes|y|Y)$ ]]; then
+ true
+ else
+ false
+ fi
+ fi
+}
+
+prompt() {
+ read -r -p "$1 [y/N] " response < /dev/tty
+ if [[ $response =~ ^(yes|y|Y)$ ]]; then
+ true
+ else
+ false
+ fi
+}
+
+success() {
+ echo -e "$(tput setaf 2)$1$(tput sgr0)"
+}
+
+inform() {
+ echo -e "$(tput setaf 6)$1$(tput sgr0)"
+}
+
+warning() {
+ echo -e "$(tput setaf 1)$1$(tput sgr0)"
+}
+
+function do_config_backup {
+ if [ ! $CONFIG_BACKUP == true ]; then
+ CONFIG_BACKUP=true
+ FILENAME="config.preinstall-$LIBRARY_NAME-$DATESTAMP.txt"
+ inform "Backing up $CONFIG to /boot/$FILENAME\n"
+ cp $CONFIG /boot/$FILENAME
+ mkdir -p $RESOURCES_TOP_DIR/config-backups/
+ cp $CONFIG $RESOURCES_TOP_DIR/config-backups/$FILENAME
+ if [ -f "$UNINSTALLER" ]; then
+ echo "cp $RESOURCES_TOP_DIR/config-backups/$FILENAME $CONFIG" >> $UNINSTALLER
+ fi
+ fi
+}
+
+function apt_pkg_install {
+ PACKAGES=()
+ PACKAGES_IN=("$@")
+ for ((i = 0; i < ${#PACKAGES_IN[@]}; i++)); do
+ PACKAGE="${PACKAGES_IN[$i]}"
+ printf "Checking for $PACKAGE\n"
+ dpkg -L $PACKAGE > /dev/null 2>&1
+ if [ "$?" == "1" ]; then
+ PACKAGES+=("$PACKAGE")
+ fi
+ done
+ PACKAGES="${PACKAGES[@]}"
+ if ! [ "$PACKAGES" == "" ]; then
+ echo "Installing missing packages: $PACKAGES"
+ if [ ! $APT_HAS_UPDATED ]; then
+ apt update
+ APT_HAS_UPDATED=true
+ fi
+ apt install -y $PACKAGES
+ if [ -f "$UNINSTALLER" ]; then
+ echo "apt uninstall -y $PACKAGES"
+ fi
+ fi
+}
+
+user_check
-if [ $(id -u) -ne 0 ]; then
- printf "Script must be run as root. Try 'sudo ./install.sh'\n"
+apt_pkg_install python-configparser
+
+CONFIG_VARS=`python - <<EOF
+from configparser import ConfigParser
+c = ConfigParser()
+c.read('library/setup.cfg')
+p = dict(c['pimoroni'])
+# Convert multi-line config entries into bash arrays
+for k in p.keys():
+ fmt = '"{}"'
+ if '\n' in p[k]:
+ p[k] = "'\n\t'".join(p[k].split('\n')[1:])
+ fmt = "('{}')"
+ p[k] = fmt.format(p[k])
+print("""
+LIBRARY_NAME="{name}"
+LIBRARY_VERSION="{version}"
+""".format(**c['metadata']))
+print("""
+PY3_DEPS={py3deps}
+PY2_DEPS={py2deps}
+SETUP_CMDS={commands}
+CONFIG_TXT={configtxt}
+""".format(**p))
+EOF`
+
+if [ $? -ne 0 ]; then
+ warning "Error parsing configuration...\n"
exit 1
fi
+eval $CONFIG_VARS
+
+RESOURCES_DIR=$RESOURCES_TOP_DIR/$LIBRARY_NAME
+UNINSTALLER=$RESOURCES_DIR/uninstall.sh
+
+mkdir -p $RESOURCES_DIR
+
+cat << EOF > $UNINSTALLER
+printf "It's recommended you run these steps manually.\n"
+printf "If you want to run the full script, open it in\n"
+printf "an editor and remove 'exit 1' from below.\n"
+exit 1
+EOF
+
+printf "$LIBRARY_NAME $LIBRARY_VERSION Python Library: Installer\n\n"
+
cd library
printf "Installing for Python 2..\n"
-python setup.py install
+apt_pkg_install "${PY2_DEPS[@]}"
+python setup.py install > /dev/null
+if [ $? -eq 0 ]; then
+ success "Done!\n"
+ echo "pip uninstall $LIBRARY_NAME" >> $UNINSTALLER
+fi
if [ -f "/usr/bin/python3" ]; then
printf "Installing for Python 3..\n"
- python3 setup.py install
+ apt_pkg_install "${PY3_DEPS[@]}"
+ python3 setup.py install > /dev/null
+ if [ $? -eq 0 ]; then
+ success "Done!\n"
+ echo "pip3 uninstall $LIBRARY_NAME" >> $UNINSTALLER
+ fi
fi
-cd ..
+cd $WD
-printf "Backing up $CONFIG\n"
-cp $CONFIG "config.preinstall-$DATESTAMP.txt"
+for ((i = 0; i < ${#SETUP_CMDS[@]}; i++)); do
+ CMD="${SETUP_CMDS[$i]}"
+ # Attempt to catch anything that touches /boot/config.txt and trigger a backup
+ if [[ "$CMD" == *"raspi-config"* ]] || [[ "$CMD" == *"$CONFIG"* ]] || [[ "$CMD" == *"\$CONFIG"* ]]; then
+ do_config_backup
+ fi
+ eval $CMD
+done
-printf "Setting up i2c and SPI..\n"
-raspi-config nonint do_spi 0
-raspi-config nonint do_i2c 0
+for ((i = 0; i < ${#CONFIG_TXT[@]}; i++)); do
+ CONFIG_LINE="${CONFIG_TXT[$i]}"
+ if ! [ "$CONFIG_LINE" == "" ]; then
+ do_config_backup
+ inform "Adding $CONFIG_LINE to $CONFIG\n"
+ sed -i "s/^#$CONFIG_LINE/$CONFIG_LINE/" $CONFIG
+ if ! grep -q "^$CONFIG_LINE" $CONFIG; then
+ printf "$CONFIG_LINE\n" >> $CONFIG
+ fi
+ fi
+done
-printf "Setting up serial for PMS5003..\n"
-# Disable serial terminal over /dev/ttyAMA0
-raspi-config nonint do_serial 1
-# Enable serial port
-raspi-config nonint set_config_var enable_uart 1 $CONFIG
-# Switch serial port to full UART for stability (may adversely affect bluetooth)
-sed -i 's/^#dtoverlay=pi3-miniuart-bt/dtoverlay=pi3-miniuart-bt/' $CONFIG
-if ! grep -q -E "^dtoverlay=pi3-miniuart-bt" $CONFIG; then
- printf "dtoverlay=pi3-miniuart-bt\n" >> $CONFIG
+if [ -d "examples" ]; then
+ if confirm "Would you like to copy examples to $RESOURCES_DIR?"; then
+ inform "Copying examples to $RESOURCES_DIR"
+ cp -r examples/ $RESOURCES_DIR
+ echo "rm -r $RESOURCES_DIR" >> $UNINSTALLER
+ fi
fi
-printf "Done!\n"
+success "\nAll done!"
+inform "If this is your first time installing you should reboot for hardware changes to take effect.\n"
+inform "Find uninstall steps in $UNINSTALLER\n"
projects / EVA-2020-02-2.git / blobdiff