internal static IEnumerable <DockerContainerInstance> GetRemoteDockerContainers(IConnection connection, string hostname, out int totalContainers)
{
totalContainers = 0;
SSHConnection sshConnection = connection as SSHConnection;
List <string> outputLines = new List <string>();
StringBuilder errorSB = new StringBuilder();
if (sshConnection == null)
{
return(null);
}
List <DockerContainerInstance> containers = new List <DockerContainerInstance>();
DockerCommandSettings settings = new DockerCommandSettings(hostname, true);
settings.SetCommand(dockerPSCommand, dockerPSArgs);
RemoteCommandRunner commandRunner = new RemoteCommandRunner(settings, sshConnection);
ManualResetEvent resetEvent = new ManualResetEvent(false);
int exitCode = 0;
commandRunner.ErrorOccured += ((sender, args) =>
{
errorSB.Append(args);
});
commandRunner.Closed += ((sender, args) =>
{
exitCode = args;
resetEvent.Set();
});
commandRunner.OutputReceived += ((sender, args) =>
{
if (!string.IsNullOrWhiteSpace(args))
{
// If it isn't json, assume its an error message
if (args.Trim()[0] != '{')
{
errorSB.Append(args);
}
// Unix line endings are '\n' so split on that for json items.
foreach (var item in args.Split('\n').ToList())
{
if (!string.IsNullOrWhiteSpace(item))
{
outputLines.Add(item);
}
}
}
});
commandRunner.Start();
bool cancellationRequested = false;
VS.VSOperationWaiter.Wait(UIResources.QueryingForContainersMessage, false, (cancellationToken) =>
{
while (!resetEvent.WaitOne(2000) && !cancellationToken.IsCancellationRequested)
{
}
cancellationRequested = cancellationToken.IsCancellationRequested;
});
if (!cancellationRequested)
{
if (exitCode != 0)
{
// if the exit code is not zero, then the output we received possibly is the error message
string exceptionMessage = UIResources.CommandExecutionErrorWithExitCodeFormat.FormatCurrentCultureWithArgs(
"{0} {1}".FormatInvariantWithArgs(settings.Command, settings.CommandArgs),
exitCode,
errorSB.ToString());
throw new CommandFailedException(exceptionMessage);
}
foreach (var item in outputLines)
{
if (DockerContainerInstance.TryCreate(item, out DockerContainerInstance containerInstance))
{
containers.Add(containerInstance);
}
totalContainers++;
}
}
return(containers);
}
private static void RunDockerCommand(DockerCommandSettings settings, Action <string> callback)
{
LocalCommandRunner commandRunner = new LocalCommandRunner(settings);
StringBuilder errorSB = new StringBuilder();
int? exitCode = null;
try
{
ManualResetEvent resetEvent = new ManualResetEvent(false);
commandRunner.ErrorOccured += ((sender, args) =>
{
if (!string.IsNullOrWhiteSpace(args.ErrorMessage))
{
errorSB.AppendLine(args.ErrorMessage);
}
resetEvent.Set();
});
commandRunner.Closed += ((sender, args) =>
{
exitCode = args;
resetEvent.Set();
});
commandRunner.OutputReceived += ((sender, args) =>
{
if (!string.IsNullOrWhiteSpace(args))
{
args = args.Trim(charsToTrim);
callback(args);
}
});
commandRunner.Start();
bool cancellationRequested = false;
VS.VSOperationWaiter.Wait(UIResources.QueryingForContainersMessage, false, (cancellationToken) =>
{
while (!resetEvent.WaitOne(2000) && !cancellationToken.IsCancellationRequested)
{
}
cancellationRequested = cancellationToken.IsCancellationRequested;
});
if (!cancellationRequested)
{
if (exitCode.GetValueOrDefault(-1) != 0)
{
string exceptionMessage = UIResources.CommandExecutionErrorWithExitCodeFormat.FormatCurrentCultureWithArgs(
"{0} {1}".FormatInvariantWithArgs(settings.Command, settings.CommandArgs),
exitCode,
errorSB.ToString());
throw new CommandFailedException(exceptionMessage);
}
if (errorSB.Length > 0)
{
throw new CommandFailedException(errorSB.ToString());
}
}
}
catch (Win32Exception ex)
{
string errorMessage = UIResources.CommandExecutionErrorFormat.FormatCurrentCultureWithArgs(settings.CommandArgs, ex.Message);
throw new CommandFailedException(errorMessage, ex);
}
}
