#!/usr/bin/env bash

# Based on https://github.com/kraxel/imagefish

# See https://github.com/kholia/OSX-KVM/ for Pull Requests
# And/Or https://github.com/sickcodes/osx-serial-generator/

help_text="

Purpose:            Create a bootable OpenCore disk image using libguestfs

Usage:             ./opencore-image-ng.sh [options]

Required options:
    --config <clover-config>    Config.plist file
    --img <disk-image>          Output image file (default: OpenCore.qcow2)

Optional options:
    --enable-voodoohda          Includes VoodooHDA kext for Catalina or older
    --catalina                  Includes VoodooHDA kext for Catalina or older
    --help, -h, help            Display this help and exit

Maintainer:                     Sick.Codes  https://github.com/sickcodes
Maintainer:                     Kholia      https://github.com/kholia
Contributor:                    Kraxel      https://github.com/kraxel

License: GPLv3+
"

# function print_help() {
# cat <<EOF
# usage: $0 [ options ]
# options:
#     --iso <iso-image>
#     --img <disk-image>
#     --cfg <clover-config>
# EOF


# gather arguments
while (( "$#" )); do
    case "${1}"  in

    --help | -h | h | help ) 
                echo "${help_text}" && exit 0
            ;;

    --img=* | image=* ) 
                export IMAGE="${1#*=}"
                shift
            ;;

    --img* | image* ) 
                export IMAGE="${2}"
                shift
                shift
            ;;

    --cfg=* | config=* ) 
                export CONFIG="${1#*=}"
                shift
            ;;

    --cfg* | config* ) 
                export IMAGE="${2}"
                shift
                shift
            ;;

    --enable-voodoohda | --enable-catalina-sound | --catalina ) 
                export VOODOOHDA=true
                shift
            ;;

    *)
                echo "Invalid option ${1}. Please check your flag options"
                exit 1
            ;;

    esac
done

function msg() {
    local txt="$1"
    local bold="\x1b[1m"
    local normal="\x1b[0m"
    echo -e "${bold}### ${txt}${normal}"
}

function do_cleanup() {
    msg "cleaning up ..."
    if test "${GUESTFISH_PID}" != ""; then
        guestfish --remote -- exit >/dev/null 2>&1 || true
    fi
    sudo rm -rf "${WORK}"
}

WORK="${TMPDIR-/var/tmp}/${0##*/}-$$"
mkdir "${WORK}" || exit 1
trap 'do_cleanup' EXIT

BASE="$(dirname "${0}")"

######################################################################
# guestfish script helpers

function fish() {
    echo "#" "$@"
    guestfish --remote -- "$@"        || exit 1
}

function fish_init() {

    case "${IMAGE}" in
        *.raw ) FORMAT="${FORMAT:=raw}"
            ;;
        * )     FORMAT="${FORMAT:=qcow2}"
            ;;
    esac

    msg "creating and adding disk image"
    fish disk-create "${IMAGE}" "${FORMAT}" 384M
    fish add "${IMAGE}"
    fish run
}

function fish_fini() {
    fish umount-all
}

# disabled by @sickcodes to allow unattended image overwrites
######################################################################
# sanity checks

# if test ! -f "${CONFIG}"; then
#     echo "ERROR: cfg not found: ${CONFIG}"
#     exit 1
# fi
# if test -f "${IMAGE}"; then
#     if test "$allow_override" = "yes"; then
#         rm -f "${IMAGE}"
#     else
#         echo "ERROR: image exists: ${IMAGE}"
#         exit 1
#     fi
# fi

######################################################################
# go!

msg "copy files from local folder"
BASE="$(dirname $0)"
cp -a "${BASE}"/EFI "${WORK}"
find "${WORK}"

#msg "[debug] list drivers in EFI/OC"
#(cd "${WORK}"/EFI/OC; find driver* -print)

export LIBGUESTFS_BACKEND=direct
eval $(guestfish --listen)
if ! [ $(ps -p 1 "${GUESTFISH_PID}" 2>/dev/null) ]; then
    echo "ERROR: starting guestfish failed"
    exit 1
fi

fish_init

msg "partition disk image: ${IMAGE}"
fish part-init /dev/sda gpt
fish part-add /dev/sda p 2048 300000
fish part-add /dev/sda p 302048 -2048
fish part-set-gpt-type /dev/sda 1 C12A7328-F81F-11D2-BA4B-00A0C93EC93B
fish part-set-bootable /dev/sda 1 true
fish mkfs vfat /dev/sda1 label:EFI
fish mkfs vfat /dev/sda2 label:OpenCore
fish mount /dev/sda2 /
fish mkdir /ESP
fish mount /dev/sda1 /ESP

msg "copy files to disk image"
cp -v "${CONFIG}" "${WORK}"/config.plist
fish mkdir /ESP/EFI
fish mkdir /ESP/EFI/OC
fish mkdir /ESP/EFI/OC/Kexts
fish mkdir /ESP/EFI/OC/ACPI
fish mkdir /ESP/EFI/OC/Resources
fish mkdir /ESP/EFI/OC/Tools

# remove VOODOOHDA if set
if [ "${VOODOOHDA}" == true ]; then
    find "${WORK}"/ -name "*VoodooHDA*" | xargs rm -f
fi

fish copy-in "${WORK}"/EFI/BOOT /ESP/EFI
fish copy-in "${WORK}"/EFI/OC/OpenCore.efi /ESP/EFI/OC
fish copy-in "${WORK}"/EFI/OC/Drivers /ESP/EFI/OC/
fish copy-in "${WORK}"/EFI/OC/Kexts /ESP/EFI/OC/
fish copy-in "${WORK}"/EFI/OC/ACPI /ESP/EFI/OC/
fish copy-in "${BASE}"/resources/OcBinaryData/Resources /ESP/EFI/OC/
fish copy-in "${WORK}"/EFI/OC/Tools /ESP/EFI/OC/

# Note
fish copy-in startup.nsh /

BASE="$(dirname $0)"
fish copy-in ""${WORK}"/config.plist"               /ESP/EFI/OC/

fish find /ESP/
fish_fini