From fa58560f71f6aa99821bef76afee0f87d96bef61 Mon Sep 17 00:00:00 2001
From: Andrew Webberley <44350096+Agwebberley@users.noreply.github.com>
Date: Sat, 8 Jun 2024 14:07:14 -0700
Subject: [PATCH] Update upgrade

---
 scripts/upgrade | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/scripts/upgrade b/scripts/upgrade
index 7aa861a..a944b23 100755
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -28,8 +28,13 @@ if [ "$upgrade_type" == "UPGRADE_APP" ]
 then
 	ynh_script_progression --message="Upgrading source files..." --weight=1
 
+	source_id_to_use="main"
+	if [[ "$release_cycle" == "edge" ]]
+	then
+	    source_id_to_use="edge"
+	fi
 	# Download, check integrity, uncompress and patch the source from app.src
-	ynh_setup_source --dest_dir="$install_dir" --keep="config.json"
+	ynh_setup_source --dest_dir="$install_dir" --keep="config.json" --source_id="$source_id_to_use"
 fi
 
 chown -R $app:www-data "$install_dir"