/// <summary>
/// Executes the given executable to process the given command.
/// </summary>
/// <param name="workingDirectory">
/// The working directory while executing the command.
/// </param>
/// <param name="executable">
/// The full path to and name of the executable to execute.
/// </param>
/// <param name="command">
/// The options to the executable.
/// </param>
/// <param name="environmentVariables">
/// An array of <see cref="System.Collections.Generic.KeyValuePair{TKey,TValue}"/> objects, containing environment variable
/// overrides to use while executing the executable.
/// </param>
/// <param name="specialArguments">
/// Any special arguments to pass to the executable, not defined by the <paramref name="command"/>
/// object, typically common arguments that should always be passed to the executable.
/// </param>
/// <exception cref="System.ArgumentNullException">
/// <para><paramref name="workingDirectory"/> is <c>null</c> or empty.</para>
/// <para>- or -</para>
/// <para><paramref name="executable"/> is <c>null</c> or empty.</para>
/// <para>- or -</para>
/// <para><paramref name="command"/> is <c>null</c>.</para>
/// <para>- or -</para>
/// <para><paramref name="environmentVariables"/> is <c>null</c>.</para>
/// <para>- or -</para>
/// <para><paramref name="specialArguments"/> is <c>null</c>.</para>
/// </exception>
/// <exception cref="MercurialException">
/// <para>The executable did not finish in the allotted time.</para>
/// </exception>
public static void Execute(
string workingDirectory,
string executable,
ICommand command,
KeyValuePair <string, string>[] environmentVariables,
IEnumerable <string> specialArguments)
{
if (StringEx.IsNullOrWhiteSpace(workingDirectory))
{
throw new ArgumentNullException(nameof(workingDirectory));
}
if (StringEx.IsNullOrWhiteSpace(executable))
{
throw new ArgumentNullException(nameof(executable));
}
if (command == null)
{
throw new ArgumentNullException(nameof(command));
}
if (environmentVariables == null)
{
throw new ArgumentNullException(nameof(environmentVariables));
}
if (specialArguments == null)
{
throw new ArgumentNullException(nameof(specialArguments));
}
command.Validate();
command.Before();
IEnumerable <string> arguments = specialArguments;
arguments = arguments.Concat(command.Arguments.Where(a => !StringEx.IsNullOrWhiteSpace(a)));
arguments = arguments.Concat(command.AdditionalArguments.Where(a => !StringEx.IsNullOrWhiteSpace(a)));
string argumentsString = string.Join(" ", arguments.ToArray());
var startInfo = new ProcessStartInfo
{
FileName = executable,
WorkingDirectory = workingDirectory,
RedirectStandardInput = true,
RedirectStandardOutput = true,
RedirectStandardError = true,
CreateNoWindow = true,
WindowStyle = ProcessWindowStyle.Hidden,
UseShellExecute = false,
ErrorDialog = false,
Arguments = command.Command + " " + argumentsString,
};
foreach (var kvp in environmentVariables)
{
startInfo.EnvironmentVariables[kvp.Key] = kvp.Value;
}
ClientExecutable.LazyInitialize();
startInfo.StandardErrorEncoding = ClientExecutable.GetMainEncoding();
startInfo.StandardOutputEncoding = ClientExecutable.GetMainEncoding();
command.Observer?.Executing(command.Command, argumentsString);
using Process process = Process.Start(startInfo);
Func <StreamReader, Action <string>, string> reader;
if (command.Observer != null)
{
reader = delegate(StreamReader streamReader, Action <string> logToObserver)
{
var output = new StringBuilder();
string line;
while ((line = streamReader.ReadLine()) != null)
{
logToObserver(line);
if (output.Length > 0)
{
output.Append(Environment.NewLine);
}
output.Append(line);
}
return(output.ToString());
};
}
else
{
reader = (streamReader, logToObserver) => streamReader.ReadToEnd();
}
string standardOutput = reader(process.StandardOutput,
line => command.Observer.Output(line));
string errorOutput = reader(process.StandardError,
line => command.Observer.ErrorOutput(line));
int timeout = Timeout.Infinite;
if (command.Timeout > 0)
{
timeout = 1000 * command.Timeout;
}
process.WaitForExit(timeout);
if (command.Observer != null)
{
command.Observer.Executed(command.Command, argumentsString, process.ExitCode, standardOutput,
errorOutput);
}
command.After(process.ExitCode, standardOutput, errorOutput);
}
/// <summary>
/// Initializes a new instance of the <see cref="RevSpec"/> class.
/// </summary>
/// <param name="value">
/// The value of this <see cref="RevSpec"/> value, can be both a hash and an expression.
/// </param>
public RevSpec(string value)
{
Debug.Assert(!StringEx.IsNullOrWhiteSpace(value), "value cannot be null or empty here");
_Value = value.Trim();
}
/// <summary>
/// Executes the given <see cref="IMercurialCommand"/> command against
/// the Mercurial repository.
/// </summary>
/// <param name="command">
/// The <see cref="IMercurialCommand"/> command to execute.
/// </param>
/// <exception cref="System.ArgumentNullException">
/// <para><paramref name="command"/> is <c>null</c>.</para>
/// </exception>
/// <exception cref="MercurialException">
/// HG did not complete within the allotted time.
/// </exception>
public void Execute(IMercurialCommand command)
{
if (command == null)
{
throw new ArgumentNullException("command");
}
if (_Process == null)
{
StartPersistentMercurialClient();
}
command.Validate();
command.Before();
IEnumerable <string> arguments = new[]
{
command.Command,
"--noninteractive",
};
arguments = arguments.Concat(command.Arguments.Where(a => !StringEx.IsNullOrWhiteSpace(a)));
arguments = arguments.Concat(command.AdditionalArguments.Where(a => !StringEx.IsNullOrWhiteSpace(a)));
var commandParts = arguments.ToArray();
string commandEncoded = string.Join("\0", commandParts.Select(p => p.Trim(new[] { '"' })).ToArray());
int length = commandEncoded.Length;
var commandBuffer = new StringBuilder();
commandBuffer.Append("runcommand\n");
commandBuffer.Append((char)((length >> 24) & 0xff));
commandBuffer.Append((char)((length >> 16) & 0xff));
commandBuffer.Append((char)((length >> 8) & 0xff));
commandBuffer.Append((char)(length & 0xff));
commandBuffer.Append(commandEncoded);
string commandArguments = null;
if (command.Observer != null)
{
commandArguments = string.Join(" ", commandParts.Skip(1).ToArray());
command.Observer.Executing(command.Command, commandArguments);
}
MemoryStream output = new MemoryStream();
MemoryStream error = new MemoryStream();
var outputs = new Dictionary <CommandChannel, Stream>()
{
{ CommandChannel.Output, output },
{ CommandChannel.Error, error },
};
var _codec = ClientExecutable.GetMainEncoding();
int resultCode = RunCommand(commandParts, outputs, null);
var result = new CommandResult(_codec.GetString(output.GetBuffer(), 0, (int)output.Length),
_codec.GetString(error.GetBuffer(), 0, (int)error.Length),
resultCode);
if (resultCode == 0 || !string.IsNullOrEmpty(result.Output))
{
if (command.Observer != null)
{
command.Observer.Output(result.Output);
command.Observer.ErrorOutput(result.Error);
command.Observer.Executed(command.Command, commandArguments, resultCode, result.Output, result.Error);
}
command.After(resultCode, result.Output, result.Error);
return;
}
StopPersistentMercurialClient();
throw new MercurialExecutionException(
string.IsNullOrEmpty(result.Error) ?
"Unable to decode output from executing command, spinning down persistent client"
: result.Error);
}
