Refactor GetPackage

anypackage · Dec 10, 2024 · 75876ad · 75876ad
1 parent 9a31bb9
commit 75876ad
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/code/MsiProvider.cs b/src/code/MsiProvider.cs
@@ -28,6 +28,14 @@ public void FindPackage(PackageRequest request)
     }
 
     public void GetPackage(PackageRequest request)
+    {
+        foreach (var package in GetPackageImpl(request))
+        {
+            request.WritePackage(package);
+        }
+    }
+
+    private IEnumerable<PackageInfo> GetPackageImpl(PackageRequest request)
     {
         using var powershell = PowerShell.Create(RunspaceMode.CurrentRunspace);
         powershell.AddCommand("Get-Package")
@@ -62,15 +70,13 @@ public void GetPackage(PackageRequest request)
                 source = null;
             }
 
-            var package = new PackageInfo(result.Name,
+            yield return new PackageInfo(result.Name,
                                           result.Version,
                                           source,
                                           result.Description,
                                           dependencies: null,
                                           result.Metadata.ToDictionary(x => x.Key, x => x.Value),
                                           ProviderInfo);
-
-            request.WritePackage(package);
         }
     }