#! /bin/sh
set -e

# grub-mkconfig helper script.
# Copyright (C) 2006,2007,2008,2009,2010  Free Software Foundation, Inc.
#
# GRUB is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# GRUB is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GRUB.  If not, see <http://www.gnu.org/licenses/>.

prefix=/usr
exec_prefix=${prefix}
bindir=${exec_prefix}/bin
libdir=${exec_prefix}/lib
. ${libdir}/grub/grub-mkconfig_lib

export TEXTDOMAIN=grub
export TEXTDOMAINDIR=${prefix}/share/locale

CLASS="--class os"

case "${GRUB_DISTRIBUTOR}" in
  Debian)
	OS="${GRUB_DISTRIBUTOR} GNU/kFreeBSD"
	CLASS="--class $(echo ${GRUB_DISTRIBUTOR} | tr '[A-Z]' '[a-z]' | cut -d' ' -f1) --class gnu-kfreebsd --class gnu ${CLASS}"
  ;;
  *)
	OS="FreeBSD"
	CLASS="--class freebsd --class bsd ${CLASS}"
  ;;
esac

kfreebsd_entry ()
{
  os="$1"
  version="$2"
  recovery="$3"	# not used yet
  args="$4"	# not used yet
  title="$(gettext_quoted "%s, with kFreeBSD %s")"
  printf "menuentry '${title}' ${CLASS} {\n" "${os}" "${version}"
  save_default_entry | sed -e "s/^/\t/"
  if [ -z "${prepare_boot_cache}" ]; then
    prepare_boot_cache="$(prepare_grub_to_access_device ${GRUB_DEVICE_BOOT} | sed -e "s/^/\t/")"
  fi
  if [ -z "${prepare_module_dir_cache}" ]; then
    prepare_module_dir_cache="$(prepare_grub_to_access_device $(grub-probe -t device "${module_dir}") | sed -e "s/^/\t/")"
  fi

  printf '%s\n' "${prepare_boot_cache}"
  message="$(gettext_printf "Loading kernel of FreeBSD %s ..." ${version})"
  cat << EOF
	echo			'$message'
	kfreebsd		${rel_dirname}/${basename}
EOF

  if test -n "${devices}" ; then
    cat << EOF
	kfreebsd_loadenv	${devices_rel_dirname}/${devices_basename}
EOF
  fi

  if test -e "${module_dir}/acpi.ko" ; then
    printf '%s\n' "${prepare_module_dir_cache}"
    cat << EOF
	kfreebsd_module_elf	${module_dir_rel}/acpi.ko
EOF
  fi

  case "${kfreebsd_fs}" in
    ext2fs)
      test -e "${module_dir}/ext2fs.ko"

      printf '%s\n' "${prepare_module_dir_cache}"
      cat << EOF
	kfreebsd_module_elf	${module_dir_rel}/ext2fs.ko
EOF
    ;;
    zfs)
      test -e "${module_dir}/opensolaris.ko"
      test -e "${module_dir}/zfs.ko"
      test -e "${dirname}/zfs/zpool.cache"

      printf '%s\n' "${prepare_module_dir_cache}"
      cat << EOF
	kfreebsd_module_elf	${module_dir_rel}/opensolaris.ko
	kfreebsd_module_elf	${module_dir_rel}/zfs.ko
EOF

      printf '%s\n' "${prepare_boot_cache}"
  cat << EOF
	kfreebsd_module		${rel_dirname}/zfs/zpool.cache type=/boot/zfs/zpool.cache
EOF
    ;;
  esac

  cat << EOF
	set kFreeBSD.vfs.root.mountfrom=${kfreebsd_fs}:${kfreebsd_device}
	set kFreeBSD.vfs.root.mountfrom.options=rw
}
EOF
}

list=`for i in /boot/kfreebsd-* /boot/kernel/kernel ; do
        if grub_file_is_not_garbage "$i" ; then echo -n "$i " ; fi
      done`
prepare_boot_cache=

while [ "x$list" != "x" ] ; do
  kfreebsd=`version_find_latest $list`
  echo "Found kernel of FreeBSD: $kfreebsd" >&2
  basename=`basename $kfreebsd`
  dirname=`dirname $kfreebsd`
  rel_dirname=`make_system_path_relative_to_its_root $dirname`

  if [ -f /boot/device.hints ] ; then
    devices=/boot/device.hints
    devices_basename=`basename $devices`
    devices_dirname=`dirname $devices`
    devices_rel_dirname=`make_system_path_relative_to_its_root $devices_dirname`
  fi

  case ${GRUB_FS} in
    ufs1 | ufs2)        kfreebsd_fs=ufs ;;
    ext2)               kfreebsd_fs=ext2fs ;;
    *)                  kfreebsd_fs=${GRUB_FS} ;;
  esac

  case ${GRUB_FS} in
    zfs)		kfreebsd_device=$(grub-probe -t fs_label --device ${GRUB_DEVICE})$(grub-mkrelpath / | sed -e "s,/*@$,,") ;;
    *)			kfreebsd_device=${GRUB_DEVICE} ;;
  esac

  version=`echo $basename | sed -e "s,^[^0-9]*-,,g;s/\.gz$//g"`
  alt_version=`echo $version | sed -e "s,\.old$,,g"`

  module_dir=
  for i in "/lib/modules/${version}" "/lib/modules/${alt_version}" \
      "/boot/kernel"; do
    if test -e "$i" ; then
      module_dir="$i"
      break
    fi
  done
  if test -n "${module_dir}" ; then
    echo "Found kernel module directory: ${module_dir}" >&2
    module_dir_rel=$(make_system_path_relative_to_its_root $module_dir)
  fi

  kfreebsd_entry "${OS}" "${version}"

  list=`echo $list | tr ' ' '\n' | grep -vx $kfreebsd | tr '\n' ' '`
done
