1 files changed, 62 insertions, 0 deletions

diff --git a/src/WixToolset.Core.Burn/Bundles/GetPackageFacadesCommand.cs b/src/WixToolset.Core.Burn/Bundles/GetPackageFacadesCommand.cs
new file mode 100644
index 00000000..7485758c
--- /dev/null
+++ b/src/WixToolset.Core.Burn/Bundles/GetPackageFacadesCommand.cs
@@ -0,0 +1,62 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Core.Burn.Bundles
+{
+    using System.Collections.Generic;
+    using WixToolset.Data;
+    using WixToolset.Data.Rows;
+
+    internal class GetPackageFacadesCommand
+    {
+        public Table PackageTable { private get; set; }
+
+        public Table ExePackageTable { private get; set; }
+
+        public Table MsiPackageTable { private get; set; }
+
+        public Table MspPackageTable { private get; set; }
+
+        public Table MsuPackageTable { private get; set; }
+
+        public IDictionary<string, PackageFacade> PackageFacades { get; private set; }
+
+        public void Execute()
+        {
+            RowDictionary<WixBundleExePackageRow> exePackages = new RowDictionary<WixBundleExePackageRow>(this.ExePackageTable);
+            RowDictionary<WixBundleMsiPackageRow> msiPackages = new RowDictionary<WixBundleMsiPackageRow>(this.MsiPackageTable);
+            RowDictionary<WixBundleMspPackageRow> mspPackages = new RowDictionary<WixBundleMspPackageRow>(this.MspPackageTable);
+            RowDictionary<WixBundleMsuPackageRow> msuPackages = new RowDictionary<WixBundleMsuPackageRow>(this.MsuPackageTable);
+
+            Dictionary<string, PackageFacade> facades = new Dictionary<string, PackageFacade>(this.PackageTable.Rows.Count);
+
+            foreach (WixBundlePackageRow package in this.PackageTable.Rows)
+            {
+                string id = package.WixChainItemId;
+                PackageFacade facade = null;
+
+                switch (package.Type)
+                {
+                    case WixBundlePackageType.Exe:
+                        facade = new PackageFacade(package, exePackages.Get(id));
+                        break;
+
+                    case WixBundlePackageType.Msi:
+                        facade = new PackageFacade(package, msiPackages.Get(id));
+                        break;
+
+                    case WixBundlePackageType.Msp:
+                        facade = new PackageFacade(package, mspPackages.Get(id));
+                        break;
+
+                    case WixBundlePackageType.Msu:
+                        facade = new PackageFacade(package, msuPackages.Get(id));
+                        break;
+                }
+
+                facades.Add(id, facade);
+            }
+
+            this.PackageFacades = facades;
+        }
+    }
+}