SSHAccountInfo::Vect_t SSHAccountManagerDlg::GetAccounts() const { SSHAccountInfo::Vect_t accounts; for(int i = 0; i < m_dvListCtrlAccounts->GetItemCount(); ++i) { accounts.push_back(*(SSHAccountInfo*)(m_dvListCtrlAccounts->GetItemData(m_dvListCtrlAccounts->RowToItem(i)))); } return accounts; }
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(); } }