# Maintainer: Jan Neumann <koglimail at gmail>
# Contributor: nl6720 <nl6720@gmail.com>
# Contributor: Keshav Amburay <(the ddoott ridikulus ddoott rat) (aatt) (gemmaeiil) (ddoott) (ccoomm)>
# Contributor: Tobias Powalowski <tpowa@archlinux.org>

_USE_GNU_EFI='1'
_PXE='0'

#######
[[ "${CARCH}" == 'x86_64' ]] && _TIANO_ARCH='X64'
[[ "${CARCH}" == 'x86_64' ]] && _TIANO_S_ARCH='x64'
#######

_pkgname='refind'
pkgname="${_pkgname}-efi-git-patched"
pkgver=0.11.2.r604.fb5631a
pkgrel=1
pkgdesc='rEFInd Boot Manager - patched git version'
url='http://www.rodsbooks.com/refind/'
arch=('x86_64')
license=('GPL3' 'custom')

makedepends=('git')
depends=('bash' 'dosfstools' 'efibootmgr')
optdepends=('sudo: privilege elevation'
            'parted: For finding ESP'
            'imagemagick: For refind-mkfont script'
            'python: For refind-mkdefault script'
            'sbsigntools: For EFI binary signing (Secure Boot support)'
            'preloader-signed: pre-signed shim loader (Secure Boot support)'
            'shim-signed: pre-signed shim loader (Secure Boot support)')

options=('!strip' '!buildflags' '!makeflags')

conflicts=("${pkgname%-patched}" "${pkgname%-git-patched}")
provides=("${pkgname%-git-patched}=${pkgver}")

install="${pkgname%-git-patched}.install"

source=("refind::git+https://git.code.sf.net/p/refind/code#commit=fb5631a"
        'refind_linux.conf'
	'0001-Use-initrd-file-with-shortest-file-name-if-same-kern.patch'
	'0002-Added-initrd-path-variable-to-refind_linux.conf.patch')

sha512sums=('SKIP'
            'c6a4fbd7604635ed9d562f018c6c426ab8b67184aad66b23a060e4e96b251fea371e6c41575a839e53baccc5093d3947c617c485f1f30efc2814f4deccd43169'
	    '61d1e211a8b84e91a503f88aa695894085f6788b236f828a7cc256e2587f944cbe5a7c50ab1fa34eeab4fa417ca3718af7b01724a5f65f3027cb12916ff1c075'
	    'b9c73c12e1dc2bde0237045f3243619cd537c39977fc447f04675bc2f6bf9ca1ef006c8555c4c53f473b26438b8cd10773c8897ad033e979392f4034be3e6065')

if [[ "${_USE_GNU_EFI}" == '1' ]]; then

	pkgdesc="${pkgdesc} - Built with GNU-EFI libs"
	makedepends+=('gnu-efi-libs')

else

	pkgdesc="${pkgdesc} - Built with Tianocore UDK libs"
	makedepends+=('python2')

	_UDK_VERSION='UDK2014'
	_TIANO_DIR_="edk2-${_UDK_VERSION}"

	_TIANOCORE_PKG='Mde'
	_TIANOCORE_TARGET='RELEASE'
	_UDK_TARGET="${_TIANOCORE_PKG}Pkg/${_TIANOCORE_PKG}Pkg.dsc"
	_COMPILER='GCC48'

	source+=("${_TIANO_DIR_}::git+https://github.com/tianocore/edk2.git#branch=${_UDK_VERSION}")
	sha512sums+=('SKIP')

fi

pkgver() {

	cd "${srcdir}/${_pkgname}/"

	printf "%s.r%s.%s" "$(grep -oP 'REFIND_VERSION L"\K[^"]+' "${srcdir}/${_pkgname}/include/version.h")" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"

}

_setup_tianocore_env_vars() {

	# Setup UDK PATH ENV variables
	export _UDK_DIR="${srcdir}/${_TIANO_DIR_}_build"
	export EDK_TOOLS_PATH="${_UDK_DIR}/BaseTools"

}

_prepare_tianocore_sources() {

	# Delete old UDK BUILD dir
	rm -rf "${_UDK_DIR}/" || true

	# Create UDK BUILD dir
	cp -r "${srcdir}/${_TIANO_DIR_}" "${_UDK_DIR}"

	cd "${_UDK_DIR}/"

	# Cleanup UDK config files
	rm -rf "${_UDK_DIR}/Build/" || true
	rm -rf "${_UDK_DIR}/Conf/" || true
	mkdir -p "${_UDK_DIR}/Conf/"
	mkdir -p "${_UDK_DIR}/Build/"

	# Use python2 for UDK BaseTools
	sed 's|python |python2 |g' -i "${EDK_TOOLS_PATH}/BinWrappers/PosixLike"/* || true
	sed 's|python |python2 |g' -i "${EDK_TOOLS_PATH}/Tests/GNUmakefile"

	# Fix GCC Warning as error
	sed 's|-Werror |-Wno-error -Wno-unused-but-set-variable |g' -i "${EDK_TOOLS_PATH}/Source/C/Makefiles/header.makefile" || true
	sed 's|-Werror |-Wno-error -Wno-unused-but-set-variable |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true

	# Fix GenFw: ERROR 3000: Invalid, bad symbol definition http://www.mail-archive.com/edk2-devel@lists.sourceforge.net/msg03625.html
	sed -e 's|_OBJCOPY_FLAGS      =|_OBJCOPY_FLAGS      = -R .eh_frame|g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true

	# Fix GCC >=4.7 error - gcc: error: unrecognized command line option '-melf_x86_64'
	sed 's| -m64 --64 -melf_x86_64| -m64|g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true
	sed 's|--64 | |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true
	sed 's| -m64 -melf_x86_64| -m64|g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true

	# Remove GCC -g debug option and add -O0 -mabi=ms -maccumulate-outgoing-args
	sed 's|DEFINE GCC_ALL_CC_FLAGS            = -g |DEFINE GCC_ALL_CC_FLAGS            = -O0 -mabi=ms -maccumulate-outgoing-args |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true
	sed 's|DEFINE GCC44_ALL_CC_FLAGS            = -g |DEFINE GCC44_ALL_CC_FLAGS            = -O0 -mabi=ms -maccumulate-outgoing-args |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true

	# Fix GenFw: ERROR 3000: Invalid, Unsupported section alignment
	sed 's|--gc-sections|--gc-sections --build-id=none|g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template"
	# sed 's|Error (NULL, 0, 3000, "Invalid", "Unsupported section alignment.");|continue;|g' -i "${EDK_TOOLS_PATH}/Source/C/GenFw/Elf64Convert.c"
	# sed 's|Error (NULL, 0, 3000, "Invalid", "Unsupported section alignment.");|continue;|g' -i "${EDK_TOOLS_PATH}/Source/C/GenFw/Elf32Convert.c"
	# sed 's|_GCC48_IA32_OBJCOPY_FLAGS        =|_GCC48_IA32_OBJCOPY_FLAGS        = --section-alignment=0x20 |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true
	# sed 's|_GCC48_X64_OBJCOPY_FLAGS        =|_GCC48_X64_OBJCOPY_FLAGS        = --section-alignment=0x20 |g' -i "${EDK_TOOLS_PATH}/Conf/tools_def.template" || true

	# Fix UDK Target Platform
	sed "s|ACTIVE_PLATFORM       = Nt32Pkg/Nt32Pkg.dsc|ACTIVE_PLATFORM       = ${_UDK_TARGET}|g" -i "${EDK_TOOLS_PATH}/Conf/target.template" || true
	sed "s|TARGET                = DEBUG|TARGET                = ${_TIANOCORE_TARGET}|g" -i "${EDK_TOOLS_PATH}/Conf/target.template" || true
	sed "s|TOOL_CHAIN_TAG        = MYTOOLS|TOOL_CHAIN_TAG        = ${_COMPILER}|g" -i "${EDK_TOOLS_PATH}/Conf/target.template" || true

	# Fix UDK Target ARCH for rEFInd
	sed "s|X64|${_TIANO_ARCH}|g" -i "${EDK_TOOLS_PATH}/Conf/target.template" || true
	sed "s|IA32|${_TIANO_ARCH}|g" -i "${EDK_TOOLS_PATH}/Conf/target.template" || true

	chmod 0755 "${_UDK_DIR}/BaseTools/BuildEnv"

}

_prepare_refind_sources() {

	rm -rf "${srcdir}/${_pkgname}_build/" || true
	cp -r "${srcdir}/${_pkgname}" "${srcdir}/${_pkgname}_build"

	cd "${srcdir}/${_pkgname}_build/"

	# Clean rEFInd git repo
	git clean -x -d -f

	git am < ../0001-Use-initrd-file-with-shortest-file-name-if-same-kern.patch ../0002-Added-initrd-path-variable-to-refind_linux.conf.patch

	if [[ "${_USE_GNU_EFI}" == '1' ]]; then
		# Enable GNU_EFI_USE_MS_ABI
		sed "s|-DEFI_FUNCTION_WRAPPER|-DEFI_FUNCTION_WRAPPER -maccumulate-outgoing-args|g" -i "${srcdir}/${_pkgname}_build/Make.common" || true
		sed "s|-DEFIX64|-DEFIX64 -maccumulate-outgoing-args|g" -i "${srcdir}/${_pkgname}_build/Make.common" || true
		sed "s|-m64|-maccumulate-outgoing-args -m64|g" -i "${srcdir}/${_pkgname}_build/filesystems/Make.gnuefi" || true
	else
		# Fix UDK Path in rEFInd Makefiles
		sed "s|^export EDK2BASE=.*$|export EDK2BASE=${_UDK_DIR}|g" -i "${srcdir}/${_pkgname}_build/Makefile" || true

		# Fix GenFw: ERROR 3000: Invalid section alignment
		sed 's|--gc-sections|--gc-sections --build-id=none|g' -i "${srcdir}/${_pkgname}_build/Make.common" || true
		# sed -e 's|--gc-sections|--gc-sections -z max-page-size=0x20|g' -i "${srcdir}/${_pkgname}_build/Make.common" || true
		# sed -e 's|--strip-unneeded|--section-alignment=0x20 --strip-unneeded|g' -i "${srcdir}/${_pkgname}_build/Make.common" || true
	fi

}

prepare() {
	if [[ "${_USE_GNU_EFI}" != '1' ]]; then
		_setup_tianocore_env_vars

		# Prepare Tianocore Sources
		_prepare_tianocore_sources
	fi

	# Prepare rEFInd Sources
	_prepare_refind_sources

}

_build_tianocore_sources() {

	_setup_tianocore_env_vars

	cd "${_UDK_DIR}/"

	# Setup UDK Environment
	source "${_UDK_DIR}/BaseTools/BuildEnv" BaseTools

	# Compile UDK BaseTools
	make -C "${EDK_TOOLS_PATH}"

	# Compile UDK Libraries
	"${EDK_TOOLS_PATH}/BinWrappers/PosixLike/build" -p "${_UDK_TARGET}" -a "${_TIANO_ARCH}" -b "${_TIANOCORE_TARGET}" -t "${_COMPILER}"

}

build() {

	if [[ "${_USE_GNU_EFI}" != '1' ]]; then
		_build_tianocore_sources
	fi

	cd "${srcdir}/${_pkgname}_build/"

	# Compile rEFInd UEFI application
	if [[ "${_USE_GNU_EFI}" == '1' ]]; then
		make gnuefi
	else
		make tiano
	fi

	# Compile UEFI FS drivers
	if [[ "${_USE_GNU_EFI}" == '1' ]]; then
		make fs_gnuefi
	else
		make fs_tiano
	fi

	if [[ "${_PXE}" == '1' ]]; then
		cd "${srcdir}/${_pkgname}_build/net"

		# Compile Network support
		make source
		make netboot
	fi

}

package() {

	# Install the rEFInd UEFI application
	install -d "${pkgdir}/usr/share/refind/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/refind/refind_${_TIANO_S_ARCH}.efi" "${pkgdir}/usr/share/refind/refind_${_TIANO_S_ARCH}.efi"

	# Install UEFI drivers built from rEFInd
	install -d "${pkgdir}/usr/share/refind/drivers_${_TIANO_S_ARCH}/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/drivers_${_TIANO_S_ARCH}"/*.efi "${pkgdir}/usr/share/refind/drivers_${_TIANO_S_ARCH}/"

	# Install UEFI applications built from rEFInd
	install -d "${pkgdir}/usr/share/refind/tools_${_TIANO_S_ARCH}"
	install -D -m0644 "${srcdir}/${_pkgname}_build/gptsync/gptsync_${_TIANO_S_ARCH}.efi" "${pkgdir}/usr/share/refind/tools_${_TIANO_S_ARCH}/gptsync_${_TIANO_S_ARCH}.efi"

	if [[ "${_PXE}" == '1' ]]; then
		install -D -m0644 "${srcdir}/${_pkgname}_build/net/bin/ipxe.efi" "${pkgdir}/usr/share/refind/tools_${_TIANO_S_ARCH}/ipxe_${_TIANO_S_ARCH}.efi"
		install -D -m0644 "${srcdir}/${_pkgname}_build/net/bin/ipxe_discovery.efi" "${pkgdir}/usr/share/refind/tools_${_TIANO_S_ARCH}/ipxe_discovery_${_TIANO_S_ARCH}.efi"
	fi

	# Install rEFInd helper scripts
	install -d "${pkgdir}/usr/bin/"
	install -D -m0755 "${srcdir}/${_pkgname}_build/refind-install" "${pkgdir}/usr/bin/refind-install"
	install -D -m0755 "${srcdir}/${_pkgname}_build/refind-mkdefault" "${pkgdir}/usr/bin/refind-mkdefault"
	install -D -m0755 "${srcdir}/${_pkgname}_build/mkrlconf" "${pkgdir}/usr/bin/mkrlconf"
	install -D -m0755 "${srcdir}/${_pkgname}_build/mvrefind" "${pkgdir}/usr/bin/mvrefind"
	install -D -m0755 "${srcdir}/${_pkgname}_build/fonts/mkfont.sh" "${pkgdir}/usr/bin/refind-mkfont"

	# Install the rEFInd sample config files
	install -D -m0644 "${srcdir}/${_pkgname}_build/refind.conf-sample" "${pkgdir}/usr/share/refind/refind.conf-sample"
	install -D -m0644 "${srcdir}/refind_linux.conf" "${pkgdir}/usr/share/refind/refind_linux.conf-sample"

	# Install the rEFInd docs
	install -d "${pkgdir}/usr/share/refind/docs/html/"
	install -d "${pkgdir}/usr/share/refind/docs/Styles/"
	install -d "${pkgdir}/usr/share/man/man8/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/docs/refind"/* "${pkgdir}/usr/share/refind/docs/html/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/docs/Styles"/* "${pkgdir}/usr/share/refind/docs/Styles/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/README.txt" "${pkgdir}/usr/share/refind/docs/README.txt"
	install -D -m0644 "${srcdir}/${_pkgname}_build/NEWS.txt" "${pkgdir}/usr/share/refind/docs/NEWS.txt"
	install -D -m0644 "${srcdir}/${_pkgname}_build/docs/man"/*.8 "${pkgdir}/usr/share/man/man8/"

	# Install the rEFInd fonts
	install -d "${pkgdir}/usr/share/refind/fonts/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/fonts"/* "${pkgdir}/usr/share/refind/fonts/"
	rm -f "${pkgdir}/usr/share/refind/fonts/mkfont.sh"

	# Install the rEFInd icons
	install -d "${pkgdir}/usr/share/refind/icons/svg/"
	install -d "${pkgdir}/usr/share/refind/icons/licenses/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/icons/svg"/* "${pkgdir}/usr/share/refind/icons/svg/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/icons"/*.png "${pkgdir}/usr/share/refind/icons/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/icons/licenses"/* "${pkgdir}/usr/share/refind/icons/licenses/"

	# Install the rEFInd images
	install -d "${pkgdir}/usr/share/refind/images/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/images"/*.{png,bmp} "${pkgdir}/usr/share/refind/images/"

	# Install the rEFInd keys
	install -d "${pkgdir}/usr/share/refind/keys/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/keys"/* "${pkgdir}/usr/share/refind/keys/"

	# Install the rEFIt license file, since rEFInd is a fork of rEFIt
	install -d "${pkgdir}/usr/share/licenses/refind/"
	install -D -m0644 "${srcdir}/${_pkgname}_build/LICENSE.txt" "${pkgdir}/usr/share/licenses/refind/LICENSE"

	# Use '#!/usr/bin/env bash' in all scripts
	sed 's|#!/bin/bash|#!/usr/bin/env bash|g' -i "${pkgdir}/usr/bin"/* || true

	# Point refind dir paths to /usr/share/refind/ in refind-install script
	sed 's|RefindDir=\"\$ThisDir/refind\"|RefindDir="/usr/share/refind/"|g' -i "${pkgdir}/usr/bin/refind-install"

}