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?

@objc enum KMSubscribeWaterMarkType: Int {
    case none       = 0
    case stamp      = 1
    case link
    case sign
    case editText
    case editImage
    case insert
    case extract
    case replace
    case split
    case delete
    case rotate
    case copy
    case toWord
    case toExcel
    case toPPT
    case toRTF
    case toCSV
    case toHTML
    case toText
    case toImage
    case compress
    case merge
    case setPassword
    case removePassword
    case crop
    case aiTranslate
    case aiRewrite
    case aiCorrect
}

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, type: KMSubscribeWaterMarkType = .none, subscribeDidClick: (()->Void)? = nil, completion: @escaping (_ isSubscribeSuccess: Bool, _ isWaterMarkExport: Bool, _ isClose: Bool) -> Void) -> KMSubscribeWaterMarkWindowController {
        var isAI = false
        if type == .aiCorrect ||
            type == .aiRewrite ||
            type == .aiTranslate {
            isAI = true
        } else {
            isAI = false
        }
        
        if KMLightMemberManager.manager.isLogin() {
            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("Save with Watermark", 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
                if let callback = subscribeDidClick {
                    callback()
                }
                
                // 数据埋点
                controller.trackEvent(type: type)
                
                controller.waterMarkView.beginLoading(backgroundColor: NSColor.black.withAlphaComponent(0.2))
                KMPurchaseManager.manager.purchaseProduct(productIdentifier: PRODUCT_1) { isSuccess, error in
                    if isSuccess {
                        subscribeWaterMarkMainWindow?.endSheet(controller.window!)
                        controller.window?.close()
                        subscribeWaterMarkController = nil
//                        subscribeWaterMarkMainWindow = nil
                        
                        if subscribeWaterMarkMainWindow != nil {
                            let controller = KMSubscribeSuccessWindowController.show(window: subscribeWaterMarkMainWindow!)
                            controller.closeAction = { controller in
                                controller.closeWindow()
                                completion(true, false, false)
                            }
                            subscribeWaterMarkMainWindow = nil
                        } else {
                            completion(true, false, false)
                        }
                        
                    } else {
                        KMComparativeTableViewController.purchasefailed(view: subscribeWaterMarkController?.waterMarkView, state: error)
                        completion(false, false, false)
                    }
                    
                    controller.waterMarkView.endLoading()
                }
            }
            
            subscribeWaterMarkWindowController.restoreAction = { controller in
                controller.waterMarkView.beginLoading(backgroundColor: NSColor.black.withAlphaComponent(0.2))
                KMPurchaseManager.manager.restorePurchases { isSuccess, error in
                    if isSuccess {
                        subscribeWaterMarkMainWindow?.endSheet(controller.window!)
                        controller.window?.close()
                        subscribeWaterMarkController = nil
                        subscribeWaterMarkMainWindow = nil
                        
                        completion(true, false, false)
                    } else {
                        if error != .restoreSuccess {
                            KMComparativeTableViewController.purchasefailed(view: subscribeWaterMarkController?.waterMarkView, state: error)
                        }
                        completion(false, false, false)
                    }
                    controller.waterMarkView.endLoading()
                }
            }
            
            subscribeWaterMarkController = subscribeWaterMarkWindowController
            subscribeWaterMarkMainWindow = window
            return subscribeWaterMarkWindowController
        } else {
            let controller = KMLoginWindowController.show(window: NSApp.mainWindow!,.equity)
            if isAI {
                let alert = NSAlert()
                alert.messageText = NSLocalizedString("Please login or register to proceed to the next step", comment: "")
                alert.informativeText = NSLocalizedString("As a PDF Master member, you can subscribe to use AI features as needed.", comment: "")
                alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
                alert.beginSheetModal(for: (controller?.window)!) { result in
//                    if (result == .alertFirstButtonReturn) {
//
//                    } else if result == .alertSecondButtonReturn {
//                        return
//                    }
                }
            }
            return KMSubscribeWaterMarkWindowController.init()
        }
    }
    
    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
            self?.trackEvent(eventName: "PUW_Sub_SaveWatermark")
            
            guard let callBack = self?.watermarkExportAction else { return }
            
            callBack((self!))
        }
        
        self.waterMarkView.subscribeAction = { [weak self] view in
            self?.trackEvent(eventName: "PUW_Sub_Subscribe")
            
            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()
    }
}

// MARK: - Analytics (埋点)

extension KMSubscribeWaterMarkWindowController {
    func trackEvent(eventName: String) -> Void {
        KMAnalytics.trackEvent(eventName: eventName, parameters: [
            KMAnalytics.Parameter.categoryKey : KMAnalytics.Category.puw,
            KMAnalytics.Parameter.labelKey : KMAnalytics.Label.sub_PUW], platform: .AppCenter, appTarget: .all)
    }
    
    func trackEvent(type: KMSubscribeWaterMarkType) -> Void {
        KMTools.trackEvent(type: type)
    }
}