Add support for tracking files to simplify file transfers

7 files changed, 73 insertions, 25 deletions

diff --git a/src/WixToolset.Extensibility/Data/BindResult.cs b/src/WixToolset.Extensibility/Data/BindResult.cs
index 0ccaa08b..e467d269 100644
--- a/src/WixToolset.Extensibility/Data/BindResult.cs
+++ b/src/WixToolset.Extensibility/Data/BindResult.cs
@@ -8,6 +8,6 @@ namespace WixToolset.Extensibility.Data
     {
         public IEnumerable<IFileTransfer> FileTransfers { get; set; }
 
-        public IEnumerable<string> ContentFilePaths { get; set; }
+        public IEnumerable<ITrackedFile> TrackedFiles { get; set; }
     }
 }
diff --git a/src/WixToolset.Extensibility/Data/FileTransferType.cs b/src/WixToolset.Extensibility/Data/FileTransferType.cs
deleted file mode 100644
index b00a00c4..00000000
--- a/src/WixToolset.Extensibility/Data/FileTransferType.cs
+++ /dev/null
@@ -1,17 +0,0 @@
-// 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.Extensibility.Data
-{
-    public enum FileTransferType
-    {
-        /// <summary>
-        /// Transfer of a file built during this build.
-        /// </summary>
-        Built,
-
-        /// <summary>
-        /// Transfer of a file contained in the output.
-        /// </summary>
-        Content,
-    }
-}
diff --git a/src/WixToolset.Extensibility/Data/IFileTransfer.cs b/src/WixToolset.Extensibility/Data/IFileTransfer.cs
index ca936219..6f521536 100644
--- a/src/WixToolset.Extensibility/Data/IFileTransfer.cs
+++ b/src/WixToolset.Extensibility/Data/IFileTransfer.cs
@@ -23,8 +23,5 @@ namespace WixToolset.Extensibility.Data
 
         /// <summary>Optional source line numbers where this file transfer orginated.</summary>
         SourceLineNumber SourceLineNumbers { get; set; }
-
-        /// <summary>Type of file this transfer is moving or copying.</summary>
-        FileTransferType Type { get; set; }
     }
 }
diff --git a/src/WixToolset.Extensibility/Data/ILayoutContext.cs b/src/WixToolset.Extensibility/Data/ILayoutContext.cs
index 7011fcf6..89aadc4d 100644
--- a/src/WixToolset.Extensibility/Data/ILayoutContext.cs
+++ b/src/WixToolset.Extensibility/Data/ILayoutContext.cs
@@ -11,7 +11,7 @@ namespace WixToolset.Extensibility.Data
 
         IEnumerable<ILayoutExtension> Extensions { get; set; }
 
-        IEnumerable<string> ContentFilePaths { get; set; }
+        IEnumerable<ITrackedFile> TrackedFiles { get; set; }
 
         IEnumerable<IFileTransfer> FileTransfers { get; set; }
 
diff --git a/src/WixToolset.Extensibility/Data/ITrackedFile.cs b/src/WixToolset.Extensibility/Data/ITrackedFile.cs
new file mode 100644
index 00000000..df36bd2b
--- /dev/null
+++ b/src/WixToolset.Extensibility/Data/ITrackedFile.cs
@@ -0,0 +1,32 @@
+// 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.Extensibility.Data
+{
+    using WixToolset.Data;
+
+    /// <summary>
+    /// Interface used to track all files processed.
+    /// </summary>
+    public interface ITrackedFile
+    {
+        /// <summary>
+        /// Indicates whether the tracked file should be cleaned by the project.
+        /// </summary>
+        bool Clean { get; set; }
+
+        /// <summary>
+        /// Path to tracked file.
+        /// </summary>
+        string Path { get; set; }
+
+        /// <summary>
+        /// Optional source line numbers where the tracked file was created.
+        /// </summary>
+        SourceLineNumber SourceLineNumbers { get; set; }
+
+        /// <summary>
+        /// Type of tracked file.
+        /// </summary>
+        TrackedFileType Type { get; set; }
+    }
+}
diff --git a/src/WixToolset.Extensibility/Data/TrackedFileType.cs b/src/WixToolset.Extensibility/Data/TrackedFileType.cs
new file mode 100644
index 00000000..195d5de9
--- /dev/null
+++ b/src/WixToolset.Extensibility/Data/TrackedFileType.cs
@@ -0,0 +1,30 @@
+// 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.Extensibility.Data
+{
+    public enum TrackedFileType
+    {
+        /// <summary>
+        /// File tracked as input (like content included in an .msi).
+        /// </summary>
+        Input,
+
+        /// <summary>
+        /// Temporary file (like an .idt or any other temporary file).
+        /// These are to be deleted before the build completes.
+        /// </summary>
+        Temporary,
+
+        /// <summary>
+        /// Intermediate file (like a .cab in the cabcache).
+        /// These are left for subsequent builds.
+        /// </summary>
+        Intermediate,
+
+        /// <summary>
+        /// Final output (like a .msi, .cab or .wixpdb).
+        /// These are the whole point of the build process.
+        /// </summary>
+        Final,
+    }
+}
diff --git a/src/WixToolset.Extensibility/Services/IBackendHelper.cs b/src/WixToolset.Extensibility/Services/IBackendHelper.cs
index 813b40ef..0622693d 100644
--- a/src/WixToolset.Extensibility/Services/IBackendHelper.cs
+++ b/src/WixToolset.Extensibility/Services/IBackendHelper.cs
@@ -17,9 +17,7 @@ namespace WixToolset.Extensibility.Services
         /// <param name="source">Source for the file transfer.</param>
         /// <param name="destination">Destiation for the file transfer.</param>
         /// <param name="move">Indicates whether to move or copy the source file.</param>
-        /// <param name="type">Type of file transfer to create.</param>
-        /// <param name="sourceLineNumbers">Optional source line numbers that requested the file transfer.</param>
-        IFileTransfer CreateFileTransfer(string source, string destination, bool move, FileTransferType type, SourceLineNumber sourceLineNumbers = null);
+        IFileTransfer CreateFileTransfer(string source, string destination, bool move, SourceLineNumber sourceLineNumbers = null);
 
         /// <summary>
         /// Creates a version 3 name-based UUID.
@@ -28,5 +26,13 @@ namespace WixToolset.Extensibility.Services
         /// <param name="value">The value.</param>
         /// <returns>The generated GUID for the given namespace and value.</returns>
         string CreateGuid(Guid namespaceGuid, string value);
+
+        /// <summary>
+        /// Creates a tracked file.
+        /// </summary>
+        /// <param name="path">Destination path for the build output.</param>
+        /// <param name="type">Type of tracked file to create.</param>
+        /// <param name="sourceLineNumbers">Optional source line numbers that requested the tracked file.</param>
+        ITrackedFile TrackFile(string path, TrackedFileType type, SourceLineNumber sourceLineNumbers = null);
     }
 }