#!/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

set -e

export LC_ALL=C

source "${CUSTOM_PI_OS_PATH}"/common.sh

echo_green -e "\nBUILD STARTED @ $(date)!\n"

function execute_chroot_script() {

}

# check prerequisites

if [ -n "$BASE_IMAGE_ENLARGEROOT" ] || [ -n "$BASE_IMAGE_RESIZEROOT" ]; then

# resizing the root partition requires 'sfdisk' in our path

which sfdisk >/dev/null 2>/dev/null || \

die "'sfdisk' not found in PATH; did you mean to run the script as root?"

fi

# start!

mkdir -p $BASE_WORKSPACE

mkdir -p $BASE_MOUNT_PATH

# This is already genrated at "build" sourced in "config", but copying here mostly for debug

if [ -f "${EXTRA_BAORD_CONFIG}" ]; then

mv -v "${EXTRA_BAORD_CONFIG}" "${BASE_WORKSPACE}"/extra_board_config

fi

# Clean exported artifacts from other builds

rm -rf "${BASE_WORKSPACE}"/*.tar.gz

install_cleanup_trap

install_fail_on_error_trap

unmount_image $BASE_MOUNT_PATH force || true

pushd "${BASE_WORKSPACE}"

if [ -e *.img ]; then

rm *.img

fi

if [ ! -f "$BASE_ZIP_IMG" ] || [ "$BASE_ZIP_IMG" == "" ]; then

echo "Error: could not find image: $BASE_ZIP_IMG"

echo "On CustomPiOS v2 you can provide -d to download the latest image of your board automatically"

exit 1

fi

if [[ $BASE_ZIP_IMG =~ \.img$ ]]; then

# if the image is already extracted copy over

cp "$BASE_ZIP_IMG" .

else

7za x -aoa "$BASE_ZIP_IMG"

fi

BASE_IMG_PATH=`ls | grep '.img$\|.raw$' | head -n 1`

if [ ! -f "$BASE_IMG_PATH" ]; then

echo "Error, can't find image path, did you place an image in the image folder?"

exit 1

fi

export CUSTOM_PI_OS_BUILDBASE=$(basename "$BASE_IMG_PATH")

if [ -n "$BASE_IMAGE_ENLARGEROOT" ]

then

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

enlarge_ext $BASE_IMG_PATH $BASE_ROOT_PARTITION $BASE_IMAGE_ENLARGEROOT

fi

# mount root and boot partition

mount_image "${BASE_IMG_PATH}" "${BASE_ROOT_PARTITION}" "${BASE_MOUNT_PATH}" "${BASE_BOOT_MOUNT_PATH}" "${BASE_BOOT_PARTITION}"

if [ -n "$BASE_APT_CACHE" ] && [ "$BASE_APT_CACHE" != "no" ]

then

mkdir -p "$BASE_APT_CACHE"

mount --bind "$BASE_APT_CACHE" $BASE_MOUNT_PATH/var/cache/apt

fi

#Edit pi filesystem

pushd $BASE_MOUNT_PATH

#make QEMU boot (remember to return)

if [ "$BASE_IMAGE_RASPBIAN" == "yes" ]; then

fixLd

fi

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

### Execute chroot scripts ###

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

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

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

execute_chroot_script $BASE_PRESCRIPT $BASE_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 $BASE_SCRIPT_PATH $BASE_CHROOT_SCRIPT_PATH

CHROOT_SCRIPT=${BASE_WORKSPACE}/chroot_script

MODULES_AFTER_PATH=${BASE_WORKSPACE}/modules_after

MODULES_BEFORE="${MODULES}"

${CUSTOM_PI_OS_PATH}/custompios_core/execution_order.py "${MODULES}" "${CHROOT_SCRIPT}" "${MODULES_AFTER_PATH}" "${REMOTE_AND_META_CONFIG}"

if [ -f "${REMOTE_AND_META_CONFIG}" ]; then

echo "Sourcing remote and submodules config"

source "${REMOTE_AND_META_CONFIG}" ${@}

MODULES_AFTER=$(cat "${MODULES_AFTER_PATH}")

load_module_config "${MODULES_AFTER}"

else

echo "No remote and submodules config detected"

fi

echo $ARMBIAN_CONFIG_TXT_FILE

# if you need anything from common running in execute_chroot_script, export it here

export -f chroot_correct_qemu

export -f execute_chroot_script

bash -x "${CHROOT_SCRIPT}"

# 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 "$BASE_POSTSCRIPT" ] && [ -f $BASE_POSTSCRIPT/chroot_script ]; then

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

execute_chroot_script $BASE_POSTSCRIPT $BASE_POSTSCRIPT/chroot_script

fi

### End Execute chroot scripts ###

if [ "$BASE_IMAGE_RASPBIAN" == "yes" ]; then

restoreLd

fi

popd

# unmount first boot, then root partition

unmount_image $BASE_MOUNT_PATH

chmod 644 $BASE_IMG_PATH

if [ -n "$BASE_IMAGE_RESIZEROOT" ]

then

# resize image to minimal size + provided size

minimize_ext $BASE_IMG_PATH $BASE_ROOT_PARTITION $BASE_IMAGE_RESIZEROOT

fi

popd

echo_green -e "\nBUILD SUCCEEDED @ $(date)!\n"