From 875833fc80f40e5305c6cd0818ac1f5af461c8d8 Mon Sep 17 00:00:00 2001
From: Dimitris Karakasilis <jimmykarily@gmail.com>
Date: Wed, 2 Oct 2024 16:26:32 +0300
Subject: [PATCH] Implement `--recovery` flag in suc-upgrade script (#1087)

To allow users to upgrade recovery from kubernetes

Part of: https://github.com/kairos-io/kairos/issues/2760

Signed-off-by: Dimitris Karakasilis <dimitris@karakasilis.me>
---
 packages/system/suc-upgrade/definition.yaml |  2 +-
 packages/system/suc-upgrade/suc-upgrade.sh  | 14 +++++++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/packages/system/suc-upgrade/definition.yaml b/packages/system/suc-upgrade/definition.yaml
index ccb9afd..9b411f5 100644
--- a/packages/system/suc-upgrade/definition.yaml
+++ b/packages/system/suc-upgrade/definition.yaml
@@ -1,3 +1,3 @@
 name: "suc-upgrade"
 category: "system"
-version: "0.2.1"
+version: "0.2.2"
diff --git a/packages/system/suc-upgrade/suc-upgrade.sh b/packages/system/suc-upgrade/suc-upgrade.sh
index 474280e..56baedf 100644
--- a/packages/system/suc-upgrade/suc-upgrade.sh
+++ b/packages/system/suc-upgrade/suc-upgrade.sh
@@ -12,6 +12,18 @@ fi
 
 mount --rbind $HOST_DIR/dev /dev
 mount --rbind $HOST_DIR/run /run
-kairos-agent upgrade --source dir:/
+
+recovery_mode=false
+while [[ "$#" -gt 0 ]]; do
+    case $1 in
+        --recovery) recovery_mode=true;;
+    esac
+    shift
+done
+if [ "$recovery_mode" = true ]; then
+    kairos-agent upgrade --recovery --source dir:/
+else
+    kairos-agent upgrade --source dir:/
+fi
 nsenter -i -m -t 1 -- reboot
 exit 1