#!/usr/bin/env bash

set -x

set -e

# OctoPI generation script

# Helper script that runs in a Raspbian chroot to create the OctoPI distro

# Written by Guy Sheffer <guysoft at gmail dot com>

# GPL V3

source /common.sh

unpack /filesystem/home/pi /home/pi pi

unpack /filesystem/home/root /root root

unpack /filesystem/boot /boot

apt-get update

apt-get remove -y --purge scratch squeak-plugins-scratch squeak-vm wolfram-engine python-minecraftpi minecraft-pi sonic-pi oracle-java8-jdk

#apt-get octoprint virtualenv

apt-get -y --force-yes install python2.7 python-virtualenv python-dev git screen libts-bin subversion cmake checkinstall avahi-daemon libavahi-compat-libdnssd1

pushd /home/pi

#build virtualenv

sudo -u pi virtualenv --system-site-packages oprint

# OctoPrint & pyserial

if [ "$OCTOPI_INCLUDE_OCTOPRINT" == "yes" ]

then

echo "--- Installing OctoPrint"

apt-get install -y --force-yes python-numpy python-netifaces

apt-get remove -y python-serial

#pyserial that can handle non-standard baud rates

sudo -u pi svn co http://pyserial.svn.sourceforge.net/svnroot/pyserial/trunk pyserial

pushd pyserial/pyserial

sudo -u pi /home/pi/oprint/bin/python setup.py install

popd

#pybonjour (for mdns discovery)

sudo -u pi /home/pi/oprint/bin/pip install pybonjour

#OctoPrint

gitclone OCTOPI_OCTOPRINT_REPO OctoPrint

pushd OctoPrint

PIP_DEFAULT_TIMEOUT=60 sudo -u pi /home/pi/oprint/bin/python setup.py install

popd

fi

#OctoPiPanel

if [ "$OCTOPI_INCLUDE_OCTOPIPANEL" == "yes" ]

then

echo "--- Installing OctoPiPanel"

gitclone OCTOPI_OCTOPIPANEL_REPO OctoPiPanel

pushd OctoPiPanel

sudo -u pi /home/pi/oprint/bin/pip install -r requirements.txt

popd

#Add fbcp for TFT screens

gitclone OCTOPI_FBCP_REPO rpi-fbcp

pushd rpi-fbcp

sudo -u pi mkdir build

pushd build

sudo -u pi cmake ..

sudo -u pi make

install fbcp /usr/local/bin/fbcp

sudo -u pi make clean

popd

popd

fi

#mjpg-streamer

if [ "$OCTOPI_INCLUDE_MJPGSTREAMER" == "yes" ]

then

echo "--- Installing mjpg-streamer"

apt-get -y --force-yes install libjpeg8-dev imagemagick libav-tools libv4l-dev

gitclone OCTOPI_MJPGSTREAMER_REPO mjpg-streamer

pushd mjpg-streamer

mv mjpg-streamer-experimental/* .

sudo -u pi make

popd

fi

#CuraEngine

if [ "$OCTOPI_INCLUDE_CURAENGINE" == "yes" ]

then

echo "--- Installing CuraEngine"

folder=CuraEngine-$OCTOPI_CURAENGINE_VERSION

zipfile=$folder.zip

apt-get -y install gcc-4.7 g++-4.7

sudo -u pi wget -O$zipfile $OCTOPI_CURAENGINE_ARCHIVE

sudo -u pi unzip $zipfile

pushd $folder

sudo -u pi wget https://gist.githubusercontent.com/foosel/f86d6a729c0187ff8a82/raw/3d4f152b924080f502d887d11269f27a2389d64d/CuraEngine.patch

sudo -u pi patch < CuraEngine.patch

sudo -u pi make CXX=g++-4.7 VERSION=$OCTOPI_CURAENGINE_VERSION

cp build/CuraEngine /usr/local/bin/cura_engine

popd

sudo -u pi rm -r $folder $zipfile

fi

#setup haproxy for http and https, and webcam

if [ "$OCTOPI_INCLUDE_HAPROXY" == "yes" ]

then

echo "--- Installing haproxy"

export HAPROXY_VERSION=OCTOPI_HAPROXY_VERSION

sudo apt-get -y --force-yes install ssl-cert libssl-dev libpcre3-dev

rm /etc/ssl/private/ssl-cert-snakeoil.key /etc/ssl/certs/ssl-cert-snakeoil.pem

sudo -u pi wget $OCTOPI_HAPROXY_ARCHIVE

sudo -u pi tar xzvf haproxy-$OCTOPI_HAPROXY_VERSION.tar.gz

rm haproxy-$OCTOPI_HAPROXY_VERSION.tar.gz

sudo -u pi mv `ls | grep haproxy | head -n 1` haproxy-ss

pushd haproxy-ss

sudo -u pi make TARGET=linux2628 USE_LINUX_SPLICE=1 USE_LINUX_TPROXY=1 USE_PCRE=1 USE_OPENSSL=1

mkdir -p /usr/local/share/man/man1 /usr/local/doc /usr/local/doc/haproxy

sudo checkinstall --default --pkgname haproxy --pkgversion 1.5

popd

rm -rf haproxy-ss

adduser --system --disabled-password --disabled-login --home /var/lib/haproxy \

--no-create-home --quiet --force-badname --group haproxy

fi

popd

#Make sure user pi has access to serial ports

usermod -a -G tty pi

usermod -a -G dialout pi

# store octopi commit used to build this image

echo "$OCTOPI_COMMIT" > /etc/octopi_commit

# allow pi user to run shutdown and service commands

echo "pi ALL=NOPASSWD: /sbin/shutdown" > /etc/sudoers.d/octoprint-shutdown

echo "pi ALL=NOPASSWD: /sbin/service" > /etc/sudoers.d/octoprint-service

#reach printer by name

echo "$OCTOPI_OVERRIDE_HOSTNAME" > /etc/hostname

sed -i -e "s@raspberrypi@$OCTOPI_OVERRIDE_HOSTNAME@g" /etc/hosts

#make sure users don't run git with sudo, thus breaking permissions, by adding /root/bin to the

#default sudo path and placing a git wrapper script there that checks if it's run as root

sed -i "s@secure_path=\"@secure_path=\"/root/bin:@g" /etc/sudoers

chmod +x /root/bin/git

# enable raspicam

echo "# enable raspicam" >> /boot/config.txt

echo "start_x=1" >> /boot/config.txt

echo "gpu_mem=128" >> /boot/config.txt

# allow network configuration via /boot/octopi-network.txt

sed -i "s@wpa-conf@# wpa-conf@g" /etc/network/interfaces

sed -i "s@iface wlan@# iface wlan@g" /etc/network/interfaces

echo "source /boot/octopi-network.txt" >> /etc/network/interfaces

#unpack root in the end, so etc file are not overwritten, might need to add two roots int he future

unpack /filesystem/root /

#####################################################################

### setup services

### OctoPrint

if [ "$OCTOPI_INCLUDE_OCTOPRINT" == "yes" ]

then

update-rc.d octoprint defaults 95

else

# let's remove the configs for system services we don't need

rm /etc/init.d/octoprint

rm /etc/default/octoprint

fi

### haproxy

if [ "$OCTOPI_INCLUDE_HAPROXY" == "yes" ]

then

update-rc.d gencert defaults

update-rc.d haproxy defaults 99

else

# let's remove the configs for system services we don't need

rm /etc/init.d/gencert

rm /etc/init.d/haproxy

rm /etc/default/haproxy

rm /etc/logrotate.d/haproxy

rm /etc/rsyslog.d/49-haproxy.conf

# also we need to make OctoPrint bind to all interfaces because otherwise

# it will be unaccessible...

[ -f /etc/default/octoprint ] && sed -i "s@HOST=127.0.0.1@HOST=0.0.0.0@" /etc/default/octoprint

fi

### CuraEngine

if [ ! "$OCTOPI_INCLUDE_CURAENGINE" == "yes" ]

then

# unconfigure the cura engine path in octoprint's config.yaml

sudo -u pi sed -i -e "s@cura_engine: /usr/local/bin/cura_engine@cura_engine:@g" /home/pi/.octoprint/config.yaml

fi

### mjpg_streamer

if [ "$OCTOPI_INCLUDE_MJPGSTREAMER" == "yes" ]

then

# make webcamd autostart

update-rc.d webcamd defaults

else

rm /etc/logrotate.d/webcamd

rm /etc/init.d/webcamd

rm /etc/default/webcamd

rm /root/bin/webcamd

fi

### OctoPiPanel

if [ ! "$OCTOPI_INCLUDE_OCTOPIPANEL" == "yes" ]

then

# we don't need the tft setup stuff

rm /home/pi/scripts/OctoPiPanel

rm /home/pi/scripts/calibrate-rpi-display

rm /home/pi/scripts/enable-adafruit-pitft

rm /home/pi/scripts/enable-rpi-display

rm /boot/cmdline-pi-tft.txt

rm /etc/udev/rules.d/95-ads7846.rules

fi

#cleanup

apt-get clean

apt-get autoremove -y