obfuscate pwd in sql and kql kernel display events

dotnet · Feb 29, 2024 · b6defd4 · b6defd4
1 parent f7dc954
commit b6defd4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/Microsoft.DotNet.Interactive.SqlServer/ToolsServiceKernel.cs b/src/Microsoft.DotNet.Interactive.SqlServer/ToolsServiceKernel.cs
@@ -5,6 +5,7 @@
 using System.Collections.Generic;
 using System.CommandLine.Parsing;
 using System.ComponentModel;
+using System.Diagnostics;
 using System.IO;
 using System.Linq;
 using System.Text;
@@ -399,7 +400,15 @@ private string PrependVariableDeclarationsToCode(SubmitCode command, KernelInvoc
         foreach (var variableNameAndValue in _variables)
         {
             var declareStatement = CreateVariableDeclaration(variableNameAndValue.Key, variableNameAndValue.Value);
-            context.Display($"Adding shared variable declaration statement : {declareStatement}");
+
+            var displayStatement = declareStatement;
+
+            if (variableNameAndValue.Value is PasswordString pwd)
+            {
+                displayStatement = displayStatement.Replace(pwd.GetClearTextPassword(), pwd.ToString());
+            }
+
+            context.Display($"Adding shared variable declaration statement : {displayStatement}");
             sb.AppendLine(declareStatement);
         }