flash_instructions() {
  echo "# echo 0 > /sys/block/mmcblk2boot0/force_ro"
  echo "# dd if=/boot/u-boot/images/u-boot.imx of=/dev/mmcblk2boot0 bs=1k seek=1 conv=notrunc"
}

extlinux_warning() {
  echo "==> WARNING: Remember to set up your kernels in /boot/extlinux/extlinux.conf"
}

flash_uboot() {
  # For now we assume that users are flashing u-boot on the eMMC
  # u-boot.img can also be loaded through USB for the frist boot with imx_usb_loader
  echo "A new U-Boot version needs to be flashed."
  echo "Do you want to flash it on the eMMC now? [y|N]"
  read -r shouldwe
  if [[ $shouldwe =~ ^([yY][eE][sS]|[yY])$ ]]; then
    echo 0 > /sys/block/mmcblk2boot0/force_ro
    dd if=/boot/u-boot/images/u-boot.imx of=/dev/mmcblk2boot0 bs=1k seek=1 conv=notrunc
  else
    echo "You can do this later by running:"
    flash_instructions
  fi
  extlinux_warning
}

post_install() {
  flash_uboot
}

post_upgrade() {
  flash_uboot
}