AI - 放开AI功能订阅限制限制

PDF Office/PDF Master/Class/Home/ViewController/KMAIRewritingVC.swift

@@ -335,31 +335,33 @@ class KMAIRewritingVC: NSViewController {
     
     @IBAction func aiRewritingAction(_ sender: NSButton) {
         if self.uiState == .CanAIRewriting || self.uiState == .CanAIErrorCorrection || self.uiState == .CanCopy || self.uiState == .Copyed || self.uiState == .Exceed || self.uiState == .Error {
-//            if !KMLightMemberManager.manager.isLogin() {
-//                KMLoginWindowController.show(window: NSApp.mainWindow!)
+            if !KMLightMemberManager.manager.isLogin() {
+                KMLoginWindowController.show(window: NSApp.mainWindow!)
+
+                return
+            }
+            self._aiRewritingAction(sender)
+            
+//            Task { @MainActor in
+//                if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                    var type: KMSubscribeWaterMarkType = .aiRewrite
+//                    if (self.state == .AIErrorCorrection) {
+//                        type = .aiCorrect
+//                    }
+//                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: type) { isSub, _, isClose in
+//                        if (isClose) {
+//                            return
+//                        }
+//                        if (isSub) {
+//                            self._aiRewritingAction(sender)
+//                            return
+//                        }
+//                    }
+//                    return
+//                }
 //
-//                return
+//                self._aiRewritingAction(sender)
 //            }
-            Task { @MainActor in
-                if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                    var type: KMSubscribeWaterMarkType = .aiRewrite
-                    if (self.state == .AIErrorCorrection) {
-                        type = .aiCorrect
-                    }
-                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: type) { isSub, _, isClose in
-                        if (isClose) {
-                            return
-                        }
-                        if (isSub) {
-                            self._aiRewritingAction(sender)
-                            return
-                        }
-                    }
-                    return
-                }
-                
-                self._aiRewritingAction(sender)
-            }
         }
     }
     
@@ -408,26 +410,27 @@ class KMAIRewritingVC: NSViewController {
     
     @IBAction func copyAction(_ sender: NSButton) {
         if self.uiState == .CanCopy || self.uiState == .Copyed {
-//            if !KMLightMemberManager.manager.isLogin() {
-//                KMLoginWindowController.show(window: NSApp.mainWindow!)
-//
-//                return
-//            }
-            Task { @MainActor in
-                if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiRewrite) { isSub, _, isClose in
-                        if (isClose) {
-                            return
-                        }
-                        if (isSub) {
-                            self._copyAction(sender)
-                            return
-                        }
-                    }
-                    return
-                }
-                self._copyAction(sender)
+            if !KMLightMemberManager.manager.isLogin() {
+                KMLoginWindowController.show(window: NSApp.mainWindow!)
+
+                return
             }
+            self._copyAction(sender)
+//            Task { @MainActor in
+//                if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiRewrite) { isSub, _, isClose in
+//                        if (isClose) {
+//                            return
+//                        }
+//                        if (isSub) {
+//                            self._copyAction(sender)
+//                            return
+//                        }
+//                    }
+//                    return
+//                }
+//                self._copyAction(sender)
+//            }
         }
     }
     

PDF Office/PDF Master/Class/Home/ViewController/KMAITranslationVC.swift

@@ -108,26 +108,27 @@ class KMAITranslationVC: NSViewController {
         }
         self.selectYourFilesBox.downCallback = { [unowned self](downEntered: Bool, mouseBox: KMBox, event) -> Void in
             if downEntered {
-//                if !KMLightMemberManager.manager.isLogin() {
-//                    KMLoginWindowController.show(window: NSApp.mainWindow!)
-//
-//                    return
-//                }
-                Task { @MainActor in
-                    if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                        let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiTranslate) { isSub, _, isClose in
-                            if (isClose) {
-                                return
-                            }
-                            if (isSub) {
-                                self._selectFiles()
-                                return
-                            }
-                        }
-                        return
-                    }
-                    self._selectFiles()
+                if !KMLightMemberManager.manager.isLogin() {
+                    KMLoginWindowController.show(window: NSApp.mainWindow!)
+
+                    return
                 }
+                self._selectFiles()
+//                Task { @MainActor in
+//                    if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                        let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiTranslate) { isSub, _, isClose in
+//                            if (isClose) {
+//                                return
+//                            }
+//                            if (isSub) {
+//                                self._selectFiles()
+//                                return
+//                            }
+//                        }
+//                        return
+//                    }
+//                    self._selectFiles()
+//                }
             }
         }
         

PDF Office/PDF Master/Class/Home/WindowController/KMAITranslationConfirmWindowController.swift

@@ -131,26 +131,27 @@ class KMAITranslationConfirmWindowController: NSWindowController {
     // MARK: Action Methods
 
     @IBAction func translateAction(_ sender: NSButton) {
-//        if !KMLightMemberManager.manager.isLogin() {
-//            KMLoginWindowController.show(window: NSApp.mainWindow!)
-//
-//            return
-//        }
-        Task { @MainActor in
-            if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
-                    if (isClose) {
-                        return
-                    }
-                    if (isSubscribeSuccess) {
-                        self._translateAction(sender)
-                        return
-                    }
-                }
-                return
-            }
-            self._translateAction(sender)
+        if !KMLightMemberManager.manager.isLogin() {
+            KMLoginWindowController.show(window: NSApp.mainWindow!)
+
+            return
         }
+        self._translateAction(sender)
+//        Task { @MainActor in
+//            if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
+//                    if (isClose) {
+//                        return
+//                    }
+//                    if (isSubscribeSuccess) {
+//                        self._translateAction(sender)
+//                        return
+//                    }
+//                }
+//                return
+//            }
+//            self._translateAction(sender)
+//        }
     }
     
     private func _translateAction(_ sender: NSButton) {

PDF Office/PDF Master/Class/Home/WindowController/KMAITranslationWindowController.swift

@@ -391,25 +391,26 @@ class KMAITranslationWindowController: NSWindowController {
     
     @IBAction func aiTranslationAction(_ sender: NSButton) {
         if self.state == .CanTranslate || self.state == .CanCopy || self.state == .Copyed || self.state == .Exceed || self.state == .Error {
-//            if !KMLightMemberManager.manager.isLogin() {
-//                KMLoginWindowController.show(window: NSApp.mainWindow!)
-//                return
-//            }
-            Task { @MainActor in
-                if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
-                        if (isClose) {
-                            return
-                        }
-                        if (isSubscribeSuccess) {
-                            self._aiTranslationAction(sender)
-                            return
-                        }
-                    }
-                    return
-                }
-                self._aiTranslationAction(sender)
+            if !KMLightMemberManager.manager.isLogin() {
+                KMLoginWindowController.show(window: NSApp.mainWindow!)
+                return
             }
+            self._aiTranslationAction(sender)
+//            Task { @MainActor in
+//                if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
+//                        if (isClose) {
+//                            return
+//                        }
+//                        if (isSubscribeSuccess) {
+//                            self._aiTranslationAction(sender)
+//                            return
+//                        }
+//                    }
+//                    return
+//                }
+//                self._aiTranslationAction(sender)
+//            }
         }
     }
     
@@ -461,27 +462,29 @@ class KMAITranslationWindowController: NSWindowController {
     
     @IBAction func copyAction(_ sender: NSButton) {
         if self.state == .CanCopy || self.state == .Copyed {
-//            if !KMLightMemberManager.manager.isLogin() {
-//                KMLoginWindowController.show(window: NSApp.mainWindow!)
-//
-//                return
-//            }
-            Task { @MainActor in
-                if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
-                        if (isClose) {
-                            return
-                        }
-                        if (isSubscribeSuccess) {
-                            self._copyAction(sender)
-                            return
-                        }
-                    }
-                    return
-                }
+            if !KMLightMemberManager.manager.isLogin() {
+                KMLoginWindowController.show(window: NSApp.mainWindow!)
 
-                self._copyAction(sender)
+                return
             }
+            self._copyAction(sender)
+//            Task { @MainActor in
+//                if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                    let _ = KMSubscribeWaterMarkWindowController.show(window: self.window!, type: .aiTranslate) { isSubscribeSuccess, _, isClose in
+//                        if (isClose) {
+//                            return
+//                        }
+//                        if (isSubscribeSuccess) {
+//                            self._copyAction(sender)
+//                            return
+//                        }
+//                    }
+//                    return
+//                }
+//
+//                self._copyAction(sender)
+//            }
+            
         }
     }
     

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

@@ -2140,28 +2140,29 @@ extension KMMainViewController : KMMainToolbarControllerDelegate {
     }
     
     func aiTranslationPDFFileAction() {
-//        if !KMLightMemberManager.manager.isLogin() && NSApp.mainWindow != nil {
-//            KMLoginWindowController.show(window: NSApp.mainWindow!)
-//
-//            return
-//        }
-        self.trackEvent_aiTranslate()
-        Task { @MainActor in
-            if await (KMLightMemberManager.manager.canPayFunction() == false) {
-                let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiTranslate) { isSub, _, isClose in
-                    if (isClose) {
-                        return
-                    }
-                    if (isSub) {
-                        self._aiTranslationPDFFileAction()
-                        return
-                    }
-                }
-                return
-            }
+        if !KMLightMemberManager.manager.isLogin() && NSApp.mainWindow != nil {
+            KMLoginWindowController.show(window: NSApp.mainWindow!)
 
-            self._aiTranslationPDFFileAction()
+            return
         }
+        self.trackEvent_aiTranslate()
+        self._aiTranslationPDFFileAction()
+//        Task { @MainActor in
+//            if await (KMLightMemberManager.manager.canPayFunction() == false) {
+//                let _ = KMSubscribeWaterMarkWindowController.show(window: self.view.window!, type: .aiTranslate) { isSub, _, isClose in
+//                    if (isClose) {
+//                        return
+//                    }
+//                    if (isSub) {
+//                        self._aiTranslationPDFFileAction()
+//                        return
+//                    }
+//                }
+//                return
+//            }
+//
+//            self._aiTranslationPDFFileAction()
+//        }
     }
     
     private func _aiTranslationPDFFileAction() {