Output correct commands for each terminal

Will output variable commands for PowerShell when the program is run from PowerShell, will output Batch script syntax otherwise.
RaeesBhatti · Oct 23, 2016 · 94e708e · 94e708e
1 parent 689dc93
commit 94e708e
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/Program.cs b/Program.cs
@@ -115,6 +115,9 @@ static void runAgent()
                 try
                 {
                     SSHAgent.Start();
+
+                    Process parent = FindParent.ParentProcess(Process.GetCurrentProcess());
+
                     while (!SSHAgent.StandardOutput.EndOfStream)
                     {
                         var line = SSHAgent.StandardOutput.ReadLine();
@@ -126,15 +129,20 @@ static void runAgent()
                         {
                             AgentSock = command[1];
                             Environment.SetEnvironmentVariable(SSH_AUTH_SOCK, command[1], EnvironmentVariableTarget.User);
-                            Console.WriteLine("set " + SSH_AUTH_SOCK + "=" + command[1]);
+                            Console.WriteLine(
+                                (parent.ProcessName == "powershell" ? "$env:" : "set ") + SSH_AUTH_SOCK + "=\"" + command[1] + "\";");
                         }
                         else if (command[0] == SSH_AGENT_PID && command.Length > 1)
                         {
                             AgentPID = command[1];
                             Environment.SetEnvironmentVariable(SSH_AGENT_PID, command[1], EnvironmentVariableTarget.User);
-                            Console.WriteLine("set " + SSH_AGENT_PID + "=" + command[1]);
+                            Console.WriteLine(
+                                (parent.ProcessName == "powershell" ? "$env:" : "set ") + SSH_AGENT_PID + "=\"" + command[1] + "\";");
                         }
                     }
+                    Console.WriteLine((parent.ProcessName == "powershell" ? "# " : "rem ") +
+                                          "Your environment has been configured. " +
+                                          "Run these commands to configure current terminal.");
                 }
                 catch (Exception e)
                 {