comparison data/fileio/RemoteFile.cpp @ 211:e2bbb58e6df6

Several changes related to referring to remote URLs for sessions and files:
* Pull file dialog wrapper functions out from MainWindow into FileFinder
* If a file referred to in a session is not found at its expected location,
  try a few other alternatives (same location as the session file or same
  location as the last audio file) before asking the user to locate it
* Allow user to give a URL when locating an audio file, not just locate on
  the filesystem
* Make wave file models remember the "original" location (e.g. URL) of
  the audio file, not just the actual location from which the data was
  loaded (e.g. local copy of that URL) -- when saving a session, use the
  original location so as not to refer to a temporary file
* Clean up incompletely-downloaded local copies of files

 
     QString scheme = url.scheme().toLower();
 
     if (scheme == "http") {
 
+        m_ok = true;
         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);
+
+    } else if (scheme == "ftp") {
+
         m_ok = true;
-
-    } else if (scheme == "ftp") {
-
         m_ftp = new QFtp;
         connect(m_ftp, SIGNAL(done(bool)), this, SLOT(done(bool)));
         connect(m_ftp, SIGNAL(dataTransferProgress(qint64, qint64)),
                 this, SLOT(dataTransferProgress(qint64, qint64)));
         m_ftp->connectToHost(url.host(), url.port(21));

                 m_ftp->cd(*i);
             } else {
                 m_ftp->get(*i, m_localFile);
             }
         }
-
-        m_ok = true;
     }
 
     if (m_ok) {
         m_progressDialog = new QProgressDialog(tr("Downloading %1...").arg(url.toString()), tr("Cancel"), 0, 100);
         m_progressDialog->hide();

     }
 }
 
 RemoteFile::~RemoteFile()
 {
-    delete m_ftp;
-    delete m_http;
-    delete m_localFile;
-    delete m_progressDialog;
-}
-
-bool
-RemoteFile::canHandleScheme(QUrl url)
-{
-    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();
-    }
-}
-
-bool
-RemoteFile::isOK() const
-{
-    return m_ok;
-}
-
-bool
-RemoteFile::isDone() const
-{
-    return m_done;
-}
-
-QString
-RemoteFile::getLocalFilename() const
-{
-    return m_localFilename;
-}
-
-QString
-RemoteFile::getErrorString() const
-{
-    return m_errorString;
-}
-
-void
-RemoteFile::dataReadProgress(int done, int total)
-{
-    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()
-{
+    cleanup();
+}
+
+void
+RemoteFile::cleanup()
+{
+//    std::cerr << "RemoteFile::cleanup" << std::endl;
+    m_done = true;
     delete m_http;
     m_http = 0;
     delete m_ftp;
     m_ftp = 0;
     delete m_progressDialog;
     m_progressDialog = 0;
     delete m_localFile;
     m_localFile = 0;
+}
+
+bool
+RemoteFile::canHandleScheme(QUrl url)
+{
+    QString scheme = url.scheme().toLower();
+    return (scheme == "http" || scheme == "ftp");
+}
+
+bool
+RemoteFile::isAvailable()
+{
+    while (m_ok && (!m_done && m_lastStatus == 0)) {
+        QApplication::processEvents();
+    }
+    bool available = true;
+    if (!m_ok) available = false;
+    else available = (m_lastStatus / 100 == 2);
+    std::cerr << "RemoteFile::isAvailable: " << (available ? "yes" : "no")
+              << std::endl;
+    return available;
+}
+
+void
+RemoteFile::wait()
+{
+    while (m_ok && !m_done) {
+        QApplication::processEvents();
+    }
+}
+
+bool
+RemoteFile::isOK() const
+{
+    return m_ok;
+}
+
+bool
+RemoteFile::isDone() const
+{
+    return m_done;
+}
+
+QString
+RemoteFile::getLocalFilename() const
+{
+    return m_localFilename;
+}
+
+QString
+RemoteFile::getErrorString() const
+{
+    return m_errorString;
+}
+
+void
+RemoteFile::dataReadProgress(int done, int total)
+{
+    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());
+        std::cerr << "RemoteFile::responseHeaderReceived: "
+                  << m_errorString.toStdString() << std::endl;
+    } else {
+        std::cerr << "RemoteFile::responseHeaderReceived: "
+                  << m_lastStatus << std::endl;
+    }        
+}
+
+void
+RemoteFile::dataTransferProgress(qint64 done, qint64 total)
+{
+    if (!m_progressDialog) return;
+
+    int percent = int((double(done) / double(total)) * 100.0 - 0.1);
+    emit progress(percent);
+
+    if (percent > 0) {
+        m_progressDialog->setValue(percent);
+        m_progressDialog->show();
+    }
+}
+
+void
+RemoteFile::cancelled()
+{
+    deleteLocalFile();
     m_done = true;
     m_ok = false;
     m_errorString = tr("Download cancelled");
 }
 
 void
 RemoteFile::done(bool error)
 {
+//    std::cerr << "RemoteFile::done(" << error << ")" << std::endl;
+
+    if (m_done) return;
+
     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;
-
-    delete m_progressDialog;
-    m_progressDialog = 0;
-
-    if (m_ok) {
+        error = true;
+    }
+
+    cleanup();
+
+    if (!error) {
         QFileInfo fi(m_localFilename);
         if (!fi.exists()) {
             m_errorString = tr("Failed to create local file %1").arg(m_localFilename);
-            m_ok = false;
+            error = true;
         } else if (fi.size() == 0) {
             m_errorString = tr("File contains no data!");
-            m_ok = false;
-        }
-    }
+            error = true;
+        }
+    }
+
+    if (error) {
+        deleteLocalFile();
+    }
+
+    m_ok = !error;
+    m_done = true;
+}
+
+void
+RemoteFile::deleteLocalFile()
+{
+//    std::cerr << "RemoteFile::deleteLocalFile" << std::endl;
+
+    cleanup();
+
+    if (m_localFilename == "") return;
+
+    m_fileCreationMutex.lock();
+
+    if (!QFile(m_localFilename).remove()) {
+        std::cerr << "RemoteFile::deleteLocalFile: ERROR: Failed to delete file \"" << m_localFilename.toStdString() << "\"" << std::endl;
+    } else {
+        m_localFilename = "";
+    }
+
+    m_fileCreationMutex.unlock();
+
     m_done = true;
 }
 
 void
 RemoteFile::showProgressDialog()

 }
 
 QString
 RemoteFile::createLocalFile(QUrl url)
 {
-    //!!! should we actually put up dialogs for these errors? or propagate an exception?
-    
     QDir dir;
     try {
         dir = TempDirectory::getInstance()->getSubDirectoryPath("download");
     } catch (DirectoryCreationFailed f) {
         std::cerr << "RemoteFile::createLocalFile: ERROR: Failed to create temporary directory: " << f.what() << std::endl;