src/SharedFilesCtrl.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 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 "SharedFilesCtrl.h"    // Interface declarations
  27
  28 #include <common/MenuIDs.h>
  29
  30 #include "muuli_wdr.h"                  // Needed for ID_SHFILELIST
  31 #include "SharedFilesWnd.h"             // Needed for CSharedFilesWnd
  32 #include "amuleDlg.h"                   // Needed for CamuleDlg
  33 #include "CommentDialog.h"              // Needed for CCommentDialog
  34 #include "PartFile.h"                   // Needed for CPartFile
  35 #include "SharedFileList.h"             // Needed for CKnownFileMap
  36 #include "amule.h"                              // Needed for theApp
  37 #include "ServerConnect.h"              // Needed for CServerConnect
  38 #include "Preferences.h"                // Needed for thePrefs
  39 #include "BarShader.h"                  // Needed for CBarShader
  40 #include "DataToText.h"                 // Needed for PriorityToStr
  41 #include "GuiEvents.h"                  // Needed for CoreNotify_*
  42 #include "MuleCollection.h"             // Needed for CMuleCollection
  43 #include "DownloadQueue.h"              // Needed for CDownloadQueue
  44 #include "TransferWnd.h"                // Needed for CTransferWnd
  45
  46
  47 BEGIN_EVENT_TABLE(CSharedFilesCtrl,CMuleListCtrl)
  48         EVT_LIST_ITEM_RIGHT_CLICK(-1, CSharedFilesCtrl::OnRightClick)
  49
  50         EVT_MENU( MP_PRIOVERYLOW,       CSharedFilesCtrl::OnSetPriority )
  51         EVT_MENU( MP_PRIOLOW,           CSharedFilesCtrl::OnSetPriority )
  52         EVT_MENU( MP_PRIONORMAL,        CSharedFilesCtrl::OnSetPriority )
  53         EVT_MENU( MP_PRIOHIGH,          CSharedFilesCtrl::OnSetPriority )
  54         EVT_MENU( MP_PRIOVERYHIGH,      CSharedFilesCtrl::OnSetPriority )
  55         EVT_MENU( MP_POWERSHARE,        CSharedFilesCtrl::OnSetPriority )
  56         EVT_MENU( MP_PRIOAUTO,          CSharedFilesCtrl::OnSetPriorityAuto )
  57
  58         EVT_MENU( MP_CMT,                       CSharedFilesCtrl::OnEditComment )
  59         EVT_MENU( MP_ADDCOLLECTION,             CSharedFilesCtrl::OnAddCollection )
  60         EVT_MENU( MP_GETMAGNETLINK,             CSharedFilesCtrl::OnCreateURI )
  61         EVT_MENU( MP_GETED2KLINK,                               CSharedFilesCtrl::OnCreateURI )
  62         EVT_MENU( MP_GETSOURCEED2KLINK,                 CSharedFilesCtrl::OnCreateURI )
  63         EVT_MENU( MP_GETCRYPTSOURCEDED2KLINK,                   CSharedFilesCtrl::OnCreateURI )
  64         EVT_MENU( MP_GETHOSTNAMESOURCEED2KLINK, CSharedFilesCtrl::OnCreateURI )
  65         EVT_MENU( MP_GETHOSTNAMECRYPTSOURCEED2KLINK,                    CSharedFilesCtrl::OnCreateURI )
  66         EVT_MENU( MP_GETAICHED2KLINK,   CSharedFilesCtrl::OnCreateURI )
  67         EVT_MENU( MP_RENAME,            CSharedFilesCtrl::OnRename )
  68         EVT_MENU( MP_WS,                CSharedFilesCtrl::OnGetFeedback )
  69
  70
  71         EVT_CHAR( CSharedFilesCtrl::OnKeyPressed )
  72 END_EVENT_TABLE()
  73
  74 enum SharedFilesListColumns {
  75         ID_SHARED_COL_NAME = 0,
  76         ID_SHARED_COL_SIZE,
  77         ID_SHARED_COL_TYPE,
  78         ID_SHARED_COL_PRIO,
  79         ID_SHARED_COL_ID,
  80         ID_SHARED_COL_REQ,
  81         ID_SHARED_COL_AREQ,
  82         ID_SHARED_COL_TRA,
  83         ID_SHARED_COL_RTIO,
  84         ID_SHARED_COL_PART,
  85         ID_SHARED_COL_CMPL,
  86         ID_SHARED_COL_PATH
  87 };
  88
  89
  90 CSharedFilesCtrl::CSharedFilesCtrl(wxWindow* parent, int id, const wxPoint& pos, wxSize size, int flags)
  91         : CMuleListCtrl(parent, id, pos, size, flags | wxLC_OWNERDRAW )
  92 {
  93         // Setting the sorter function.
  94         SetSortFunc( SortProc );
  95
  96         // Set the table-name (for loading and saving preferences).
  97         SetTableName( wxT("Shared") );
  98
  99         m_menu=NULL;
 100
 101         InsertColumn(ID_SHARED_COL_NAME, _("File Name"),                wxLIST_FORMAT_LEFT, 250, wxT("N") );
 102         InsertColumn(ID_SHARED_COL_SIZE, _("Size"),                     wxLIST_FORMAT_LEFT, 100, wxT("Z") );
 103         InsertColumn(ID_SHARED_COL_TYPE, _("Type"),                     wxLIST_FORMAT_LEFT,  50, wxT("Y") );
 104         InsertColumn(ID_SHARED_COL_PRIO, _("Priority"),                 wxLIST_FORMAT_LEFT,  70, wxT("p") );
 105         InsertColumn(ID_SHARED_COL_ID,   _("FileID"),                   wxLIST_FORMAT_LEFT, 220, wxT("I") );
 106         InsertColumn(ID_SHARED_COL_REQ,  _("Requests"),                 wxLIST_FORMAT_LEFT, 100, wxT("Q") );
 107         InsertColumn(ID_SHARED_COL_AREQ, _("Accepted Requests"),        wxLIST_FORMAT_LEFT, 100, wxT("A") );
 108         InsertColumn(ID_SHARED_COL_TRA,  _("Transferred Data"),         wxLIST_FORMAT_LEFT, 120, wxT("T") );
 109         InsertColumn(ID_SHARED_COL_RTIO, _("Share Ratio"),              wxLIST_FORMAT_LEFT, 100, wxT("R") );
 110         InsertColumn(ID_SHARED_COL_PART, _("Obtained Parts"),           wxLIST_FORMAT_LEFT, 120, wxT("P") );
 111         InsertColumn(ID_SHARED_COL_CMPL, _("Complete Sources"),         wxLIST_FORMAT_LEFT, 120, wxT("C") );
 112         InsertColumn(ID_SHARED_COL_PATH, _("Directory Path"),           wxLIST_FORMAT_LEFT, 220, wxT("D") );
 113
 114         LoadSettings();
 115 }
 116
 117
 118 wxString CSharedFilesCtrl::GetOldColumnOrder() const
 119 {
 120         return wxT("N,Z,Y,p,I,Q,A,T,R,P,C,D");
 121 }
 122
 123
 124 CSharedFilesCtrl::~CSharedFilesCtrl()
 125 {
 126 }
 127
 128
 129 void CSharedFilesCtrl::OnRightClick(wxListEvent& event)
 130 {
 131         long item_hit = CheckSelection(event);
 132
 133         if ( (m_menu == NULL) && (item_hit != -1)) {
 134                 m_menu = new wxMenu(_("Shared Files"));
 135                 wxMenu* prioMenu = new wxMenu();
 136                 prioMenu->AppendCheckItem(MP_PRIOVERYLOW, _("Very low"));
 137                 prioMenu->AppendCheckItem(MP_PRIOLOW, _("Low"));
 138                 prioMenu->AppendCheckItem(MP_PRIONORMAL, _("Normal"));
 139                 prioMenu->AppendCheckItem(MP_PRIOHIGH, _("High"));
 140                 prioMenu->AppendCheckItem(MP_PRIOVERYHIGH, _("Very High"));
 141                 prioMenu->AppendCheckItem(MP_POWERSHARE, _("Release"));
 142                 prioMenu->AppendCheckItem(MP_PRIOAUTO, _("Auto"));
 143
 144                 m_menu->Append(0,_("Priority"),prioMenu);
 145                 m_menu->AppendSeparator();
 146
 147                 CKnownFile* file = (CKnownFile*)GetItemData(item_hit);
 148                 if (file->GetFileComment().IsEmpty() && !file->GetFileRating()) {
 149                         m_menu->Append(MP_CMT, _("Add Comment/Rating"));
 150                 } else {
 151                         m_menu->Append(MP_CMT, _("Edit Comment/Rating"));
 152                 }
 153
 154                 m_menu->AppendSeparator();
 155                 m_menu->Append(MP_RENAME, _("Rename"));
 156                 m_menu->AppendSeparator();
 157
 158                 if (file->GetFileName().GetExt() == wxT("emulecollection")) {
 159                         m_menu->Append( MP_ADDCOLLECTION, _("Add files in collection to transfer list"));
 160                         m_menu->AppendSeparator();
 161                 }
 162                 m_menu->Append(MP_GETMAGNETLINK,_("Copy magnet &URI to clipboard"));
 163                 m_menu->Append(MP_GETED2KLINK,_("Copy eD2k &link to clipboard"));
 164                 m_menu->Append(MP_GETSOURCEED2KLINK,_("Copy eD2k link to clipboard (&Source)"));
 165                 m_menu->Append(MP_GETCRYPTSOURCEDED2KLINK,_("Copy eD2k link to clipboard (Source) (&With Crypt options)"));
 166                 m_menu->Append(MP_GETHOSTNAMESOURCEED2KLINK,_("Copy eD2k link to clipboard (&Hostname)"));
 167                 m_menu->Append(MP_GETHOSTNAMECRYPTSOURCEED2KLINK,_("Copy eD2k link to clipboard (Hostname) (With &Crypt options)"));
 168                 m_menu->Append(MP_GETAICHED2KLINK,_("Copy eD2k link to clipboard (&AICH info)"));
 169                 m_menu->Append(MP_WS,_("Copy feedback to clipboard"));
 170
 171                 m_menu->Enable(MP_GETAICHED2KLINK, file->HasProperAICHHashSet());
 172                 m_menu->Enable(MP_GETHOSTNAMESOURCEED2KLINK, !thePrefs::GetYourHostname().IsEmpty());
 173                 m_menu->Enable(MP_GETHOSTNAMECRYPTSOURCEED2KLINK, !thePrefs::GetYourHostname().IsEmpty());
 174                 m_menu->Enable(MP_RENAME, file->IsPartFile());
 175
 176                 int priority = file->IsAutoUpPriority() ? PR_AUTO : file->GetUpPriority();
 177
 178                 prioMenu->Check(MP_PRIOVERYLOW, priority == PR_VERYLOW);
 179                 prioMenu->Check(MP_PRIOLOW,     priority == PR_LOW);
 180                 prioMenu->Check(MP_PRIONORMAL,  priority == PR_NORMAL);
 181                 prioMenu->Check(MP_PRIOHIGH,    priority == PR_HIGH);
 182                 prioMenu->Check(MP_PRIOVERYHIGH,priority == PR_VERYHIGH);
 183                 prioMenu->Check(MP_POWERSHARE,  priority == PR_POWERSHARE);
 184                 prioMenu->Check(MP_PRIOAUTO,    priority == PR_AUTO);
 185
 186                 PopupMenu( m_menu, event.GetPoint() );
 187
 188                 delete m_menu;
 189
 190                 m_menu = NULL;
 191         }
 192 }
 193
 194
 195 void CSharedFilesCtrl::OnGetFeedback(wxCommandEvent& WXUNUSED(event))
 196 {
 197         wxString feed;
 198         long index = GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 199         while (index != -1) {
 200                 if (feed.IsEmpty()) {
 201                         feed = CFormat(_("Feedback from: %s (%s)\n\n")) % thePrefs::GetUserNick() % GetFullMuleVersion();
 202                 } else {
 203                         feed += wxT("\n");
 204                 }
 205                 feed += ((CKnownFile*)GetItemData(index))->GetFeedback();
 206                 index = GetNextItem(index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
 207         }
 208
 209         if (!feed.IsEmpty()) {
 210                 theApp->CopyTextToClipboard(feed);
 211         }
 212 }
 213
 214
 215 #ifndef CLIENT_GUI
 216 void CSharedFilesCtrl::ShowFileList()
 217 {
 218         Freeze();
 219         DeleteAllItems();
 220
 221         std::vector<CKnownFile*> files;
 222         theApp->sharedfiles->CopyFileList(files);
 223         for (unsigned i = 0; i < files.size(); ++i) {
 224                 DoShowFile(files[i], true);
 225         }
 226
 227         SortList();
 228         ShowFilesCount();
 229
 230         Thaw();
 231 }
 232 #endif
 233
 234
 235 void CSharedFilesCtrl::RemoveFile(CKnownFile *toRemove)
 236 {
 237         long index = FindItem( -1, reinterpret_cast<wxUIntPtr>(toRemove) );
 238
 239         if ( index != -1 ) {
 240                 DeleteItem( index );
 241
 242                 ShowFilesCount();
 243         }
 244 }
 245
 246
 247 void CSharedFilesCtrl::ShowFile(CKnownFile* file)
 248 {
 249         DoShowFile(file, false);
 250 }
 251
 252
 253 void CSharedFilesCtrl::DoShowFile(CKnownFile* file, bool batch)
 254 {
 255         wxUIntPtr ptr = reinterpret_cast<wxUIntPtr>(file);
 256         if ((!batch) && (FindItem(-1, ptr) > -1)) {
 257                 return;
 258         }
 259
 260         const long insertPos = (batch ? GetItemCount() : GetInsertPos(ptr));
 261
 262         long newitem = InsertItem(insertPos, wxEmptyString);
 263         SetItemPtrData( newitem, ptr );
 264
 265         if (!batch) {
 266                 ShowFilesCount();
 267         }
 268 }
 269
 270 void CSharedFilesCtrl::OnSetPriority( wxCommandEvent& event )
 271 {
 272         int priority = 0;
 273
 274         switch ( event.GetId() ) {
 275                 case MP_PRIOVERYLOW:    priority = PR_VERYLOW;  break;
 276                 case MP_PRIOLOW:                priority = PR_LOW;              break;
 277                 case MP_PRIONORMAL:             priority = PR_NORMAL;   break;
 278                 case MP_PRIOHIGH:               priority = PR_HIGH;             break;
 279                 case MP_PRIOVERYHIGH:   priority = PR_VERYHIGH; break;
 280                 case MP_POWERSHARE:             priority = PR_POWERSHARE; break;
 281         }
 282
 283         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 284
 285         while( index != -1 ) {
 286                 CKnownFile* file = (CKnownFile*)GetItemData( index );
 287                 CoreNotify_KnownFile_Up_Prio_Set( file, priority );
 288
 289                 RefreshItem( index );
 290
 291                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 292         }
 293 }
 294
 295
 296 void CSharedFilesCtrl::OnSetPriorityAuto( wxCommandEvent& WXUNUSED(event) )
 297 {
 298         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 299
 300         while( index != -1 ) {
 301                 CKnownFile* file = (CKnownFile*)GetItemData( index );
 302                 CoreNotify_KnownFile_Up_Prio_Auto(file);
 303
 304                 RefreshItem( index );
 305
 306                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 307         }
 308 }
 309
 310
 311 void CSharedFilesCtrl::OnCreateURI( wxCommandEvent& event )
 312 {
 313         wxString URIs;
 314
 315         if ( event.GetId() == MP_GETSOURCEED2KLINK || event.GetId() == MP_GETCRYPTSOURCEDED2KLINK) {
 316                 if ( !theApp->IsConnectedED2K() || theApp->serverconnect->IsLowID() ) {
 317                         wxMessageBox(_("You need a HighID to create a valid sourcelink"), _("WARNING"), wxOK | wxICON_ERROR, this);
 318
 319                         return;
 320                 }
 321         }
 322
 323         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 324
 325         while( index != -1 ) {
 326                 CKnownFile* file = (CKnownFile*)GetItemData( index );
 327
 328                 switch ( event.GetId() ) {
 329                         case MP_GETMAGNETLINK:                          URIs += theApp->CreateMagnetLink( file ) + wxT("\n");                           break;
 330                         case MP_GETED2KLINK:                            URIs += theApp->CreateED2kLink( file ) + wxT("\n");                                     break;
 331                         case MP_GETSOURCEED2KLINK:                      URIs += theApp->CreateED2kLink( file , true) + wxT("\n");                       break;
 332                         case MP_GETCRYPTSOURCEDED2KLINK:                        URIs += theApp->CreateED2kLink( file , true, false, true) + wxT("\n");                  break;
 333                         case MP_GETHOSTNAMESOURCEED2KLINK:      URIs += theApp->CreateED2kLink( file , true, true) + wxT("\n"); break;
 334                         case MP_GETHOSTNAMECRYPTSOURCEED2KLINK:                 URIs += theApp->CreateED2kLink( file, true, true, true ) + wxT("\n");                   break;
 335                         case MP_GETAICHED2KLINK:                        URIs += theApp->CreateED2kAICHLink( file ) + wxT("\n");                         break;
 336                 }
 337
 338                 index = GetNextItem( index, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 339         }
 340
 341         if ( !URIs.IsEmpty() ) {
 342                 theApp->CopyTextToClipboard( URIs.RemoveLast() );
 343         }
 344 }
 345
 346
 347 void CSharedFilesCtrl::OnEditComment( wxCommandEvent& WXUNUSED(event) )
 348 {
 349         long index = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 350
 351         if ( index != -1 ) {
 352                 CKnownFile* file = (CKnownFile*)GetItemData( index );
 353
 354                 CCommentDialog dialog( this, file );
 355
 356                 dialog.ShowModal();
 357         }
 358 }
 359
 360
 361 int CSharedFilesCtrl::SortProc(wxUIntPtr item1, wxUIntPtr item2, long sortData)
 362 {
 363         CKnownFile* file1 = (CKnownFile*)item1;
 364         CKnownFile* file2 = (CKnownFile*)item2;
 365
 366         int mod = (sortData & CMuleListCtrl::SORT_DES) ? -1 : 1;
 367         bool altSorting = (sortData & CMuleListCtrl::SORT_ALT) > 0;
 368
 369         switch (sortData & CMuleListCtrl::COLUMN_MASK) {
 370                 // Sort by filename.
 371                 case  ID_SHARED_COL_NAME:
 372                         return mod * CmpAny(file1->GetFileName(), file2->GetFileName());
 373
 374                 // Sort by filesize.
 375                 case  ID_SHARED_COL_SIZE:
 376                         return mod * CmpAny( file1->GetFileSize(), file2->GetFileSize() );
 377
 378                 // Sort by filetype.
 379                 case  ID_SHARED_COL_TYPE:
 380                         return mod * GetFiletypeByName(file1->GetFileName()).CmpNoCase(GetFiletypeByName( file2->GetFileName()) );
 381
 382                 // Sort by priority.
 383                 case  ID_SHARED_COL_PRIO: {
 384                         int8 prioA = file1->GetUpPriority();
 385                         int8 prioB = file2->GetUpPriority();
 386
 387                         // Work-around for PR_VERYLOW which has value 4. See KnownFile.h for that stupidity ...
 388                         return mod * CmpAny( ( prioB != PR_VERYLOW ? prioB : -1 ), ( prioA != PR_VERYLOW ? prioA : -1 ) );
 389                 }
 390
 391                 // Sort by fileID.
 392                 case  ID_SHARED_COL_ID:
 393                         return mod * file1->GetFileHash().Encode().Cmp( file2->GetFileHash().Encode() );
 394
 395                 // Sort by Requests this session.
 396                 case  ID_SHARED_COL_REQ:
 397                         if (altSorting) {
 398                                 return mod * CmpAny( file1->statistic.GetAllTimeRequests(), file2->statistic.GetAllTimeRequests() );
 399                         } else {
 400                                 return mod * CmpAny( file1->statistic.GetRequests(), file2->statistic.GetRequests() );
 401                         }
 402
 403                 // Sort by accepted requests. Ascending.
 404                 case  ID_SHARED_COL_AREQ:
 405                         if (altSorting) {
 406                                 return mod * CmpAny( file1->statistic.GetAllTimeAccepts(), file2->statistic.GetAllTimeAccepts() );
 407                         } else {
 408                                 return mod * CmpAny( file1->statistic.GetAccepts(), file2->statistic.GetAccepts() );
 409                         }
 410
 411                 // Sort by transferred. Ascending.
 412                 case  ID_SHARED_COL_TRA:
 413                         if (altSorting) {
 414                                 return mod * CmpAny( file1->statistic.GetAllTimeTransferred(), file2->statistic.GetAllTimeTransferred() );
 415                         } else {
 416                                 return mod * CmpAny( file1->statistic.GetTransferred(), file2->statistic.GetTransferred() );
 417                         }
 418
 419                 // Sort by Share Ratio. Ascending.
 420                 case  ID_SHARED_COL_RTIO:
 421                         return mod * CmpAny( (double)file1->statistic.GetAllTimeTransferred() / file1->GetFileSize(),
 422                                         (double)file2->statistic.GetAllTimeTransferred() / file2->GetFileSize() );
 423
 424                 // Complete sources asc
 425                 case ID_SHARED_COL_CMPL:
 426                         return mod * CmpAny( file1->m_nCompleteSourcesCount, file2->m_nCompleteSourcesCount );
 427
 428                 // Folders ascending
 429                 case ID_SHARED_COL_PATH: {
 430                         if ( file1->IsPartFile() && file2->IsPartFile() )
 431                                 return mod *  0;
 432                         if ( file1->IsPartFile() )
 433                                 return mod * -1;
 434                         if ( file2->IsPartFile() )
 435                                 return mod *  1;
 436
 437                         return mod * CmpAny(file1->GetFilePath(), file2->GetFilePath());
 438                 }
 439
 440                 default:
 441                         return 0;
 442         }
 443 }
 444
 445
 446 void CSharedFilesCtrl::UpdateItem(CKnownFile* toupdate)
 447 {
 448         long result = FindItem( -1, reinterpret_cast<wxUIntPtr>(toupdate) );
 449
 450         if ( result > -1 ) {
 451                 RefreshItem(result);
 452
 453                 if ( GetItemState( result, wxLIST_STATE_SELECTED ) ) {
 454                         theApp->amuledlg->m_sharedfileswnd->SelectionUpdated();
 455                 }
 456         }
 457 }
 458
 459
 460 void CSharedFilesCtrl::ShowFilesCount()
 461 {
 462         wxString str = wxString::Format(_("Shared Files (%i)"), GetItemCount());
 463         wxStaticText* label = CastByName( wxT("sharedFilesLabel"), GetParent(), wxStaticText );
 464
 465         label->SetLabel( str );
 466         label->GetParent()->Layout();
 467 }
 468
 469
 470 void CSharedFilesCtrl::OnDrawItem( int item, wxDC* dc, const wxRect& rect, const wxRect& rectHL, bool highlighted )
 471 {
 472         CKnownFile *file = (CKnownFile*)GetItemData(item);
 473         wxASSERT( file );
 474
 475         if ( highlighted ) {
 476                 CMuleColour newcol(GetFocus() ? wxSYS_COLOUR_HIGHLIGHT : wxSYS_COLOUR_BTNSHADOW);
 477                 dc->SetBackground(newcol.Blend(125).GetBrush());
 478                 dc->SetTextForeground( CMuleColour(wxSYS_COLOUR_HIGHLIGHTTEXT));
 479                 // The second blending goes over the first one.
 480                 dc->SetPen(newcol.Blend(65).GetPen());
 481         } else {
 482                 dc->SetBackground( CMuleColour(wxSYS_COLOUR_LISTBOX).GetBrush() );
 483                 dc->SetTextForeground(CMuleColour(wxSYS_COLOUR_WINDOWTEXT));
 484                 dc->SetPen(*wxTRANSPARENT_PEN);
 485         }
 486
 487         dc->SetBrush(dc->GetBackground());
 488         dc->DrawRectangle(rectHL);
 489         dc->SetPen(*wxTRANSPARENT_PEN);
 490
 491         // Offset based on the height of the fonts
 492         const int textVOffset = ( rect.GetHeight() - dc->GetCharHeight() ) / 2;
 493         // Empty space to each side of a column
 494         const int SPARE_PIXELS_HORZ     = 4;
 495
 496         // The leftmost position of the current column
 497         int columnLeft = 0;
 498
 499         for ( int i = 0; i < GetColumnCount(); ++i ) {
 500                 const int columnWidth = GetColumnWidth(i);
 501
 502                 if (columnWidth > 2*SPARE_PIXELS_HORZ) {
 503                         wxRect columnRect(
 504                                 columnLeft + SPARE_PIXELS_HORZ, rect.y,
 505                                 columnWidth - 2 * SPARE_PIXELS_HORZ, rect.height);
 506
 507                         wxDCClipper clipper(*dc, columnRect);
 508
 509                         wxString textBuffer;
 510                         switch ( i ) {
 511                                 case ID_SHARED_COL_NAME:
 512                                         textBuffer = file->GetFileName().GetPrintable();
 513
 514                                         if (file->GetFileRating() || file->GetFileComment().Length()) {
 515                                                 int image = Client_CommentOnly_Smiley;
 516                                                 if (file->GetFileRating()) {
 517                                                         image = Client_InvalidRating_Smiley + file->GetFileRating() - 1;
 518                                                 }
 519
 520                                                 wxASSERT(image >= Client_InvalidRating_Smiley);
 521                                                 wxASSERT(image <= Client_CommentOnly_Smiley);
 522
 523                                                 int imgWidth = 16;
 524
 525                                                 theApp->amuledlg->m_imagelist.Draw(image, *dc, columnRect.x,
 526                                                                 columnRect.y + 1, wxIMAGELIST_DRAW_TRANSPARENT);
 527
 528                                                 // Move the text to the right
 529                                                 columnRect.x += (imgWidth + 4);
 530                                         }
 531
 532                                         break;
 533
 534                                 case ID_SHARED_COL_SIZE:
 535                                         textBuffer = CastItoXBytes(file->GetFileSize());
 536                                         break;
 537
 538                                 case ID_SHARED_COL_TYPE:
 539                                         textBuffer = GetFiletypeByName(file->GetFileName());
 540                                         break;
 541
 542                                 case ID_SHARED_COL_PRIO:
 543                                         textBuffer = PriorityToStr(file->GetUpPriority(), file->IsAutoUpPriority());
 544                                         break;
 545
 546                                 case ID_SHARED_COL_ID:
 547                                         textBuffer = file->GetFileHash().Encode();
 548                                         break;
 549
 550                                 case ID_SHARED_COL_REQ:
 551                                         textBuffer = wxString::Format(wxT("%u (%u)"),
 552                                                         file->statistic.GetRequests(),
 553                                                         file->statistic.GetAllTimeRequests());
 554                                         break;
 555
 556                                 case ID_SHARED_COL_AREQ:
 557                                         textBuffer = wxString::Format(wxT("%u (%u)"),
 558                                                         file->statistic.GetAccepts(),
 559                                                         file->statistic.GetAllTimeAccepts());
 560                                         break;
 561
 562                                 case ID_SHARED_COL_TRA:
 563                                         textBuffer = CastItoXBytes(file->statistic.GetTransferred())
 564                                                 + wxT(" (") + CastItoXBytes(file->statistic.GetAllTimeTransferred()) + wxT(")");
 565                                         break;
 566
 567                                 case ID_SHARED_COL_RTIO:
 568                                         textBuffer = wxString::Format(wxT("%.2f"),
 569                                                         (double)file->statistic.GetAllTimeTransferred() / file->GetFileSize() );
 570                                         break;
 571
 572                                 case ID_SHARED_COL_PART:
 573                                         if ( file->GetPartCount() ) {
 574                                                 wxRect barRect(columnRect.x, columnRect. y + 1,
 575                                                         columnRect.width, columnRect.height - 2);
 576
 577                                                 DrawAvailabilityBar(file, dc, barRect);
 578                                         }
 579                                         break;
 580
 581                                 case ID_SHARED_COL_CMPL:
 582                                         if ( file->m_nCompleteSourcesCountLo == 0 ) {
 583                                                 if ( file->m_nCompleteSourcesCountHi ) {
 584                                                         textBuffer = wxString::Format(wxT("< %u"), file->m_nCompleteSourcesCountHi );
 585                                                 } else {
 586                                                         textBuffer = wxT("0");
 587                                                 }
 588                                         } else if (file->m_nCompleteSourcesCountLo == file->m_nCompleteSourcesCountHi) {
 589                                                 textBuffer = wxString::Format(wxT("%u"), file->m_nCompleteSourcesCountLo);
 590                                         } else {
 591                                                 textBuffer = wxString::Format(wxT("%u - %u"),
 592                                                                 file->m_nCompleteSourcesCountLo,
 593                                                                 file->m_nCompleteSourcesCountHi);
 594                                         }
 595
 596                                         break;
 597
 598                                 case ID_SHARED_COL_PATH:
 599                                         if ( file->IsPartFile() ) {
 600                                                 textBuffer = _("[PartFile]");
 601                                         } else {
 602                                                 textBuffer = file->GetFilePath().GetPrintable();
 603                                         }
 604                         }
 605
 606                         if (!textBuffer.IsEmpty()) {
 607                                 dc->DrawText(textBuffer, columnRect.x, columnRect.y + textVOffset);
 608                         }
 609                 }
 610
 611                 // Move to the next column
 612                 columnLeft += columnWidth;
 613         }
 614 }
 615
 616
 617 wxString CSharedFilesCtrl::GetTTSText(unsigned item) const
 618 {
 619         return reinterpret_cast<CKnownFile*>(GetItemData(item))->GetFileName().GetPrintable();
 620 }
 621
 622
 623 bool CSharedFilesCtrl::AltSortAllowed(unsigned column) const
 624 {
 625         switch ( column ) {
 626                 case ID_SHARED_COL_REQ:
 627                 case ID_SHARED_COL_AREQ:
 628                 case ID_SHARED_COL_TRA:
 629                         return true;
 630
 631                 default:
 632                         return false;
 633         }
 634 }
 635
 636
 637 void CSharedFilesCtrl::DrawAvailabilityBar(CKnownFile* file, wxDC* dc, const wxRect& rect ) const
 638 {
 639         // Reference to the availability list
 640         const ArrayOfUInts16& list = file->IsPartFile() ?
 641                 ((CPartFile*)file)->m_SrcpartFrequency :
 642                 file->m_AvailPartFrequency;
 643         wxPen   old_pen   = dc->GetPen();
 644         wxBrush old_brush = dc->GetBrush();
 645         bool bFlat = thePrefs::UseFlatBar();
 646
 647         wxRect barRect = rect;
 648         if (!bFlat) { // round bar has a black border, the bar itself is 1 pixel less on each border
 649                 barRect.x ++;
 650                 barRect.y ++;
 651                 barRect.height -= 2;
 652                 barRect.width -= 2;
 653         }
 654         static CBarShader s_ChunkBar;
 655         s_ChunkBar.SetFileSize( file->GetFileSize() );
 656         s_ChunkBar.SetHeight( barRect.GetHeight() );
 657         s_ChunkBar.SetWidth( barRect.GetWidth() );
 658         s_ChunkBar.Set3dDepth( CPreferences::Get3DDepth() );
 659         uint64 end = 0;
 660         for ( unsigned int i = 0; i < list.size(); ++i ) {
 661                 uint64 start = PARTSIZE * static_cast<uint64>(i);
 662                 end   = PARTSIZE * static_cast<uint64>(i + 1);
 663                 s_ChunkBar.FillRange(start, end, CMuleColour(list[i] ? 0 : 255, list[i] ? ((210-(22*( list[i] - 1 ) ) < 0) ? 0 : (210-(22*( list[i] - 1 ) ))) : 0, list[i] ? 255 : 0));
 664         }
 665         s_ChunkBar.FillRange(end + 1, file->GetFileSize() - 1, CMuleColour(255, 0, 0));
 666         s_ChunkBar.Draw(dc, barRect.x, barRect.y, bFlat);
 667
 668         if (!bFlat) {
 669                 // Draw black border
 670                 dc->SetPen( *wxBLACK_PEN );
 671                 dc->SetBrush( *wxTRANSPARENT_BRUSH );
 672                 dc->DrawRectangle(rect);
 673         }
 674
 675         dc->SetPen( old_pen );
 676         dc->SetBrush( old_brush );
 677 }
 678
 679 void CSharedFilesCtrl::OnRename( wxCommandEvent& WXUNUSED(event) )
 680 {
 681         int item = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 682         if ( item != -1 ) {
 683                 CKnownFile* file = (CKnownFile*)GetItemData(item);
 684
 685                 // Currently renaming of completed files causes problem with kad
 686                 if (file->IsPartFile()) {
 687                         wxString strNewName = ::wxGetTextFromUser(
 688                                 _("Enter new name for this file:"),
 689                                 _("File rename"), file->GetFileName().GetPrintable());
 690
 691                         CPath newName = CPath(strNewName);
 692                         if (newName.IsOk() && (newName != file->GetFileName())) {
 693                                 theApp->sharedfiles->RenameFile(file, newName);
 694                         }
 695                 }
 696         }
 697 }
 698
 699
 700 void CSharedFilesCtrl::OnKeyPressed( wxKeyEvent& event )
 701 {
 702         if (event.GetKeyCode() == WXK_F2) {
 703                 wxCommandEvent evt;
 704                 OnRename(evt);
 705
 706                 return;
 707         }
 708         event.Skip();
 709 }
 710
 711
 712 void CSharedFilesCtrl::OnAddCollection( wxCommandEvent& WXUNUSED(evt) )
 713 {
 714         int item = GetNextItem( -1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED );
 715         if (item != -1) {
 716                 CKnownFile *file = (CKnownFile*)GetItemData(item);
 717                 wxString CollectionFile = file->GetFilePath().JoinPaths(file->GetFileName()).GetRaw();
 718                 CMuleCollection my_collection;
 719                 if (my_collection.Open( (std::string)CollectionFile.mb_str() )) {
 720 //#warning This is probably not working on Unicode
 721                         for (size_t e = 0; e < my_collection.GetFileCount(); ++e) {
 722                                 theApp->downloadqueue->AddLink(
 723                                         wxString(my_collection.GetEd2kLink(e).c_str(), wxConvUTF8));
 724                         }
 725
 726                 }
 727         }
 728 }
 729
 730 // File_checked_for_headers