kget/core/transfer.cpp

   1 /* This file is part of the KDE project
   2
   3    Copyright (C) 2004 Dario Massarin <nekkar@libero.it>
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public
   7    License as published by the Free Software Foundation; version 2
   8    of the License.
   9 */
  10
  11 #include "core/transfer.h"
  12
  13 #include "core/transferhandler.h"
  14 #include "core/plugin/transferfactory.h"
  15 #include "core/transfergroup.h"
  16 #include "core/scheduler.h"
  17
  18 #include <klocale.h>
  19 #include <kiconloader.h>
  20
  21 #include <QDomElement>
  22 #include <QTime>
  23
  24 Transfer::Transfer(TransferGroup * parent, TransferFactory * factory,
  25                    Scheduler * scheduler, const KUrl & source, const KUrl & dest,
  26                    const QDomElement * e)
  27     : Job(parent, scheduler),
  28       m_source(source), m_dest(dest),
  29       m_totalSize(0), m_processedSize(0),
  30       m_percent(0), m_speed(0),
  31       m_isSelected(false),
  32       m_handler(0), m_factory(factory)
  33 {
  34     if( e )
  35         load( *e );
  36     else
  37     {
  38         setStatus(status(), i18n("Stopped"), SmallIcon("process-stop"));
  39     }
  40 }
  41
  42 Transfer::~Transfer()
  43 {
  44     if(status() == Job::Delayed)
  45         m_scheduler->stopDelayTimer(this);
  46
  47     delete(m_handler);
  48 }
  49
  50 void Transfer::setDelay(int seconds)
  51 {
  52     m_scheduler->startDelayTimer(this, seconds);
  53
  54     setStatus(Job::Delayed, i18n("Delayed"), SmallIcon("history"));
  55
  56     setTransferChange(Tc_Status, true);
  57 }
  58
  59 void Transfer::delayTimerEvent()
  60 {
  61     setStatus(Job::Stopped, i18n("Stopped"), SmallIcon("process-stop"));
  62
  63     setTransferChange(Tc_Status, true);
  64 }
  65
  66 void Transfer::setLog(const QString& message, LogLevel level)
  67 {
  68     QString msg("<font color=\"blue\">" + QTime::currentTime().toString() + "</font> : ");
  69     if (level == error)
  70     {
  71         msg += "<font color=\"red\">" + message + "</font>";
  72     }
  73     if (level == warning)
  74     {
  75         msg += "<font color=\"yellow\">" + message + "</font>";
  76     } else {
  77         msg += message;
  78     }
  79     m_log << msg;
  80 }
  81
  82 TransferHandler * Transfer::handler()
  83 {
  84     if(!m_handler)
  85         m_handler = m_factory->createTransferHandler(this, scheduler());
  86
  87     return m_handler;
  88 }
  89
  90 TransferTreeModel * Transfer::model()
  91 {
  92     return group()->model();
  93 }
  94
  95 void Transfer::save(QDomElement e) // krazy:exclude=passbyvalue
  96 {
  97     e.setAttribute("Source", m_source.url());
  98     e.setAttribute("Dest", m_dest.url());
  99     e.setAttribute("TotalSize", (qulonglong) m_totalSize);
 100     e.setAttribute("ProcessedSize", (qulonglong) m_processedSize);
 101 }
 102
 103 void Transfer::load(const QDomElement &e)
 104 {
 105     m_source = KUrl(e.attribute("Source"));
 106     m_dest = KUrl(e.attribute("Dest"));
 107
 108     m_totalSize = e.attribute("TotalSize").toInt();
 109     m_processedSize = e.attribute("ProcessedSize").toInt();
 110
 111     if( m_totalSize != 0)
 112         m_percent = (int)((100.0 * m_processedSize) / m_totalSize);
 113     else
 114         m_percent = 0;
 115
 116     if((m_totalSize == m_processedSize) && (m_totalSize != 0))
 117     {
 118         setStatus(Job::Finished, i18n("Finished"), SmallIcon("ok"));
 119     }
 120     else
 121     {
 122         setStatus(status(), i18n("Stopped"), SmallIcon("process-stop"));
 123     }
 124 }
 125
 126 void Transfer::setStatus(Job::Status jobStatus, const QString &text, const QPixmap &pix)
 127 {
 128     //If a job is finished don't let it to be changed
 129     if((status() == Job::Finished) && (jobStatus != Job::Finished))
 130         return;
 131
 132     m_statusText = text;
 133     m_statusPixmap = pix;
 134
 135     /**
 136     * It's important to call job::setStatus AFTER having changed the
 137     * icon or the text or whatever.
 138     * This because this function also notifies about this change
 139     * the scheduler which could also decide to change it another time
 140     * as well. For example if a job status is set to Aborted, the scheduler
 141     * could mark it to Delayed. This could trigger another icon or text
 142     * change which would be the right one since the status of the Job
 143     * has changed. If we set the icon or text after calling setStatus(),
 144     * we can overwrite the last icon or text change.
 145     */
 146     Job::setStatus(jobStatus);
 147 }
 148
 149 void Transfer::setTransferChange(ChangesFlags change, bool postEvent)
 150 {
 151     handler()->setTransferChange(change, postEvent);
 152 }