From 93d3a9e1f9cdc05456a612e238a83738cfd1750f Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Sun, 10 May 2020 19:31:53 +1000
Subject: Track files in BindBundleCommand.

---
 src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs')

diff --git a/src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs b/src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs
index 683c3c50..bb9ab844 100644
--- a/src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs
+++ b/src/WixToolset.Core/Bind/ExtractEmbeddedFilesCommand.cs
@@ -7,18 +7,25 @@ namespace WixToolset.Core.Bind
     using System.Linq;
     using WixToolset.Data;
     using WixToolset.Extensibility.Data;
+    using WixToolset.Extensibility.Services;
 
     public class ExtractEmbeddedFilesCommand
     {
-        public ExtractEmbeddedFilesCommand(IEnumerable<IExpectedExtractFile> embeddedFiles)
+        public ExtractEmbeddedFilesCommand(IBackendHelper backendHelper, IEnumerable<IExpectedExtractFile> embeddedFiles)
         {
+            this.BackendHelper = backendHelper;
             this.FilesWithEmbeddedFiles = embeddedFiles;
         }
 
+        public IEnumerable<ITrackedFile> TrackedFiles { get; private set; }
+
+        private IBackendHelper BackendHelper { get; }
+
         private IEnumerable<IExpectedExtractFile> FilesWithEmbeddedFiles { get; }
 
         public void Execute()
         {
+            var trackedFiles = new List<ITrackedFile>();
             var group = this.FilesWithEmbeddedFiles.GroupBy(e => e.Uri);
 
             foreach (var expectedEmbeddedFileByUri in group)
@@ -34,10 +41,13 @@ namespace WixToolset.Core.Bind
                         if (uniqueIds.Add(embeddedFile.EmbeddedFileId))
                         {
                             wixout.ExtractEmbeddedFile(embeddedFile.EmbeddedFileId, embeddedFile.OutputPath);
+                            trackedFiles.Add(this.BackendHelper.TrackFile(embeddedFile.OutputPath, TrackedFileType.Temporary));
                         }
                     }
                 }
             }
+
+            this.TrackedFiles = trackedFiles;
         }
     }
 }
-- 
cgit v1.2.3-55-g6feb