feat: Add to my kDrive floating action.

Infomaniak · Dec 12, 2024 · abd2bae · abd2bae
1 parent 384b52d
commit abd2bae
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 5 deletions.
diff --git a/kDrive/UI/Controller/Files/FileActionsFloatingPanelViewController+Actions.swift b/kDrive/UI/Controller/Files/FileActionsFloatingPanelViewController+Actions.swift
@@ -165,6 +165,8 @@ extension FileActionsFloatingPanelViewController {
             leaveShareAction()
         case .cancelImport:
             cancelImportAction()
+        case .addToMyDrive:
+            addToMyKDrive()
         default:
             break
         }
@@ -520,4 +522,32 @@ extension FileActionsFloatingPanelViewController {
             }
         }
     }
+
+    private func addToMyKDrive() {
+        guard accountManager.currentAccount != nil else {
+            let upsaleFloatingPanelController = UpsaleViewController.instantiateInFloatingPanel(rootViewController: self)
+            present(upsaleFloatingPanelController, animated: true, completion: nil)
+            return
+        }
+
+        guard let currentUserDriveFileManager = accountManager.currentDriveFileManager,
+              let publicShareProxy = driveFileManager.publicShareProxy else {
+            return
+        }
+
+        let itemId = [file.id]
+        let saveViewController = SaveFileViewController.instantiate(driveFileManager: currentUserDriveFileManager)
+        let saveNavigationViewController = SaveFileViewController
+            .setInNavigationController(saveViewController: saveViewController)
+
+        saveViewController.completionClosure = { [weak self] in
+            guard let self else { return }
+            self.dismiss(animated: true)
+        }
+
+        saveViewController.publicShareFileIds = itemId
+        saveViewController.publicShareProxy = publicShareProxy
+
+        present(saveNavigationViewController, animated: true, completion: nil)
+    }
 }
diff --git a/kDrive/UI/Controller/Files/FileActionsFloatingPanelViewController.swift b/kDrive/UI/Controller/Files/FileActionsFloatingPanelViewController.swift
@@ -204,6 +204,12 @@ public class FloatingPanelAction: Equatable {
         image: KDriveResourcesAsset.colorBucket.image
     )
 
+    static let addToMyDrive = FloatingPanelAction(
+        id: 22,
+        name: KDriveResourcesStrings.Localizable.buttonAddToKDrive,
+        image: KDriveResourcesAsset.drive.image
+    )
+
     static var quickActions: [FloatingPanelAction] {
         return [informations, sendCopy, shareAndRights, shareLink].map { $0.reset() }
     }
@@ -213,7 +219,7 @@ public class FloatingPanelAction: Equatable {
     }
 
     static var publicShareActions: [FloatingPanelAction] {
-        return [openWith, sendCopy, download].map { $0.reset() }
+        return [openWith, sendCopy, download, addToMyDrive].map { $0.reset() }
     }
 
     static var publicShareFolderActions: [FloatingPanelAction] {

diff --git a/kDrive/UI/Controller/Menu/Share/PublicShareViewModel.swift b/kDrive/UI/Controller/Menu/Share/PublicShareViewModel.swift
@@ -152,10 +152,6 @@ final class PublicShareViewModel: InMemoryFileListViewModel {
     }
 
     private func addToMyDrive(sender: Any?, publicShareProxy: PublicShareProxy) {
-        guard let currentUserDriveFileManager = accountManager.currentDriveFileManager else {
-            return
-        }
-
         let selectedItemsIds = multipleSelectionViewModel?.selectedItems.map(\.id) ?? [] + [rootProxy.id]
         let exceptItemIds = multipleSelectionViewModel?.exceptItemIds.map { $0 } ?? []