fix(qBittorrent): api version state handling (#826)

jesec · Nov 25, 2024 · 80825db · 80825db
1 parent 843b9c9
commit 80825db
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 21 deletions.
diff --git a/server/services/qBittorrent/clientGatewayService.ts b/server/services/qBittorrent/clientGatewayService.ts
@@ -85,11 +85,12 @@ class QBittorrentClientGatewayService extends ClientGatewayService {
       throw new Error();
     }
 
+    const method = isApiVersionAtLeast(await this.clientRequestManager.apiVersion, '2.11.0') ? 'stopped' : 'paused';
     await this.clientRequestManager
       .torrentsAddFiles(fileBuffers, {
         savepath: destination,
         tags: tags.join(','),
-        [isApiVersionAtLeast(this.clientRequestManager.apiVersion, '2.11.0') ? 'stopped' : 'paused']: !start,
+        [method]: !start,
         root_folder: !isBasePath,
         contentLayout: isBasePath ? 'NoSubfolder' : undefined,
         sequentialDownload: isSequential,
@@ -119,11 +120,12 @@ class QBittorrentClientGatewayService extends ClientGatewayService {
       throw new Error();
     }
 
+    const method = isApiVersionAtLeast(await this.clientRequestManager.apiVersion, '2.11.0') ? 'stopped' : 'paused';
     await this.clientRequestManager
       .torrentsAddURLs(urls, {
         savepath: destination,
         tags: tags.join(','),
-        [isApiVersionAtLeast(this.clientRequestManager.apiVersion, '2.11.0') ? 'stopped' : 'paused']: !start,
+        [method]: !start,
         root_folder: !isBasePath,
         contentLayout: isBasePath ? 'NoSubfolder' : undefined,
         sequentialDownload: isSequential,
@@ -526,7 +528,7 @@ class QBittorrentClientGatewayService extends ClientGatewayService {
 
   async testGateway(): Promise<void> {
     return this.clientRequestManager
-      .updateAuthCookie()
+      .updateConnection()
       .then(() => this.processClientRequestSuccess(undefined), this.processClientRequestError);
   }
 }

diff --git a/server/services/qBittorrent/clientRequestManager.ts b/server/services/qBittorrent/clientRequestManager.ts
@@ -36,7 +36,7 @@ const EMPTY_SERVER_STATE = {
 class ClientRequestManager {
   private connectionSettings: QBittorrentConnectionSettings;
   private apiBase: string;
-  apiVersion: string | null = null;
+  apiVersion: Promise<string | undefined> = Promise.resolve(undefined);
   private authCookie: Promise<string | undefined> = Promise.resolve(undefined);
   private isMainDataPending = false;
 
@@ -80,17 +80,27 @@ class ClientRequestManager {
       });
   }
 
-  async updateAuthCookie(connectionSettings?: QBittorrentConnectionSettings): Promise<void> {
-    let authFailed = false;
+  async updateConnection(connectionSettings?: QBittorrentConnectionSettings): Promise<void> {
+    let failed = false;
 
     this.authCookie = this.authenticate(connectionSettings).catch(() => {
-      authFailed = true;
+      failed = true;
       return undefined;
     });
 
+    this.apiVersion = this.authCookie
+      .then(() => {
+        return !failed ? this.getApiVersion() : Promise.resolve(undefined);
+      })
+      .catch(() => {
+        failed = true;
+        return undefined;
+      });
+
     await this.authCookie;
+    await this.apiVersion;
 
-    if (authFailed) {
+    if (failed) {
       throw new Error();
     }
   }
@@ -120,15 +130,12 @@ class ClientRequestManager {
       });
   }
 
-  async getApiVersion(): Promise<void> {
-    try {
-      const {data} = await axios.get(`${this.apiBase}/app/webapiVersion`, {
+  async getApiVersion(): Promise<string> {
+    return axios
+      .get<string>(`${this.apiBase}/app/webapiVersion`, {
         headers: await this.getRequestHeaders(),
-      });
-      this.apiVersion = data;
-    } catch (error) {
-      this.apiVersion = null;
-    }
+      })
+      .then((res) => res.data);
   }
 
   async getTorrentInfos(): Promise<QBittorrentTorrentInfos> {
@@ -306,7 +313,7 @@ class ClientRequestManager {
   }
 
   async torrentsPause(hashes: Array<string>): Promise<void> {
-    const method = isApiVersionAtLeast(this.apiVersion, '2.11.0') ? 'stop' : 'pause';
+    const method = isApiVersionAtLeast(await this.apiVersion, '2.11.0') ? 'stop' : 'pause';
     return axios
       .post(
         `${this.apiBase}/torrents/${method}`,
@@ -323,7 +330,7 @@ class ClientRequestManager {
   }
 
   async torrentsResume(hashes: Array<string>): Promise<void> {
-    const method = isApiVersionAtLeast(this.apiVersion, '2.11.0') ? 'start' : 'resume';
+    const method = isApiVersionAtLeast(await this.apiVersion, '2.11.0') ? 'start' : 'resume';
     return axios
       .post(
         `${this.apiBase}/torrents/${method}`,
@@ -622,8 +629,7 @@ class ClientRequestManager {
   constructor(connectionSettings: QBittorrentConnectionSettings) {
     this.connectionSettings = connectionSettings;
     this.apiBase = `${connectionSettings.url}/api/v2`;
-    this.updateAuthCookie().catch(() => undefined);
-    this.getApiVersion();
+    this.updateConnection().catch(() => undefined);
   }
 }
 

diff --git a/server/services/qBittorrent/util/apiVersionCheck.ts b/server/services/qBittorrent/util/apiVersionCheck.ts
@@ -1,4 +1,4 @@
-export function isApiVersionAtLeast(currentVersion: string | null, targetVersion: string): boolean {
+export function isApiVersionAtLeast(currentVersion: string | undefined, targetVersion: string): boolean {
   if (!currentVersion) {
     return false;
   }