#!/bin/bash

stime=$(date +'%Y%m%d-%H%M%S')
exec 3>&1 4>&2
exec > >(tee -a "/tmp/linux_build_${stime}.log") 2>&1

#set -e
echo  "build ARCH=[${ARCH}]"
echo "AP100_DBG_ON=[$AP100_DBG_ON]"
#Use library
export ALEXPRO100_LIB_LOCATION="${ALEXPRO100_LIB_LOCATION:-./lib/alexpro100_lib.sh}"
# shellcheck disable=SC1091
source ./lib/common/lib_connect.sh

msg_print note "AP100_DBG_ON=[$AP100_DBG_ON]"

# shellcheck disable=SC1091
if [[ -f ./public_parameters ]]; then
  source ./public_parameters
else
  return_err "Public parameters not found!"
fi
# shellcheck disable=SC1091
[[ -f ./private_parameters ]] && source ./private_parameters

if [[ -n "$CUSTOM_PROFILE" ]]; then
	export cust_param="`mktemp -u`_custom_parameters"
	msg_print note "${cust_param}" "$CUSTOM_PROFILE/custom_parameters"
	get_file_s "${cust_param}" "$CUSTOM_PROFILE/custom_parameters" &
	show_progress train $! "Downloading custom_parameters..."
	[[ -f "${cust_param}" ]] && source "${cust_param}"
	msg_print note "mirror_alpine=[${mirror_alpine}]"
fi

LI_TYPE="${LI_TYPE:-private}" LI_DEBUG="${LI_DEBUG:-0}"
LI_VERSION="$(cat ./version_install)"
if [[ "$LI_DEBUG" == "1" ]]; then
    LI_VERSION="${LI_VERSION}-dbg"
    msg_print warn "Building DEBUG version..."
fi
msg_print note "AP100_DBG_ON=[$AP100_DBG_ON]"
msg_print note "build ARCH=[${ARCH}]"
ARCH="${ARCH:-x86_64}"
msg_print note "build ARCH=[${ARCH}]"
ALPINE_VERSION="${ALPINE_VERSION:-3.16}"
#export ALPINE_VERSION="3.16" # used in install_sys.sh
#export ALPINE_VERSION="3.14" # used in install_sys.sh
##export ALPINE_VERSION="3.7" # used in install_sys.sh
export ALPINE_VERSION
ALPINE_REVISION="${ALPINE_REVISION:-0}"
ALPINE_NETBOOT_VERSION="lts"
BUILDS_DIR="${BUILDS_DIR:-releases}"
ALPINE_ISO="${ALPINE_ISO:-$mirror_alpine/v$ALPINE_VERSION/releases/$ARCH/alpine-standard-$ALPINE_VERSION.$ALPINE_REVISION-$ARCH.iso}"
ALPINE_NETBOOT="${ALPINE_NETBOOT:-$mirror_alpine/v$ALPINE_VERSION/releases/$ARCH/netboot-$ALPINE_VERSION.$ALPINE_REVISION}"
LI_ISO="$BUILDS_DIR/linux_install-$ARCH-$LI_VERSION-$ALPINE_VERSION-$LI_TYPE.iso"
LI_NETBOOT="$BUILDS_DIR/linux_install-$ARCH-$LI_VERSION-$ALPINE_VERSION-$LI_TYPE.pxe.tgz"
LI_BUILD_ISO="${LI_BUILD_ISO:-1}" LI_BUILD_NETBOOT="${LI_BUILD_NETBOOT:-1}"
msg_print note "build arch=[${arch}]"
function make_bootable_iso() (
    local iso_location
    iso_location="$(realpath "$2")"
    cd -- "$1" || return_err "No directory $1!"
    xorriso -as mkisofs -o "$iso_location"  -b boot/syslinux/isolinux.bin -c boot/syslinux/boot.cat -no-emul-boot -boot-load-size 4 -boot-info-table \
      -eltorito-alt-boot -eltorito-platform 0xEF -eltorito-boot boot/grub/efi.img -no-emul-boot -lJR .; 
)

function prepare_initfs() {
    arccat gz "$1" | unpack_cpio "${make_build:?}/initfs"
    mv "$make_build/initfs/init" "$make_build/initfs/init_orig"
    cp "$build_files/init" "$make_build/initfs/init"
    pack_initfs_cpio "$make_build/initfs" | zstd -T"$(nproc)" -19 > "$2"
}
msg_print note "make_build=[${make_build}]"
[[ -d "$BUILDS_DIR" ]] || mkdir -p "$BUILDS_DIR"
build_files="./bin/make_images/linux_install_files"
create_tmp_dir make_build

#Prepare rootfs
msg_print note "make_build=[${make_build}]"
mkdir "$make_build/rootfs" "$make_build/custom" 
cp -r "$build_files/custom/." "$make_build/custom"
mount -t tmpfs tmpfs "$make_build/rootfs"
cp "$build_files/linux_install_$ARCH.sh" "$make_build/config.sh"
msg_print note "build arch=[${arch}]"
msg_print note "cust_param=[${cust_param}]"
msg_print note "AP100_DBG_ON=[$AP100_DBG_ON]"
CUSTOM_DIR="$make_build/custom" default_dir="$make_build/rootfs" "./install_sys.sh" "$make_build/config.sh"

[[ "$LI_DEBUG" == "1" ]] && sed -i '6s/=0/=1/' "$make_build/rootfs/root/installer_debian.sh"
cp -Rf . "$make_build/rootfs/root/linux_install"
if [[ -n "$old_gpgkey" ]]; then
	msg_print note "$make_build/rootfs/root/debian-archive-removed-keys.gpg" "$old_gpgkey"
	get_file_s "$make_build/rootfs/root/debian-archive-removed-keys.gpg" "$old_gpgkey" &
	show_progress train $! "Downloading debian-archive-removed-keys.gpg..."
fi

rm -rf "$make_build/rootfs/root/linux_install/.git" \
    "$make_build/rootfs/root/linux_install/.github" \
    "$make_build/rootfs/root/linux_install/_config.yml" \
    "$make_build/rootfs/root/linux_install/releases" \
    "$make_build/rootfs/root/linux_install/bin/make_images"
if [[ "$LI_TYPE" == "public" ]]; then
    rm -rf "$make_build/rootfs/root/linux_install/private_parameters" \
    "$make_build/rootfs/root/linux_install/custom/custom_script.sh" \
    "$make_build/rootfs/root/linux_install/custom/rootfs"
fi
echo "v$ALPINE_VERSION" > "$make_build/rootfs/root/linux_install/alpine_version"

if [[ "$ALPINE_VERSION" < "3.16" ]]; then
  msg_print note "squashfs_rootfs_pack_old  $make_build/rootfs $make_build/rootfs.img xz"
  squashfs_rootfs_pack_old "$make_build/rootfs" "$make_build/rootfs.img" xz
else
   msg_print note "squashfs_rootfs_pack  $make_build/rootfs $make_build/rootfs.img -comp zstd -Xcompression-level 22"
  squashfs_rootfs_pack "$make_build/rootfs" "$make_build/rootfs.img" -comp zstd -Xcompression-level 22
fi

umount "$make_build/rootfs"
msg_print note "LI_BUILD_ISO=[${LI_BUILD_ISO}]"
if [[ $LI_BUILD_ISO == "1" ]]; then
    msg_print note "Building ISO..."
    [[ -e "$LI_ISO" ]] && rm -rf "$LI_ISO"
    mkdir "$make_build/orig_iso" "$make_build/final_iso" "$make_build/initfs"
    msg_print note "$make_build/alpine.iso" "$ALPINE_ISO"
    get_file_s "$make_build/alpine.iso" "$ALPINE_ISO" &
    show_progress train $! "Downloading alpine.iso..."
    7z x -o"$make_build/final_iso" "$make_build/alpine.iso"
    rm -rf "$make_build/final_iso/[BOOT]"
    msg_print note "Preparing initfs..."
    msg_print note "$(find "$make_build/final_iso/boot" -name "*initramfs-*" | head -n 1)" "$(find "$make_build/final_iso/boot" -name "*initramfs-*" | head -n 1)"
    prepare_initfs "$(find "$make_build/final_iso/boot" -name "*initramfs-*" | head -n 1)" "$(find "$make_build/final_iso/boot" -name "*initramfs-*" | head -n 1)"
    #rm -f $make_build/final_iso/apks/rootfs.img
    msg_print note "cp $make_build/rootfs.img $make_build/final_iso/apks/rootfs.img"
    cp "$make_build/rootfs.img" "$make_build/final_iso/apks/rootfs.img"
	if [[ -n "$CUSTOM_PROFILE" ]]; then
	    sed -ie "s|quiet|quiet rootfs_net=rootfs.img CUSTOM_PROFILE=${CUSTOM_PROFILE}/customize.sh|" "$make_build/final_iso/boot/syslinux/syslinux.cfg"
	    sed -ie "s|quiet|quiet rootfs_net=rootfs.img CUSTOM_PROFILE=${CUSTOM_PROFILE}/customize.sh|" "$make_build/final_iso/boot/grub/grub.cfg"
	else
	    sed -ie 's|quiet|quiet rootfs_net=rootfs.img CUSTOM_PROFILE=http://xxx.xxx.xxx.xxx/customize.sh|' "$make_build/final_iso/boot/syslinux/syslinux.cfg"
	    sed -ie 's|quiet|quiet rootfs_net=rootfs.img CUSTOM_PROFILE=http://xxx.xxx.xxx.xxx/customize.sh|' "$make_build/final_iso/boot/grub/grub.cfg"
	fi
   msg_print note "Making ISO..."
    msg_print note "$make_build/final_iso  $LI_ISO"
    make_bootable_iso "$make_build/final_iso" "$LI_ISO"
    rm -rf "$make_build/alpine.iso" "$make_build/final_iso" "$make_build/initfs"
fi

msg_print note "LI_BUILD_NETBOOT=[${LI_BUILD_NETBOOT}]"
if [[ $LI_BUILD_NETBOOT == "1" ]]; then
    msg_print note "Building NETBOOT..."
    mkdir "$make_build/final_netboot" "$make_build/initfs"
    msg_print note "$make_build/final_netboot/vmlinuz  $ALPINE_NETBOOT/vmlinuz-$ALPINE_NETBOOT_VERSION"
    get_file_s "$make_build/final_netboot/vmlinuz" "$ALPINE_NETBOOT/vmlinuz-$ALPINE_NETBOOT_VERSION" &
    show_progress train $! "Downloading vmlinuz..."
    msg_print note "$make_build/final_netboot/modloop  $ALPINE_NETBOOT/modloop-$ALPINE_NETBOOT_VERSION"
    get_file_s "$make_build/final_netboot/modloop" "$ALPINE_NETBOOT/modloop-$ALPINE_NETBOOT_VERSION" &
    show_progress train $! "Downloading modloop..."
    msg_print note "$make_build/initramfs $ALPINE_NETBOOT/initramfs-$ALPINE_NETBOOT_VERSION"
    get_file_s "$make_build/initramfs" "$ALPINE_NETBOOT/initramfs-$ALPINE_NETBOOT_VERSION" &
    show_progress train $! "Downloading initramfs..."
    msg_print note "Preparing initfs..."
    msg_print note  "$make_build/initramfs $make_build/final_netboot/initfs.img"
    prepare_initfs "$make_build/initramfs" "$make_build/final_netboot/initfs.img"
    cp "$make_build/rootfs.img" "$make_build/final_netboot/rootfs.img"
    msg_print note "Archiving NETBOOT..."
    (
        cd "$make_build/final_netboot"
        tar czf "../arc.tgz" ./*
    )
    cp -a "$make_build/arc.tgz" "$LI_NETBOOT"
fi
[[ -f "${cust_param}" ]] && rm -f "${cust_param}"
rm -rf "$make_build"

msg_print note "Builds are located at $BUILDS_DIR."
exec 1>&3 2>&4
exec 3>&- 4>&-

mv  /tmp/linux_build_${stime}.log  ~/