Update SessionExtensions.cs

nefarius · Jul 7, 2024 · 7412042 · 7412042
1 parent 7c8a0c6
commit 7412042
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/setup/Util/SessionExtensions.cs b/setup/Util/SessionExtensions.cs
@@ -1,4 +1,5 @@
-using System;
+#nullable enable
+using System;
 using System.Linq;
 
 using Microsoft.Deployment.WindowsInstaller;
@@ -9,8 +10,13 @@ public static class SessionExtensions
 {
     public static bool IsFeatureEnabledPartial(this Session session, string partialName)
     {
-        FeatureInfo featureInfo =
-            session.Features.Single(f => f.Name.Contains(partialName, StringComparison.OrdinalIgnoreCase));
+        FeatureInfo? featureInfo =
+            session.Features.SingleOrDefault(f => f.Name.Contains(partialName, StringComparison.OrdinalIgnoreCase));
+
+        if (featureInfo is null)
+        {
+            return false;
+        }
 
         return featureInfo.RequestState != InstallState.Unknown;
     }