Add new properties in the IOutputSanitizer (#425)

Azure · Jul 23, 2024 · ccb149a · ccb149a
1 parent c4f1529
commit ccb149a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/Common/AzurePSCmdlet.cs b/src/Common/AzurePSCmdlet.cs
@@ -548,7 +548,9 @@ protected void WriteSurvey()
 
         private void SanitizeOutput(object sendToPipeline)
         {
-            if (OutputSanitizer?.RequireSecretsDetection == true)
+            if (OutputSanitizer != null && OutputSanitizer.RequireSecretsDetection
+                && !OutputSanitizer.IgnoredModules.Contains(MyInvocation?.MyCommand?.ModuleName)
+                && !OutputSanitizer.IgnoredCmdlets.Contains(MyInvocation?.MyCommand?.Name))
             {
                 OutputSanitizer.Sanitize(sendToPipeline, out var telemetry);
                 _qosEvent?.SanitizerInfo.Combine(telemetry);

diff --git a/src/Common/Sanitizer/IOutputSanitizer.cs b/src/Common/Sanitizer/IOutputSanitizer.cs
@@ -12,12 +12,18 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System.Collections.Generic;
+
 namespace Microsoft.WindowsAzure.Commands.Common.Sanitizer
 {
     public interface IOutputSanitizer
     {
         bool RequireSecretsDetection { get; }
 
+        IEnumerable<string> IgnoredModules { get; }
+
+        IEnumerable<string> IgnoredCmdlets { get; }
+
         void Sanitize(object sanitizingObject, out SanitizerTelemetry telemetryData);
     }
 }