Support sending custom status codes via WebAPI

qbittorrent · Jul 15, 2024 · 3da5c73 · 3da5c73
1 parent 3acd540
commit 3da5c73
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/src/webui/api/apicontroller.cpp b/src/webui/api/apicontroller.cpp
@@ -42,6 +42,7 @@ void APIResult::clear()
     data.clear();
     mimeType.clear();
     filename.clear();
+    statusCode = 0;
 }
 
 APIController::APIController(IApplication *app, QObject *parent)
@@ -105,3 +106,8 @@ void APIController::setResult(const QByteArray &result, const QString &mimeType,
     m_result.mimeType = mimeType;
     m_result.filename = filename;
 }
+
+void APIController::setStatus(const int statusCode)
+{
+    m_result.statusCode = statusCode;
+}
diff --git a/src/webui/api/apicontroller.h b/src/webui/api/apicontroller.h
@@ -43,6 +43,7 @@ struct APIResult
     QVariant data;
     QString mimeType;
     QString filename;
+    int statusCode;
 
     void clear();
 };
@@ -67,6 +68,8 @@ class APIController : public ApplicationComponent<QObject>
     void setResult(const QJsonObject &result);
     void setResult(const QByteArray &result, const QString &mimeType = {}, const QString &filename = {});
 
+    void setStatus(int statusCode);
+
 private:
     StringMap m_params;
     DataMap m_data;

diff --git a/src/webui/webapplication.cpp b/src/webui/webapplication.cpp
@@ -356,6 +356,9 @@ void WebApplication::doProcessRequest()
     try
     {
         const APIResult result = controller->run(action, m_params, data);
+        if (result.statusCode != 0)
+            status(result.statusCode);
+
         switch (result.data.userType())
         {
         case QMetaType::QJsonDocument: