#!/usr/bin/env bash

# OctoPi generation script

# This script takes a Raspbian image and adds to it octoprint and verions addons

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

# GPL V3

source $OCTOPI_SCRIPT_PATH/common.sh

function execute_chroot_script() {

}

mkdir -p $OCTOPI_WORKSPACE

mkdir -p $OCTOPI_MOUNT_PATH

install_cleanup_trap

install_fail_on_error_trap $OCTOPI_MOUNT_PATH

unmount_image $OCTOPI_MOUNT_PATH force || true

pushd $OCTOPI_WORKSPACE

if [ -e *.img ]; then

rm *.img

fi

unzip $OCTOPI_ZIP_IMG

OCTOPI_IMG_PATH=`ls | grep .img`

export OCTOPI_BUILDBASE=$(basename $OCTOPI_IMG_PATH)

if [ -n "$OCTOPI_IMAGE_ENLARGEROOT" ]

then

# make our image a bit larger so we don't run into size problems...

enlarge_ext $OCTOPI_IMG_PATH 2 $OCTOPI_IMAGE_ENLARGEROOT

fi

# mount root and boot partition

mount_image $OCTOPI_IMG_PATH $OCTOPI_MOUNT_PATH

if [ -n "$OCTOPI_APT_CACHE" ]

then

mkdir -p "$OCTOPI_APT_CACHE"

mount --bind "$OCTOPI_APT_CACHE" $OCTOPI_MOUNT_PATH/var/cache/apt

fi

#Edit pi filesystem

pushd $OCTOPI_MOUNT_PATH

#make QEMU boot (remember to return)

fixLd

#sed -i 's@include /etc/ld.so.conf.d/\*.conf@\#include /etc/ld.so.conf.d/\*.conf@' etc/ld.so.conf

# if an additional pre-script is defined, execute that now

if [ -n "$OCTOPI_PRESCRIPT" ] && [ -f $OCTOPI_PRESCRIPT/chroot_script ]; then

echo "Injecting environment pre script from $OCTOPI_PRESCRIPT..."

execute_chroot_script $OCTOPI_PRESCRIPT $OCTOPI_PRESCRIPT/chroot_script

fi

# if building a variant, execute its pre-chroot script

if [ -n "$VARIANT_BASE" ] && [ -f $VARIANT_BASE/pre_chroot_script ]; then

echo "Injecting variant pre script from $VARIANT_BASE..."

execute_chroot_script $VARIANT_BASE $VARIANT_BASE/pre_chroot_script

fi

# execute the base chroot script

execute_chroot_script $OCTOPI_SCRIPT_PATH $OCTOPI_CHROOT_SCRIPT_PATH

# if building a variant, execute its post-chroot script

if [ -n "$VARIANT_BASE" ] && [ -f $VARIANT_BASE/post_chroot_script ]; then

echo "Injecting variant post script from $VARIANT_BASE..."

execute_chroot_script $VARIANT_BASE $VARIANT_BASE/post_chroot_script

fi

# if an additional post-script is defined, execute that now

if [ -n "$OCTOPI_POSTSCRIPT" ] && [ -f $OCTOPI_POSTSCRIPT/chroot_script ]; then

echo "Injecting environment post script from $OCTOPI_POSTSCRIPT..."

execute_chroot_script $OCTOPI_POSTSCRIPT $OCTOPI_POSTSCRIPT/chroot_script

fi

restoreLd

popd

# unmount first boot, then root partition

unmount_image $OCTOPI_MOUNT_PATH

chmod 777 $OCTOPI_IMG_PATH

if [ -n "$OCTOPI_IMAGE_RESIZEROOT" ]

then

# resize image to minimal size + provided size

minimize_ext $OCTOPI_IMG_PATH 2 $OCTOPI_IMAGE_RESIZEROOT

fi

popd