【编辑工具】添加密码移除密码逻辑完善

PDF Office/PDF Master/Class/PDFTools/AddPassword/New/KMSecurityWindowController.swift

@@ -24,7 +24,13 @@ class KMSecurityWindowController: KMBaseWindowController {
         }
         
         securityView.batchAction = { [unowned self] view, files in
-            self.batchAction?(self, files)
+            if files.count == 0 {
+                let file: KMFileAttribute = KMFileAttribute()
+                file.filePath = self.pdfDocument?.documentURL.path ?? ""
+                self.batchAction?(self, [file])
+            } else {
+                self.batchAction?(self, files)
+            }
         }
         
         securityView.doneAction = { [unowned self] view, model, files in

PDF Office/PDF Master/Class/PDFTools/RemovePassword/New/KMRemovePasswordWindowController.swift

@@ -32,7 +32,13 @@ class KMRemovePasswordWindowController: KMBaseWindowController {
         }
         
         self.removePasswordView.batchAction = { [unowned self] view, files in
-            self.batchAction?(self, files)
+            if files.count == 0 {
+                let file: KMFileAttribute = KMFileAttribute()
+                file.filePath = self.pdfDocument?.documentURL.path ?? ""
+                self.batchAction?(self, [file])
+            } else {
+                self.batchAction?(self, files)
+            }
         }
     }
 }

PDF Office/PDF Master/Class/PDFWindowController/ViewController/KMMainViewController+Action.swift

@@ -3325,6 +3325,7 @@ extension KMMainViewController : KMMainToolbarControllerDelegate {
                     return
                 } else if index == 2 {
                     let controller = KMRemovePasswordWindowController(windowNibName: "KMRemovePasswordWindowController")
+                    controller.pdfDocument = self.document
                     self.currentWindowController = controller
                     
                     controller.batchAction = { [unowned self] controller, files in