Fix Mastodon sync and transfer reliability edge cases

2 files changed, 18 insertions, 31 deletions

diff --git a/transferengine-plugins/mastodontransferplugin/mastodonapi.cpp b/transferengine-plugins/mastodontransferplugin/mastodonapi.cpp
index baf9dd8..24bc6f5 100644
--- a/transferengine-plugins/mastodontransferplugin/mastodonapi.cpp
+++ b/transferengine-plugins/mastodontransferplugin/mastodonapi.cpp
@@ -3,6 +3,7 @@
  */
 
 #include "mastodonapi.h"
+#include "mastodonauthutils.h"
 
 #include <QtCore/QFile>
 #include <QtCore/QFileInfo>
@@ -18,6 +19,7 @@
 
 MastodonApi::MastodonApi(QNetworkAccessManager *qnam, QObject *parent)
     : QObject(parent)
+    , m_cancelRequested(false)
     , m_qnam(qnam)
 {
 }
@@ -26,33 +28,6 @@ MastodonApi::~MastodonApi()
 {
 }
 
-QString MastodonApi::normalizeApiHost(const QString &rawHost)
-{
-    QString host = rawHost.trimmed();
-    if (host.isEmpty()) {
-        host = QStringLiteral("https://mastodon.social");
-    }
-
-    if (!host.startsWith(QLatin1String("https://"))
-            && !host.startsWith(QLatin1String("http://"))) {
-        host.prepend(QStringLiteral("https://"));
-    }
-
-    QUrl url(host);
-    if (!url.isValid() || url.host().isEmpty()) {
-        return QStringLiteral("https://mastodon.social");
-    }
-
-    QString normalized = QString::fromLatin1(url.toEncoded(QUrl::RemovePath
-                                                            | QUrl::RemoveQuery
-                                                            | QUrl::RemoveFragment));
-    if (normalized.endsWith(QLatin1Char('/'))) {
-        normalized.chop(1);
-    }
-
-    return normalized;
-}
-
 bool MastodonApi::uploadImage(const QString &filePath,
                               const QString &statusText,
                               const QString &mimeType,
@@ -65,7 +40,8 @@ bool MastodonApi::uploadImage(const QString &filePath,
         return false;
     }
 
-    m_apiHost = normalizeApiHost(apiHost);
+    m_cancelRequested = false;
+    m_apiHost = MastodonAuthUtils::normalizeApiHost(apiHost);
     m_accessToken = accessToken;
     m_statusText = statusText;
 
@@ -116,7 +92,8 @@ bool MastodonApi::postStatus(const QString &statusText,
                              const QString &apiHost,
                              const QString &accessToken)
 {
-    m_apiHost = normalizeApiHost(apiHost);
+    m_cancelRequested = false;
+    m_apiHost = MastodonAuthUtils::normalizeApiHost(apiHost);
     m_accessToken = accessToken;
     m_statusText = statusText;
 
@@ -172,11 +149,11 @@ void MastodonApi::cancelUpload()
         return;
     }
 
+    m_cancelRequested = true;
     const QList<QNetworkReply*> replies = m_replies.keys();
     Q_FOREACH (QNetworkReply *reply, replies) {
         reply->abort();
     }
-    m_replies.clear();
 }
 
 void MastodonApi::replyError(QNetworkReply::NetworkError error)
@@ -205,6 +182,14 @@ void MastodonApi::finished()
 
     reply->deleteLater();
 
+    if (m_cancelRequested && error == QNetworkReply::OperationCanceledError) {
+        if (m_replies.isEmpty()) {
+            m_cancelRequested = false;
+            emit transferCanceled();
+        }
+        return;
+    }
+
     if (apiCall == UPLOAD_MEDIA) {
         if (error != QNetworkReply::NoError || httpCode < 200 || httpCode >= 300) {
             finishTransfer(error == QNetworkReply::NoError ? QNetworkReply::UnknownNetworkError : error,
@@ -241,6 +226,8 @@ void MastodonApi::finished()
 
 void MastodonApi::finishTransfer(QNetworkReply::NetworkError error, int httpCode, const QByteArray &data)
 {
+    m_cancelRequested = false;
+
     if (httpCode == 401) {
         emit credentialsExpired();
     }
diff --git a/transferengine-plugins/mastodontransferplugin/mastodonapi.h b/transferengine-plugins/mastodontransferplugin/mastodonapi.h
index 4ac3d80..e24914d 100644
--- a/transferengine-plugins/mastodontransferplugin/mastodonapi.h
+++ b/transferengine-plugins/mastodontransferplugin/mastodonapi.h
@@ -49,11 +49,11 @@ private Q_SLOTS:
     void uploadProgress(qint64 received, qint64 total);
 
 private:
-    static QString normalizeApiHost(const QString &rawHost);
     bool postStatusInternal(const QString &mediaId);
     void finishTransfer(QNetworkReply::NetworkError error, int httpCode, const QByteArray &data);
 
     QMap<QNetworkReply*, API_CALL> m_replies;
+    bool m_cancelRequested;
     QNetworkAccessManager *m_qnam;
     QString m_accessToken;
     QString m_apiHost;