[EVA-2020-02-2.git] / install.sh

#!/bin/bash

CONFIG=/boot/config.txt
DATESTAMP=`date "+%Y-%M-%d-%H-%M-%S"`
CONFIG_BACKUP=false

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
	printf "Error parsing configuration...\n"
	exit 1
fi

function do_config_backup {
	if [ ! $CONFIG_BACKUP == true ]; then
		CONFIG_BACKUP=true
		FILENAME="/boot/config.preinstall-$LIBRARY_NAME-$DATESTAMP.txt"
		printf "Backing up $CONFIG to $FILENAME\n"
		cp $CONFIG $FILENAME
	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"
		sudo apt update
		sudo apt install -y $PACKAGES
	fi
}

eval $CONFIG_VARS

printf "$LIBRARY_NAME $LIBRARY_VERSION Python Library: Installer\n\n"

if [ $(id -u) -ne 0 ]; then
	printf "Script must be run as root. Try 'sudo ./install.sh'\n"
	exit 1
fi

cd library

printf "Installing for Python 2..\n"
apt_pkg_install "${PY2_DEPS[@]}"
python setup.py install

if [ -f "/usr/bin/python3" ]; then
	printf "Installing for Python 3..\n"
	apt_pkg_install "${PY3_DEPS[@]}"
	python3 setup.py install
fi

cd ..

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

for ((i = 0; i < ${#CONFIG_TXT[@]}; i++)); do
	CONFIG_LINE="${CONFIG_TXT[$i]}"
	if ! [ "$CONFIG_LINE" == "" ]; then
		do_config_backup
		printf "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 "Done!\n"
Commit	Line	Data
651e1c20 PH	1	#!/bin/bash
651e1c20 PH	2
d7ce2531 PH	3	CONFIG=/boot/config.txt
d7ce2531 PH	4	DATESTAMP=`date "+%Y-%M-%d-%H-%M-%S"`
ab1eb580 PH	5	CONFIG_BACKUP=false
	6
	7	CONFIG_VARS=`python - <<EOF
	8	from configparser import ConfigParser
	9	c = ConfigParser()
	10	c.read('library/setup.cfg')
	11	p = dict(c['pimoroni'])
	12	# Convert multi-line config entries into bash arrays
	13	for k in p.keys():
	14	fmt = '"{}"'
	15	if '\n' in p[k]:
	16	p[k] = "'\n\t'".join(p[k].split('\n')[1:])
	17	fmt = "('{}')"
	18	p[k] = fmt.format(p[k])
	19	print("""
	20	LIBRARY_NAME="{name}"
	21	LIBRARY_VERSION="{version}"
	22	""".format(**c['metadata']))
	23	print("""
	24	PY3_DEPS={py3deps}
	25	PY2_DEPS={py2deps}
	26	SETUP_CMDS={commands}
	27	CONFIG_TXT={configtxt}
	28	""".format(**p))
	29	EOF`
	30
	31	if [ $? -ne 0 ]; then
	32	printf "Error parsing configuration...\n"
	33	exit 1
	34	fi
	35
	36	function do_config_backup {
	37	if [ ! $CONFIG_BACKUP == true ]; then
	38	CONFIG_BACKUP=true
db36c95e	39	FILENAME="/boot/config.preinstall-$LIBRARY_NAME-$DATESTAMP.txt"
ab1eb580 PH	40	printf "Backing up $CONFIG to $FILENAME\n"
	41	cp $CONFIG $FILENAME
	42	fi
	43	}
	44
	45	function apt_pkg_install {
	46	PACKAGES=()
	47	PACKAGES_IN=("$@")
	48	for ((i = 0; i < ${#PACKAGES_IN[@]}; i++)); do
	49	PACKAGE="${PACKAGES_IN[$i]}"
	50	printf "Checking for $PACKAGE\n"
	51	dpkg -L $PACKAGE > /dev/null 2>&1
	52	if [ "$?" == "1" ]; then
	53	PACKAGES+=("$PACKAGE")
	54	fi
	55	done
	56	PACKAGES="${PACKAGES[@]}"
	57	if ! [ "$PACKAGES" == "" ]; then
	58	echo "Installing missing packages: $PACKAGES"
	59	sudo apt update
	60	sudo apt install -y $PACKAGES
	61	fi
	62	}
	63
	64	eval $CONFIG_VARS
1b2cabb8 PH	65
1b2cabb8 PH	66	printf "$LIBRARY_NAME $LIBRARY_VERSION Python Library: Installer\n\n"
651e1c20 PH	67
	68	if [ $(id -u) -ne 0 ]; then
	69	printf "Script must be run as root. Try 'sudo ./install.sh'\n"
	70	exit 1
	71	fi
	72
	73	cd library
	74
	75	printf "Installing for Python 2..\n"
ab1eb580	76	apt_pkg_install "${PY2_DEPS[@]}"
651e1c20 PH	77	python setup.py install
	78
	79	if [ -f "/usr/bin/python3" ]; then
	80	printf "Installing for Python 3..\n"
ab1eb580	81	apt_pkg_install "${PY3_DEPS[@]}"
651e1c20 PH	82	python3 setup.py install
	83	fi
	84
	85	cd ..
	86
ab1eb580 PH	87	for ((i = 0; i < ${#SETUP_CMDS[@]}; i++)); do
	88	CMD="${SETUP_CMDS[$i]}"
	89	# Attempt to catch anything that touches /boot/config.txt and trigger a backup
	90	if [[ "$CMD" == "raspi-config" ]] \|\| [[ "$CMD" == "$CONFIG" ]] \|\| [[ "$CMD" == "\$CONFIG" ]]; then
	91	do_config_backup
	92	fi
	93	eval $CMD
	94	done
	95
	96	for ((i = 0; i < ${#CONFIG_TXT[@]}; i++)); do
	97	CONFIG_LINE="${CONFIG_TXT[$i]}"
	98	if ! [ "$CONFIG_LINE" == "" ]; then
	99	do_config_backup
	100	printf "Adding $CONFIG_LINE to $CONFIG\n"
	101	sed -i "s/^#$CONFIG_LINE/$CONFIG_LINE/" $CONFIG
	102	if ! grep -q "^$CONFIG_LINE" $CONFIG; then
	103	printf "$CONFIG_LINE\n" >> $CONFIG
	104	fi
	105	fi
	106	done
d7ce2531	107
651e1c20	108	printf "Done!\n"