---
- name: Configure Bootloader
  vars:
    _efi_vendor: >-
      {{
        "redhat" if os == "rhel"
        else ("ubuntu" if os in ["ubuntu", "ubuntu-lts"] else os)
      }}
    _efi_loader: "{{ _configuration_platform.efi_loader }}"
  block:
    - name: Install GRUB EFI binary
      when: _configuration_platform.grub_install
      ansible.builtin.command: >-
        {{ chroot_command }} /usr/sbin/grub-install --target=x86_64-efi
        --efi-directory={{ partitioning_efi_mountpoint }}
        --bootloader-id={{ _efi_vendor }}
        --no-nvram
      register: configuration_bootloader_result
      changed_when: configuration_bootloader_result.rc == 0

    - name: Check existing EFI boot entries
      ansible.builtin.command: efibootmgr
      register: configuration_efi_entries
      changed_when: false

    - name: Ensure EFI boot entry exists
      when: ('* ' + _efi_vendor) not in configuration_efi_entries.stdout
      ansible.builtin.command: >-
        efibootmgr -c
        -L '{{ _efi_vendor }}'
        -d '{{ install_drive }}'
        -p 1
        -l '\EFI\{{ _efi_vendor }}\{{ _efi_loader }}'
      register: configuration_efi_entry_result
      changed_when: configuration_efi_entry_result.rc == 0

    - name: Set installed OS as first EFI boot entry
      ansible.builtin.shell:
        cmd: >-
          set -o pipefail &&
          efibootmgr | grep -i '{{ _efi_vendor }}' | grep -oP 'Boot\K[0-9A-F]+' | head -1
          | xargs -I{} efibootmgr -o {}
        executable: /bin/bash
      register: _efi_bootorder_result
      changed_when: _efi_bootorder_result.rc == 0

    - name: Ensure lvm2 for non btrfs filesystems
      when: os == "archlinux" and system_cfg.filesystem != "btrfs"
      ansible.builtin.lineinfile:
        path: /mnt/etc/mkinitcpio.conf
        regexp: "^(HOOKS=.*block)(?!.*lvm2)(.*)"
        line: "\\1 lvm2\\2"
        backrefs: true

    - name: Regenerate initramfs
      when: _configuration_platform.initramfs_cmd | length > 0
      ansible.builtin.command: "{{ chroot_command }} {{ _configuration_platform.initramfs_cmd }}"
      register: configuration_initramfs_result
      changed_when: configuration_initramfs_result.rc == 0

    - name: Generate grub config (RedHat)
      when: os_family == 'RedHat'
      ansible.builtin.command: >-
        {{ chroot_command }} /usr/sbin/{{ _configuration_platform.grub_mkconfig_prefix }}
        -o /boot/grub2/grub.cfg
      register: configuration_grub_result
      changed_when: configuration_grub_result.rc == 0

    - name: Fix btrfs BLS boot variable in grub config
      when:
        - os_family == 'RedHat'
        - system_cfg.filesystem == 'btrfs'
      ansible.builtin.replace:
        path: /mnt/boot/grub2/grub.cfg
        regexp: 'search --no-floppy --fs-uuid --set=boot \S+'
        replace: 'set boot=$root'

    - name: Create EFI grub.cfg wrapper for RedHat
      when: os_family == 'RedHat'
      vars:
        _grub2_path: >-
          {{
            '/grub2'
            if (partitioning_separate_boot | bool)
            else ('/@/boot/grub2' if system_cfg.filesystem == 'btrfs' else '/boot/grub2')
          }}
      ansible.builtin.shell:
        cmd: |
          set -o pipefail
          uuid=$(grep -m1 'search.*--set=root' /mnt/boot/grub2/grub.cfg | grep -oP '[\da-f]{8}(-[\da-f]{4}){3}-[\da-f]{12}')
          cat > /mnt{{ partitioning_efi_mountpoint }}/EFI/{{ _efi_vendor }}/grub.cfg <<GRUBEOF
          search --no-floppy --fs-uuid --set=dev $uuid
          set prefix=(\$dev){{ _grub2_path }}
          export \$prefix
          configfile \$prefix/grub.cfg
          GRUBEOF
        executable: /bin/bash
      register: _grub_wrapper_result
      changed_when: _grub_wrapper_result.rc == 0

    - name: Generate grub config (non-RedHat)
      when: os_family != 'RedHat'
      ansible.builtin.command: "{{ chroot_command }} /usr/sbin/grub-mkconfig -o /boot/grub/grub.cfg"
      register: configuration_grub_result
      changed_when: configuration_grub_result.rc == 0