protected async Task <TfsVCPorcelainCommandResult> TryRunPorcelainCommandAsync(FormatFlags formatFlags, bool ignoreStderr, params string[] args)
{
// Validation.
ArgUtil.NotNull(args, nameof(args));
ArgUtil.NotNull(ExecutionContext, nameof(ExecutionContext));
// Invoke tf.
using (var processInvoker = HostContext.CreateService <IProcessInvoker>())
{
var result = new TfsVCPorcelainCommandResult();
var outputLock = new object();
processInvoker.OutputDataReceived += (object sender, ProcessDataReceivedEventArgs e) =>
{
lock (outputLock)
{
ExecutionContext.Debug(e.Data);
result.Output.Add(e.Data);
}
};
processInvoker.ErrorDataReceived += (object sender, ProcessDataReceivedEventArgs e) =>
{
lock (outputLock)
{
if (ignoreStderr)
{
ExecutionContext.Output(e.Data);
}
else
{
ExecutionContext.Debug(e.Data);
result.Output.Add(e.Data);
}
}
};
string arguments = FormatArguments(formatFlags, args);
ExecutionContext.Debug($@"tf {arguments}");
// TODO: Test whether the output encoding needs to be specified on a non-Latin OS.
try
{
await processInvoker.ExecuteAsync(
workingDirectory : SourcesDirectory,
fileName : FilePath,
arguments : arguments,
environment : AdditionalEnvironmentVariables,
requireExitCodeZero : true,
outputEncoding : OutputEncoding,
cancellationToken : CancellationToken);
}
catch (ProcessExitCodeException ex)
{
result.Exception = ex;
}
return(result);
}
}
public async Task <ITfsVCWorkspace[]> WorkspacesAsync(bool matchWorkspaceNameOnAnyComputer = false)
{
// Build the args.
var args = new List <string>();
args.Add("workspaces");
if (matchWorkspaceNameOnAnyComputer)
{
args.Add(WorkspaceName);
args.Add($"-computer:*");
}
args.Add("-format:xml");
// Run the command.
TfsVCPorcelainCommandResult result = await TryRunPorcelainCommandAsync(FormatFlags.None, args.ToArray());
ArgUtil.NotNull(result, nameof(result));
if (result.Exception != null)
{
// Check if the workspace name was specified and the command returned exit code 1.
if (matchWorkspaceNameOnAnyComputer && result.Exception.ExitCode == 1)
{
// Ignore the error. This condition can indicate the workspace was not found.
return(new ITfsVCWorkspace[0]);
}
// Dump the output and throw.
result.Output?.ForEach(x => ExecutionContext.Output(x ?? string.Empty));
throw result.Exception;
}
// Note, string.join gracefully handles a null element within the IEnumerable<string>.
string xml = string.Join(Environment.NewLine, result.Output ?? new List <string>()) ?? string.Empty;
// The command returns a non-XML message preceeding the XML if there are no workspaces.
if (!xml.StartsWith("<?xml"))
{
return(null);
}
// Deserialize the XML.
var serializer = new XmlSerializer(typeof(TeeWorkspaces));
using (var reader = new StringReader(xml))
{
return((serializer.Deserialize(reader) as TeeWorkspaces)
?.Workspaces
?.Cast <ITfsVCWorkspace>()
.ToArray());
}
}
protected async Task <string> RunPorcelainCommandAsync(FormatFlags formatFlags, bool ignoreStderr, params string[] args)
{
// Run the command.
TfsVCPorcelainCommandResult result = await TryRunPorcelainCommandAsync(formatFlags, ignoreStderr, args);
ArgUtil.NotNull(result, nameof(result));
if (result.Exception != null)
{
// The command failed. Dump the output and throw.
result.Output?.ForEach(x => ExecutionContext.Output(x ?? string.Empty));
throw result.Exception;
}
// Return the output.
// Note, string.join gracefully handles a null element within the IEnumerable<string>.
return(string.Join(Environment.NewLine, result.Output ?? new List <string>()));
}
