#!/bin/bash
set -e
img="$1"
zippath="$2"
zipname=$(basename "$zippath")

build_dir=$(mktemp -d ./tmp.XXXXX)
systemfile="$build_dir/system.PARTITION"
systemfile_ext4="$build_dir/system.PARTITION.ext4"
smallprcfile="$(pwd)/services/smallp.rc"
starthugepfile="$(pwd)/services/starthugep.sh"
uninstallpkgrc="$(pwd)/services/uninstallpkg.rc"
uninstallpkgfile="$(pwd)/services/uninstallpkg.sh"
setmacrc="$(pwd)/services/setmac.rc"
setmacfile="$(pwd)/services/set-mac.sh"

aml_image_v2_packer -d "$img" "./$build_dir"
if file "$systemfile" | grep "Android sparse image"
then
    simg2img "$systemfile" "$systemfile_ext4"
    loopdev=$(sudo losetup -P -f --show "$systemfile_ext4")
else
    loopdev=$(sudo losetup -P -f --show "$systemfile")
fi
mkdir "$build_dir/system"
sudo mount "$loopdev" "$build_dir/system"

sudo wget -P "$build_dir/system/" -O "$build_dir/system/bin/hugep-all.zip"

sudo chmod 777 "$starthugepfile"
sudo cp -f "$smallprcfile" "$build_dir/system/etc/init/"
sudo cp -f "$starthugepfile" "$build_dir/system/bin/"
sudo chcon u:object_r:drmserver_exec:s0 "$build_dir/system/bin/$(basename "$starthugepfile")"

sudo cp -f "$uninstallpkgrc" "$build_dir/system/etc/init"
sudo cp -f "$uninstallpkgfile" "$build_dir/system/bin"
sudo chcon u:object_r:drmserver_exec:s0 "$build_dir/system/bin/$(basename "$uninstallpkgfile")"

sudo cp -f "$setmacrc" "$build_dir/system/etc/init"
sudo cp -f "$setmacfile" "$build_dir/system/bin"
sudo chcon u:object_r:drmserver_exec:s0 "$build_dir/system/bin/$(basename "$setmacfile")"

sudo umount "$loopdev"
sudo losetup -d "$loopdev"
rm -rf "$build_dir/system"
sync

if [[ -f "$systemfile_ext4" ]]
then
    rm -f "$systemfile"
    img2simg "$systemfile_ext4" "$systemfile"
    rm -f "$systemfile_ext4"
fi
aml_image_v2_packer -r "./$build_dir/image.cfg" "$build_dir" "$(basename "$img")"
rm -rf "./$build_dir/"