PDF_Office/PDF Office/PDF Master/Class/KMLightMember/Controller/SubscriptionView/WaterMark/KMSubscribeWaterMarkWindowController.swift

//
//  KMSubscribeWaterMarkWindowController.swift
//  PDF Master
//
//  Created by lizhe on 2023/6/12.
//

import Cocoa
var subscribeWaterMarkController: KMSubscribeWaterMarkWindowController?
var subscribeWaterMarkMainWindow: NSWindow?

typealias KMSubscribeWaterMarkWindowControllerCloseAction = (_ controller: KMSubscribeWaterMarkWindowController) -> Void
typealias KMSubscribeWaterMarkWindowControllerWatermarkExportAction = (_ controller: KMSubscribeWaterMarkWindowController) -> Void
typealias KMSubscribeWaterMarkWindowControllerSubscribeAction = (_ controller: KMSubscribeWaterMarkWindowController) -> Void
typealias KMSubscribeWaterMarkWindowControllerRestoreAction = (_ controller: KMSubscribeWaterMarkWindowController) -> Void

class KMSubscribeWaterMarkWindowController: NSWindowController {

    @IBOutlet weak var waterMarkView: KMSubscribeWaterMarkView!
    @IBOutlet weak var waterViewHeightConstrain: NSLayoutConstraint!
    
    var closeAction: KMSubscribeWaterMarkWindowControllerCloseAction?
    var subscribeAction: KMSubscribeWaterMarkWindowControllerSubscribeAction?
    var watermarkExportAction: KMSubscribeWaterMarkWindowControllerWatermarkExportAction?
    var restoreAction: KMSubscribeWaterMarkWindowControllerRestoreAction?
    
    var isContinue: Bool?
    var isAI: Bool?
    
    deinit {
        KMPrint("KMSubscribeWaterMarkWindowController 释放")
    }
    
    override func windowDidLoad() {
        super.windowDidLoad()
        self.setup()
        self.reloadData()
        // Implement this method to handle any initialization after your window controller's window has been loaded from its nib file.
    }
    
    override func awakeFromNib() {
        super.awakeFromNib()
        
        if let data = self.isAI, data {
            self.waterMarkView.waterExportButton.isHidden = true
            self.waterMarkView.subTrailingConst.constant = 60
            self.waterMarkView.subWidthConst.constant = 280
            self.waterMarkView.preferentialImageTrailingConst.constant = 92
        } else {
            self.waterMarkView.waterExportButton.isHidden = false
            self.waterMarkView.subTrailingConst.constant = 0
            self.waterMarkView.subWidthConst.constant = 192
            self.waterMarkView.preferentialImageTrailingConst.constant = 32
        }
    }
    
    //MARK: 打开文件
    static func show(window: NSWindow, isContinue: Bool = false, isAI: Bool = false, completion: @escaping (_ isSubscribeSuccess: Bool, _ isWaterMarkExport: Bool, _ isClose: Bool) -> Void) -> KMSubscribeWaterMarkWindowController {
        let subscribeWaterMarkWindowController: KMSubscribeWaterMarkWindowController = KMSubscribeWaterMarkWindowController.init(windowNibName: "KMSubscribeWaterMarkWindowController")

        subscribeWaterMarkWindowController.isContinue = isContinue
        subscribeWaterMarkWindowController.isAI = isAI
        window.beginSheet(subscribeWaterMarkWindowController.window!)
        subscribeWaterMarkWindowController.window?.center()
        
        if isContinue {
            subscribeWaterMarkWindowController.waterMarkView.waterExportButton.title = NSLocalizedString("Continue", comment: "")
        } else {
            subscribeWaterMarkWindowController.waterMarkView.waterExportButton.title = NSLocalizedString("带水印保存", comment: "")
        }
        
        subscribeWaterMarkWindowController.closeAction = { controller in
            completion(false, false, true)
            subscribeWaterMarkMainWindow?.endSheet(controller.window!)
            controller.window?.close()
            
            subscribeWaterMarkController = nil
            subscribeWaterMarkMainWindow = nil
        }
        
        subscribeWaterMarkWindowController.watermarkExportAction = { controller in
            completion(false, true, false)
            subscribeWaterMarkMainWindow?.endSheet(controller.window!)
            controller.window?.close()
            subscribeWaterMarkController = nil
            subscribeWaterMarkMainWindow = nil
        }
        
        subscribeWaterMarkWindowController.subscribeAction = { controller in
            KMPrint("购买")
            controller.waterMarkView.beginLoading(backgroundColor: NSColor.black.withAlphaComponent(0.2))
            KMPurchaseManager.manager.purchaseProduct(productIdentifier: PRODUCT_1) { isSuccess, error in
                if isSuccess {
                    KMPrint("购买成功")
                    subscribeWaterMarkMainWindow?.endSheet(controller.window!)
                    controller.window?.close()
                    subscribeWaterMarkController = nil
                    subscribeWaterMarkMainWindow = nil
                    
                    DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.1) {
                        if NSApp.mainWindow != nil {
                            let controller = KMSubscribeSuccessWindowController.show(window: NSApp.mainWindow!)
                            controller.closeAction = { controller in
                                controller.closeWindow()
                                completion(true, false, false)
                            }
                        }
                    }
                } else {
                    KMPrint("购买失败")
                    controller.purchasefailed()
                    completion(false, false, false)
                }
                
                controller.waterMarkView.endLoading()
            }
        }
        
        subscribeWaterMarkWindowController.restoreAction = { controller in
            KMPrint("restore")
            controller.waterMarkView.beginLoading(backgroundColor: NSColor.black.withAlphaComponent(0.2))
            KMPurchaseManager.manager.restorePurchases { isSuccess in
                if isSuccess {
                    KMPrint("restore成功")
                    subscribeWaterMarkMainWindow?.endSheet(controller.window!)
                    controller.window?.close()
                    subscribeWaterMarkController = nil
                    subscribeWaterMarkMainWindow = nil
                    
                    if NSApp.mainWindow != nil {
                        let controller = KMSubscribeSuccessWindowController.show(window: NSApp.mainWindow!)
                        controller.closeAction = { controller in
                            controller.closeWindow()
                            completion(true, false, false)
                        }
                    }
                } else {
                    KMPrint("restore失败")
                    controller.purchasefailed()
                    completion(false, false, false)
                }
                controller.waterMarkView.endLoading()
            }
        }
        
        subscribeWaterMarkController = subscribeWaterMarkWindowController
        subscribeWaterMarkMainWindow = window
        return subscribeWaterMarkWindowController
    }
    
    func setup() {
        self.window?.contentView?.backgroundColor(NSColor(hex: "#FFFFFF"))
        self.waterMarkView.waterExportButton.title = ""
        self.waterMarkView.closeAction = { [weak self] view in
            guard let callBack = self?.closeAction else { return }
            
            callBack((self!))
        }
        
        self.waterMarkView.waterMarkExportAction = { [weak self] view in
            guard let callBack = self?.watermarkExportAction else { return }
            
            callBack((self!))
        }
        
        self.waterMarkView.subscribeAction = { [weak self] view in
            guard let callBack = self?.subscribeAction else { return }
            
            callBack((self!))
        }
        
        self.waterMarkView.restoreAction = { [weak self] view in
            guard let callBack = self?.restoreAction else { return }
            
            callBack((self!))
        }
    }
    
    func closeWindow() {
        if subscribeWaterMarkMainWindow != nil {
            subscribeWaterMarkMainWindow?.endSheet((subscribeWaterMarkController?.window)!)
        }
        
        if subscribeWaterMarkController != nil {
            subscribeWaterMarkController?.window?.close()
        }
        subscribeWaterMarkController = nil
        subscribeWaterMarkMainWindow = nil
    }
    
    func reloadData() {
        #if VERSION_FREE
        self.waterViewHeightConstrain.constant = 511
        #endif
        
        #if VERSION_DMG
        self.waterViewHeightConstrain.constant = 511 - 80
        #endif
    }
    
    static func isSampleController() -> KMSubscribeWaterMarkWindowController {
        for window in NSApp.windows {
            let controller = window.windowController
            if controller is KMSubscribeWaterMarkWindowController {
                return controller as! KMSubscribeWaterMarkWindowController
            }
        }
        return KMSubscribeWaterMarkWindowController()
    }
    
    func purchasefailed() {
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.1, execute: {
            let alert = NSAlert()
            alert.messageText = NSLocalizedString("购买失败", comment: "")
            alert.informativeText = NSLocalizedString("", comment: "")
//            alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
            alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
            var window = NSApp.mainWindow
            if NSApp.mainWindow?.sheets.first != nil {
                window = NSApp.mainWindow?.sheets.first
            }
            if window != nil {
                alert.beginSheetModal(for: window!) { result in
                    if (result == .alertFirstButtonReturn) {

                    } else if result == .alertSecondButtonReturn {
                        return
                    }
                }
            }
        })
    }
}