kget/core/kget.cpp

   1 /* This file is part of the KDE project
   2
   3    Copyright (C) 2005 Dario Massarin <nekkar@libero.it>
   4    Copyright (C) 2007 Lukas Appelhans <l.appelhans@gmx.de>
   5    Copyright (C) 2008 Urs Wolfer <uwolfer @ kde.org>
   6
   7    This program is free software; you can redistribute it and/or
   8    modify it under the terms of the GNU General Public
   9    License as published by the Free Software Foundation; either
  10    version 2 of the License, or (at your option) any later version.
  11 */
  12
  13 #include "core/kget.h"
  14
  15 #include "mainwindow.h"
  16 #include "core/transfer.h"
  17 #include "core/transferdatasource.h"
  18 #include "core/transfergroup.h"
  19 #include "core/transfergrouphandler.h"
  20 #include "core/transfertreemodel.h"
  21 #include "core/transfertreeselectionmodel.h"
  22 #include "core/plugin/plugin.h"
  23 #include "core/plugin/transferfactory.h"
  24 #include "core/observer.h"
  25 #include "core/kuiserverjobs.h"
  26 #include "settings.h"
  27
  28 #include <kio/netaccess.h>
  29 #include <kinputdialog.h>
  30 #include <kfiledialog.h>
  31 #include <kmessagebox.h>
  32 #include <klocale.h>
  33 #include <kstandarddirs.h>
  34 #include <kservicetypetrader.h>
  35 #include <kiconloader.h>
  36 #include <kactioncollection.h>
  37 #include <kio/renamedialog.h>
  38 #include <KPassivePopup>
  39 #include <KSystemTrayIcon>
  40
  41 #include <QTextStream>
  42 #include <QDomElement>
  43 #include <QApplication>
  44 #include <QClipboard>
  45 #include <QAbstractItemView>
  46
  47 /**
  48  * This is our KGet class. This is where the user's transfers and searches are
  49  * stored and organized.
  50  * Use this class from the views to add or remove transfers or searches
  51  * In order to organize the transfers inside categories we have a TransferGroup
  52  * class. By definition, a transfer must always belong to a TransferGroup. If we
  53  * don't want it to be displayed by the gui inside a specific group, we will put
  54  * it in the group named "Not grouped" (better name?).
  55  **/
  56
  57 KGet& KGet::self( MainWindow * mainWindow )
  58 {
  59     if(mainWindow)
  60     {
  61         m_mainWindow = mainWindow;
  62     }
  63
  64     static KGet m;
  65     return m;
  66 }
  67
  68 void KGet::addObserver(ModelObserver * observer)
  69 {
  70     kDebug(5001);
  71
  72     m_observers.append(observer);
  73
  74     //Update the new observer with the TransferGroups objects of the model
  75     QList<TransferGroup *>::const_iterator it = m_transferTreeModel->transferGroups().begin();
  76     QList<TransferGroup *>::const_iterator itEnd = m_transferTreeModel->transferGroups().end();
  77
  78     for( ; it!=itEnd ; ++it )
  79     {
  80         postAddedTransferGroupEvent(*it, observer);
  81     }
  82
  83     kDebug(5001) << " >>> EXITING";
  84 }
  85
  86 void KGet::delObserver(ModelObserver * observer)
  87 {
  88     m_observers.removeAll(observer);
  89 }
  90
  91 bool KGet::addGroup(const QString& groupName)
  92 {
  93     kDebug(5001);
  94
  95     // Check if a group with that name already exists
  96     if(m_transferTreeModel->findGroup(groupName))
  97         return false;
  98
  99     TransferGroup * group = new TransferGroup(m_transferTreeModel, m_scheduler, groupName);
 100     m_transferTreeModel->addGroup(group);
 101
 102     //post notifications
 103     postAddedTransferGroupEvent(group);
 104
 105     return true;
 106 }
 107
 108 void KGet::delGroup(const QString& groupName)
 109 {
 110     TransferGroup * group = m_transferTreeModel->findGroup(groupName);
 111
 112     if(group)
 113     {
 114         m_transferTreeModel->delGroup(group);
 115         postRemovedTransferGroupEvent(group);
 116         delete(group);
 117     }
 118 }
 119
 120 void KGet::renameGroup(const QString& oldName, const QString& newName)
 121 {
 122     TransferGroup *group = m_transferTreeModel->findGroup(oldName);
 123
 124     if(group)
 125     {
 126         group->handler()->setName(newName);
 127     }
 128 }
 129
 130 QStringList KGet::transferGroupNames()
 131 {
 132     QStringList names;
 133
 134     foreach(TransferGroup *group, m_transferTreeModel->transferGroups()) {
 135         names << group->name();
 136     }
 137
 138     return names;
 139 }
 140
 141 void KGet::addTransfer(KUrl srcUrl, QString destDir, // krazy:exclude=passbyvalue
 142                        const QString& groupName, bool start)
 143 {
 144     kDebug(5001) << "Source:" << srcUrl.url();
 145
 146     KUrl destUrl;
 147
 148     if ( srcUrl.isEmpty() )
 149     {
 150         //No src location: we let the user insert it manually
 151         srcUrl = urlInputDialog();
 152         if( srcUrl.isEmpty() )
 153             return;
 154     }
 155
 156     if ( !isValidSource( srcUrl ) )
 157         return;
 158
 159     if (destDir.isEmpty())
 160     {
 161         if (Settings::useDefaultDirectory())
 162 #ifdef Q_OS_WIN //krazy:exclude=cpp
 163             destDir = Settings::defaultDirectory().remove("file:///");
 164 #else
 165             destDir = Settings::defaultDirectory().remove("file://");
 166 #endif
 167
 168         QString checkExceptions = getSaveDirectoryFromExceptions(srcUrl);
 169         if (Settings::enableExceptions() && !checkExceptions.isEmpty())
 170             destDir = checkExceptions;
 171     }
 172
 173     if (!isValidDestDirectory(destDir))
 174         destDir = destInputDialog();
 175
 176     if( (destUrl = getValidDestUrl( destDir, srcUrl )).isEmpty() )
 177         return;
 178
 179     if(m_transferTreeModel->findTransferByDestination(destUrl) != 0 || (destUrl.isLocalFile() && QFile::exists(destUrl.path()))) {
 180         KIO::RenameDialog dlg( m_mainWindow, i18n("Rename transfer"), srcUrl,
 181                                destUrl, KIO::M_MULTI);
 182         if (dlg.exec() == KIO::R_RENAME)
 183             destUrl = dlg.newDestUrl();
 184         else
 185             return;
 186     }
 187     createTransfer(srcUrl, destUrl, groupName, start);
 188 }
 189
 190 void KGet::addTransfer(const QDomElement& e, const QString& groupName)
 191 {
 192     //We need to read these attributes now in order to know which transfer
 193     //plugin to use.
 194     KUrl srcUrl = KUrl( e.attribute("Source") );
 195     KUrl destUrl = KUrl( e.attribute("Dest") );
 196
 197     kDebug(5001) << " src= " << srcUrl.url()
 198               << " dest= " << destUrl.url()
 199               << " group= "<< groupName << endl;
 200
 201     if ( srcUrl.isEmpty() || !isValidSource(srcUrl)
 202          || !isValidDestDirectory(destUrl.directory()) )
 203         return;
 204
 205     createTransfer(srcUrl, destUrl, groupName, false, &e);
 206 }
 207
 208 void KGet::addTransfer(KUrl::List srcUrls, QString destDir, // krazy:exclude=passbyvalue
 209                        const QString& groupName, bool start)
 210 {
 211     KUrl::List urlsToDownload;
 212
 213     KUrl::List::ConstIterator it = srcUrls.begin();
 214     KUrl::List::ConstIterator itEnd = srcUrls.end();
 215
 216     for(; it!=itEnd ; ++it)
 217     {
 218         if ( isValidSource( *it ) )
 219             urlsToDownload.append( *it );
 220     }
 221
 222     if ( urlsToDownload.count() == 0 )
 223         return;
 224
 225     if ( urlsToDownload.count() == 1 )
 226     {
 227         // just one file -> ask for filename
 228         addTransfer(srcUrls.first(), destDir + srcUrls.first().fileName(), groupName, start);
 229         return;
 230     }
 231
 232     KUrl destUrl;
 233
 234     // multiple files -> ask for directory, not for every single filename
 235     if (!isValidDestDirectory(destDir) && !Settings::useDefaultDirectory())
 236         destDir = destInputDialog();
 237
 238     it = urlsToDownload.begin();
 239     itEnd = urlsToDownload.end();
 240
 241     for ( ; it != itEnd; ++it )
 242     {
 243         if (destDir.isEmpty())
 244         {
 245             if (Settings::useDefaultDirectory())
 246 #ifdef Q_OS_WIN //krazy:exclude=cpp
 247                 destDir = Settings::defaultDirectory().remove("file:///");
 248 #else
 249                 destDir = Settings::defaultDirectory().remove("file://");
 250 #endif
 251
 252             QString checkExceptions = getSaveDirectoryFromExceptions(*it);
 253             if (Settings::enableExceptions() && !checkExceptions.isEmpty())
 254                 destDir = checkExceptions;
 255         }
 256         destUrl = getValidDestUrl(destDir, *it);
 257
 258         if(!isValidDestUrl(destUrl))
 259             continue;
 260
 261         createTransfer(*it, destUrl, groupName, start);
 262     }
 263 }
 264
 265
 266 void KGet::delTransfer(TransferHandler * transfer)
 267 {
 268     Transfer * t = transfer->m_transfer;
 269
 270     m_transferTreeModel->delTransfer(t);
 271
 272     //Here I delete the Transfer. The other possibility is to move it to a list
 273     //and to delete all these transfers when kget gets closed. Obviously, after
 274     //the notification to the views that the transfer has been removed, all the
 275     //pointers to it are invalid.
 276     transfer->postDeleteEvent();
 277 // TODO: why does it crash if a download is going to be deleted which is not the last in the list?
 278 // there are always no problems with the last download.
 279 //     delete( t );
 280 }
 281
 282 void KGet::moveTransfer(TransferHandler * transfer, const QString& groupName)
 283 {
 284   Q_UNUSED(transfer);
 285   Q_UNUSED(groupName);
 286 }
 287
 288 QList<TransferHandler *> KGet::selectedTransfers()
 289 {
 290 //     kDebug(5001) << "KGet::selectedTransfers";
 291
 292     QList<TransferHandler *> selectedTransfers;
 293
 294     QModelIndexList selectedIndexes = m_selectionModel->selectedRows();
 295
 296     foreach(QModelIndex currentIndex, selectedIndexes)
 297     {
 298         if(!m_transferTreeModel->isTransferGroup(currentIndex))
 299             selectedTransfers.append(static_cast<TransferHandler *> (currentIndex.internalPointer()));
 300     }
 301
 302     return selectedTransfers;
 303
 304
 305 // This is the code that was used in the old selectedTransfers function
 306 /*    QList<TransferGroup *>::const_iterator it = m_transferTreeModel->transferGroups().begin();
 307     QList<TransferGroup *>::const_iterator itEnd = m_transferTreeModel->transferGroups().end();
 308
 309     for( ; it!=itEnd ; ++it )
 310     {
 311         TransferGroup::iterator it2 = (*it)->begin();
 312         TransferGroup::iterator it2End = (*it)->end();
 313
 314         for( ; it2!=it2End ; ++it2 )
 315         {
 316             Transfer * transfer = (Transfer*) *it2;
 317
 318             if( transfer->isSelected() )
 319                 selectedTransfers.append( transfer->handler() );
 320         }
 321     }
 322     return selectedTransfers;*/
 323 }
 324
 325 QList<TransferGroupHandler *> KGet::selectedTransferGroups()
 326 {
 327     QList<TransferGroupHandler *> selectedTransferGroups;
 328
 329     QModelIndexList selectedIndexes = m_selectionModel->selectedRows();
 330
 331     foreach(QModelIndex currentIndex, selectedIndexes)
 332     {
 333         if(m_transferTreeModel->isTransferGroup(currentIndex))
 334             selectedTransferGroups.append(static_cast<TransferGroupHandler *> (currentIndex.internalPointer()));
 335     }
 336
 337     return selectedTransferGroups;
 338 }
 339
 340 TransferTreeSelectionModel * KGet::selectionModel()
 341 {
 342     return m_selectionModel;
 343 }
 344
 345 void KGet::addTransferView(QAbstractItemView * view)
 346 {
 347     view->setModel(m_transferTreeModel);
 348 }
 349
 350 void KGet::load( QString filename ) // krazy:exclude=passbyvalue
 351 {
 352     kDebug(5001) << "(" << filename << ")";
 353
 354     if(filename.isEmpty())
 355         filename = KStandardDirs::locateLocal("appdata", "transfers.kgt");
 356
 357     QString tmpFile;
 358
 359     //Try to save the transferlist to a temporary location
 360     if(!KIO::NetAccess::download(KUrl(filename), tmpFile, 0))
 361         return;
 362
 363     QFile file(tmpFile);
 364     QDomDocument doc;
 365
 366     kDebug(5001) << "file:" << filename;
 367
 368     if(doc.setContent(&file))
 369     {
 370         QDomElement root = doc.documentElement();
 371
 372         QDomNodeList nodeList = root.elementsByTagName("TransferGroup");
 373         int nItems = nodeList.length();
 374
 375         for( int i = 0 ; i < nItems ; i++ )
 376         {
 377             TransferGroup * foundGroup = m_transferTreeModel->findGroup( nodeList.item(i).toElement().attribute("Name") );
 378
 379             kDebug(5001) << "KGet::load  -> group = " << nodeList.item(i).toElement().attribute("Name");
 380
 381             if( !foundGroup )
 382             {
 383                 kDebug(5001) << "KGet::load  -> group not found";
 384
 385                 TransferGroup * newGroup = new TransferGroup(m_transferTreeModel, m_scheduler);
 386
 387                 m_transferTreeModel->addGroup(newGroup);
 388
 389                 newGroup->load(nodeList.item(i).toElement());
 390
 391                 //Post notifications
 392                 postAddedTransferGroupEvent(newGroup);
 393             }
 394             else
 395             {
 396                 kDebug(5001) << "KGet::load  -> group found";
 397
 398                 //A group with this name already exists.
 399                 //Integrate the group's transfers with the ones read from file
 400                 foundGroup->load(nodeList.item(i).toElement());
 401             }
 402         }
 403     }
 404     else
 405     {
 406         kWarning(5001) << "Error reading the transfers file";
 407     }
 408 }
 409
 410 void KGet::save( QString filename ) // krazy:exclude=passbyvalue
 411 {
 412     if ( !filename.isEmpty()
 413         && QFile::exists( filename )
 414         && (KMessageBox::questionYesNoCancel(0,
 415                 i18n("The file %1 already exists.\nOverwrite?", filename),
 416                 i18n("Overwrite existing file?"), KStandardGuiItem::yes(),
 417                 KStandardGuiItem::no(), KStandardGuiItem::cancel(), "QuestionFilenameExists" )
 418                 != KMessageBox::Yes) )
 419         return;
 420
 421     if(filename.isEmpty())
 422         filename = KStandardDirs::locateLocal("appdata", "transfers.kgt");
 423
 424     QDomDocument doc(QString("KGetTransfers"));
 425     QDomElement root = doc.createElement("Transfers");
 426     doc.appendChild(root);
 427
 428     QList<TransferGroup *>::const_iterator it = m_transferTreeModel->transferGroups().begin();
 429     QList<TransferGroup *>::const_iterator itEnd = m_transferTreeModel->transferGroups().end();
 430
 431     for ( ; it!=itEnd ; ++it )
 432     {
 433         QDomElement e = doc.createElement("TransferGroup");
 434         root.appendChild(e);
 435         (*it)->save(e);
 436     }
 437     QFile file(filename);
 438     if ( !file.open( QIODevice::WriteOnly ) )
 439     {
 440         //kWarning(5001)<<"Unable to open output file when saving";
 441         KMessageBox::error(0,
 442                            i18n("Unable to save to: %1", filename),
 443                            i18n("Error"));
 444         return;
 445     }
 446
 447     QTextStream stream( &file );
 448     doc.save( stream, 0 );
 449     file.close();
 450 }
 451
 452 TransferFactory * KGet::factory(TransferHandler * transfer)
 453 {
 454     return transfer->m_transfer->factory();
 455 }
 456
 457 KActionCollection * KGet::actionCollection()
 458 {
 459     return m_mainWindow->actionCollection();
 460 }
 461
 462 void KGet::setSchedulerRunning(bool running)
 463 {
 464     if(running)
 465     {
 466         m_scheduler->stop(); //stopall first, to have a clean startingpoint
 467         m_scheduler->start();
 468     }
 469     else
 470         m_scheduler->stop();
 471 }
 472
 473 bool KGet::schedulerRunning()
 474 {
 475     return (m_scheduler->countRunningJobs() > 0);
 476 }
 477
 478 void KGet::setPluginsSettingsWidget(KTabWidget * widget)
 479 {
 480     kDebug(5001);
 481     QList<TransferFactory *>::iterator it = m_transferFactories.begin();
 482     QList<TransferFactory *>::iterator itEnd = m_transferFactories.end();
 483
 484     QWidget * settingsWidget;
 485     for( ; it!=itEnd ; ++it)
 486     {
 487         KDialog *dialog = static_cast<KDialog*>(widget->parent()->parent());
 488         if (!dialog)
 489             return;
 490
 491         settingsWidget = (*it)->createSettingsWidget(dialog);
 492         if(settingsWidget)
 493             widget->addTab( settingsWidget, (*it)->displayName() );
 494     }
 495 }
 496
 497 QList<TransferHandler*> KGet::allTransfers()
 498 {
 499     QList<TransferHandler*> transfers;
 500
 501     foreach (TransferGroup *group, KGet::m_transferTreeModel->transferGroups())
 502     {
 503         transfers << group->handler()->transfers();
 504
 505     }
 506     return transfers;
 507 }
 508
 509 void KGet::checkSystemTray()
 510 {
 511     kDebug(5001);
 512     bool running = false;
 513
 514     foreach (TransferHandler *handler, KGet::allTransfers())
 515     {
 516         if (handler->status() == Job::Running)
 517             running = true;
 518
 519         if (running)
 520             continue;
 521     }
 522
 523     m_mainWindow->setSystemTrayDownloading(running);
 524 }
 525
 526 void KGet::settingsChanged()
 527 {
 528     kDebug(5001);
 529
 530     QList<TransferFactory*>::const_iterator it = m_transferFactories.begin();
 531     QList<TransferFactory*>::const_iterator itEnd = m_transferFactories.end();
 532
 533     for( ; it!=itEnd ; ++it )
 534     {
 535         (*it)->settingsChanged();
 536     }
 537 }
 538
 539 void KGet::registerKJob(KJob *job)
 540 {
 541     m_jobManager->registerJob(job);
 542 }
 543
 544 void KGet::unregisterKJob(KJob *job)
 545 {
 546     m_jobManager->unregisterJob(job);
 547 }
 548
 549 void KGet::reloadKJobs()
 550 {
 551     m_jobManager->reload();
 552 }
 553
 554 QStringList KGet::defaultFolders(const KUrl &filename, const QString &groupname)
 555 {
 556     kDebug(5001) << filename << groupname;
 557
 558     QStringList list;
 559     if (Settings::enableExceptions())
 560         list.append(KGet::getSaveDirectoryFromExceptions(filename));
 561
 562     TransferGroup * group = KGet::m_transferTreeModel->findGroup(groupname);
 563
 564     if (!group->defaultFolder().isEmpty())
 565         list.append(group->defaultFolder());
 566
 567     if (Settings::useDefaultDirectory())
 568         list.append(Settings::defaultDirectory());
 569
 570     if (!Settings::lastDirectory().isEmpty() &&
 571         QString::compare(Settings::lastDirectory(), Settings::defaultDirectory()) != 0)
 572         list.append(Settings::lastDirectory());
 573
 574     foreach(QString string, list)
 575     {
 576 #ifdef Q_OS_WIN //krazy:exclude=cpp
 577         string.remove("file:///");
 578 #else
 579         string.remove("file://");
 580 #endif
 581     }
 582
 583     return list;
 584 }
 585
 586 // ------ STATIC MEMBERS INITIALIZATION ------
 587 QList<ModelObserver *> KGet::m_observers;
 588 TransferTreeModel * KGet::m_transferTreeModel;
 589 TransferTreeSelectionModel * KGet::m_selectionModel;
 590 QList<TransferFactory *> KGet::m_transferFactories;
 591 Scheduler * KGet::m_scheduler = new Scheduler();
 592 MainWindow * KGet::m_mainWindow = 0;
 593 KUiServerJobs * KGet::m_jobManager = new KUiServerJobs();
 594
 595 // ------ PRIVATE FUNCTIONS ------
 596 KGet::KGet()
 597 {
 598     m_transferTreeModel = new TransferTreeModel(m_scheduler);
 599     m_selectionModel = new TransferTreeSelectionModel(m_transferTreeModel);
 600
 601     //Load all the available plugins
 602     loadPlugins();
 603
 604     //Create the default group
 605     addGroup(i18n("My Downloads"));
 606 }
 607
 608 KGet::~KGet()
 609 {
 610     delete m_scheduler;
 611 }
 612
 613 bool KGet::createTransfer(const KUrl &src, const KUrl &dest, const QString& groupName,
 614                           bool start, const QDomElement * e)
 615 {
 616     kDebug(5001) << "srcUrl= " << src.url() << "  "
 617                              << "destUrl= " << dest.url()
 618                              << "group= _" << groupName << "_" << endl;
 619
 620     TransferGroup * group = m_transferTreeModel->findGroup(groupName);
 621     if (group==0)
 622     {
 623         kDebug(5001) << "KGet::createTransfer  -> group not found";
 624         group = m_transferTreeModel->transferGroups().first();
 625     }
 626     Transfer * newTransfer;
 627
 628     QList<TransferFactory *>::iterator it = m_transferFactories.begin();
 629     QList<TransferFactory *>::iterator itEnd = m_transferFactories.end();
 630
 631     for( ; it!=itEnd ; ++it)
 632     {
 633         kDebug(5001) << "Trying plugin   n.plugins=" << m_transferFactories.size();
 634         if((newTransfer = (*it)->createTransfer(src, dest, group, m_scheduler, e)))
 635         {
 636 //             kDebug(5001) << "KGet::createTransfer   ->   CREATING NEW TRANSFER ON GROUP: _" << group->name() << "_";
 637             m_transferTreeModel->addTransfer(newTransfer, group);
 638
 639             if(start)
 640                 newTransfer->handler()->start();
 641
 642             if (newTransfer->percent() != 100) //Don't add a finished observer if the Transfer has already been finished
 643                 newTransfer->handler()->addObserver(new TransferFinishedObserver());
 644
 645             return true;
 646         }
 647     }
 648     kDebug(5001) << "Warning! No plugin found to handle the given url";
 649     return false;
 650 }
 651
 652 TransferDataSource * KGet::createTransferDataSource(const KUrl &src)
 653 {
 654     kDebug(5001);
 655     QList<TransferFactory *>::iterator it = m_transferFactories.begin();
 656     QList<TransferFactory *>::iterator itEnd = m_transferFactories.end();
 657
 658     TransferDataSource *dataSource;
 659     for( ; it!=itEnd ; ++it)
 660     {
 661         dataSource = (*it)->createTransferDataSource(src);
 662         if(dataSource)
 663             return dataSource;
 664     }
 665     return 0;
 666 }
 667
 668 void KGet::postAddedTransferGroupEvent(TransferGroup * group, ModelObserver * observer)
 669 {
 670     kDebug(5001);
 671     if(observer)
 672     {
 673         observer->addedTransferGroupEvent(group->handler());
 674         return;
 675     }
 676
 677     QList<ModelObserver *>::iterator it = m_observers.begin();
 678     QList<ModelObserver *>::iterator itEnd = m_observers.end();
 679
 680     for(; it!=itEnd; ++it)
 681     {
 682         kDebug(5001) << "message posted";
 683
 684         (*it)->addedTransferGroupEvent(group->handler());
 685     }
 686 }
 687
 688 void KGet::postRemovedTransferGroupEvent(TransferGroup * group, ModelObserver * observer)
 689 {
 690     if(observer)
 691     {
 692         observer->removedTransferGroupEvent(group->handler());
 693         return;
 694     }
 695
 696     QList<ModelObserver *>::iterator it = m_observers.begin();
 697     QList<ModelObserver *>::iterator itEnd = m_observers.end();
 698
 699     for(; it!=itEnd; ++it)
 700     {
 701         (*it)->removedTransferGroupEvent(group->handler());
 702     }
 703 }
 704
 705 KUrl KGet::urlInputDialog()
 706 {
 707     QString newtransfer;
 708     bool ok = false;
 709
 710     KUrl clipboardUrl = KUrl(QApplication::clipboard()->text(QClipboard::Clipboard).trimmed());
 711     if (clipboardUrl.isValid())
 712         newtransfer = clipboardUrl.url();
 713
 714     while (!ok)
 715     {
 716         newtransfer = KInputDialog::getText(i18n("New Download"), i18n("Enter URL:"), newtransfer, &ok, 0);
 717
 718         if (!ok)
 719         {
 720             //user pressed cancel
 721             return KUrl();
 722         }
 723
 724         KUrl src = KUrl(newtransfer);
 725         if(src.isValid())
 726             return src;
 727         else
 728             ok = false;
 729     }
 730     return KUrl();
 731 }
 732
 733 QString KGet::destInputDialog()
 734 {
 735     QString destDir = KFileDialog::getExistingDirectory(Settings::lastDirectory());
 736
 737     Settings::setLastDirectory( destDir );
 738     return destDir;
 739 }
 740
 741 QString KGet::getSaveDirectoryFromExceptions(const KUrl &filename)
 742 {
 743     QString destDir;
 744
 745     QStringList list = Settings::extensionsFolderList();
 746     QStringList::Iterator it = list.begin();
 747     QStringList::Iterator end = list.end();
 748     while (it != end) {
 749         // odd list items are regular expressions for extensions
 750         QString ext = *it;
 751         ++it;
 752         QString path = *it;
 753         ++it;
 754
 755         if (!ext.startsWith('*'))
 756             ext = '*' + ext;
 757
 758         QRegExp rexp(ext);
 759         rexp.setPatternSyntax(QRegExp::Wildcard);
 760
 761         if (rexp.exactMatch(filename.url())) {
 762             destDir = path;
 763             break;
 764         }
 765     }
 766
 767 #ifdef Q_OS_WIN //krazy:exclude=cpp
 768     destDir = destDir.remove("file:///");
 769 #endif
 770     return destDir.remove("file://");
 771 }
 772
 773 bool KGet::isValidSource(const KUrl &source)
 774 {
 775     if (!source.isValid())
 776     {
 777         KMessageBox::error(0,
 778                            i18n("Malformed URL:\n%1", source.prettyUrl()),
 779                            i18n("Error"));
 780         return false;
 781     }
 782     // Check if a transfer with the same url already exists
 783     Transfer * transfer = m_transferTreeModel->findTransfer( source );
 784     if ( transfer )
 785     {
 786         if ( transfer->status() == Job::Finished )
 787         {
 788             // transfer is finished, ask if we want to download again
 789             if (KMessageBox::questionYesNoCancel(0,
 790                 i18n("URL already saved:\n%1\nDownload again?", source.prettyUrl()),
 791                 i18n("Download URL again?"), KStandardGuiItem::yes(),
 792                 KStandardGuiItem::no(), KStandardGuiItem::cancel(), "QuestionUrlAlreadySaved" )
 793                 == KMessageBox::Yes)
 794             {
 795                 //TODO reimplement this
 796                 //transfer->slotRemove();
 797                 //checkQueue();
 798                 return true;
 799             }
 800         }
 801         else
 802         {
 803             //Transfer is already in list and not finished, ...
 804             return false;
 805         }
 806         return false;
 807     }
 808     return true;
 809 }
 810
 811 bool KGet::isValidDestDirectory(const QString & destDir)
 812 {
 813     kDebug(5001) << destDir;
 814     if (!QFileInfo(destDir).isDir())
 815     {
 816         if (QFileInfo(KUrl(destDir).directory()).isWritable())
 817             return (!destDir.isEmpty());
 818         if (!QFileInfo(KUrl(destDir).directory()).isWritable() && !destDir.isEmpty())
 819             KMessageBox::error(0, i18n("Directory is not writable"));
 820     }
 821     else
 822     {
 823         if (QFileInfo(destDir).isWritable())
 824             return (!destDir.isEmpty());
 825         if (!QFileInfo(destDir).isWritable() && !destDir.isEmpty())
 826             KMessageBox::error(0, i18n("Directory is not writable"));
 827     }
 828     return false;
 829 }
 830
 831 bool KGet::isValidDestUrl(const KUrl &destUrl)
 832 {
 833     if(KIO::NetAccess::exists(destUrl, KIO::NetAccess::DestinationSide, 0))
 834     {
 835         if (KMessageBox::warningYesNoCancel(0,
 836             i18n("Destination file \n%1\nalready exists.\n"
 837                  "Do you want to overwrite it?", destUrl.prettyUrl()))
 838             == KMessageBox::Yes)
 839         {
 840             safeDeleteFile( destUrl );
 841             return true;
 842         }
 843         else
 844             return false;
 845     }
 846     if (m_transferTreeModel->findTransferByDestination(destUrl) || destUrl.isEmpty())
 847         return false;
 848
 849     return true;
 850    /*
 851     KIO::open_RenameDlg(i18n("File already exists"),
 852     (*it).url(), destUrl.url(),
 853     KIO::M_MULTI);
 854    */
 855 }
 856
 857 KUrl KGet::getValidDestUrl(const QString& destDir, const KUrl &srcUrl)
 858 {
 859     if ( !isValidDestDirectory(destDir) )
 860         return KUrl();
 861
 862     // create a proper destination file from destDir
 863     KUrl destUrl = KUrl( destDir );
 864     QString filename = srcUrl.fileName();
 865
 866     if ( filename.isEmpty() )
 867     {
 868         // simply use the full url as filename
 869         filename = KUrl::toPercentEncoding( srcUrl.prettyUrl(), "/" );
 870         kDebug(5001) << " Filename is empty. Setting to  " << filename;
 871         kDebug(5001) << "   srcUrl = " << srcUrl.url();
 872         kDebug(5001) << "   prettyUrl = " << srcUrl.prettyUrl();
 873     }
 874     if (QFileInfo(destUrl.path()).isDir())
 875     {
 876         kDebug(5001) << " Filename is not empty";
 877         destUrl.adjustPath( KUrl::AddTrailingSlash );
 878         destUrl.setFileName( filename );
 879     }
 880     if (!isValidDestUrl(destUrl))
 881     {
 882         kDebug(5001) << "   destUrl " << destUrl.path() << " is not valid";
 883         return KUrl();
 884     }
 885     return destUrl;
 886 }
 887
 888 void KGet::loadPlugins()
 889 {
 890     // Add versioning constraint
 891     QString
 892     str  = "[X-KDE-KGet-framework-version] == ";
 893     str += QString::number( FrameworkVersion );
 894     str += " and ";
 895     str += "[X-KDE-KGet-rank] > 0";
 896     str += " and ";
 897     str += "[X-KDE-KGet-plugintype] == ";
 898
 899     KService::List offers;
 900
 901     //TransferFactory plugins
 902     offers = KServiceTypeTrader::self()->query( "KGet/Plugin", str + "'TransferFactory'" );
 903
 904     //Here we use a QMap only to easily sort the plugins by rank
 905     QMap<int, KService::Ptr> services;
 906     QMap<int, KService::Ptr>::iterator it;
 907
 908     for ( int i = 0; i < offers.count(); ++i )
 909     {
 910         services[ offers[i]->property( "X-KDE-KGet-rank" ).toInt() ] = offers[i];
 911         kDebug(5001) << " TransferFactory plugin found:" << endl <<
 912          "  rank = " << offers[i]->property( "X-KDE-KGet-rank" ).toInt() << endl <<
 913          "  plugintype = " << offers[i]->property( "X-KDE-KGet-plugintype" ) << endl;
 914     }
 915
 916     //I must fill this pluginList before and my m_transferFactories list after.
 917     //This because calling the KLibLoader::globalLibrary() erases the static
 918     //members of this class (why?), such as the m_transferFactories list.
 919     QList<KGetPlugin *> pluginList;
 920
 921     for( it = services.begin(); it != services.end(); ++it )
 922     {
 923         KGetPlugin * plugin;
 924         if( (plugin = createPluginFromService(*it)) != 0 )
 925         {
 926             pluginList.prepend(plugin);
 927             kDebug(5001) << "TransferFactory plugin (" << (*it)->library()
 928                       << ") found and added to the list of available plugins" << endl;
 929         }
 930         else
 931             kDebug(5001) << "Error loading TransferFactory plugin ("
 932                       << (*it)->library() << ")" << endl;
 933     }
 934
 935     QList<KGetPlugin *>::iterator it2 = pluginList.begin();
 936     QList<KGetPlugin *>::iterator it2End = pluginList.end();
 937
 938     for( ; it2!=it2End ; ++it2 )
 939         m_transferFactories.append( static_cast<TransferFactory *>(*it2) );
 940
 941     kDebug(5001) << "Number of factories = " << m_transferFactories.size();
 942 }
 943
 944 KGetPlugin * KGet::createPluginFromService( const KService::Ptr &service )
 945 {
 946     //try to load the specified library
 947     KPluginFactory *factory = KPluginLoader(service->library()).factory();
 948
 949     if (!factory)
 950     {
 951         KMessageBox::error(0, i18n("<html><p>KPluginFactory could not load the plugin:<br/><i>%1</i></p></html>",
 952                                    service->library()));
 953         kError(5001) << "KPluginFactory could not load the plugin:" << service->library();
 954         return 0;
 955     }
 956     KGetPlugin * plugin = factory->create< TransferFactory >(0);
 957
 958     return plugin;
 959 }
 960
 961 bool KGet::safeDeleteFile( const KUrl& url )
 962 {
 963     if ( url.isLocalFile() )
 964     {
 965         QFileInfo info( url.path() );
 966         if ( info.isDir() )
 967         {
 968             KMessageBox::information(0L,i18n("Not deleting\n%1\nas it is a "
 969                                      "directory.", url.prettyUrl()),
 970                                      i18n("Not Deleted"));
 971             return false;
 972         }
 973         KIO::NetAccess::del( url, 0L );
 974         return true;
 975     }
 976
 977     else
 978         KMessageBox::information( 0L,
 979                                   i18n("Not deleting\n%1\nas it is not a local"
 980                                           " file.", url.prettyUrl()),
 981                                   i18n("Not Deleted") );
 982     return false;
 983 }
 984
 985 TransferFinishedObserver::TransferFinishedObserver()
 986      : TransferObserver()
 987 {
 988 }
 989
 990 void TransferFinishedObserver::transferChangedEvent(TransferHandler * transfer)
 991 {
 992     kDebug(5001);
 993
 994     if (transfer->status() == Job::Finished && Settings::quitAfterCompletedTransfer())
 995     {
 996         checkAndFinish();
 997     }
 998
 999     if (prevStatus != transfer->statusText())//FIXME: HACK: better: check statusFlags if it
1000     {                                                                 //contains Tc_Status (flags & Transfer::Tc_Status <-doesn't work)
1001         prevStatus = transfer->statusText();
1002         KGet::checkSystemTray();
1003     }
1004 }
1005
1006 void TransferFinishedObserver::checkAndFinish()
1007 {
1008     bool quitFlag = true;
1009     foreach(TransferGroup *transferGroup, KGet::m_transferTreeModel->transferGroups()) {
1010         foreach(TransferHandler *transfer, transferGroup->handler()->transfers()) {
1011             if(transfer->status() != Job::Finished) {
1012                 quitFlag = false;
1013             }
1014         }
1015     }
1016
1017     // check if there is some unfinished transfer in scheduler queues
1018     if(quitFlag) {
1019         KPassivePopup *message;
1020         // we have to call diferent message from kpassivePopup
1021         // one with parent as QWidget for the mainWindow
1022         // and another with parent as QSystemTrayIcon if the parent is a systemTray
1023         // so passing the QSystemTrayIcon as QWidget don't work
1024         if(Settings::enableSystemTray())
1025         {
1026             message = KPassivePopup::message(5000, KGET_QUIT_MESSAGE_TITLE,
1027                     KGET_QUIT_MESSAGE,
1028                     KGet::m_mainWindow->systemTray());
1029         }
1030         else
1031         {
1032             message = KPassivePopup::message(5000, KGET_QUIT_MESSAGE_TITLE,
1033                     KGET_QUIT_MESSAGE,
1034                     KGet::m_mainWindow);
1035         }
1036
1037         QObject::connect(message, SIGNAL(destroyed()), KGet::m_mainWindow, SLOT(slotQuit()));
1038     }
1039 }