-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProcessRunner.cs
More file actions
102 lines (85 loc) · 2.99 KB
/
ProcessRunner.cs
File metadata and controls
102 lines (85 loc) · 2.99 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
using System.Diagnostics;
internal static class ProcessRunner
{
public static async Task<ProcessResult> RunAsync(
string fileName,
IReadOnlyList<string> arguments,
string workingDirectory,
bool whatIf,
IReadOnlyList<string> sensitiveValues,
bool printOutputOnSuccess = true)
{
var commandForLog = BuildCommandForLog(fileName, arguments, sensitiveValues);
Console.WriteLine($"> {commandForLog}");
if (whatIf)
{
return ProcessResult.Ok();
}
var processStartInfo = new ProcessStartInfo
{
FileName = fileName,
WorkingDirectory = workingDirectory,
RedirectStandardOutput = true,
RedirectStandardError = true,
UseShellExecute = false,
CreateNoWindow = true
};
foreach (var arg in arguments)
{
processStartInfo.ArgumentList.Add(arg);
}
using var process = new Process { StartInfo = processStartInfo };
process.Start();
var stdOutTask = process.StandardOutput.ReadToEndAsync();
var stdErrTask = process.StandardError.ReadToEndAsync();
await process.WaitForExitAsync();
var stdOut = await stdOutTask;
var stdErr = await stdErrTask;
if (process.ExitCode == 0)
{
if (printOutputOnSuccess)
{
if (!string.IsNullOrWhiteSpace(stdOut))
{
Console.Write(MaskSensitive(stdOut, sensitiveValues));
}
if (!string.IsNullOrWhiteSpace(stdErr))
{
Console.Error.Write(MaskSensitive(stdErr, sensitiveValues));
}
}
return ProcessResult.Ok();
}
if (!string.IsNullOrWhiteSpace(stdOut))
{
Console.Write(MaskSensitive(stdOut, sensitiveValues));
}
if (!string.IsNullOrWhiteSpace(stdErr))
{
Console.Error.Write(MaskSensitive(stdErr, sensitiveValues));
}
return ProcessResult.Fail($"Command failed with exit code {process.ExitCode}: {fileName}");
}
private static string BuildCommandForLog(string fileName, IReadOnlyList<string> args, IReadOnlyList<string> sensitiveValues)
{
var command = fileName + " " + string.Join(" ", args.Select(QuoteArgForLog));
return MaskSensitive(command, sensitiveValues);
}
private static string QuoteArgForLog(string arg)
{
if (arg.Contains(' ') || arg.Contains('"'))
{
return "\"" + arg.Replace("\"", "\\\"") + "\"";
}
return arg;
}
private static string MaskSensitive(string text, IReadOnlyList<string> sensitiveValues)
{
var value = text;
foreach (var secret in sensitiveValues.Where(s => !string.IsNullOrWhiteSpace(s)))
{
value = value.Replace(secret, "***", StringComparison.Ordinal);
}
return value;
}
}