From dee03e1359598678f639ed1d969bee6048238484 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Tue, 15 Jun 2021 08:41:33 -0400
Subject: [PATCH] Attempt to updateboot at end of pack

---
 imgutil/imgutil | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/imgutil/imgutil b/imgutil/imgutil
index 05b6e527..9af1d00c 100644
--- a/imgutil/imgutil
+++ b/imgutil/imgutil
@@ -343,6 +343,7 @@ def pack_image(opts, args):
     initrdname = os.path.join(args[0], 'boot/initramfs-{0}.img'.format(mostrecent))
     mkdirp(os.path.join(outdir, 'boot/efi/boot'))
     mkdirp(os.path.join(outdir, 'boot/initramfs'))
+    profname = os.path.basename(outdir)
     os.symlink(
         '/var/lib/confluent/public/site/initramfs.cpio',
         os.path.join(outdir, 'boot/initramfs/site.cpio'))
@@ -358,6 +359,7 @@ def pack_image(opts, args):
     subprocess.check_call(['mksquashfs', args[0],
                            os.path.join(outdir, 'rootimg.sfs'), '-comp', 'xz'])
     oshandler = fingerprint_host(args[0])
+    tryupdateboot = False
     if oshandler:
         prettyname = oshandler.osname
         with open(os.path.join(args[0], 'etc/os-release')) as osr:
@@ -375,9 +377,12 @@ def pack_image(opts, args):
             os.path.join(outdir, 'boot/initramfs/addons.cpio'))
         if os.path.exists('{}/profiles/default'.format(confdir)):
             copy_tree('{}/profiles/default'.format(confdir), outdir)
+            tryupdate = True
     try:
         pwd.getpwnam('confluent')
         subprocess.check_call(['chown', '-R', 'confluent', outdir])
+        if tryupdate:
+            subprocess.check_call(['osdeploy', 'updateboot', profname])
     except KeyError:
         pass