void SFTPTreeView::DoOpenSession()
{
DoCloseSession();
wxString accountName = m_choiceAccount->GetStringSelection();
if(accountName.IsEmpty()) {
return;
}
SFTPSettings settings;
settings.Load();
m_account = SSHAccountInfo();
if(!settings.GetAccount(accountName, m_account)) {
::wxMessageBox(wxString() << _("Could not find account: ") << accountName, "codelite", wxICON_ERROR | wxOK);
return;
}
clSSH::Ptr_t ssh(
new clSSH(m_account.GetHost(), m_account.GetUsername(), m_account.GetPassword(), m_account.GetPort()));
try {
wxString message;
m_plugin->GetManager()->SetStatusMessage(wxString() << _("Connecting to: ") << accountName << "...");
ssh->Connect();
if(!ssh->AuthenticateServer(message)) {
if(::wxMessageBox(message, "SSH", wxYES_NO | wxCENTER | wxICON_QUESTION) == wxYES) {
ssh->AcceptServerAuthentication();
}
}
ssh->Login();
m_sftp.reset(new clSFTP(ssh));
m_sftp->Initialize();
m_sftp->SetAccount(m_account.GetAccountName());
m_plugin->GetManager()->SetStatusMessage(wxString() << _("Done!"));
DoBuildTree("/");
} catch(clException& e) {
::wxMessageBox(e.What(), "codelite", wxICON_ERROR | wxOK);
DoCloseSession();
}
}
void SFTPTreeView::OnOpenAccountManager(wxCommandEvent& event)
{
SSHAccountManagerDlg dlg(this);
if(dlg.ShowModal() == wxID_OK) {
SFTPSettings settings;
settings.Load();
settings.SetAccounts(dlg.GetAccounts());
settings.Save();
// Update the selections at the top
wxString curselection = m_choiceAccount->GetStringSelection();
m_choiceAccount->Clear();
const SSHAccountInfo::Vect_t& accounts = settings.GetAccounts();
if(accounts.empty()) {
DoCloseSession();
return;
} else {
SSHAccountInfo::Vect_t::const_iterator iter = accounts.begin();
for(; iter != accounts.end(); ++iter) {
m_choiceAccount->Append(iter->GetAccountName());
}
int where = m_choiceAccount->FindString(curselection);
if(where == wxNOT_FOUND) {
// Our previous session is no longer available, close the session
DoCloseSession();
where = 0;
}
m_choiceAccount->SetSelection(where);
}
}
}
void SFTPTreeView::OnConnection(wxCommandEvent& event)
{
SFTPImages images;
if(IsConnected()) {
// Disconnect
DoCloseSession();
// Update toobar image
m_auibar->FindTool(ID_SFTP_CONNECT)->SetBitmap(images.Bitmap("sftp_disconnected"));
m_auibar->FindTool(ID_SFTP_CONNECT)->SetShortHelp(_("Disconnected. Click to connect"));
} else {
DoOpenSession();
// Update toobar image
m_auibar->FindTool(ID_SFTP_CONNECT)->SetBitmap(images.Bitmap("sftp_connected"));
m_auibar->FindTool(ID_SFTP_CONNECT)->SetShortHelp(_("Connected. Click to disconnect"));
}
}
void SFTPTreeView::DoOpenSession()
{
DoCloseSession();
wxString accountName = m_choiceAccount->GetStringSelection();
if(accountName.IsEmpty()) {
return;
}
SFTPSettings settings;
settings.Load();
m_account = SSHAccountInfo();
if(!settings.GetAccount(accountName, m_account)) {
::wxMessageBox(wxString() << _("Could not find account: ") << accountName, "codelite", wxICON_ERROR | wxOK);
return;
}
wxString message;
#ifndef _WIN64
wxProgressDialog dlg(_("SFTP"), wxString(' ', 100) + "\n\n", 10);
dlg.Show();
dlg.Update(1, wxString() << _("Connecting to: ") << accountName << "..." << _("\n(this may take a few seconds)"));
#else
wxBusyCursor bc;
clGetManager()->SetStatusMessage(wxString() << _("Connecting to: ") << accountName);
#endif
// We know that there is a bug that libssh succeeded on connecting only on the second attempt..
// to workaround it, we issue first connect with 1 second timeout, and then re-open the connection
try {
clSSH::Ptr_t ssh(
new clSSH(m_account.GetHost(), m_account.GetUsername(), m_account.GetPassword(), m_account.GetPort()));
ssh->Connect(wxNOT_FOUND);
} catch(...) {
}
try {
clSSH::Ptr_t ssh(
new clSSH(m_account.GetHost(), m_account.GetUsername(), m_account.GetPassword(), m_account.GetPort()));
ssh->Connect(5);
#ifndef _WIN64
dlg.Update(5, _("Connected!"));
dlg.Update(6, _("Authenticating server..."));
#endif
if(!ssh->AuthenticateServer(message)) {
if(::wxMessageBox(message, "SSH", wxYES_NO | wxCENTER | wxICON_QUESTION) == wxYES) {
#ifndef _WIN64
dlg.Update(7, _("Accepting server authentication server..."));
#endif
ssh->AcceptServerAuthentication();
}
} else {
#ifndef _WIN64
dlg.Update(7, _("Server authenticated"));
#endif
}
#ifndef _WIN64
dlg.Update(8, _("Logging..."));
#endif
ssh->Login();
m_sftp.reset(new clSFTP(ssh));
m_sftp->Initialize();
m_sftp->SetAccount(m_account.GetAccountName());
m_plugin->GetManager()->SetStatusMessage(wxString() << _("Done!"));
#ifndef _WIN64
dlg.Update(9, _("Fetching directory list..."));
#endif
DoBuildTree(m_account.GetDefaultFolder().IsEmpty() ? "/" : m_account.GetDefaultFolder());
#ifndef _WIN64
dlg.Update(10, _("Done"));
#endif
} catch(clException& e) {
::wxMessageBox(e.What(), "codelite", wxICON_ERROR | wxOK);
DoCloseSession();
}
}
void SFTPTreeView::OnDisconnect(wxCommandEvent& event) { DoCloseSession(); }
void SFTPTreeView::DoOpenSession()
{
DoCloseSession();
if(!GetAccountFromUser(m_account)) { return; }
wxString message;
wxProgressDialog dlg(_("SFTP"), wxString(' ', 100) + "\n\n", 10);
dlg.Show();
dlg.Update(1, wxString() << _("Connecting to: ") << m_account.GetAccountName() << "..."
<< _("\n(this may take a few seconds)"));
// We know that there is a bug that libssh succeeded on connecting only on the second attempt..
// to workaround it, we issue first connect with 1 second timeout, and then re-open the connection
try {
clSSH::Ptr_t ssh(
new clSSH(m_account.GetHost(), m_account.GetUsername(), m_account.GetPassword(), m_account.GetPort()));
ssh->Connect(wxNOT_FOUND);
} catch(...) {
}
try {
clSSH::Ptr_t ssh(
new clSSH(m_account.GetHost(), m_account.GetUsername(), m_account.GetPassword(), m_account.GetPort()));
ssh->Connect(5);
dlg.Update(5, _("Connected!"));
dlg.Update(6, _("Authenticating server..."));
if(!ssh->AuthenticateServer(message)) {
if(::wxMessageBox(message, "SSH", wxYES_NO | wxCENTER | wxICON_QUESTION) == wxYES) {
dlg.Update(7, _("Accepting server authentication server..."));
ssh->AcceptServerAuthentication();
}
} else {
dlg.Update(7, _("Server authenticated"));
}
dlg.Update(8, _("Logging in.."));
ssh->Login();
m_sftp.reset(new clSFTP(ssh));
m_sftp->Initialize();
m_sftp->SetAccount(m_account.GetAccountName());
m_plugin->GetManager()->SetStatusMessage(wxString() << _("Done!"));
dlg.Update(9, _("Fetching directory list..."));
DoBuildTree(m_account.GetDefaultFolder().IsEmpty() ? "/" : m_account.GetDefaultFolder());
dlg.Update(10, _("Done"));
CallAfter(&SFTPTreeView::DoLoadSession);
// If this is a new account, offer the user to save it
SFTPSettings s;
s.Load();
SSHAccountInfo dummy;
if(!s.GetAccount(m_account.GetAccountName(), dummy)) {
wxString message;
message << _("Would you like to save this account?\n") << _("It will be saved as '")
<< m_account.GetAccountName() << "'";
wxStandardID res = ::PromptForYesNoCancelDialogWithCheckbox(message, "SFTPQuickConnectSaveDlg");
if(res == wxID_YES) {
// This 'Connect' was via Quick Connect option
SSHAccountInfo::Vect_t accounts = s.GetAccounts();
accounts.push_back(m_account);
s.SetAccounts(accounts);
s.Save();
}
}
} catch(clException& e) {
::wxMessageBox(e.What(), "CodeLite", wxICON_ERROR | wxOK);
DoCloseSession();
}
}
