1 files changed, 102 insertions, 0 deletions
diff --git a/src/DotNetOpenAuth.BuildTasks/CopyWithTokenSubstitution.cs b/src/DotNetOpenAuth.BuildTasks/CopyWithTokenSubstitution.cs
new file mode 100644
index 0000000..e17d8f2
--- /dev/null
+++ b/src/DotNetOpenAuth.BuildTasks/CopyWithTokenSubstitution.cs
@@ -0,0 +1,102 @@
+//-----------------------------------------------------------------------
+// <copyright file="CopyWithTokenSubstitution.cs" company="Andrew Arnott">
+//     Copyright (c) Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace DotNetOpenAuth.BuildTasks {
+	using System;
+	using System.Collections.Generic;
+	using System.IO;
+	using System.Linq;
+	using System.Text;
+	using Microsoft.Build.Framework;
+	using Microsoft.Build.Utilities;
+
+	/// <summary>
+	/// Copies files and performs a search and replace for given tokens in their contents during the process.
+	/// </summary>
+	public class CopyWithTokenSubstitution : Task {
+		/// <summary>
+		/// Gets or sets the files to copy.
+		/// </summary>
+		/// <value>The files to copy.</value>
+		[Required]
+		public ITaskItem[] SourceFiles { get; set; }
+
+		/// <summary>
+		/// Gets or sets a list of files to copy the source files to.
+		/// </summary>
+		/// <value>The list of files to copy the source files to.</value>
+		[Required]
+		public ITaskItem[] DestinationFiles { get; set; }
+
+		/// <summary>
+		/// Gets or sets the destination files actually copied to.
+		/// </summary>
+		/// <remarks>
+		/// In the case of error partway through, or files not copied due to already being up to date,
+		/// this can be a subset of the <see cref="DestinationFiles"/> array.
+		/// </remarks>
+		[Output]
+		public ITaskItem[] CopiedFiles { get; set; }
+
+		/// <summary>
+		/// Executes this instance.
+		/// </summary>
+		/// <returns><c>true</c> if the operation was successful.</returns>
+		public override bool Execute() {
+			if (this.SourceFiles.Length != this.DestinationFiles.Length) {
+				Log.LogError("{0} inputs and {1} outputs given.", this.SourceFiles.Length, this.DestinationFiles.Length);
+				return false;
+			}
+
+			var copiedFiles = new List<ITaskItem>(this.DestinationFiles.Length);
+
+			for (int i = 0; i < this.SourceFiles.Length; i++) {
+				string sourcePath = this.SourceFiles[i].ItemSpec;
+				string destPath = this.DestinationFiles[i].ItemSpec;
+				bool skipUnchangedFiles = bool.Parse(this.SourceFiles[i].GetMetadata("SkipUnchangedFiles"));
+
+				// We deliberably consider newer destination files to be up-to-date rather than
+				// requiring equality because this task modifies the destination file while copying.
+				if (skipUnchangedFiles && File.GetLastWriteTimeUtc(sourcePath) < File.GetLastWriteTimeUtc(destPath)) {
+					Log.LogMessage(MessageImportance.Low, "Skipping \"{0}\" -> \"{1}\" because the destination is up to date.", sourcePath, destPath);
+					continue;
+				}
+
+				Log.LogMessage(MessageImportance.Normal, "Transforming \"{0}\" -> \"{1}\"", sourcePath, destPath);
+
+				string[] beforeTokens = this.SourceFiles[i].GetMetadata("BeforeTokens").Split(';');
+				string[] afterTokens = this.SourceFiles[i].GetMetadata("AfterTokens").Split(';');
+				if (beforeTokens.Length != afterTokens.Length) {
+					Log.LogError("Unequal number of before and after tokens.  Before: \"{0}\". After \"{1}\".", beforeTokens, afterTokens);
+					return false;
+				}
+
+				using (StreamReader sr = File.OpenText(sourcePath)) {
+					if (!Directory.Exists(Path.GetDirectoryName(destPath))) {
+						Directory.CreateDirectory(Path.GetDirectoryName(destPath));
+					}
+					using (StreamWriter sw = File.CreateText(destPath)) {
+						StringBuilder line = new StringBuilder();
+						while (!sr.EndOfStream) {
+							line.Length = 0;
+							line.Append(sr.ReadLine());
+							for (int j = 0; j < beforeTokens.Length; j++) {
+								line.Replace(beforeTokens[j], afterTokens[j]);
+							}
+
+							sw.WriteLine(line);
+						}
+					}
+				}
+
+				copiedFiles.Add(this.DestinationFiles[i]);
+			}
+
+			this.CopiedFiles = copiedFiles.ToArray();
+			return true;
+		}
+	}
+}