public static int Main(string[] args)
{
if (args.Length < 1)
{
Console.WriteLine("Invalid command line");
return((int)Result.InvalidCommandLine);
}
var command = args[0];
if (string.Compare(command, "add", StringComparison.OrdinalIgnoreCase) == 0)
{
if (args.Length < 5)
{
Console.WriteLine("Invalid command line");
return((int)Result.InvalidCommandLine);
}
var host = args[1];
var port = int.Parse(args[2]);
var user = args[3];
var password = args[4];
Console.WriteLine($"Adding credentials for {host}");
var cis = new ConnectionInfoStore();
var passwordSecure = new SecureString();
foreach (var c in password)
{
passwordSecure.AppendChar(c);
}
cis.Add(new PasswordConnectionInfo(host, port, TimeSpan.MaxValue, user, passwordSecure));
Console.WriteLine($"Saving credentials");
cis.Save();
Console.WriteLine($"Credentials saved successfully");
return((int)Result.Success);
}
return((int)Result.UnsupportedCommand);
}
internal static Connection GetInstance(string name, ConnectionReason reason)
{
UnixSystem remoteSystem = null;
ConnectionInfoStore store = new ConnectionInfoStore();
ConnectionInfo connectionInfo = null;
StoredConnectionInfo storedConnectionInfo = store.Connections.FirstOrDefault(connection =>
{
return(name.Equals(GetFormattedConnectionName((ConnectionInfo)connection), StringComparison.OrdinalIgnoreCase));
});
if (storedConnectionInfo != null)
{
connectionInfo = (ConnectionInfo)storedConnectionInfo;
}
if (connectionInfo == null)
{
IVsConnectionManager connectionManager = (IVsConnectionManager)ServiceProvider.GlobalProvider.GetService(typeof(IVsConnectionManager));
string userName;
string hostName;
int atSignIndex = name.IndexOf('@');
if (atSignIndex > 0)
{
userName = name.Substring(0, atSignIndex);
hostName = name.Substring(atSignIndex + 1);
}
else
{
userName = string.Format(CultureInfo.CurrentCulture, StringResources.UserName_PlaceHolder);
hostName = name;
}
PasswordConnectionInfo newConnectionInfo = new PasswordConnectionInfo(hostName, userName, new System.Security.SecureString());
IConnectionManagerResult result = connectionManager.ShowDialog(newConnectionInfo);
if (result.DialogResult == ConnectionManagerDialogResult.Succeeded)
{
// Retrieve the newly added connection
store.Load();
connectionInfo = store.Connections.First(info => info.Id == result.StoredConnectionId);
}
}
if (connectionInfo != null)
{
remoteSystem = new UnixSystem();
while (true)
{
try
{
VSOperationWaiter.Wait(string.Format(CultureInfo.CurrentCulture, StringResources.WaitingOp_Connecting, name), throwOnCancel: false, action: () =>
{
remoteSystem.Connect(connectionInfo);
});
break;
}
catch (RemoteAuthenticationException)
{
IVsConnectionManager connectionManager = (IVsConnectionManager)ServiceProvider.GlobalProvider.GetService(typeof(IVsConnectionManager));
IConnectionManagerResult result = connectionManager.ShowDialog(StringResources.AuthenticationFailureHeader, StringResources.AuthenticationFailureDescription, connectionInfo);
if (result.DialogResult == ConnectionManagerDialogResult.Succeeded)
{
// Update the credentials in the store
store.Load();
store.RemoveById(result.StoredConnectionId);
store.Add(result.ConnectionInfo);
store.Save();
connectionInfo = result.ConnectionInfo;
}
else
{
return(null);
}
}
catch (Exception ex)
{
VsShellUtilities.ShowMessageBox(ServiceProvider.GlobalProvider, ex.Message, null,
OLEMSGICON.OLEMSGICON_CRITICAL, OLEMSGBUTTON.OLEMSGBUTTON_OK, OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST);
return(null);
}
}
// NOTE: This will be null if connect is canceled
if (remoteSystem != null)
{
return(new Connection(remoteSystem));
}
}
return(null);
}
