#! /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="/usr"
datarootdir="/usr/share"

. "$pkgdatadir/grub-mkconfig_lib"

export TEXTDOMAIN=grub
export TEXTDOMAINDIR="${datarootdir}/locale"

CLASS="--class gnu-linux --class gnu --class os --unrestricted"

OS_NAME=""
OS_VERSION=""

if [ -r /etc/os-release ]; then
  OS_NAME=$(grep '^NAME=' /etc/os-release | sed 's/^NAME=//; s/^"//; s/"$//')
  _variant=$(grep '^VARIANT=' /etc/os-release | sed 's/^VARIANT=//; s/^"//; s/"$//')
  OS_VERSION=$(grep '^VERSION_ID=' /etc/os-release | sed 's/^VERSION_ID=//; s/^"//; s/"$//')
  [ -n "$_variant" ] && OS_NAME="$OS_NAME $_variant"
fi

if [ -z "$OS_NAME" ] && [ -r /etc/altlinux-release ]; then
  _release=$(sed 's/ (.*$//; s/ $//' /etc/altlinux-release)
  OS_VERSION=$(echo "$_release" | sed -n 's/.* \([0-9][0-9.]*\)$/\1/p')
  OS_NAME=$(echo "$_release" | sed "s/ ${OS_VERSION}$//")
fi

if [ -z "$OS_NAME" ]; then
  OS_NAME="ALT Linux"
  OS_VERSION=""
fi

OS="$(gettext "$OS_NAME")"
[ -n "$OS_VERSION" ] && OS="$OS $OS_VERSION"

CLASS="${CLASS} --class altlinux"

# loop-AES arranges things so that /dev/loop/X can be our root device, but
# the initrds that Linux uses don't like that.
case ${GRUB_DEVICE} in
  /dev/loop/*|/dev/loop[0-9])
    GRUB_DEVICE=`losetup ${GRUB_DEVICE} | sed -e "s/^[^(]*(\([^)]\+\)).*/\1/"`
  ;;
esac

: ${GRUB_CMDLINE_LINUX_RECOVERY:=single}
: ${GRUB_TOP_LEVEL:=/boot/vmlinuz}

# Default to disabling partition uuid support to maintian compatibility with
# older kernels.
: ${GRUB_DISABLE_LINUX_PARTUUID=true}

# btrfs may reside on multiple devices. We cannot pass them as value of root= parameter
# and mounting btrfs requires user space scanning, so force UUID in this case.
if ( [ "x${GRUB_DEVICE_UUID}" = "x" ] && [ "x${GRUB_DEVICE_PARTUUID}" = "x" ] ) \
    || ( [ "x${GRUB_DISABLE_LINUX_UUID}" = "xtrue" ] \
	&& [ "x${GRUB_DISABLE_LINUX_PARTUUID}" = "xtrue" ] ) \
    || ( ! findfs UUID="${GRUB_DEVICE_UUID}" > /dev/null 2>&1 \
	&& ! findfs PARTUUID="${GRUB_DEVICE_PARTUUID}" > /dev/null 2>&1 ) \
    || ( test -e "${GRUB_DEVICE}" && uses_abstraction "${GRUB_DEVICE}" lvm && ! uses_abstraction "${GRUB_DEVICE}" luks && ! uses_abstraction "${GRUB_DEVICE}" luks2); then
  LINUX_ROOT_DEVICE=${GRUB_DEVICE}
elif [ "x${GRUB_DEVICE_UUID}" = "x" ] \
    || [ "x${GRUB_DISABLE_LINUX_UUID}" = "xtrue" ]; then
  LINUX_ROOT_DEVICE=PARTUUID=${GRUB_DEVICE_PARTUUID}
else
  LINUX_ROOT_DEVICE=UUID=${GRUB_DEVICE_UUID}
fi

case x"$GRUB_FS" in
    xbtrfs)
	rootsubvol="`make_system_path_relative_to_its_root /`"
	rootsubvol="${rootsubvol#/}"
	if [ "x${rootsubvol}" != x ] \
	       && [ "x${GRUB_BTRFS_USE_DEFAULT_SUBVOL}" != "xtrue" ] ; then
	    GRUB_CMDLINE_LINUX="rootflags=subvol=${rootsubvol} ${GRUB_CMDLINE_LINUX}"
	fi;;
    xzfs)
	rpool=`${grub_probe} --device ${GRUB_DEVICE} --target=fs_label 2>/dev/null || zdb -l ${GRUB_DEVICE} | awk -F \' '/ name/ { print $2 }'`
	bootfs="`make_system_path_relative_to_its_root / | sed -e "s,@$,,"`"
	LINUX_ROOT_DEVICE="ZFS=${rpool}${bootfs%/}"
	;;
esac

title_correction_code=

linux_entry ()
{
  os="$1"
  version="$2"
  type="$3"
  args="$4"

  if [ -z "$boot_device_id" ]; then
      boot_device_id="$(grub_get_device_id "${GRUB_DEVICE}")"
  fi
  if [ x$type != xsimple ] ; then
      case $type in
	  recovery)
	      title="$(gettext_printf "%s, %s (recovery)" "${os}" "${version}")" ;;
	  *)
	      title="$(gettext_printf "%s, %s" "${os}" "${version}")" ;;
      esac
      if [ x"$title" = x"$GRUB_ACTUAL_DEFAULT" ] || [ x"Previous Linux versions>$title" = x"$GRUB_ACTUAL_DEFAULT" ]; then
	  replacement_title="$(echo "Advanced options for ${OS}" | sed 's,>,>>,g')>$(echo "$title" | sed 's,>,>>,g')"
	  quoted="$(echo "$GRUB_ACTUAL_DEFAULT" | grub_quote)"
	  title_correction_code="${title_correction_code}if [ \"x\$default\" = '$quoted' ]; then default='$(echo "$replacement_title" | grub_quote)'; fi;"
	  grub_warn "$(gettext_printf "Please don't use old title \`%s' for GRUB_DEFAULT, use \`%s' (for versions before 2.00) or \`%s' (for 2.00 or later)" "$GRUB_ACTUAL_DEFAULT" "$replacement_title" "gnulinux-advanced-$boot_device_id>gnulinux-$version-$type-$boot_device_id")"
      fi
      echo "menuentry '$(echo "$title" | grub_quote)' ${CLASS} \$menuentry_id_option 'gnulinux-$version-$type-$boot_device_id' {" | sed "s/^/$submenu_indentation/"
  else
      echo "menuentry '$(echo "$os" | grub_quote)' ${CLASS} \$menuentry_id_option 'gnulinux-simple-$boot_device_id' {" | sed "s/^/$submenu_indentation/"
  fi      
  if [ x$type != xrecovery ] ; then
      save_default_entry | grub_add_tab
  fi

  # Use ELILO's generic "efifb" when it's known to be available.
  # FIXME: We need an interface to select vesafb in case efifb can't be used.
  if [ "x$GRUB_GFXPAYLOAD_LINUX" = x ]; then
      echo "	load_video" | sed "s/^/$submenu_indentation/"
      if grep -qx "CONFIG_FB_EFI=y" "${config}" 2> /dev/null \
	  && grep -qx "CONFIG_VT_HW_CONSOLE_BINDING=y" "${config}" 2> /dev/null; then
	  echo "	set gfxpayload=keep" | sed "s/^/$submenu_indentation/"
      fi
  else
      if [ "x$GRUB_GFXPAYLOAD_LINUX" != xtext ]; then
	  echo "	load_video" | sed "s/^/$submenu_indentation/"
      fi
      echo "	set gfxpayload=$GRUB_GFXPAYLOAD_LINUX" | sed "s/^/$submenu_indentation/"
  fi

  echo "	insmod gzio" | sed "s/^/$submenu_indentation/"

  if [ x$dirname = x/ ]; then
    if [ -z "${prepare_root_cache}" ]; then
      prepare_root_cache="$(prepare_grub_to_access_device ${GRUB_DEVICE} | grub_add_tab)"
    fi
    printf '%s\n' "${prepare_root_cache}" | sed "s/^/$submenu_indentation/"
  else
    if [ -z "${prepare_boot_cache}" ]; then
      prepare_boot_cache="$(prepare_grub_to_access_device ${GRUB_DEVICE_BOOT} | grub_add_tab)"
    fi
    printf '%s\n' "${prepare_boot_cache}" | sed "s/^/$submenu_indentation/"
  fi
  sed "s/^/$submenu_indentation/" << EOF
	linux	${rel_dirname}/${basename} root=${linux_root_device_thisversion} ro ${args}
EOF
  if test -n "${initrd}" ; then
    initrd_path=
    for i in ${initrd}; do
      initrd_path="${initrd_path} ${rel_dirname}/${i}"
    done
    sed "s/^/$submenu_indentation/" << EOF
	initrd	$(echo $initrd_path)
EOF
  fi
  sed "s/^/$submenu_indentation/" << EOF
}
EOF
}

machine=`uname -m`
case "x$machine" in
    xi?86 | xx86_64)
	list=
	for i in /boot/vmlinuz /boot/vmlinuz-* ; do
	    if grub_file_is_not_garbage "$i" ; then list="$list $i" ; fi
	done ;;
    *) 
	list=
	for i in /boot/vmlinuz /boot/vmlinuz-* /boot/vmlinux-* ; do
                  if grub_file_is_not_garbage "$i" ; then list="$list $i" ; fi
	done ;;
esac

case "$machine" in
    i?86) GENKERNEL_ARCH="x86" ;;
    mips|mips64) GENKERNEL_ARCH="mips" ;;
    mipsel|mips64el) GENKERNEL_ARCH="mipsel" ;;
    arm*) GENKERNEL_ARCH="arm" ;;
    *) GENKERNEL_ARCH="$machine" ;;
esac

prepare_boot_cache=
prepare_root_cache=
boot_device_id=
title_correction_code=

# Extra indentation to add to menu entries in a submenu. We're not in a submenu
# yet, so it's empty. In a submenu it will be equal to '\t' (one tab).
submenu_indentation=""

# Perform a reverse version sort on the entire list.
# Temporarily replace the '.old' suffix by ' 1' and append ' 2' for all
# other files to order the '.old' files after their non-old counterpart
# in reverse-sorted order.

reverse_sorted_list=$(echo $list | tr ' ' '\n' | sed -e 's/\.old$/ 1/; / 1$/! s/$/ 2/' | version_sort -r | sed -e 's/ 1$/.old/; s/ 2$//')

# Reorder the list into ALT Linux specific well-defined order:
#   1. /boot/vmlinuz default entry
#   2. Flavour symlinks (sorted by version_sort from above)
#   3. Real kernel files (sorted by version_sort from above)
default_entry=""
flavour_syms=""
real_kernels=""
for entry in $reverse_sorted_list; do
    if [ "$entry" = "/boot/vmlinuz" ]; then
        default_entry="$entry"
    elif [ -L "$entry" ]; then
        flavour_syms="$flavour_syms $entry"
    else
        real_kernels="$real_kernels $entry"
    fi
done
reverse_sorted_list="$default_entry $flavour_syms $real_kernels"

if [ "x$GRUB_TOP_LEVEL" != x ]; then
  reverse_sorted_list=$(grub_move_to_front "$GRUB_TOP_LEVEL" ${reverse_sorted_list})
fi

is_top_level=true
for linux in ${reverse_sorted_list}; do
  gettext_printf "Found linux image: %s\n" "$linux" >&2
  basename=$(basename "$linux")
  dirname=$(dirname "$linux")
  rel_dirname=$(make_system_path_relative_to_its_root "$dirname")

  # Determine kernel type
  if [ "$linux" = "/boot/vmlinuz" ]; then
      kernel_type="default"
  elif [ -L "$linux" ]; then
      kernel_type="flavour"
  else
      kernel_type="real"
  fi

  # Check GRUB_VMLINUZ_SYMLINKS setting
  if [ "$kernel_type" != "real" ]; then
      case "$GRUB_VMLINUZ_SYMLINKS" in
          no|false)
              gettext_printf "Skipping symlink: %s\n" "$linux" >&2
              continue
              ;;
          vmlinuz)
              if [ "$kernel_type" = "flavour" ]; then
                  gettext_printf "Skipping symlink: %s\n" "$linux" >&2
                  continue
              fi
              ;;
      esac
  fi

  # Compute version and initrd_pattern based on kernel type
  case "$kernel_type" in
      default)
          version="vmlinuz"
          initrd_pattern="initrd.img"
          ;;
      flavour)
          flavour="${basename#vmlinuz-}"
          version="$flavour"
          initrd_pattern="initrd-${flavour}.img"
          ;;
      real)
          version="${basename#vmlinuz-}"
          initrd_pattern="initrd-${version}.img"
          ;;
  esac

  linux_root_device_thisversion="${LINUX_ROOT_DEVICE}"

  # Early initrd (microcode, etc.)
  initrd_early=
  for i in ${GRUB_EARLY_INITRD_LINUX_STOCK} \
           ${GRUB_EARLY_INITRD_LINUX_CUSTOM}; do
      if [ -e "${dirname}/${i}" ]; then
          initrd_early="${initrd_early} ${i}"
      fi
  done

  # Main initrd based on kernel type
  initrd_real=
  if [ -e "${dirname}/${initrd_pattern}" ]; then
      initrd_real="$initrd_pattern"
  elif [ "$kernel_type" = "real" ] && [ -e "${dirname}/initramfs-${version}.img" ]; then
      initrd_real="initramfs-${version}.img"
  fi

  initrd=
  if test -n "${initrd_early}" || test -n "${initrd_real}"; then
    initrd="${initrd_early} ${initrd_real}"

    initrd_display=
    for i in ${initrd}; do
      initrd_display="${initrd_display} ${dirname}/${i}"
    done
    gettext_printf "Found initrd image: %s\n" "$(echo $initrd_display)" >&2
  fi

  # Config file (only for real kernels)
  config=
  if [ "$kernel_type" = "real" ]; then
      for i in "${dirname}/config-${version}" "/etc/kernels/kernel-config-${version}"; do
          if [ -e "$i" ]; then
              config="$i"
              break
          fi
      done
  fi

  # Check for builtin initramfs
  initramfs=
  if [ -n "${config}" ]; then
      initramfs=$(grep CONFIG_INITRAMFS_SOURCE= "${config}" 2>/dev/null | cut -f2 -d= | tr -d \")
  fi

  # Fallback root device if no initrd/initramfs
  if [ -z "${initramfs}" ] && [ -z "${initrd_real}" ]; then
      if [ "x${GRUB_DEVICE_PARTUUID}" = "x" ] \
          || [ "x${GRUB_DISABLE_LINUX_PARTUUID}" = "xtrue" ]; then
          linux_root_device_thisversion=${GRUB_DEVICE}
      else
          linux_root_device_thisversion=PARTUUID=${GRUB_DEVICE_PARTUUID}
      fi
  fi

  # Normalize GRUB_DISABLE_SUBMENU
  if [ "x${GRUB_DISABLE_SUBMENU}" = xyes ] || [ "x${GRUB_DISABLE_SUBMENU}" = xy ]; then
      GRUB_DISABLE_SUBMENU="true"
  fi

  # Simple entry (first kernel only, creates top-level menu item)
  if [ "x$is_top_level" = xtrue ] && [ "x${GRUB_DISABLE_SUBMENU}" != xtrue ]; then
      linux_entry "${OS}" "${version}" simple \
          "${GRUB_CMDLINE_LINUX} ${GRUB_CMDLINE_LINUX_DEFAULT}"

      submenu_indentation="$grub_tab"

      if [ -z "$boot_device_id" ]; then
          boot_device_id="$(grub_get_device_id "${GRUB_DEVICE}")"
      fi
      # TRANSLATORS: %s is replaced with an OS name
      echo "submenu '$(gettext_printf "Advanced options for %s" "${OS}" | grub_quote)' \$menuentry_id_option 'gnulinux-advanced-$boot_device_id' {"
      is_top_level=false
  fi

  # Advanced entry (all kernels)
  linux_entry "${OS}" "${version}" advanced \
      "${GRUB_CMDLINE_LINUX} ${GRUB_CMDLINE_LINUX_DEFAULT}"

  # Recovery entry (only for real kernels)
  if [ "$kernel_type" = "real" ] && [ "x${GRUB_DISABLE_RECOVERY}" != "xtrue" ]; then
      linux_entry "${OS}" "${version}" recovery \
          "${GRUB_CMDLINE_LINUX_RECOVERY} ${GRUB_CMDLINE_LINUX}"
  fi
done

# If at least one kernel was found, then we need to
# add a closing '}' for the submenu command.
if [ x"$is_top_level" != xtrue ]; then
  echo '}'
fi

echo "$title_correction_code"
