src/FriendListCtrl.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
  27 #include "FriendListCtrl.h"     // Interface declarations
  28
  29 #include <common/MenuIDs.h>
  30 #include <common/Macros.h>
  31
  32 #include "amule.h"              // Needed for theApp
  33 #include "amuleDlg.h"           // Needed for CamuleDlg
  34 #include "ClientDetailDialog.h" // Needed for CClientDetailDialog
  35 #include "AddFriend.h"          // Needed for CAddFriend
  36 #include "ChatWnd.h"            // Needed for CChatWnd
  37 #include "updownclient.h"       // Needed for CUpDownClient
  38 #include "Friend.h"             // Needed for CFriend
  39 #include "muuli_wdr.h"
  40 #include "SafeFile.h"
  41
  42 //#warning REMOVE WHEN EC IS CODED!
  43 #include "FriendList.h"         // Needed for the friends list
  44
  45 BEGIN_EVENT_TABLE(CFriendListCtrl, CMuleListCtrl)
  46         EVT_RIGHT_DOWN(CFriendListCtrl::OnRightClick)
  47         EVT_LIST_ITEM_ACTIVATED(ID_FRIENDLIST, CFriendListCtrl::OnItemActivated)
  48
  49         EVT_MENU(MP_MESSAGE, CFriendListCtrl::OnSendMessage)
  50         EVT_MENU(MP_REMOVEFRIEND, CFriendListCtrl::OnRemoveFriend)
  51         EVT_MENU(MP_ADDFRIEND, CFriendListCtrl::OnAddFriend)
  52         EVT_MENU(MP_DETAIL, CFriendListCtrl::OnShowDetails)
  53         EVT_MENU(MP_SHOWLIST, CFriendListCtrl::OnViewFiles)
  54         EVT_MENU(MP_FRIENDSLOT, CFriendListCtrl::OnSetFriendslot)
  55
  56         EVT_CHAR(CFriendListCtrl::OnKeyPressed)
  57 END_EVENT_TABLE()
  58
  59
  60 CDlgFriend::CDlgFriend(const CMD4Hash& hash, const wxString& name, uint32 ip, uint16 port, bool IsLinked, bool HasFriendSlot)
  61 {
  62         m_hash = hash;
  63
  64         if (name.IsEmpty()) {
  65                 m_name = wxT("?");
  66         } else {
  67                 m_name = name;
  68         }
  69
  70         m_ip = ip;
  71         m_port = port;
  72         islinked = IsLinked;
  73         hasfriendslot = HasFriendSlot;
  74 }
  75
  76
  77 CFriendListCtrl::CFriendListCtrl(wxWindow* parent, int id, const wxPoint& pos, wxSize siz, int flags)
  78 : CMuleListCtrl(parent, id, pos, siz, flags)
  79 {
  80   InsertColumn(0, _("Username"), wxLIST_FORMAT_LEFT, siz.GetWidth() - 4);
  81
  82   LoadList();
  83 }
  84
  85 CFriendListCtrl::~CFriendListCtrl()
  86 {
  87         for ( int i = 0; i < GetItemCount(); i++ )  {
  88                 delete (CDlgFriend*)GetItemData(i);
  89         }
  90 }
  91
  92
  93 void CFriendListCtrl::AddFriend(CDlgFriend* toadd, bool send_to_core)
  94 {
  95         uint32 itemnr = InsertItem(GetItemCount(), wxEmptyString);
  96         SetItemPtrData(itemnr, reinterpret_cast<wxUIntPtr>(toadd));
  97         RefreshFriend(toadd);   // set name and colour
  98
  99         //#warning CORE/GUI
 100         if (send_to_core) {
 101         #ifndef CLIENT_GUI
 102                 theApp->friendlist->AddFriend(toadd->m_hash, 0, toadd->m_ip, toadd->m_port, 0,toadd->m_name);
 103         #endif
 104         }
 105 }
 106
 107
 108 void CFriendListCtrl::AddFriend(const CMD4Hash& userhash, const wxString& name, uint32 lastUsedIP, uint32 lastUsedPort, bool IsLinked, bool HasFriendSlot, bool send_to_core)
 109 {
 110         CDlgFriend* NewFriend = new CDlgFriend( userhash, name, lastUsedIP, lastUsedPort, IsLinked, HasFriendSlot);
 111
 112         AddFriend( NewFriend, send_to_core);
 113 }
 114
 115
 116 void CFriendListCtrl::AddFriend(CUpDownClient* toadd)
 117 {
 118         if ( toadd->IsFriend() ) {
 119                 return;
 120         }
 121
 122         //#warning CORE/GUI
 123         // This links the friend to the client also
 124         #ifndef CLIENT_GUI
 125         theApp->friendlist->AddFriend(toadd);
 126         #endif
 127
 128         CDlgFriend* NewFriend = new CDlgFriend( toadd->GetUserHash(), toadd->GetUserName(), toadd->GetIP(), toadd->GetUserPort(), true, false);
 129
 130         AddFriend( NewFriend, false/*already sent to core*/ );
 131 }
 132
 133
 134 void CFriendListCtrl::RemoveFriend(CDlgFriend* toremove)
 135 {
 136         if (!toremove) {
 137                 return;
 138         }
 139
 140         sint32 itemnr = FindItem(-1, reinterpret_cast<wxUIntPtr>(toremove));
 141
 142         if ( itemnr == -1 )
 143                 return;
 144
 145         //#warning CORE/GUI
 146         #ifndef CLIENT_GUI
 147         theApp->friendlist->RemoveFriend(toremove->m_hash, toremove->m_ip, toremove->m_port);
 148         #endif
 149
 150         DeleteItem(itemnr);
 151 }
 152
 153
 154 void CFriendListCtrl::RefreshFriend(CDlgFriend* toupdate)
 155 {
 156         sint32 itemnr = FindItem(-1, reinterpret_cast<wxUIntPtr>(toupdate));
 157         if (itemnr != -1) {
 158                 SetItem(itemnr, 0, toupdate->m_name);
 159                 SetItemTextColour(itemnr, toupdate->islinked ? *wxBLUE : *wxBLACK);
 160         }
 161         //#warning CORE/GUI
 162         #ifndef CLIENT_GUI
 163         theApp->friendlist->UpdateFriendName(toupdate->m_hash, toupdate->m_name, toupdate->m_ip, toupdate->m_port);
 164         #endif
 165 }
 166
 167
 168 void CFriendListCtrl::OnItemActivated(wxListEvent& WXUNUSED(event))
 169 {
 170         int cursel = GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 171
 172         CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(cursel);
 173
 174         /* ignore this one, it is not activated anymore :) */
 175         if (cur_friend == NULL) {
 176                 return;
 177         }
 178
 179         theApp->amuledlg->m_chatwnd->StartSession(cur_friend);
 180
 181 }
 182
 183
 184 void CFriendListCtrl::LoadList()
 185 {
 186         //#warning EC: ASK THE LIST TO CORE!
 187
 188         #ifndef CLIENT_GUI
 189         for(FriendList::iterator it = theApp->friendlist->m_FriendList.begin(); it != theApp->friendlist->m_FriendList.end(); ++it) {
 190                 CFriend* core_friend = *it;
 191                 AddFriend(core_friend->GetUserHash(), core_friend->GetName(), core_friend->GetIP(), core_friend->GetPort(), (core_friend->GetLinkedClient() != NULL), core_friend->HasFriendSlot(), false);
 192         }
 193         #endif
 194
 195 }
 196
 197 CDlgFriend* CFriendListCtrl::FindFriend(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort)
 198 {
 199         for ( int i = 0; i < GetItemCount(); i++ ) {
 200                 CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(i);
 201
 202                 // to avoid that unwanted clients become a friend, we have to distinguish between friends with
 203                 // a userhash and of friends which are identified by IP+port only.
 204                 if ( !userhash.IsEmpty() && !cur_friend->m_hash.IsEmpty() ) {
 205                         // check for a friend which has the same userhash as the specified one
 206                         if (cur_friend->m_hash == userhash) {
 207                                 return cur_friend;
 208                         }
 209                 }
 210                 else if (cur_friend->m_ip == dwIP && cur_friend->m_port == nPort) {
 211                                 return cur_friend;
 212                 }
 213         }
 214
 215         return NULL;
 216 }
 217
 218 bool CFriendListCtrl::IsAlreadyFriend( uint32 dwLastUsedIP, uint32 nLastUsedPort )
 219 {
 220         return (FindFriend( CMD4Hash(), dwLastUsedIP, nLastUsedPort ) != NULL);
 221 }
 222
 223 void CFriendListCtrl::OnRightClick(wxMouseEvent& event)
 224 {
 225         int cursel = CheckSelection(event);
 226
 227         CDlgFriend* cur_friend = NULL;
 228
 229         wxMenu* menu = new wxMenu(_("Friends"));
 230
 231         if ( cursel != -1 ) {
 232                 cur_friend = (CDlgFriend*)GetItemData(cursel);
 233                 menu->Append(MP_DETAIL, _("Show &Details"));
 234                 menu->Enable(MP_DETAIL, cur_friend->islinked);
 235         }
 236
 237         menu->Append(MP_ADDFRIEND, _("Add a friend"));
 238
 239         if (cursel != (-1)) {
 240                 menu->Append(MP_REMOVEFRIEND, _("Remove Friend"));
 241                 menu->Append(MP_MESSAGE, _("Send &Message"));
 242                 menu->Append(MP_SHOWLIST, _("View Files"));
 243                 menu->AppendCheckItem(MP_FRIENDSLOT, _("Establish Friend Slot"));
 244
 245                 if (cur_friend->islinked) {
 246                         menu->Enable(MP_FRIENDSLOT, true);
 247                         menu->Check(MP_FRIENDSLOT, cur_friend->hasfriendslot);
 248                 } else {
 249                         menu->Enable(MP_FRIENDSLOT, false);
 250                 }
 251         }
 252
 253         PopupMenu(menu, event.GetPosition());
 254         delete menu;
 255 }
 256
 257 void CFriendListCtrl::OnSendMessage(wxCommandEvent& WXUNUSED(event)) {
 258         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 259
 260         while( index != -1 ) {
 261                 CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(index);
 262                 theApp->amuledlg->m_chatwnd->StartSession(cur_friend);
 263                 //#warning CORE/GUI!
 264                 #ifndef CLIENT_GUI
 265                 theApp->friendlist->StartChatSession(cur_friend->m_hash, cur_friend->m_ip, cur_friend->m_port);
 266                 #endif
 267
 268                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 269         }
 270 }
 271
 272
 273 void CFriendListCtrl::OnRemoveFriend(wxCommandEvent& WXUNUSED(event))
 274 {
 275         wxString question;
 276         if (GetSelectedItemCount() == 1) {
 277                 question = _("Are you sure that you wish to delete the selected friend?");
 278         } else {
 279                 question = _("Are you sure that you wish to delete the selected friends?");
 280         }
 281
 282         if ( wxMessageBox( question, _("Cancel"), wxICON_QUESTION | wxYES_NO, this) == wxYES ) {
 283                 long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 284
 285                 while( index != -1 ) {
 286                         CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(index);
 287                         RemoveFriend(cur_friend);
 288                         // -1 because we changed the list and removed that item.
 289                         index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 290                 }
 291         }
 292 }
 293
 294
 295 void CFriendListCtrl::OnAddFriend(wxCommandEvent& WXUNUSED(event))
 296 {
 297         CAddFriend(this).ShowModal();
 298 }
 299
 300
 301 void CFriendListCtrl::OnShowDetails(wxCommandEvent& WXUNUSED(event))
 302 {
 303         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 304
 305         while( index != -1 ) {
 306                 CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(index);
 307                 if (cur_friend->islinked) {
 308                         //#warning EC: We need a reply packet with a full CUpDownClient
 309                         #ifndef CLIENT_GUI
 310                         CClientDetailDialog
 311                                 (
 312                                 this,
 313                                 theApp->friendlist->FindFriend
 314                                         (
 315                                         cur_friend->m_hash,
 316                                         cur_friend->m_ip,
 317                                         cur_friend->m_port
 318                                         )->GetLinkedClient()
 319                                 ).ShowModal();
 320                         #endif
 321                 }
 322                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 323         }
 324
 325 }
 326
 327
 328 void CFriendListCtrl::OnViewFiles(wxCommandEvent& WXUNUSED(event))
 329 {
 330         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 331
 332         while( index != -1 ) {
 333                 //#warning CORE/GUI!
 334                 #ifndef CLIENT_GUI
 335                         CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(index);
 336                         theApp->friendlist->RequestSharedFileList(cur_friend->m_hash, cur_friend->m_ip, cur_friend->m_port);
 337                 #endif
 338                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 339         }
 340
 341 }
 342
 343
 344 void CFriendListCtrl::OnSetFriendslot(wxCommandEvent& NOT_ON_REMOTEGUI(event))
 345 {
 346         // Clean friendslots
 347         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_DONTCARE);
 348         while (index != -1) {
 349                 CDlgFriend* friend_item = (CDlgFriend*)GetItemData(index);
 350                 friend_item->hasfriendslot = false;
 351                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_DONTCARE);
 352         }
 353         // Now set the proper one
 354         index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 355         //#warning CORE/GUI!
 356         #ifndef CLIENT_GUI
 357                 CDlgFriend* cur_friend = (CDlgFriend*)GetItemData(index);
 358                 cur_friend->hasfriendslot = event.IsChecked();
 359                 theApp->friendlist->SetFriendSlot(cur_friend->m_hash, cur_friend->m_ip, cur_friend->m_port, cur_friend->hasfriendslot);
 360         #endif
 361         index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 362         if (index != -1) {
 363                 wxMessageBox(_("You are not allowed to set more than one friendslot.\n Only one slot was assigned."), _("Multiple selection"), wxOK | wxICON_ERROR, this);
 364         }
 365 }
 366
 367
 368 void CFriendListCtrl::SetLinked(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort, bool new_state)
 369 {
 370         CDlgFriend* client = FindFriend(CMD4Hash(), dwIP, nPort);
 371         if (client) {
 372                 client->m_hash = userhash;
 373                 client->islinked = new_state;
 374                 RefreshFriend(client);
 375         }
 376 }
 377
 378
 379 void CFriendListCtrl::OnKeyPressed(wxKeyEvent& event)
 380 {
 381         // Check if delete was pressed
 382         if ((event.GetKeyCode() == WXK_DELETE) || (event.GetKeyCode() == WXK_NUMPAD_DELETE)) {
 383                 if (GetItemCount()) {
 384                         wxCommandEvent evt;
 385                         evt.SetId( MP_REMOVEFRIEND );
 386                         OnRemoveFriend( evt );
 387                 }
 388         } else {
 389                 event.Skip();
 390         }
 391 }
 392 // File_checked_for_headers