1 files changed, 143 insertions, 0 deletions
diff --git a/src/sq_downloader.cpp b/src/sq_downloader.cpp
new file mode 100644
index 0000000..e877850
--- /dev/null
+++ b/src/sq_downloader.cpp
@@ -0,0 +1,143 @@
+/***************************************************************************
+                          sq_downloader.cpp  -  description
+                             -------------------
+    begin                : Fri Jun 07 2007
+    copyright            : (C) 2007 by Baryshev Dmitry
+    email                : ksquirrel.iv@gmail.com
+ ***************************************************************************/
+
+/***************************************************************************
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ ***************************************************************************/
+
+#include <tqfile.h>
+
+#include <tdeio/job.h>
+#include <tdefileitem.h>
+#include <tdetempfile.h>
+
+#include "sq_libraryhandler.h"
+#include "sq_downloader.h"
+
+#ifndef KSQUIRREL_PART
+#include "sq_archivehandler.h"
+#endif
+
+#define SQ_PREDOWNLOAD_SIZE 50
+
+SQ_Downloader::SQ_Downloader(TQObject *parent, const char *name) : TQObject(parent, name), job(0), m_error(false)
+{
+    tmp = new KTempFile;
+    tmp->setAutoDelete(true);
+    tmp->close();
+}
+
+SQ_Downloader::~SQ_Downloader()
+{
+    clean();
+
+    delete tmp;
+}
+
+void SQ_Downloader::start(KFileItem *fi)
+{
+    m_error = false;
+    mURL = fi->url();
+
+    emitPercents = false;
+    startTime = TQTime::currentTime();
+    size = 0;
+    totalSize = fi->size();
+
+#ifndef KSQUIRREL_PART
+    nomime = SQ_ArchiveHandler::instance()->findProtocolByMime(fi->mimetype()).isEmpty();
+#else
+    nomime = true;
+#endif
+
+    job = TDEIO::get(mURL, false, false);
+
+    clean();
+
+    continueDownload = false;
+
+    connect(job, TQ_SIGNAL(data(TDEIO::Job *, const TQByteArray &)), this, TQ_SLOT(slotData(TDEIO::Job *, const TQByteArray &)));
+    connect(job, TQ_SIGNAL(result(TDEIO::Job *)), this, TQ_SLOT(slotDataResult(TDEIO::Job *)));
+}
+
+void SQ_Downloader::slotData(TDEIO::Job *job, const TQByteArray &ba)
+{
+    size += ba.size();
+
+    TQFile f(tmp->name());
+
+    if(f.open(IO_WriteOnly | IO_Append))
+    {
+        f.writeBlock(ba);
+        f.close();
+    }
+
+    if(emitPercents || startTime.msecsTo(TQTime::currentTime()) > 1000)
+    {
+        emit percents(size);
+        emitPercents = true;
+    }
+
+    // 50 bytes are enough to determine file type
+    if(size >= SQ_PREDOWNLOAD_SIZE && !continueDownload && totalSize != size)
+    {
+        // cancel download (file type is not supported)
+        SQ_LIBRARY *lib = SQ_LibraryHandler::instance()->libraryForFile(tmp->name());
+
+        if(nomime && !lib)
+        {
+            job->kill(false); // kill job & emit result
+        }
+        else
+        {
+            // nice, we can open this image/archive - continue download
+            continueDownload = true;
+        }
+    }
+}
+
+void SQ_Downloader::slotDataResult(TDEIO::Job *cpjob)
+{
+    job = 0;
+
+    // job error
+    if(cpjob->error() && cpjob->error() != TDEIO::ERR_USER_CANCELED)
+    {
+        m_error = true;
+        emit result(mEmptyURL);
+    }
+    else if(cpjob->error() == TDEIO::ERR_USER_CANCELED) // not supported image/archive type - 
+                                                      // emit empty url without errors
+    {
+        emit result(mEmptyURL);
+    }
+    else // supported image type/archive type (no errors or job killed)
+    {
+        emit result(KURL::fromPathOrURL(tmp->name()));
+    }
+}
+
+void SQ_Downloader::clean()
+{
+    TQFile f(tmp->name());
+
+    if(f.open(IO_WriteOnly))
+        f.close();
+}
+
+void SQ_Downloader::kill()
+{
+    if(job) job->kill();
+}
+
+#include "sq_downloader.moc"