1 files changed, 90 insertions, 0 deletions

diff --git a/src/WixToolset.Core.WindowsInstaller/Bind/CreatePatchTransformsCommand.cs b/src/WixToolset.Core.WindowsInstaller/Bind/CreatePatchTransformsCommand.cs
new file mode 100644
index 00000000..854d973e
--- /dev/null
+++ b/src/WixToolset.Core.WindowsInstaller/Bind/CreatePatchTransformsCommand.cs
@@ -0,0 +1,90 @@
+// 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.WindowsInstaller.Bind
+{
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Linq;
+    using WixToolset.Core.WindowsInstaller.Msi;
+    using WixToolset.Core.WindowsInstaller.Unbind;
+    using WixToolset.Data;
+    using WixToolset.Data.Tuples;
+    using WixToolset.Data.WindowsInstaller;
+    using WixToolset.Extensibility.Services;
+
+    internal class CreatePatchTransformsCommand
+    {
+        public CreatePatchTransformsCommand(IMessaging messaging, Intermediate intermediate, string intermediateFolder)
+        {
+            this.Messaging = messaging;
+            this.Intermediate = intermediate;
+            this.IntermediateFolder = intermediateFolder;
+        }
+
+        private IMessaging Messaging { get; }
+
+        private Intermediate Intermediate { get; }
+
+        private string IntermediateFolder { get; }
+
+        public IEnumerable<PatchTransform> PatchTransforms { get; private set; }
+
+        public IEnumerable<PatchTransform> Execute()
+        {
+            var patchTransforms = new List<PatchTransform>();
+
+            var tuples = this.Intermediate.Sections.SelectMany(s => s.Tuples).OfType<WixPatchBaselineTuple>();
+
+            foreach (var tuple in tuples)
+            {
+                WindowsInstallerData transform;
+
+                if (tuple.TransformFile is null)
+                {
+                    var baselineData = this.GetData(tuple.BaselineFile.Path);
+                    var updateData = this.GetData(tuple.UpdateFile.Path);
+
+                    var command = new GenerateTransformCommand(this.Messaging, baselineData, updateData, false);
+                    transform = command.Execute();
+                }
+                else
+                {
+                    var exportBasePath = Path.Combine(this.IntermediateFolder, "_trans"); // TODO: come up with a better path.
+
+                    var command = new UnbindTransformCommand(this.Messaging, tuple.TransformFile.Path, exportBasePath, this.IntermediateFolder);
+                    transform = command.Execute();
+                }
+
+                patchTransforms.Add(new PatchTransform(tuple.Id.Id, transform));
+            }
+
+            this.PatchTransforms = patchTransforms;
+
+            return this.PatchTransforms;
+        }
+
+        private WindowsInstallerData GetData(string path)
+        {
+            var ext = Path.GetExtension(path);
+
+            if (".msi".Equals(ext, StringComparison.OrdinalIgnoreCase))
+            {
+                using (var database = new Database(path, OpenDatabase.ReadOnly))
+                {
+                    var exportBasePath = Path.Combine(this.IntermediateFolder, "_msi"); // TODO: come up with a better path.
+
+                    var isAdminImage = false; // TODO: need a better way to set this
+
+                    var command = new UnbindDatabaseCommand(this.Messaging, database, path, OutputType.Product, exportBasePath, this.IntermediateFolder, isAdminImage, suppressDemodularization: true, skipSummaryInfo: true);
+                    return command.Execute();
+                }
+            }
+            else // assume .wixpdb (or .wixout)
+            {
+                var data = WindowsInstallerData.Load(path, true);
+                return data;
+            }
+        }
+    }
+}