feat: Use common Drive fetch method

Author: adrien-coye

kDrive/AppRouter.swift

@@ -606,7 +606,7 @@ public struct AppRouter: AppNavigable {
             return
         }
 
-        let driveFileManager = try accountManager.getFirstAvailableDriveFileManager(for: account.userId)
+        let driveFileManager = try await accountManager.getBestAvailableDriveFileManager(for: account)
         let drive = driveFileManager.drive
         accountManager.setCurrentDriveForCurrentAccount(for: drive.id, userId: drive.userId)
         showMainViewController(driveFileManager: driveFileManager, selectedIndex: nil)

kDrive/UI/Controller/Menu/SwitchUserViewController.swift

@@ -86,12 +86,14 @@ class SwitchUserViewController: UIViewController {
 
     private func switchToConnectedAccount(_ account: Account) {
         do {
-            let driveFileManager = try accountManager.getFirstAvailableDriveFileManager(for: account.userId)
-            MatomoUtils.track(eventWithCategory: .account, name: "switch")
-            MatomoUtils.connectUser()
+            Task { @MainActor in
+                let driveFileManager = try await accountManager.getBestAvailableDriveFileManager(for: account)
+                MatomoUtils.track(eventWithCategory: .account, name: "switch")
+                MatomoUtils.connectUser()
 
-            accountManager.switchAccount(newAccount: account)
-            appNavigable.showMainViewController(driveFileManager: driveFileManager, selectedIndex: nil)
+                accountManager.switchAccount(newAccount: account)
+                appNavigable.showMainViewController(driveFileManager: driveFileManager, selectedIndex: nil)
+            }
         } catch DriveError.NoDriveError.noDrive {
             let driveErrorNavigationViewController = DriveErrorViewController.instantiateInNavigationController(
                 errorType: .noDrive,

kDrive/UI/Controller/PreloadingViewController.swift

@@ -94,7 +94,7 @@ class PreloadingViewController: UIViewController {
         Task {
             do {
                 _ = try await accountManager.updateUser(for: currentAccount, registerToken: true)
-                _ = try accountManager.getFirstAvailableDriveFileManager(for: currentAccount.userId)
+                _ = try await accountManager.getBestAvailableDriveFileManager(for: currentAccount)
 
                 if let currentDriveFileManager = self.accountManager.currentDriveFileManager {
                     let state = RootViewControllerState.mainViewController(driveFileManager: currentDriveFileManager)

kDriveCore/Data/Cache/AccountManager.swift

@@ -67,7 +67,7 @@ public protocol AccountManageable: AnyObject {
     func forceReload()
     func reloadTokensAndAccounts()
     func getDriveFileManager(for driveId: Int, userId: Int) -> DriveFileManager?
-    func getFirstAvailableDriveFileManager(for userId: Int) throws -> DriveFileManager
+    func getBestAvailableDriveFileManager(for account: Account) async throws -> DriveFileManager
 
     /// Create on the fly an "in memory" DriveFileManager for a specific share
     func getInMemoryDriveFileManager(for publicShareId: String, driveId: Int, rootFileId: Int) -> DriveFileManager?
@@ -227,22 +227,22 @@ public class AccountManager: RefreshTokenDelegate, AccountManageable {
         return DriveFileManager(drive: frozenPublicShareDrive, apiFetcher: DriveApiFetcher(), context: context)
     }
 
-    public func getFirstAvailableDriveFileManager(for userId: Int) throws -> DriveFileManager {
-        let userDrives = driveInfosManager.getDrives(for: userId)
+    public func getBestAvailableDriveFileManager(for account: Account) async throws -> DriveFileManager {
+        let userDrives = Array(driveInfosManager.getDrives(for: account.userId))
 
         guard !userDrives.isEmpty else {
             throw DriveError.NoDriveError.noDrive
         }
 
-        guard let firstAvailableDrive = userDrives.first(where: { !$0.inMaintenance }) else {
+        guard let mainDrive = try? await bestAvailableDrive(from: userDrives, account: account) else {
             if userDrives[0].isInTechnicalMaintenance {
                 throw DriveError.NoDriveError.maintenance(drive: userDrives[0])
             } else {
                 throw DriveError.NoDriveError.blocked(drive: userDrives[0])
             }
         }
 
-        guard let driveFileManager = getDriveFileManager(for: firstAvailableDrive.id, userId: firstAvailableDrive.userId) else {
+        guard let driveFileManager = getDriveFileManager(for: mainDrive.id, userId: mainDrive.userId) else {
             // We should always have a driveFileManager here
             throw DriveError.NoDriveError.noDriveFileManager
         }

kDriveTests/kDrive/Launch/MockAccountManager.swift

@@ -49,7 +49,7 @@ class MockAccountManager: AccountManageable, RefreshTokenDelegate {
 
     func getDriveFileManager(for driveId: Int, userId: Int) -> DriveFileManager? { currentDriveFileManager }
 
-    func getFirstAvailableDriveFileManager(for userId: Int) throws -> DriveFileManager { fatalError("Not implemented") }
+    func getBestAvailableDriveFileManager(for account: Account) async throws -> DriveFileManager { fatalError("Not implemented") }
 
     func getApiFetcher(for userId: Int, token: ApiToken) -> DriveApiFetcher { fatalError("Not implemented") }