src/ChatWnd.cpp

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2008 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2002-2008 Merkur ( devs@emule-project.net / http://www.emule-project.net )
   6 //
   7 // Any parts of this program derived from the xMule, lMule or eMule project,
   8 // or contributed by third-party developers are copyrighted by their
   9 // respective authors.
  10 //
  11 // This program is free software; you can redistribute it and/or modify
  12 // it under the terms of the GNU General Public License as published by
  13 // the Free Software Foundation; either version 2 of the License, or
  14 // (at your option) any later version.
  15 //
  16 // This program is distributed in the hope that it will be useful,
  17 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 // GNU General Public License for more details.
  20 //
  21 // You should have received a copy of the GNU General Public License
  22 // along with this program; if not, write to the Free Software
  23 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  24 //
  25
  26 #include <common/MenuIDs.h>  // IDs for the chat-popup menu
  27
  28 #include <wx/app.h>
  29
  30 #include "ChatWnd.h"            // Interface declarations
  31
  32 #include "amule.h"              // Needed for theApp
  33 #include "amuleDlg.h"           // Needed for CamuleDlg
  34 #include "ClientList.h"         // Needed for CClientList
  35 #include "updownclient.h"       // Needed for CUpDownClient
  36 #include "FriendListCtrl.h"     // Needed for CFriendListCtrl
  37 #include "Friend.h"                     // Needed for CFriend
  38 #include "ChatSelector.h"       // Needed for CChatSelector
  39 #include "muuli_wdr.h"          // Needed for messagePage
  40 #include "OtherFunctions.h"
  41
  42 BEGIN_EVENT_TABLE(CChatWnd, wxPanel)
  43         EVT_RIGHT_DOWN(CChatWnd::OnNMRclickChatTab)
  44
  45         EVT_MENU(MP_CLOSE_TAB,                  CChatWnd::OnPopupClose)
  46         EVT_MENU(MP_CLOSE_ALL_TABS,             CChatWnd::OnPopupCloseAll)
  47         EVT_MENU(MP_CLOSE_OTHER_TABS,   CChatWnd::OnPopupCloseOthers)
  48         EVT_MENU(MP_ADDFRIEND,                  CChatWnd::OnAddFriend )
  49
  50         EVT_TEXT_ENTER(IDC_CMESSAGE, CChatWnd::OnBnClickedCsend)
  51         EVT_BUTTON(IDC_CSEND, CChatWnd::OnBnClickedCsend)
  52         EVT_BUTTON(IDC_CCLOSE, CChatWnd::OnBnClickedCclose)
  53         EVT_MULENOTEBOOK_ALL_PAGES_CLOSED(IDC_CHATSELECTOR, CChatWnd::OnAllPagesClosed)
  54 END_EVENT_TABLE()
  55
  56
  57 CChatWnd::CChatWnd(wxWindow* pParent)
  58 : wxPanel(pParent, -1)
  59 {
  60         wxSizer* content = messagePage(this, true);
  61         content->Show(this, true);
  62
  63         chatselector = CastChild( IDC_CHATSELECTOR, CChatSelector );
  64         // We want to use our own popup menu
  65         chatselector->SetPopupHandler(this);
  66         m_menu = NULL;
  67
  68         friendlistctrl = CastChild( ID_FRIENDLIST, CFriendListCtrl );
  69 }
  70
  71 void CChatWnd::StartSession(CDlgFriend* friend_client, bool setfocus)
  72 {
  73
  74         if ( !friend_client->m_name.IsEmpty() ) {
  75                 if (setfocus) {
  76                         theApp->amuledlg->SetActiveDialog(CamuleDlg::DT_CHAT_WND, this);
  77                 }
  78                 chatselector->StartSession(GUI_ID(friend_client->m_ip, friend_client->m_port), friend_client->m_name, true);
  79         }
  80
  81         // Check to enable the window controls if needed
  82         CheckNewButtonsState();
  83 }
  84
  85
  86 void CChatWnd::OnNMRclickChatTab(wxMouseEvent& evt)
  87 {
  88         // Only handle events from the chat-notebook
  89         if (evt.GetEventObject() != (wxObject*)chatselector)
  90                 return;
  91
  92         if (chatselector->GetSelection() == -1) {
  93                 return;
  94         }
  95
  96         // Avoid opening another menu when it's already open
  97         if (m_menu == NULL) {
  98                 m_menu = new wxMenu(_("Chat"));
  99
 100                 m_menu->Append(MP_CLOSE_TAB, wxString(_("Close tab")));
 101                 m_menu->Append(MP_CLOSE_ALL_TABS, wxString(_("Close all tabs")));
 102                 m_menu->Append(MP_CLOSE_OTHER_TABS, wxString(_("Close other tabs")));
 103
 104                 m_menu->AppendSeparator();
 105
 106                 wxMenuItem * addFriend = m_menu->Append(MP_ADDFRIEND, _("Add to Friends"));
 107
 108                 // Disable this client if it is already a friend
 109                 CUpDownClient* client = chatselector->GetCurrentClient();
 110                 if (client && client->IsFriend()) {
 111                         addFriend->Enable(false);
 112                 }
 113
 114                 PopupMenu(m_menu, evt.GetPosition());
 115
 116                 delete m_menu;
 117                 m_menu = NULL;
 118         }
 119 }
 120
 121
 122 void CChatWnd::OnPopupClose(wxCommandEvent& WXUNUSED(evt))
 123 {
 124         chatselector->DeletePage(chatselector->GetSelection());
 125 }
 126
 127
 128 void CChatWnd::OnPopupCloseAll(wxCommandEvent& WXUNUSED(evt))
 129 {
 130         chatselector->DeleteAllPages();
 131 }
 132
 133
 134 void CChatWnd::OnPopupCloseOthers(wxCommandEvent& WXUNUSED(evt))
 135 {
 136         wxNotebookPage* current = chatselector->GetPage(chatselector->GetSelection());
 137
 138         for (int i = chatselector->GetPageCount() - 1; i >= 0; i--) {
 139                 if (current != chatselector->GetPage(i))
 140                         chatselector->DeletePage( i );
 141         }
 142 }
 143
 144
 145 void CChatWnd::OnAddFriend(wxCommandEvent& WXUNUSED(evt))
 146 {
 147         // Get the client that the session is open to
 148         CUpDownClient* client = chatselector->GetCurrentClient();
 149
 150         // Add the client as friend unless it's already a friend
 151         if (client && !client->IsFriend()) {
 152                 AddFriend(client);
 153         }
 154 }
 155
 156
 157 void CChatWnd::OnBnClickedCsend(wxCommandEvent& WXUNUSED(evt))
 158 {
 159         wxString message = CastChild(IDC_CMESSAGE, wxTextCtrl)->GetValue();
 160
 161         SendMessage(message);
 162 }
 163
 164
 165 void CChatWnd::OnBnClickedCclose(wxCommandEvent& WXUNUSED(evt))
 166 {
 167         chatselector->EndSession();
 168 }
 169
 170
 171 void CChatWnd::OnAllPagesClosed(wxNotebookEvent& WXUNUSED(evt))
 172 {
 173         CastChild(IDC_CMESSAGE, wxTextCtrl)->Clear();
 174         // Check to disable the window controls
 175         CheckNewButtonsState();
 176 }
 177
 178
 179 CDlgFriend* CChatWnd::FindFriend(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort)
 180 {
 181         return friendlistctrl->FindFriend(userhash, dwIP, nPort);
 182 }
 183
 184
 185 void CChatWnd::AddFriend(CUpDownClient* toadd)
 186 {
 187         friendlistctrl->AddFriend(toadd);
 188 }
 189
 190
 191 void CChatWnd::AddFriend(const CMD4Hash& userhash, const wxString& name, uint32 lastUsedIP, uint32 lastUsedPort)
 192 {
 193         friendlistctrl->AddFriend( userhash, name, lastUsedIP, lastUsedPort);
 194 }
 195
 196 void CChatWnd::RemoveFriend(const CMD4Hash& userhash, uint32 lastUsedIP, uint32 lastUsedPort)
 197 {
 198         friendlistctrl->RemoveFriend(friendlistctrl->FindFriend(userhash, lastUsedIP, lastUsedPort));
 199 }
 200
 201 void CChatWnd::RefreshFriend(CFriend* Friend, bool connected)
 202 {
 203         CDlgFriend* toupdate = friendlistctrl->FindFriend(Friend->GetUserHash(), Friend->GetIP(), Friend->GetPort());
 204         if (toupdate) {
 205                 toupdate->m_name = Friend->GetName();
 206                 toupdate->islinked = connected;
 207                 if (toupdate->m_ip == Friend->GetIP() && toupdate->m_port == Friend->GetPort()) {
 208                         chatselector->RefreshFriend(GUI_ID(toupdate->m_ip, toupdate->m_port), toupdate->m_name);
 209                 } else {
 210                         // Friend changed IP - drop Chat session
 211                         chatselector->EndSession(GUI_ID(toupdate->m_ip, toupdate->m_port));
 212                         // and update IP
 213                         toupdate->m_ip = Friend->GetIP();
 214                         toupdate->m_port = Friend->GetPort();
 215                 }
 216                 friendlistctrl->RefreshFriend(toupdate);
 217         }
 218 }
 219
 220 void CChatWnd::ProcessMessage(uint64 sender, const wxString& message)
 221 {
 222         if ( !theApp->amuledlg->IsDialogVisible(CamuleDlg::DT_CHAT_WND) ) {
 223                 theApp->amuledlg->SetMessageBlink(true);
 224         }
 225         if (chatselector->ProcessMessage(sender, message)) {
 226                 // Check to enable the window controls if needed
 227                 CheckNewButtonsState();
 228         }
 229 }
 230
 231 void CChatWnd::ConnectionResult(bool success, const wxString& message, uint64 id)
 232 {
 233         chatselector->ConnectionResult(success, message, id);
 234 }
 235
 236 void CChatWnd::SendMessage(const wxString& message, const wxString& client_name, uint64 to_id)
 237 {
 238
 239         if (chatselector->SendMessage( message, client_name, to_id )) {
 240                 CastChild(IDC_CMESSAGE, wxTextCtrl)->Clear();
 241         }
 242
 243         // Check to enable the window controls if needed
 244         CheckNewButtonsState();
 245         CastChild(IDC_CMESSAGE, wxTextCtrl)->SetFocus();
 246 }
 247
 248
 249 void CChatWnd::CheckNewButtonsState()
 250 {
 251         switch (chatselector->GetPageCount()) {
 252                         case 0:
 253                                 GetParent()->FindWindow(IDC_CSEND)->Enable(false);
 254                                 GetParent()->FindWindow(IDC_CCLOSE)->Enable(false);
 255                                 GetParent()->FindWindow(IDC_CMESSAGE)->Enable(false);
 256                                 break;
 257                         case 1:
 258                                 GetParent()->FindWindow(IDC_CSEND)->Enable(true);
 259                                 GetParent()->FindWindow(IDC_CCLOSE)->Enable(true);
 260                                 GetParent()->FindWindow(IDC_CMESSAGE)->Enable(true);
 261                                 break;
 262                         default:
 263                                 // Nothing to be done here. Keep current state, which should be enabled.
 264                                 wxASSERT(GetParent()->FindWindow(IDC_CSEND)->IsEnabled());
 265                                 wxASSERT(GetParent()->FindWindow(IDC_CCLOSE)->IsEnabled());
 266                                 wxASSERT(GetParent()->FindWindow(IDC_CMESSAGE)->IsEnabled());
 267                                 break;
 268         }
 269 }
 270
 271
 272 bool CChatWnd::IsIdValid(uint64 id)
 273 {
 274         return chatselector->GetTabByClientID(id) >= 0;
 275 }
 276
 277
 278 void CChatWnd::ShowCaptchaResult(uint64 id, bool ok)
 279 {
 280         chatselector->ShowCaptchaResult(id, ok);
 281 }
 282
 283 void CChatWnd::EndSession(uint64 id)
 284 {
 285         chatselector->EndSession(id);
 286 }
 287
 288 // File_checked_for_headers