Merge branch 'v3.1' into v3.2origin/v3.2

1 files changed, 64 insertions, 0 deletions

diff --git a/src/DotNetOpenAuth.BuildTasks/FixupShippingToolSamples.cs b/src/DotNetOpenAuth.BuildTasks/FixupShippingToolSamples.cs
new file mode 100644
index 0000000..6c71740
--- /dev/null
+++ b/src/DotNetOpenAuth.BuildTasks/FixupShippingToolSamples.cs
@@ -0,0 +1,64 @@
+//-----------------------------------------------------------------------
+// <copyright file="FixupShippingToolSamples.cs" company="Andrew Arnott">
+//     Copyright (c) Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace DotNetOpenAuth.BuildTasks {
+	using System;
+	using System.Collections;
+	using System.Collections.Generic;
+	using System.IO;
+	using System.Linq;
+	using System.Text;
+	using Microsoft.Build.BuildEngine;
+	using Microsoft.Build.Framework;
+	using Microsoft.Build.Utilities;
+
+	/// <summary>
+	/// Removes imports that only apply when a shipping tool sample builds as part of
+	/// the entire project, but not when it's part of a source code sample.
+	/// </summary>
+	public class FixupShippingToolSamples : Task {
+		[Required]
+		public ITaskItem[] Projects { get; set; }
+
+		public string[] RemoveImportsStartingWith { get; set; }
+
+		public ITaskItem[] AddReferences { get; set; }
+
+		/// <summary>
+		/// Executes this instance.
+		/// </summary>
+		/// <returns></returns>
+		public override bool Execute() {
+			foreach (ITaskItem projectTaskItem in this.Projects) {
+				this.Log.LogMessage("Fixing up the {0} sample for shipping as source code.", Path.GetFileNameWithoutExtension(projectTaskItem.ItemSpec));
+
+				var project = new Project();
+				Uri projectUri = new Uri(projectTaskItem.GetMetadata("FullPath"));
+				project.Load(projectTaskItem.ItemSpec, ProjectLoadSettings.IgnoreMissingImports);
+
+				if (this.RemoveImportsStartingWith != null && this.RemoveImportsStartingWith.Length > 0) {
+					project.Imports.Cast<Import>()
+						.Where(import => this.RemoveImportsStartingWith.Any(start => import.ProjectPath.StartsWith(start, StringComparison.OrdinalIgnoreCase)))
+						.ToList()
+						.ForEach(import => project.Imports.RemoveImport(import));
+				}
+
+				if (this.AddReferences != null) {
+					foreach (var reference in this.AddReferences) {
+						BuildItem item = project.AddNewItem("Reference", reference.ItemSpec);
+						foreach (DictionaryEntry metadata in reference.CloneCustomMetadata()) {
+							item.SetMetadata((string)metadata.Key, (string)metadata.Value);
+						}
+					}
+				}
+
+				project.Save(projectTaskItem.ItemSpec);
+			}
+
+			return !this.Log.HasLoggedErrors;
+		}
+	}
+}