Do not generate clean block in proxy function when the feature is disabled (#17112)

daxian-dbw · web-flow · commit 7cc9c874730e · 2022-04-05T12:00:41.000-07:00
diff --git a/src/System.Management.Automation/engine/CommandMetadata.cs b/src/System.Management.Automation/engine/CommandMetadata.cs
@@ -864,6 +864,15 @@ internal string GetProxyCommand(string helpComment, bool generateDynamicParamete
 ", GetDynamicParamBlock());
             }
 
+            string cleanBlock = string.Empty;
+            if (ExperimentalFeature.IsEnabled(ExperimentalFeature.PSCleanBlockFeatureName))
+            {
+                cleanBlock = string.Format(CultureInfo.InvariantCulture, @"
+clean
+{{{0}}}
+", GetCleanBlock());
+            }
+
             string result = string.Format(CultureInfo.InvariantCulture, @"{0}
 param({1})
 
@@ -875,9 +884,7 @@ internal string GetProxyCommand(string helpComment, bool generateDynamicParamete
 
 end
 {{{5}}}
-
-clean
-{{{6}}}
+{6}
 <#
 {7}
 #>
@@ -888,7 +895,7 @@ internal string GetProxyCommand(string helpComment, bool generateDynamicParamete
                 GetBeginBlock(),
                 GetProcessBlock(),
                 GetEndBlock(),
-                GetCleanBlock(),
+                cleanBlock,
                 CodeGeneration.EscapeBlockCommentContent(helpComment));
 
             return result;
