116 lines
3.6 KiB
C#
116 lines
3.6 KiB
C#
// Copyright (c) .NET Foundation and contributors. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
using Microsoft.DotNet.Cli.Utils;
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Collections.Generic;
|
|
|
|
namespace Microsoft.DotNet.Tools.Test.Utilities
|
|
{
|
|
public class TestCommand
|
|
{
|
|
protected string _command;
|
|
|
|
public Dictionary<string, string> Environment { get; } = new Dictionary<string, string>();
|
|
|
|
public TestCommand(string command)
|
|
{
|
|
_command = command;
|
|
}
|
|
|
|
public virtual CommandResult Execute(string args = "")
|
|
{
|
|
var commandPath = _command;
|
|
ResolveCommand(ref commandPath, ref args);
|
|
|
|
Console.WriteLine($"Executing - {_command} {args}");
|
|
|
|
var stdOut = new StreamForwarder();
|
|
var stdErr = new StreamForwarder();
|
|
|
|
stdOut.ForwardTo(writeLine: Reporter.Output.WriteLine);
|
|
stdErr.ForwardTo(writeLine: Reporter.Output.WriteLine);
|
|
|
|
return RunProcess(commandPath, args, stdOut, stdErr);
|
|
}
|
|
|
|
public virtual CommandResult ExecuteWithCapturedOutput(string args = "")
|
|
{
|
|
Console.WriteLine($"Executing (Captured Output) - {_command} {args}");
|
|
|
|
var command = _command;
|
|
ResolveCommand(ref command, ref args);
|
|
var commandPath = Env.GetCommandPath(command, ".exe", ".cmd", "") ??
|
|
Env.GetCommandPathFromRootPath(AppContext.BaseDirectory, command, ".exe", ".cmd", "");
|
|
|
|
var stdOut = new StreamForwarder();
|
|
var stdErr = new StreamForwarder();
|
|
|
|
stdOut.Capture();
|
|
stdErr.Capture();
|
|
|
|
return RunProcess(commandPath, args, stdOut, stdErr);
|
|
}
|
|
|
|
private void ResolveCommand(ref string executable, ref string args)
|
|
{
|
|
if (executable.EndsWith(".dll", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
var newArgs = ArgumentEscaper.EscapeSingleArg(executable);
|
|
if (!string.IsNullOrEmpty(args))
|
|
{
|
|
newArgs += " " + args;
|
|
}
|
|
args = newArgs;
|
|
executable = "corehost";
|
|
}
|
|
|
|
if (!Path.IsPathRooted(executable))
|
|
{
|
|
executable = Env.GetCommandPath(executable) ??
|
|
Env.GetCommandPathFromRootPath(AppContext.BaseDirectory, executable);
|
|
}
|
|
}
|
|
|
|
private CommandResult RunProcess(string executable, string args, StreamForwarder stdOut, StreamForwarder stdErr)
|
|
{
|
|
var psi = new ProcessStartInfo
|
|
{
|
|
FileName = executable,
|
|
Arguments = args,
|
|
RedirectStandardError = true,
|
|
RedirectStandardOutput = true
|
|
};
|
|
|
|
foreach (var item in Environment)
|
|
{
|
|
psi.Environment[item.Key] = item.Value;
|
|
}
|
|
|
|
var process = new Process
|
|
{
|
|
StartInfo = psi,
|
|
};
|
|
|
|
process.EnableRaisingEvents = true;
|
|
process.Start();
|
|
|
|
var threadOut = stdOut.BeginRead(process.StandardOutput);
|
|
var threadErr = stdErr.BeginRead(process.StandardError);
|
|
|
|
process.WaitForExit();
|
|
threadOut.Join();
|
|
threadErr.Join();
|
|
|
|
var result = new CommandResult(
|
|
process.StartInfo,
|
|
process.ExitCode,
|
|
stdOut.CapturedOutput,
|
|
stdErr.CapturedOutput);
|
|
|
|
return result;
|
|
}
|
|
}
|
|
}
|