From 67b3f6fc3093f27f22f671b060f2979641b73577 Mon Sep 17 00:00:00 2001
From: Martin Whitaker <foss@martin-whitaker.me.uk>
Date: Sat, 18 May 2019 17:54:48 +0100
Subject: [PATCH] Fix Mageia support.

Works cleanly with Mageia 7, which no longer uses isolinux. Mageia 6.1
can be booted using grub2, albeit only with a text menu, but syslinux
boot is still broken.
---
 scripts/grub.py            | 6 ++++--
 scripts/update_cfg_file.py | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/scripts/grub.py b/scripts/grub.py
index 01383d6..24ed205 100644
--- a/scripts/grub.py
+++ b/scripts/grub.py
@@ -55,7 +55,7 @@ def mbusb_update_grub_cfg():
             (loopback_cfg_list, lambda x: 'efi' in x.lower()),
             (loopback_cfg_list, lambda x: 'boot' in x.lower()),
             (grub_cfg_list, lambda x: 'efi' in x.lower()),
-            (grub_cfg_list, lambda x: 'boot' in x.lower()),
+            (grub_cfg_list, lambda x: 'boot' in x.lower() and 'efi' not in x.lower()),
             (loopback_cfg_list,
              lambda x: 'efi' not in x.lower() and 'boot' not in x.lower()),
             (grub_cfg_list,
@@ -66,7 +66,9 @@ def mbusb_update_grub_cfg():
             # We could 'break' here but will let the iteration continue
             # in order to lower the chance of keeping latent bugs.
 
-    if 0<len(candidates):
+    if config.distro == 'mageialive' and 1<len(candidates):
+        grub_cfg_path = candidates[1].replace('\\', '/')
+    elif 0<len(candidates):
         grub_cfg_path = candidates[0].replace('\\', '/')
     else :
         # No suitable grub configuration file is provided by distro.
diff --git a/scripts/update_cfg_file.py b/scripts/update_cfg_file.py
index 58ad3c0..22fcbab 100644
--- a/scripts/update_cfg_file.py
+++ b/scripts/update_cfg_file.py
@@ -313,7 +313,8 @@ def update_distro_cfg_files(iso_link, usb_disk, distro, persistence=0):
                                     'from=/multibootusb/' + iso_basename(iso_link) + '/' + iso_name(iso_link) + ' initrd=',
                                     string)
                 elif distro == "mageialive":
-                    string = re.sub(r'LABEL=\S*', 'LABEL=' + usb_label, string)
+                    string = re.sub(r'LABEL=\S*', 'UUID=' + usb_uuid + ' mgalive.basedir=/multibootusb/' + iso_basename(iso_link),
+                                    string)
                 elif distro == "solydx":
                     string = re.sub(r'live-media-path=', 'live-media-path=/multibootusb/' + iso_basename(iso_link),
                                     string)