From 891244fbdeaac9341f0f2c82298b012da9b46d7b Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Sat, 15 Feb 2025 03:01:14 +0100
Subject: [PATCH] feat: Make edition selectable

---
 src/install.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/install.sh b/src/install.sh
index e22b99e..2f4dbd2 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -668,6 +668,13 @@ updateXML() {
     sed -z "s/<AdministratorPassword>...............<Value \/>/<AdministratorPassword>\n              <Value>$admin<\/Value>/g" -i "$asset"
   fi
 
+  if [ -n "$EDITION" ]; then
+    sed -i "s/<Value>Windows Server 2025 SERVERSTANDARD<\/Value>/<Value>Windows Server 2025 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2022 SERVERSTANDARD<\/Value>/<Value>Windows Server 2022 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2019 SERVERSTANDARD<\/Value>/<Value>Windows Server 2019 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2016 SERVERSTANDARD<\/Value>/<Value>Windows Server 2016 SERVER${EDITION^^}<\/Value>/g" "$asset"
+  fi
+
   return 0
 }