1 files changed, 41 insertions, 0 deletions

diff --git a/src/tools/heat/AssemblyHarvester.cs b/src/tools/heat/AssemblyHarvester.cs
new file mode 100644
index 00000000..2304d827
--- /dev/null
+++ b/src/tools/heat/AssemblyHarvester.cs
@@ -0,0 +1,41 @@
+// 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.Harvesters
+{
+    using System;
+    using System.Reflection;
+    using System.Runtime.InteropServices;
+    using Wix = WixToolset.Harvesters.Serialize;
+
+    /// <summary>
+    /// Harvest WiX authoring from an assembly file.
+    /// </summary>
+    public sealed class AssemblyHarvester
+    {
+        /// <summary>
+        /// Harvest the registry values written by RegisterAssembly.
+        /// </summary>
+        /// <param name="path">The file to harvest registry values from.</param>
+        /// <returns>The harvested registry values.</returns>
+        public Wix.RegistryValue[] HarvestRegistryValues(string path)
+        {
+#if NETCOREAPP
+            throw new PlatformNotSupportedException();
+#else
+            RegistrationServices regSvcs = new RegistrationServices();
+            Assembly assembly = Assembly.LoadFrom(path);
+
+            // must call this before overriding registry hives to prevent binding failures
+            // on exported types during RegisterAssembly
+            assembly.GetExportedTypes();
+
+            using (RegistryHarvester registryHarvester = new RegistryHarvester(true))
+            {
+                regSvcs.RegisterAssembly(assembly, AssemblyRegistrationFlags.SetCodeBase);
+
+                return registryHarvester.HarvestRegistry();
+            }
+#endif
+        }
+    }
+}