1 files changed, 71 insertions, 0 deletions

diff --git a/src/wix/WixToolset.BuildTasks/ReattachSignedBundleEngine.cs b/src/wix/WixToolset.BuildTasks/ReattachSignedBundleEngine.cs
new file mode 100644
index 00000000..c07c12b9
--- /dev/null
+++ b/src/wix/WixToolset.BuildTasks/ReattachSignedBundleEngine.cs
@@ -0,0 +1,71 @@
+// 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.BuildTasks
+{
+    using Microsoft.Build.Framework;
+
+    /// <summary>
+    /// An MSBuild task to run WiX to reattach a (signed) bundle engine to its bundle.
+    /// </summary>
+    public sealed partial class ReattachSignedBundleEngine : WixExeBaseTask
+    {
+        /// <summary>
+        /// The bundle to which to attach the bundle engine.
+        /// </summary>
+        [Required]
+        public ITaskItem BundleFile { get; set; }
+
+        /// <summary>
+        /// The bundle engine file to reattach to the bundle.
+        /// </summary>
+        [Required]
+        public ITaskItem BundleEngineFile { get; set; }
+        
+        /// <summary>
+        /// Gets or sets the intermedidate folder to use.
+        /// </summary>
+        public ITaskItem IntermediateDirectory { get; set; }
+
+        /// <summary>
+        /// Gets or sets the path to the output detached bundle.
+        /// </summary>
+        [Required]
+        public ITaskItem OutputFile { get; set; }
+
+        /// <summary>
+        /// Gets or sets the output. Only set if the task does work.
+        /// </summary>
+        [Output]
+        public ITaskItem Output { get; set; }
+
+        protected override void BuildCommandLine(WixCommandLineBuilder commandLineBuilder)
+        {
+            commandLineBuilder.AppendTextUnquoted("burn reattach");
+
+            commandLineBuilder.AppendFileNameIfNotNull(this.BundleFile);
+            commandLineBuilder.AppendSwitchIfNotNull("-engine ", this.BundleEngineFile);
+            commandLineBuilder.AppendSwitchIfNotNull("-out ", this.OutputFile);
+            commandLineBuilder.AppendSwitchIfNotNull("-intermediatefolder ", this.IntermediateDirectory);
+
+            base.BuildCommandLine(commandLineBuilder);
+
+            commandLineBuilder.AppendTextIfNotWhitespace(this.AdditionalOptions);
+        }
+
+        protected override int ExecuteTool(string pathToTool, string responseFileCommands, string commandLineCommands)
+        {
+            var exitCode = base.ExecuteTool(pathToTool, responseFileCommands, commandLineCommands);
+
+            if (exitCode == 0) // successfully did work.
+            {
+                this.Output = this.OutputFile;
+            }
+            else if (exitCode == -1000) // no work done.
+            {
+                exitCode = 0;
+            }
+
+            return exitCode;
+        }
+    }
+}