From f3bffa9952c6c995fb7dea0b4a44fccfcfdf882b Mon Sep 17 00:00:00 2001
From: MyDrift <personal@mdiana.ch>
Date: Wed, 6 Nov 2024 11:38:19 +0100
Subject: [PATCH] restore usage of EditionID

TODO: fetch EditionID foreach index without looping Get-WindowsImage
---
 functions/public/Invoke-WPFGetIso.ps1 | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

diff --git a/functions/public/Invoke-WPFGetIso.ps1 b/functions/public/Invoke-WPFGetIso.ps1
index e43312ad..dc706aa0 100644
--- a/functions/public/Invoke-WPFGetIso.ps1
+++ b/functions/public/Invoke-WPFGetIso.ps1
@@ -261,32 +261,21 @@ function Invoke-WPFGetIso {
         }
         $sync.MicrowinWindowsFlavors.Items.Clear()
 
-        # Store the results of Get-WindowsImage in a variable
         $images = Get-WindowsImage -ImagePath $wimFile
 
-        $proEditionIndex = -1
-        $proeditionfound = $false
-
         # Populate the list of Windows flavors and find the Pro edition
         $images | ForEach-Object {
             $imageIdx = $_.ImageIndex
             $imageName = $_.ImageName
+
             $sync.MicrowinWindowsFlavors.Items.Add("$imageIdx : $imageName")
 
-            # Check if the image name contains 'Pro'
-            if ($imageName -like "*Pro*" -and !$proeditionfound) {
-                $proEditionIndex = $imageIdx
-                $proeditionfound = $true
+            if ((Get-WindowsImage -ImagePath $wimFile -Index $_.ImageIndex).EditionId -eq "Professional") {
+                # We have found the Pro edition
+                $sync.MicrowinWindowsFlavors.SelectedIndex = $_.ImageIndex - 1
             }
         }
 
-        # Set the selected index to the Pro edition if found, otherwise default to the first item
-        if ($proEditionIndex -ne -1) {
-            $sync.MicrowinWindowsFlavors.SelectedIndex = $proEditionIndex - 1
-        } else {
-            $sync.MicrowinWindowsFlavors.SelectedIndex = 0
-        }
-
         Get-Volume $driveLetter | Get-DiskImage | Dismount-DiskImage
         Write-Host "Selected value '$($sync.MicrowinWindowsFlavors.SelectedValue)'....."