From c36d07591c3a8b9412f7ec3eb3081fe6e2ea9e96 Mon Sep 17 00:00:00 2001
From: liangshiwei <deawback@hotmail.com>
Date: Mon, 3 Jul 2023 14:13:19 +0800
Subject: [PATCH] Add ConnectionStringRenameStep

---
 .../Building/ConnectionStringRenameStep.cs    | 21 +++++++++++++++++++
 .../Templates/App/AppTemplateBase.cs          |  5 +++++
 .../MicroserviceServiceTemplateBase.cs        |  8 ++++++-
 .../Templates/Module/ModuleTemplateBase.cs    |  7 ++++++-
 4 files changed, 39 insertions(+), 2 deletions(-)
 create mode 100644 framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ConnectionStringRenameStep.cs

diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ConnectionStringRenameStep.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ConnectionStringRenameStep.cs
new file mode 100644
index 0000000000..ea71eb1c51
--- /dev/null
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Building/ConnectionStringRenameStep.cs
@@ -0,0 +1,21 @@
+using System;
+using System.Linq;
+using Volo.Abp.Cli.ProjectBuilding.Files;
+
+namespace Volo.Abp.Cli.ProjectBuilding.Building;
+
+public class ConnectionStringRenameStep : ProjectBuildPipelineStep
+{
+    public override void Execute(ProjectBuildContext context)
+    {
+        foreach (var fileEntry in context.Files.Where(file => file.Name.EndsWith(CliConsts.AppSettingsJsonFileName, StringComparison.OrdinalIgnoreCase)))
+        {
+            RenameDatabaseName(fileEntry);
+        }
+    }
+
+    private void RenameDatabaseName(FileEntry fileEntry)
+    {
+        fileEntry.SetContent(fileEntry.Content.Replace("MyProjectNamePro", "MyProjectName"));
+    }
+}
\ No newline at end of file
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/App/AppTemplateBase.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/App/AppTemplateBase.cs
index 8777291c24..5bd0614206 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/App/AppTemplateBase.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/App/AppTemplateBase.cs
@@ -645,6 +645,11 @@ public abstract class AppTemplateBase : TemplateInfo
         {
             steps.Add(new ConnectionStringChangeStep());
         }
+
+        if (IsPro())
+        {
+            steps.Add(new ConnectionStringRenameStep());
+        }
     }
 
     protected void CleanupFolderHierarchy(ProjectBuildContext context, List<ProjectBuildPipelineStep> steps)
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Microservice/MicroserviceServiceTemplateBase.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Microservice/MicroserviceServiceTemplateBase.cs
index 3e396e8113..13674458ee 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Microservice/MicroserviceServiceTemplateBase.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Microservice/MicroserviceServiceTemplateBase.cs
@@ -39,7 +39,8 @@ public abstract class MicroserviceServiceTemplateBase : TemplateInfo
         SetRandomPortForHostProject(context, steps);
         RandomizeStringEncryption(context, steps);
         RandomizeAuthServerPassPhrase(context, steps);
-
+        ChangeConnectionString(context, steps);
+        
         return steps;
     }
 
@@ -75,4 +76,9 @@ public abstract class MicroserviceServiceTemplateBase : TemplateInfo
     {
         steps.Add(new RandomizeAuthServerPassPhraseStep());
     }
+
+    private static void ChangeConnectionString(ProjectBuildContext context, List<ProjectBuildPipelineStep> steps)
+    {
+        steps.Add(new ConnectionStringRenameStep());
+    }
 }
diff --git a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Module/ModuleTemplateBase.cs b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Module/ModuleTemplateBase.cs
index 5fc263afc3..5229bb167f 100644
--- a/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Module/ModuleTemplateBase.cs
+++ b/framework/src/Volo.Abp.Cli.Core/Volo/Abp/Cli/ProjectBuilding/Templates/Module/ModuleTemplateBase.cs
@@ -102,12 +102,17 @@ public abstract class ModuleTemplateBase : TemplateInfo
         steps.Add(new UpdateNuGetConfigStep("/NuGet.Config"));
     }
 
-    private static void ChangeConnectionString(ProjectBuildContext context, List<ProjectBuildPipelineStep> steps)
+    private void ChangeConnectionString(ProjectBuildContext context, List<ProjectBuildPipelineStep> steps)
     {
         if (context.BuildArgs.ConnectionString != null)
         {
             steps.Add(new ConnectionStringChangeStep());
         }
+
+        if (IsPro())
+        {
+            steps.Add(new ConnectionStringRenameStep());
+        }
     }
 
     private void CleanupFolderHierarchy(ProjectBuildContext context, List<ProjectBuildPipelineStep> steps)