PDF_Office/PDF Office/PDF Master/Class/DigtalSignature/CPDFDigtalView/KMPDFDigitalSignViewController.swift

//
//  KMPDFDigitalSignViewController.swift
//  PDF Reader Pro Edition
//
//  Created by Niehaoyu on 2023/10/19.
//

import Cocoa

class KMPDFDigitalSignViewController: NSViewController, CPDFViewDelegate {

    @IBOutlet weak var contendView: NSView!
    @IBOutlet weak var headerView: NSView!
    @IBOutlet weak var tipLabel: NSTextField!
    @IBOutlet weak var exitBox: NSBox!
    @IBOutlet weak var exitButton: KMCustomButton!
    @IBOutlet weak var saveBox: NSBox!
    @IBOutlet weak var saveButton: KMCustomButton!
    
    @IBOutlet weak var stateBGView: NSView!
    @IBOutlet weak var stateLbl: NSTextField!
    @IBOutlet weak var stateButton: NSButton!
    
    @IBOutlet weak var pdfContendView: NSView!
    @IBOutlet weak var contentTopConst: NSLayoutConstraint!
    
    
    var editWidgeAnnotation: CPDFSignatureWidgetAnnotation!
    var stateVC: CDSignatureCertificateStateViewController!
    
    @objc var scaleFactor: CGFloat = 0
    @objc var url: URL!
    @objc var password = String()
    @objc var currentPageIndex: Int = 0
    
    var signatures = NSArray()
    var type:CPromptSignaturesState = .failure
    
    @objc var titleChangeBlock: ((_ title: String, _ pageIndex: NSInteger)->Void)?
    @objc var buttonActionBlock: ((_ actionType: DSignatureActionType, _ isChanged: Bool)->Void)?
    
    var pdfView: CPDFDigtalView!
    
    override func viewDidAppear() {
        super.viewDidAppear()
        
        let contextString = NSLocalizedString("Please box an area for the signature to be added, then you can add a new digital signature.", comment: "")
        _ = CustomAlertView.alertView(message: contextString, fromView: self.view, withStyle: .black)
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do view setup here.
        
        self.contendView.wantsLayer = true
        self.contendView.layer?.backgroundColor = KMAppearance.Layout.bgColor().cgColor
        
        self.headerView.wantsLayer = true
        self.headerView.layer?.backgroundColor = KMAppearance.Interactive.a0Color().cgColor
        
        self.tipLabel.stringValue = NSLocalizedString("You are Under Digital Sign Mode", comment: "")
        
        self.exitBox.borderColor = KMAppearance.Layout.w70Color()
        self.exitBox.borderWidth = 1.0
        self.exitButton.title = NSLocalizedString("Exit", comment: "")
        
        self.saveBox.fillColor = NSColor.white
        self.saveButton.title = NSLocalizedString("Exit", comment: "")
        self.saveButton.setTitleColor(KMAppearance.Interactive.a0Color())
        
        self.exitBox.isHidden = true
        
        self.pdfView = CPDFDigtalView.init(frame: self.pdfContendView.bounds)
        self.pdfView.autoresizingMask = [.width, .height]
        
        let document = CPDFDocument.init(url: self.url)
        if self.password.count > 0 {
            document?.unlock(withPassword: self.password)
        }
        self.pdfView.delegate = self
        self.pdfView.document = document
        self.pdfView.autoScales = true
        self.pdfView.pdfListViewDelegate = self
        self.pdfContendView.addSubview(self.pdfView)
        
        self.stateBGView.wantsLayer = true
        self.stateBGView.layer?.backgroundColor = KMAppearance.Else.textHighlightColor().cgColor
        self.stateLbl.textColor = NSColor.labelColor
        
        self.reloadState()
        
        NotificationCenter.default.addObserver(self, selector: #selector(signatureStateChangeNoti), name: NSNotification.Name(rawValue: "CSignatureTrustCerDidChangeNotification"), object: nil)

    }
    
    //MARK: Public Method
    
    func reloadState() {
        if self.pdfView.signatures.count > 0 {
            self.stateBGView.isHidden = false
            self.contentTopConst.constant = 44
        } else {
            self.stateBGView.isHidden = true
            self.contentTopConst.constant = 0
        }
        
        self.signatures = self.pdfView.signatures! as NSArray
        
        var isSignVerified = false
        var isCertTrusted = false
        
        for item in self.signatures {
            let signature: CPDFSignature = item as! CPDFSignature
            if signature.signers != nil {
                let signer = signature.signers.first
                if signer?.isCertTrusted == false {
                    isCertTrusted = false
                    break
                } else {
                    isCertTrusted = true
                }
            }
        }
        
        for item in self.signatures {
            let signature: CPDFSignature = item as! CPDFSignature
            if signature.signers != nil {
                let signer = signature.signers.first
                if signer?.isSignVerified == false {
                    isSignVerified = false
                    break
                } else {
                    isSignVerified = true
                }
            }
        }
        
        self.stateBGView.isHidden = false
        
        if (isSignVerified && isCertTrusted) {
            self.type = .Success;
        } else if(isSignVerified && !isCertTrusted) {
            self.type = .Unknown;
        } else {
            self.type = .failure;
        }
        
        if (.Success == self.type) {
            self.stateButton.image = NSImage(named: "ImageNameSigntureVerifySuccess")
            self.stateLbl.stringValue = NSLocalizedString("Signature is valid.", comment: "")
        } else if(.Unknown == self.type) {
            self.stateButton.image = NSImage(named: "ImageNameSigntureTrustedFailure")
            if(self.signatures.count > 1) {
                self.stateLbl.stringValue = NSLocalizedString("At least one signature is invalid.", comment: "")
            } else {
                self.stateLbl.stringValue = NSLocalizedString("Signature is invalid", comment: "")
            }
        } else {
            self.stateButton.image = NSImage(named: "ImageNameSigntureVerifyFailure")
            if(self.signatures.count > 1) {
                self.stateLbl.stringValue = NSLocalizedString("At least one signature is invalid.", comment: "")
            } else {
                self.stateLbl.stringValue = NSLocalizedString("Signature is invalid", comment: "")
            }
        }
    }
    
    func writeSignatureToWidget(_ widget: CPDFSignatureWidgetAnnotation, _ path: String, _ password: String, _ config: CPDFSignatureConfig, _ isLock: Bool) ->() {
        
        let fileName = self.pdfView.document.documentURL?.lastPathComponent
        let fileNameWithoutExtension = URL(fileURLWithPath: fileName!).deletingPathExtension().lastPathComponent

        let outputSavePanel = NSSavePanel()
        outputSavePanel.directoryURL = self.pdfView.document.documentURL.deletingLastPathComponent()
        outputSavePanel.title = NSLocalizedString("", comment: "Save as PDF")
        outputSavePanel.allowedFileTypes = ["pdf"]
        outputSavePanel.nameFieldStringValue = fileNameWithoutExtension + "_" + NSLocalizedString("Signed", comment: "")
        let result = outputSavePanel.runModal()
        if result == .OK {
            let contentArr = NSMutableArray()
            var locationStr = ""
            var reasonStr = NSLocalizedString("none", comment: "")
            for item in config.contents {
                if item.key == NSLocalizedString("Reason", comment: "") {
                    if item.value == NSLocalizedString("<your signing reason here>", comment: "") {
                        item.value = " " + NSLocalizedString("none", comment: "")
                    }
                    reasonStr = item.value
                } else if item.key == NSLocalizedString("Location", comment: "") {
                    if item.value == NSLocalizedString("<your signing location here>", comment: "") {
                        item.value = " "
                    }
                    locationStr = item.value
                }
                contentArr.add(item)
            }
            config.contents = contentArr as? [CPDFSignatureConfigItem]
            widget.signAppearanceConfig(config)
            
            let success = self.pdfView.document.writeSignature(to: outputSavePanel.url, withWidget: widget, pkcs12Cert: path, password: password, location: locationStr, reason: reasonStr, permissions: .forbidChange)
            widget.removeSignature()
            if success {
                DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.3) {
                    NSDocumentController.shared.openDocument(withContentsOf: outputSavePanel.url!, display: true) { document, documentWasAlreadyOpen, error in
                        if error != nil {
                            NSApp.presentError(error!)
                            return
                        }
                    }
                }
            } else {
                let alert = NSAlert.init()
                alert.messageText = NSLocalizedString("Save failed!", comment: "")
                alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
                alert.runModal()
            }
            widget.page.removeAnnotation(widget)
            self.pdfView.setNeedsDisplayAnnotationViewFor(widget.page)
        } else {
            widget.page.removeAnnotation(widget)
            self.pdfView.setNeedsDisplayAnnotationViewFor(widget.page)
        }
    }
 
    //MARK: Setter
    
    
    
    //MARK: IBAction
    @IBAction func exitButtonAction(_ sender: NSButton) {
        
    }
    
    @IBAction func saveButtonAction(_ sender: NSButton) {
        
//        var documentArray = NSDocumentController.shared.documents
        var didFileEdit = false
//        for i in 0...documentArray.count-1 {
//            let document = documentArray[i] 
//            if document.fileURL!.path == self.pdfView.document.documentURL.path {
//                didFileEdit = document.isDocumentEdited
//            
//                break
//            }
//        }
        guard let callBack = self.buttonActionBlock else {
            return
        }
        callBack(.cancel, didFileEdit)
        
    }
 
    
    //MARK: CPDFViewDelegate
    func pdfViewDocumentDidLoaded(_ pdfView: CPDFView!) {
        self.pdfView.go(toPageIndex: self.currentPageIndex, animated: true)
    }
    
    func setCurrentPageIndex(_ currentPageIndex: Int) {
        self.pdfView.go(toPageIndex: currentPageIndex, animated: false)
    }
    
    func pdfViewCurrentPageDidChanged(_ pdfView: CPDFView!) {
        let fileName = pdfView.document.documentURL.deletingPathExtension().lastPathComponent
        let title = String(format: "%@ (page %ld / %ld)", fileName, pdfView.currentPageIndex+1, pdfView.document.pageCount)
        
        self.currentPageIndex = pdfView.currentPageIndex
        guard let callBack = self.titleChangeBlock else {
            return
        }
        callBack(title, self.currentPageIndex)
    }

    func popUpSignatureWidgetState(_ signature: CPDFSignature, _ pdfListView: CPDFDigtalView) ->(){
        if self.stateVC == nil {
            self.stateVC = CDSignatureCertificateStateViewController.init()
        }
        self.stateVC.signature = signature
        self.stateVC.pdfListView = pdfListView
        self.stateVC.actionBlock = { [weak self, weak stateVC] stateVCSelf, actionType in
            guard let self = self, let stateVC = stateVC else { return }
            if actionType == .cancel {
                stateVC.dismiss(stateVCSelf)
            } else if actionType == .confirm {
                if let signer = signature.signers.first, let data = signer.certificates {
                    let signatureDetail = DSignatureDetailsViewController.init()
                    signatureDetail.certificates = data
                    signatureDetail.signature = signature
                    signatureDetail.pdfListView = pdfListView
                    stateVCSelf.presentAsSheet(signatureDetail)
                } else {
                    NSSound.beep()
                }
            }
        }
        if let stateVC = self.stateVC {
            self.presentAsSheet(stateVC)
            stateVC.reloadData()
        }
    }
     
    func signElectronicSignature(_ signatureWidgetAnnotation: CPDFSignatureWidgetAnnotation) -> () {
        
//        PDFCustomSignatureWindowController *signatureWindowController = [[PDFCustomSignatureWindowController alloc] init];
//        __block typeof(self) blockSelf = self;
//        [signatureWindowController beginSheetModalForWindow:[NSApp mainWindow] completionHandler:^(PDFSignature *signature){
//            if (signature) {
//                [signatureWidgetAnnotation signWithImage:signature.pathsImage];
//                [blockSelf.PDFListView setNeedsDisplayAnnotationViewForPage:signatureWidgetAnnotation.page];
//            }
//        }];
//        [signatureWindowController release];
    }
    
    //MARK: Notification
    @objc fileprivate func signatureStateChangeNoti() {
         
        let signatures = self.pdfView.document.signatures()!
        
        let tempArr = NSMutableArray()
        for signature in signatures {
            if signature.signers != nil {
                if signature.signers.count > 0 {
                    signature.verifySignature(with: self.pdfView.document)
                    tempArr.add(signature)
                }
            }
        }
        self.pdfView.signatures = tempArr as? [Any]
        
        self.reloadState()
        
        if self.stateVC != nil {
            let signatureWidget = self.editWidgeAnnotation!
            var signature = signatureWidget.signature()
            if signature == nil {
                return
            }
            self.stateVC.signature = signature
            self.stateVC.reloadData()
        }
    }
}

extension KMPDFDigitalSignViewController: CPDFDigtalViewDelegate {
    func pdfListViewAddAnnotation(_ pdfListView: CPDFDigtalView!, forAdd annotation: CPDFAnnotation!, in pdfPage: CPDFPage!) {
        
        let widget = CPDFSignatureWidgetAnnotation.init(PDFListViewNoteWith: self.pdfView.document)
        widget.bounds = CGRectMake(-1000, -1000, 545, 178);
        annotation.page.addAnnotation(widget)
        
        let configWindowVC = DSignatureConfigWindowController.init(windowNibName: "DSignatureConfigWindowController")
        configWindowVC.viewType = .fileList;
        configWindowVC.appearanceWidget = widget;
        configWindowVC.isCreatDS = false
        configWindowVC.complentionHandle = {[weak self] isSign, dic, config, isLock in
            widget.page.removeAnnotation(widget)
            if isSign {
                if (dic.object(forKey: SAVEFILEPATH_KEY) != nil) {
                    let p12Path = dic.object(forKey: SAVEFILEPATH_KEY) as! String
                    let password = dic.object(forKey: PASSWORD_KEY)
                    if p12Path.count > 0 {
                        self?.writeSignatureToWidget(annotation as! CPDFSignatureWidgetAnnotation, p12Path, password as! String, config, isLock)
                    }
                }
            }
        }
        configWindowVC.actionBlock = { controller, type in
            if (type == .cancel) {
                NSApplication.shared.stopModal()
                controller.window?.orderOut(nil)
                controller.window?.close()
            
                annotation.page.removeAnnotation(annotation)
                widget.page.removeAnnotation(widget)
                self.pdfView.setNeedsDisplayAnnotationViewFor(annotation.page)
                
            } else if (type == .confirm) {
                NSApplication.shared.stopModal()
                controller.window?.orderOut(nil)
                controller.window?.close()
            }
        }
        configWindowVC.window?.center()
        NSApplication.shared.runModal(for: configWindowVC.window!)
    }
    
    func pdfListViewEditAnnotation(_ pdfListView: CPDFDigtalView!, for anotation: CPDFAnnotation!) {
        if anotation.className == CPDFSignatureWidgetAnnotation.className() {
            let signatureWidget = anotation as! CPDFSignatureWidgetAnnotation
            var signature = signatureWidget.signature()
            if signature == nil {
                return
            }
            if signature?.signers != nil {
                self.editWidgeAnnotation = signatureWidget
                self.popUpSignatureWidgetState(signature!, self.pdfView)
            } else if signatureWidget.isSigned() {
//                self.signElectronicSignature(signatureWidget)
            } else {
                
            }
            
////            __block WindowController *weakself = self;
//            CPDFSignatureWidgetAnnotation *signatureWidget = (CPDFSignatureWidgetAnnotation *)annotation;
//            CPDFSignature *signature = [signatureWidget signature];
//            if (signature.signers.count > 0) {
//                [self popUpSignatureWidgetState:signature];
//            } else if (signatureWidget.isSigned) {
//                [self signElectronicSignature:signatureWidget];
//            } else {
//                CDSignatureSignTypeWindowController *signType = [[CDSignatureSignTypeWindowController alloc] init];
//                signType.callback = ^(CDSignType type) {
//                    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.3 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
//                        if(type == CDSignType_ElectronicSignature) {
//                            [weakself signElectronicSignature:signatureWidget];
//                        } else {
//                            [weakself signDigitalSignature:signatureWidget];
//                        }
//                    });
//                };
//                [signType startModal:nil];
//                [signType release];
//            }
        }
    }
    
    func pdfListViewDeleteAnnotation(_ pdfListView: CPDFDigtalView!, for anotation: CPDFAnnotation!) {
        let fileName = self.pdfView.document.documentURL?.lastPathComponent
        let fileNameWithoutExtension = URL(fileURLWithPath: fileName!).deletingPathExtension().lastPathComponent
        
        let outputSavePanel = NSSavePanel()
        outputSavePanel.directoryURL = self.pdfView.document.documentURL.deletingLastPathComponent()
        outputSavePanel.title = NSLocalizedString("", comment: "Save as PDF")
        outputSavePanel.allowedFileTypes = ["pdf"]
        outputSavePanel.nameFieldStringValue = fileNameWithoutExtension + "_" + NSLocalizedString("Signed", comment: "")
        let result = outputSavePanel.runModal()
        if result == .OK {
            let success = self.pdfView.document.write(to: outputSavePanel.url)
            if success {
                DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.3) {
                    NSDocumentController.shared.openDocument(withContentsOf: outputSavePanel.url!, display: true) { document, documentWasAlreadyOpen, error in
                        if error != nil {
                            NSApp.presentError(error!)
                            return
                        }
                    }
                }
            } else {
                let alert = NSAlert.init()
                alert.messageText = NSLocalizedString("Save failed!", comment: "")
                alert.addButton(withTitle: NSLocalizedString("OK", comment: ""))
                alert.runModal()
            }
        }
    }
}