#!/bin/bash
# Filename:      ${GRML_FAI_CONFIG}/config/scripts/GRMLBASE/45-efi
# Purpose:       create grub image for use in ISO for EFI boot
# Authors:       grml-team (grml.org), (c) Michael Prokop <mika@grml.org>
# Bug-Reports:   see http://grml.org/bugs/
# License:       This file is licensed under the GPL v2 or any later version.
################################################################################

set -e


BOOTX64="${target}/boot/bootx64.efi"
EFI_IMG="${target}/boot/efi.img"
TMP_CONFIG="${target}/tmp/grub_config_efi"

rm -f "$BOOTX64" "$EFI_IMG" "$TMP_CONFIG"

cat > "$TMP_CONFIG" <<EOF
search --set -f /conf/bootid.txt root
if [ -e /boot/grub/grub.cfg ]; then
 set prefix=(\$root)/boot/grub
 configfile /boot/grub/grub.cfg
else
 echo "E: Could not find root device!"
fi
EOF

BOOTX64="${BOOTX64##${target}}"
EFI_IMG="${EFI_IMG##${target}}"
TMP_CONFIG="${TMP_CONFIG##${target}}"


declare -A ADDITIONAL_MODULES
ARCHS=()

if [ -r "${target}"/usr/lib/grub/i386-pc/moddep.lst ] ;  then
  ARCHS+=(i386-pc)
  ADDITIONAL_MODULES[i386-pc]="biosdisk"
fi

if ifclass AMD64 ; then
  if [ -r "${target}"/usr/lib/grub/x86_64-efi/moddep.lst ] ; then
    ARCHS+=(x86_64-efi)
    ADDITIONAL_MODULES[x86_64-efi]="efi_gop efi_uga"
  else
    echo "/usr/lib/grub/x86_64-efi/moddep.lst could not be found, skipping."
    echo "NOTE: grub-efi-amd64-bin not installed?"
  fi
fi

BOOTX64="${BOOTX64##${target}}"
EFI_IMG="${EFI_IMG##${target}}"
TMP_CONFIG="${TMP_CONFIG##${target}}"


if [ -z "$ARCHS" ] ; then
  echo "WARNING: No grub modules found, exiting" >&2
  exit 0
fi

for arch in ${ARCHS[@]} ; do
$ROOTCMD grub-mkimage -O $arch -o /boot/$arch.img --prefix=/boot/grub/ --config="$TMP_CONFIG" \
  bitmap boot btrfs cat chain cmp configfile cpio echo elf ext2    \
  fat gfxmenu gfxterm gzio help iso9660 jpeg linux loopback lvm    \
  memdisk minicmd multiboot normal part_gpt part_msdos play png    \
  probe raid regexp reiserfs search search_fs_file search_fs_uuid  \
  search_label squash4 terminal test video videoinfo xfs ${ADDITIONAL_MODULES[$arch]}
done

if [ -f "${target}/boot/i386-pc.img" ] ; then
  mv "${target}/boot/i386-pc.img" "${target}/boot/grub/grub.img"
fi

if [ -f "${target}/boot/x86_64-efi.img" ] ; then
  mv "${target}/boot/x86_64-efi.img" "${target}/${BOOTX64}"
fi

if ifclass AMD64 ; then
  if ! [ -r "${target}/${BOOTX64}" ] ; then
    echo "Can not access grub efi image." >&2
    exit 0
  fi

  SIZE=$(du -sk "${target}/${BOOTX64}" | awk -F" " '{print $1'})
  SIZE=$(((($SIZE / 32 )+2)*32))

  dd if=/dev/zero of="${target}/${EFI_IMG}" bs=1k count="$SIZE" 2>/dev/null
  $ROOTCMD mkfs.vfat -n GRML "$EFI_IMG" >/dev/null
  $ROOTCMD mmd -i "$EFI_IMG" ::EFI
  $ROOTCMD mmd -i "$EFI_IMG" ::EFI/BOOT
  $ROOTCMD mcopy -i "$EFI_IMG" "$BOOTX64" ::EFI/BOOT/bootx64.efi >/dev/null
  echo "Generated EFI image $BOOTX64"
  echo "Generated bootx64 image $EFI_IMG"
fi

rm -f "${target}/${TMP_CONFIG}"
echo "Generated Grub images"


## END OF FILE #################################################################
# vim:ft=sh expandtab ai tw=80 tabstop=4 shiftwidth=2