#!/bin/bash 

bootpartition=$(mount | awk '/ \/boot / {print $1}')
if [ -n "${bootpartition}" ] && ! echo ${bootpartition} | grep -q "/dev/mapper/"; then
  # On a une partition de type /dev/sda2 /dev/nvme0p1 -> on récupère le device
  bootdevice=/dev/$(lsblk -ndo pkname $bootpartition)
else
  # On est sans doute sur du lvm
  bootdevice=${bootpartition}
fi
rootpartition=$(mount | awk '/ \/ / {print $1}')
if [ -n "${rootpartition}" ] && ! echo ${rootpartition} | grep -q "/dev/mapper/"; then
  # On a une partition de type /dev/sda2 /dev/nvme0p1 -> on récupère le device
  rootdevice=/dev/$(lsblk -ndo pkname $rootpartition)
else
  # On est sans doute sur du lvm
  rootdevice=${rootpartition}
fi

if [[ -n "${bootdevice}" ]]
then
  grub-install ${bootdevice}
  exit $?
elif [[ -n "${rootdevice}" ]]
then
  grub-install ${rootdevice}
  exit $?
else
  echo "Le disque d'installation de grub ne peut pas être déterminé."
  exit 127
fi
