From db6233e620ec247dbdbd5c2bc2905cdb050ffc0d Mon Sep 17 00:00:00 2001
From: Yunus Emre Kalkan <emre.kalkan@volosoft.com>
Date: Wed, 26 Feb 2020 09:30:00 +0300
Subject: [PATCH] added ReplaceConfigureAwaitPropsStep

---
 .../ModuleProjectBuildPipelineBuilder.cs      |  1 +
 .../Steps/ReplaceConfigureAwaitPropsStep.cs   | 84 +++++++++++++++++++
 2 files changed, 85 insertions(+)
 create mode 100644 framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ReplaceConfigureAwaitPropsStep.cs

diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ModuleProjectBuildPipelineBuilder.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ModuleProjectBuildPipelineBuilder.cs
index 5eba01b388..9b66f17834 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ModuleProjectBuildPipelineBuilder.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ModuleProjectBuildPipelineBuilder.cs
@@ -11,6 +11,7 @@ namespace Volo.Abp.Cli.ProjectBuilding.Building
             pipeline.Steps.Add(new FileEntryListReadStep());
             pipeline.Steps.Add(new ProjectReferenceReplaceStep());
             pipeline.Steps.Add(new ReplaceCommonPropsStep());
+            pipeline.Steps.Add(new ReplaceConfigureAwaitPropsStep());
             pipeline.Steps.Add(new CreateProjectResultZipStep());
 
             return pipeline;
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ReplaceConfigureAwaitPropsStep.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ReplaceConfigureAwaitPropsStep.cs
new file mode 100644
index 0000000000..6d96a889bd
--- /dev/null
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/Steps/ReplaceConfigureAwaitPropsStep.cs
@@ -0,0 +1,84 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Text.RegularExpressions;
+using System.Xml;
+using Volo.Abp.Cli.ProjectBuilding.Files;
+
+namespace Volo.Abp.Cli.ProjectBuilding.Building.Steps
+{
+    public class ReplaceConfigureAwaitPropsStep : ProjectBuildPipelineStep
+    {
+        public override void Execute(ProjectBuildContext context)
+        {
+            new ConfigureAwaitPropsReplacer(context.Files).Run();
+        }
+
+        private class ConfigureAwaitPropsReplacer
+        {
+            private readonly List<FileEntry> _entries;
+
+            public ConfigureAwaitPropsReplacer(
+                List<FileEntry> entries)
+            {
+                _entries = entries;
+            }
+
+            public void Run()
+            {
+                foreach (var fileEntry in _entries)
+                {
+                    if (fileEntry.Name.EndsWith(".csproj"))
+                    {
+                        fileEntry.SetContent(ProcessFileContent(fileEntry.Content));
+                    }
+                }
+            }
+
+            private string ProcessFileContent(string content)
+            {
+                Check.NotNull(content, nameof(content));
+
+                var doc = new XmlDocument() { PreserveWhitespace = true };
+
+                doc.Load(GenerateStreamFromString(content));
+
+                return ProcessReferenceNodes(doc, content);
+            }
+
+            private string ProcessReferenceNodes(XmlDocument doc, string content)
+            {
+                Check.NotNull(content, nameof(content));
+
+                var importNodes = doc.SelectNodes("/Project/Import[@Project]");
+
+                if (importNodes == null)
+                {
+                    return doc.OuterXml;
+                }
+
+                foreach (XmlNode node in importNodes)
+                {
+                    if (!(node.Attributes?["Project"]?.Value?.EndsWith("\\configureawait.props") ?? false))
+                    {
+                        continue;
+                    }
+
+                    node.ParentNode?.RemoveChild(node);
+                }
+
+                return doc.OuterXml;
+            }
+
+            private static Stream GenerateStreamFromString(string s)
+            {
+                var stream = new MemoryStream();
+                var writer = new StreamWriter(stream);
+                writer.Write(s);
+                writer.Flush();
+                stream.Position = 0;
+                return stream;
+            }
+        }
+    }
+}