comparison data/fileio/RemoteFile.cpp @ 210:a06afefe45ee

* Cancel when downloading file
* Handle status codes (404 etc)
* Add RemoteFile::isAvailable
* Start on FileFinder for looking up files referred to in distant sessions

 #include <QFtp>
 #include <QFileInfo>
 #include <QDir>
 #include <QApplication>
 #include <QProgressDialog>
+#include <QHttpResponseHeader>
 
 #include <iostream>
 
 int
 RemoteFile::m_count = 0;

 RemoteFile::RemoteFile(QUrl url) :
     m_ftp(0),
     m_http(0),
     m_localFile(0),
     m_ok(false),
+    m_lastStatus(0),
     m_done(false),
-    m_progressDialog(0)
+    m_progressDialog(0),
+    m_progressShowTimer(this)
 {
     if (!canHandleScheme(url)) {
         std::cerr << "RemoteFile::RemoteFile: ERROR: Unsupported scheme in URL \"" << url.toString().toStdString() << "\"" << std::endl;
         return;
     }

 
         m_http = new QHttp(url.host(), url.port(80));
         connect(m_http, SIGNAL(done(bool)), this, SLOT(done(bool)));
         connect(m_http, SIGNAL(dataReadProgress(int, int)),
                 this, SLOT(dataReadProgress(int, int)));
+        connect(m_http, SIGNAL(responseHeaderReceived(const QHttpResponseHeader &)),
+                this, SLOT(responseHeaderReceived(const QHttpResponseHeader &)));
         m_http->get(url.path(), m_localFile);
         m_ok = true;
 
     } else if (scheme == "ftp") {
 

         m_ok = true;
     }
 
     if (m_ok) {
         m_progressDialog = new QProgressDialog(tr("Downloading %1...").arg(url.toString()), tr("Cancel"), 0, 100);
-        m_progressDialog->show();
+        m_progressDialog->hide();
+        connect(&m_progressShowTimer, SIGNAL(timeout()),
+                this, SLOT(showProgressDialog()));
+        connect(m_progressDialog, SIGNAL(canceled()), this, SLOT(cancelled()));
+        m_progressShowTimer.setSingleShot(true);
+        m_progressShowTimer.start(2000);
     }
 }
 
 RemoteFile::~RemoteFile()
 {

 {
     QString scheme = url.scheme().toLower();
     return (scheme == "http" || scheme == "ftp");
 }
 
+bool
+RemoteFile::isAvailable()
+{
+    while (!m_done && m_lastStatus == 0) {
+        QApplication::processEvents();
+    }
+    return (m_lastStatus / 100 == 2);
+}
+
 void
 RemoteFile::wait()
 {
     while (!m_done) {
         QApplication::processEvents();

 {
     dataTransferProgress(done, total);
 }
 
 void
+RemoteFile::responseHeaderReceived(const QHttpResponseHeader &resp)
+{
+    m_lastStatus = resp.statusCode();
+    if (m_lastStatus / 100 >= 4) {
+        m_errorString = QString("%1 %2")
+            .arg(resp.statusCode()).arg(resp.reasonPhrase());
+    }
+}
+
+void
 RemoteFile::dataTransferProgress(qint64 done, qint64 total)
 {
     int percent = int((double(done) / double(total)) * 100.0 - 0.1);
     emit progress(percent);
 
     m_progressDialog->setValue(percent);
+    m_progressDialog->show();
+}
+
+void
+RemoteFile::cancelled()
+{
+    delete m_http;
+    m_http = 0;
+    delete m_ftp;
+    m_ftp = 0;
+    delete m_progressDialog;
+    m_progressDialog = 0;
+    delete m_localFile;
+    m_localFile = 0;
+    m_done = true;
+    m_ok = false;
+    m_errorString = tr("Download cancelled");
 }
 
 void
 RemoteFile::done(bool error)
 {
-    //!!! need to identify 404s etc in the return headers
-
     emit progress(100);
     m_ok = !error;
+
     if (error) {
         if (m_http) {
             m_errorString = m_http->errorString();
         } else if (m_ftp) {
             m_errorString = m_ftp->errorString();
         }
+    }
+
+    if (m_lastStatus / 100 >= 4) {
+        m_ok = false;
     }
 
     delete m_localFile;
     m_localFile = 0;
 

         }
     }
     m_done = true;
 }
 
+void
+RemoteFile::showProgressDialog()
+{
+    if (m_progressDialog) m_progressDialog->show();
+}
+
 QString
 RemoteFile::createLocalFile(QUrl url)
 {
     //!!! should we actually put up dialogs for these errors? or propagate an exception?