Merge pull request #10 from ariasemis/fix-env-vars

Fix recorder so that it doesnt capture all environment variables
Watfaq · Oct 15, 2020 · 615903a · 615903a
2 parents b093005 + a4f549b
commit 615903a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/PowerSession.Main/Commands/RecordCommand.cs b/PowerSession.Main/Commands/RecordCommand.cs
@@ -2,6 +2,7 @@
 {
     using System;
     using System.Collections;
+    using System.Collections.Generic;
     using System.IO;
     using ConPTY;
     using Newtonsoft.Json;
@@ -67,7 +68,7 @@ private void _record(string filename, string command = null)
         {
             if (string.IsNullOrEmpty(command)) command = "powershell.exe";
 
-            if (_env == null) _env = Environment.GetEnvironmentVariables();
+            if (_env == null) _env = new Dictionary<string, string>();
             _env.Add("POWERSESSION_RECORDING", "1");
             _env.Add("SHELL", "powershell.exe");
             _env.Add("TERM", Environment.GetEnvironmentVariable("TERMINAL_EMULATOR") ?? "NotSure");
@@ -81,7 +82,7 @@ private void _record(string filename, string command = null)
                 Environment = _env
             };
             writer.SetHeader(headerInfo);
-            
+
             var terminal = new Terminal(writer.GetInputStream(), writer.GetWriteStream(), width: headerInfo.Width, height: headerInfo.Height);
             terminal.Record(command, _env);
             Console.WriteLine("Record Finished");