From faeaa6cea6973ffc435e4dc63aebc6aa4d89c9cf Mon Sep 17 00:00:00 2001
From: Tyler <71953103+fntyler@users.noreply.github.com>
Date: Tue, 2 Jun 2026 10:33:45 -0400
Subject: [PATCH] BRE-1893 fix(marketplace): preserve 001_onboot on cloud-init
 clean (#519)

---
 CommonMarketplace/scripts/90-cleanup.sh | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/CommonMarketplace/scripts/90-cleanup.sh b/CommonMarketplace/scripts/90-cleanup.sh
index 975afac..2be7471 100644
--- a/CommonMarketplace/scripts/90-cleanup.sh
+++ b/CommonMarketplace/scripts/90-cleanup.sh
@@ -78,11 +78,18 @@ done
 
 find /var/log -mtime -1 -type f -exec truncate -s 0 {} \;
 rm -rf /var/log/*.gz /var/log/*.[0-9] /var/log/*-????????
-# Reset cloud-init state for redeploy. `rm -rf /var/lib/cloud/instances/*` alone
-# leaves the /var/lib/cloud/instance symlink dangling, which causes cloud-init
-# to enter `degraded done` on the deployed VM's first boot
 if command -v cloud-init >/dev/null 2>&1; then
+  ONBOOT_SCRIPT=/var/lib/cloud/scripts/per-instance/001_onboot
+  if [ -f "$ONBOOT_SCRIPT" ]; then
+    cp -a "$ONBOOT_SCRIPT" /root/001_onboot.bak
+  fi
   cloud-init clean --logs --machine-id
+  if [ -f /root/001_onboot.bak ]; then
+    mkdir -p /var/lib/cloud/scripts/per-instance
+    mv /root/001_onboot.bak "$ONBOOT_SCRIPT"
+    chown root:root "$ONBOOT_SCRIPT"
+    chmod +x "$ONBOOT_SCRIPT"
+  fi
 fi
 rm -rf /var/lib/cloud/instance /var/lib/cloud/instances/* /var/lib/cloud/data/* /var/lib/cloud/sem/*
 rm -f /root/.ssh/authorized_keys /home/ubuntu/.ssh/authorized_keys