PDF_Office/PDF Office/PDF Master/Class/PDFTools/AddHeaderFooter/New/View/KMHeaderFooterView.swift

//
//  KMHeaderFooterView.swift
//  PDF Master
//
//  Created by lizhe on 2023/11/24.
//

import Cocoa

import Foundation

let kHeaderFooterFileSavePath = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.applicationDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).last?.stringByAppendingPathComponent(Bundle.main.bundleIdentifier!).stringByAppendingPathComponent("HeaderFooterpdf.pdf")

let dateFormatArray: [String] = [
    "m/d",
    "m/d/yy",
    "m/d/yyyy",
    "mm/dd/yy",
    "mm/dd/yyyy",
    "d/m/yy",
    "d/m/yyyy",
    "dd/mm/yy",
    "dd/mm/yyyy",
    "mm/yy",
    "mm/yyyy",
    "m.d.yy",
    "m.d.yyyy",
    "mm.dd.yy",
    "mm.dd.yyyy",
    "mm.yy",
    "mm.yyyy",
    "d.m.yy",
    "d.m.yyyy",
    "dd.mm.yy",
    "dd.mm.yyyy",
    "yy-mm-dd",
    "yyyy-mm-dd"
]

let pageFormatArray: [String] = [
    "1",
    "1 of n",
    "1/n",
    "Page 1",
    "Page 1 of n"
]


let fontSizeArray: [String] = [
    "8",
    "9",
    "10",
    "11",
    "12",
    "14",
    "16",
    "18",
    "20",
    "22",
    "24",
    "26",
    "28",
    "36",
    "48",
    "72",
    "144",
    "288",
]

func convertDateFormat(_ oldString: String) -> String {
    var newString = oldString
    
    for dateFormat in dateFormatArray {
        if newString.contains(dateFormat) {
            let formatString = dateFormat.replacingOccurrences(of: "m", with: "M")
            let replace = "<<\(dateFormat)>>"
            
            let date = Date()
            let dateFormatter = DateFormatter()
            dateFormatter.dateFormat = formatString
            let dateString = dateFormatter.string(from: date)
            
            newString = newString.replacingOccurrences(of: replace, with: dateString)
        }
    }
    
    return newString
}

typealias KMHeaderFooterViewDoneAction = ( _ model: KMHeaderFooterModel) -> ()
typealias KMHeaderFooterViewCancelAction = (_ view: KMHeaderFooterView) -> Void

class KMHeaderFooterView: KMBaseXibView, NSTextViewDelegate, NSComboBoxDelegate {
    
    @IBOutlet weak var pdfView: KMWatermarkPDFView!
    @IBOutlet weak var currentPageIndexTextField: NSTextField!
    @IBOutlet weak var totalPageCountLabel: NSTextField!

    @IBOutlet weak var textBox: NSBox!
    @IBOutlet weak var textLabel: NSTextField!
    @IBOutlet weak var fontSizeLabel: NSTextField!
    @IBOutlet weak var fontSizeCombobox: NSComboBox!
    @IBOutlet weak var textColorLabel: NSTextField!
    @IBOutlet weak var textColorWell: NSColorWell!

    @IBOutlet weak var marginBox: NSBox!
    @IBOutlet weak var marginLabel: NSTextField!
    @IBOutlet weak var topMarginLabel: NSTextField!
    @IBOutlet weak var topMargintextField: NSTextField!
    @IBOutlet weak var topMarginStepper: NSStepper!
    @IBOutlet weak var bottomMarginLabel: NSTextField!
    @IBOutlet weak var bottomMarginTextField: NSTextField!
    @IBOutlet weak var bottomMarginStepper: NSStepper!
    @IBOutlet weak var leftMarginLabel: NSTextField!
    @IBOutlet weak var leftMarginTextField: NSTextField!
    @IBOutlet weak var leftMarginStepper: NSStepper!
    @IBOutlet weak var rightMarginLabel: NSTextField!
    @IBOutlet weak var rightMarginTextField: NSTextField!
    @IBOutlet weak var rightMarginStepper: NSStepper!

    @IBOutlet weak var formatBox: NSBox!
    @IBOutlet weak var formatLabel: NSTextField!
    @IBOutlet weak var dateFormateLabel: NSTextField!
    @IBOutlet weak var dateFormateCombobox: NSComboBox!
    @IBOutlet weak var pageLabel: NSTextField!
    @IBOutlet weak var pageCombobox: NSComboBox!
    @IBOutlet weak var beginPageLabel: NSTextField!
    @IBOutlet weak var beginPageCombobox: NSComboBox!

    @IBOutlet weak var topLeftLabel: NSTextField!
    @IBOutlet weak var topLeftTextView: NSTextView!
    @IBOutlet weak var topCenterLabel: NSTextField!
    @IBOutlet weak var topCenterTextView: NSTextView!
    @IBOutlet weak var topRightLabel: NSTextField!
    @IBOutlet weak var topRightTextView: NSTextView!
    @IBOutlet weak var bottomLeftLabel: NSTextField!
    @IBOutlet weak var bottomLeftTextView: NSTextView!
    @IBOutlet weak var bottomCenterLabel: NSTextField!
    @IBOutlet weak var BottomCenterTextView: NSTextView!
    @IBOutlet weak var bottomRightLabel: NSTextField!
    @IBOutlet weak var bottomRightTextView: NSTextView!
    @IBOutlet weak var pageRangeLabel: NSTextField!
    @IBOutlet weak var pageRangeCombobox: NSComboBox!
    @IBOutlet weak var saveToTemplateButton: NSButton!
    @IBOutlet weak var templateNameLabel: NSTextField!
    @IBOutlet weak var templateNameTextField: NSTextField!

    @IBOutlet weak var topLeftButton: NSButton!
    @IBOutlet weak var topCentButton: NSButton!
    @IBOutlet weak var topRightButton: NSButton!
    @IBOutlet weak var bottomLeftButton: NSButton!
    @IBOutlet weak var bottomCenterButton: NSButton!
    @IBOutlet weak var bottomRightButton: NSButton!

    @IBOutlet weak var batchButton: NSButton!
    @IBOutlet weak var cancelButton: NSButton!
    @IBOutlet weak var applyButton: NSButton!

    @IBOutlet weak var batesBox: NSBox!
    @IBOutlet weak var prefixLabel: NSTextField!
    @IBOutlet weak var prefixTextField: NSTextField!
    @IBOutlet weak var suffixLabel: NSTextField!
    @IBOutlet weak var suffixTextField: NSTextField!
    @IBOutlet weak var batesNumberLabel: NSTextField!
    @IBOutlet weak var batesNumberTextField: NSTextField!
    @IBOutlet weak var batesNumberStepper: NSStepper!

    @IBOutlet weak var batesStartPageLabel: NSTextField!
    @IBOutlet weak var batesStartpageCombobox: NSComboBox!
    @IBOutlet weak var constraint: NSLayoutConstraint!
    var isHiddenBatchBtn: Bool = false
    
    var initialID: String = ""
    
    var filePath: String = Bundle.main.path(forResource: "Quick Start Guide", ofType: "pdf") ?? ""
    var password: String = ""
    var type: KMBatchModifyTemplateType = .Use // Replace with the actual type
    var pdfDocument: CPDFDocument? {
        didSet {
            self.pdfView.document = self.pdfDocument
            self.pdfView.headerFooter = self.headerFooterObj
            self.reloadData()
        }
    }
    var headerFooterObj: KMHeaderFooterModel = KMHeaderFooterModel()
    var originalHeaderFooterObj: KMHeaderFooterModel = KMHeaderFooterModel()
    var dateFormatArray: [Any] {
        get {
            return KMHeaderFooterManager.defaultManager.dateFormatArray
        }
    }
    
    var onlyManagerTemplate: Bool = true
    var isBates: Bool = false {
        didSet {
            if (self.isBates) {
                headerFooterObj.id = KMHeaderFooterManager.defaultManager.fetchBatesAvailableName()
            } else {
                headerFooterObj.id = KMHeaderFooterManager.defaultManager.fetchHeaderFooterAvailableName()
            }
            headerFooterObj.isBates = self.isBates
            self.reloadData()
        }
    }
    
    var cancelAction: KMHeaderFooterViewCancelAction?
    var operateCallBack: KMHeaderFooterViewDoneAction?
    
    deinit {
        NotificationCenter.default.removeObserver(self)
    }
    
    convenience init?(baseFile filePath: String, headerFooter object: KMHeaderFooterModel, password: String, type: KMBatchModifyTemplateType) {
        self.init()
        
        self.filePath = filePath
        self.password = password
        self.headerFooterObj = object.copy() as! KMHeaderFooterModel
        self.originalHeaderFooterObj = object
        self.initialID = object.id

        self.type = type
        self.pdfDocument = CPDFDocument(url: URL(fileURLWithPath: self.filePath))

        if self.pdfDocument!.isLocked {
            self.pdfDocument!.unlock(withPassword: password)
        }

        if self.pdfDocument!.isLocked {
            return nil
        }
    }
    
    override func setup() {
        let buttons = [
            topLeftButton, topCentButton, topRightButton,
            bottomLeftButton, bottomCenterButton, bottomRightButton
        ]

        for button in buttons {
            button?.wantsLayer = true
            button?.layer?.borderWidth = 1.0
            button?.layer?.cornerRadius = 1.0
        }

        let textViewArr: [NSTextView] = [
            self.topLeftTextView, self.topCenterTextView, self.topRightTextView,
            self.bottomLeftTextView, self.BottomCenterTextView, self.bottomRightTextView
        ]

        for textView in textViewArr {
            textView.wantsLayer = true
            // textView?.layer?.borderWidth = 1.0
            // textView?.layer?.borderColor = NSColor.gridColor.cgColor
        }

        self.textColorWell.color = self.headerFooterObj.getTextColor()
        self.pdfView.autoScales = true
//        self.pdfView.documentView?.enclosingScrollView?.hasVerticalScroller = false
//        self.pdfView.documentView?.enclosingScrollView?.hasHorizontalScroller = false

        
        self.topMarginLabel.allowsExpansionToolTips = true
        self.bottomMarginLabel.allowsExpansionToolTips = true
        self.leftMarginLabel.allowsExpansionToolTips = true
        self.rightMarginLabel.allowsExpansionToolTips = true


        self.formatLabel.allowsExpansionToolTips = true
        self.dateFormateLabel.allowsExpansionToolTips = true
        self.pageLabel.allowsExpansionToolTips = true
        self.beginPageLabel.allowsExpansionToolTips = true
        self.topLeftLabel.allowsExpansionToolTips = true
        self.topCenterLabel.allowsExpansionToolTips = true
        self.topRightLabel.allowsExpansionToolTips = true
        self.bottomLeftLabel.allowsExpansionToolTips = true
        self.bottomCenterLabel.allowsExpansionToolTips = true
        self.bottomRightLabel.allowsExpansionToolTips = true

        self.saveToTemplateButton.isEnabled = onlyManagerTemplate

        self.pageRangeCombobox.removeAllItems()
        self.pageRangeCombobox.addItems(withObjectValues: [
            NSLocalizedString("All Pages", comment: ""),
            NSLocalizedString("Odd Pages Only", comment: ""),
            NSLocalizedString("Even Pages Only", comment: ""),
            NSLocalizedString("e.g. 1,3-5,10", comment: "")
        ])
        self.pageRangeCombobox.placeholderString = NSLocalizedString("e.g. 1,3-5,10", comment: "")
        self.pageRangeCombobox.selectItem(at: 0)
        self.pageRangeCombobox.isEnabled = true
        self.pageRangeCombobox.delegate = self

        pageRangeCombobox.wantsLayer = true
        pageRangeCombobox.layer?.cornerRadius = 3.0

        self.batchButton.isHidden = true
        if self.type == .Use {
            self.batchButton.isHidden = false
        }

        self.fontSizeCombobox.removeAllItems()
        self.fontSizeCombobox.addItems(withObjectValues: fontSizeArray)
        self.fontSizeCombobox.isEditable = false
        self.fontSizeCombobox.delegate = self
    }

    override func reloadData() {
        guard let pdfDocument = pdfDocument else { return }
        updateViewColor()
        
        self.totalPageCountLabel.stringValue = String(format: "/ %ld", pdfDocument.pageCount)
        
        self.leftMarginTextField.stringValue = self.headerFooterObj.leftMargin.description
        self.rightMarginTextField.stringValue = self.headerFooterObj.rightMargin.description
        self.topMargintextField.stringValue = self.headerFooterObj.topMargin.description
        self.bottomMarginTextField.stringValue = self.headerFooterObj.bottomMargin.description

        self.leftMarginStepper.integerValue = Int(self.headerFooterObj.leftMargin)
        self.rightMarginStepper.integerValue = Int(self.headerFooterObj.rightMargin)
        self.topMarginStepper.integerValue = Int(self.headerFooterObj.topMargin)
        self.bottomMarginStepper.integerValue = Int(self.headerFooterObj.bottomMargin)

        let string = Int(self.headerFooterObj.getTextFontSize()).description
        let index = self.fontSizeCombobox.indexOfItem(withObjectValue: string)
        if index != NSNotFound {
            self.fontSizeCombobox.selectItem(withObjectValue: string)
        } else {
            self.fontSizeCombobox.stringValue = string
        }

        self.dateFormateCombobox.removeAllItems()
        self.dateFormateCombobox.addItems(withObjectValues: self.dateFormatArray)
        self.dateFormateCombobox.selectItem(withObjectValue: self.headerFooterObj.dateFormatString)

        let containsItem = self.pageCombobox.objectValues.contains { item in
            return item as? String == self.headerFooterObj.pageFormatString
        }

        if containsItem {
            self.pageCombobox.selectItem(withObjectValue: self.headerFooterObj.pageFormatString)
        } else {
            self.pageCombobox.stringValue = ""
        }

        var pages = [Any]()
        for i in 0..<self.pdfDocument!.pageCount {
            pages.append(String(i + 1))
        }

        self.beginPageCombobox.formatter = TextFieldFormatter()
        self.beginPageCombobox.isEditable = false
        self.beginPageCombobox.removeAllItems()
        self.beginPageCombobox.addItems(withObjectValues: pages)

        let indexOfObject = self.beginPageCombobox.objectValues.contains { item in
            return item as? String == self.headerFooterObj.startString
        }
        if indexOfObject {
            self.beginPageCombobox.selectItem(withObjectValue: self.headerFooterObj.startString)
        }

        self.batesStartpageCombobox.formatter = TextFieldFormatter()
        self.batesStartpageCombobox.isEditable = false
        self.batesStartpageCombobox.removeAllItems()
        self.batesStartpageCombobox.addItems(withObjectValues: pages)

        let batesIndexOfObject = self.batesStartpageCombobox.objectValues.contains { item in
            return item as? String == self.headerFooterObj.startString
        }
        
        if batesIndexOfObject {
            self.batesStartpageCombobox.selectItem(withObjectValue: self.headerFooterObj.startString)
        }

        switch self.headerFooterObj.pageRangeType {
        case .all:
//            self.pageRangeCombobox.isEditable = false
            self.pageRangeCombobox.selectItem(at: 0)
        case .odd:
//            self.pageRangeCombobox.isEditable = false
            self.pageRangeCombobox.selectItem(at: 1)
        case .even:
//            self.pageRangeCombobox.isEditable = false
            self.pageRangeCombobox.selectItem(at: 2)
        case .other:
//            self.pageRangeCombobox.isEditable = true
            self.pageRangeCombobox.selectItem(at: 3)
            self.pageRangeCombobox.stringValue = self.headerFooterObj.pageRangeString 
            let sting = self.headerFooterObj.pageRangeString
            if sting.count > 0 {
                let array = sting.components(separatedBy: ",")
                var pagesString = ""
                for i in 0..<(array.count - 1) {
                    let index = array[i]
                    if i == array.count - 1 {
                        pagesString = String(format: "%@%ld", pagesString, (Int(index ) ?? 0) + 1)
                    } else {
                        pagesString = String(format: "%@%ld,", pagesString, (Int(index) ?? 0 ) + 1)
                    }
                }
                self.pageRangeCombobox.stringValue = pagesString
            }
        }

        self.topLeftTextView.string = self.headerFooterObj.topLeftString
        self.topCenterTextView.string = self.headerFooterObj.topCenterString
        self.topRightTextView.string = self.headerFooterObj.topRightString
        self.bottomLeftTextView.string = self.headerFooterObj.bottomLeftString
        self.BottomCenterTextView.string = self.headerFooterObj.bottomCenterString
        self.bottomRightTextView.string = self.headerFooterObj.bottomRightString

        self.templateNameTextField.stringValue = self.headerFooterObj.id

        if self.topLeftTextView.string.count <= 0 &&
            self.topCenterTextView.string.count <= 0 &&
            self.topRightTextView.string.count <= 0 &&
            self.bottomLeftTextView.string.count <= 0 &&
            self.BottomCenterTextView.string.count <= 0 &&
            self.bottomRightTextView.string.count <= 0 {
            self.applyButton.isEnabled = false
        } else {
            self.applyButton.isEnabled = true
        }
        
        if self.isBates {
            self.constraint.constant = 49
            self.formatBox.isHidden = true
            self.batesBox.isHidden = false
        } else {
            self.constraint.constant = 20
            self.formatBox.isHidden = false
            self.batesBox.isHidden = true
        }
        self.batchButton.isHidden = isHiddenBatchBtn
    }
    
    func updateViewColor() {
        let borderColor: CGColor
        let backgroundColor: CGColor

        if KMAppearance.isDarkMode() {
            borderColor = NSColor.white.withAlphaComponent(0.3).cgColor
            backgroundColor = NSColor(red: 54 / 255.0, green: 54 / 255.0, blue: 54 / 255.0, alpha: 0.85).cgColor
        } else {
            borderColor = NSColor.white.withAlphaComponent(0.3).cgColor
            backgroundColor = NSColor(red: 238 / 255.0, green: 238 / 255.0, blue: 238 / 255.0, alpha: 0.85).cgColor
        }

        let buttons = [
            topLeftButton, topCentButton, topRightButton,
            bottomLeftButton, bottomCenterButton, bottomRightButton
        ]

        for button in buttons {
            button?.layer?.borderColor = borderColor
            button?.layer?.backgroundColor = backgroundColor
        }
    }
    
    override func updateLanguage() {
        self.currentPageIndexTextField.stringValue = NSLocalizedString("1", comment: "")
        self.batesStartPageLabel.stringValue = String(format: "%@:", NSLocalizedString("Start Page Number", comment: ""))

        self.textLabel.stringValue = NSLocalizedString("Font Size", comment: "")
        self.fontSizeLabel.stringValue = String(format: "%@:", NSLocalizedString("Size", comment: ""))
        self.textColorLabel.stringValue = String(format: "%@:", NSLocalizedString("Color", comment: ""))

        self.marginLabel.stringValue = NSLocalizedString("Margin", comment: "")
        self.topMarginLabel.stringValue = String(format: "%@:", NSLocalizedString("Top", comment: ""))
        self.bottomMarginLabel.stringValue = String(format: "%@:", NSLocalizedString("Bottom", comment: ""))
        self.leftMarginLabel.stringValue = String(format: "%@:", NSLocalizedString("Left", comment: ""))
        self.rightMarginLabel.stringValue = String(format: "%@:", NSLocalizedString("Right", comment: ""))

        self.topMarginLabel.toolTip = String(format: "%@:", NSLocalizedString("Top", comment: ""))
        self.bottomMarginLabel.toolTip = String(format: "%@:", NSLocalizedString("Bottom", comment: ""))
        self.leftMarginLabel.toolTip = String(format: "%@:", NSLocalizedString("Left", comment: ""))
        self.rightMarginLabel.toolTip = String(format: "%@:", NSLocalizedString("Right", comment: ""))
        
        self.formatLabel.stringValue = NSLocalizedString("Page Number and Date Format", comment: "")
        self.dateFormateLabel.stringValue = String(format: "%@:", NSLocalizedString("Date", comment: ""))
        self.pageLabel.stringValue = String(format: "%@:", NSLocalizedString("Page", comment: ""))
        self.beginPageLabel.stringValue = String(format: "%@:", NSLocalizedString("Start Page Number", comment: ""))

        self.formatLabel.toolTip = NSLocalizedString("Page Number and Date Format", comment: "")
        self.dateFormateLabel.toolTip = String(format: "%@:", NSLocalizedString("Date", comment: ""))
        self.pageLabel.toolTip = String(format: "%@:", NSLocalizedString("Page", comment: ""))
        self.beginPageLabel.toolTip = String(format: "%@:", NSLocalizedString("Start Page Number", comment: ""))

        self.topLeftLabel.stringValue = NSLocalizedString("Left Header Text", comment: "")
        self.topCenterLabel.stringValue = NSLocalizedString("Center Header Text", comment: "")
        self.topRightLabel.stringValue = NSLocalizedString("Right Header Text", comment: "")
        self.bottomLeftLabel.stringValue = NSLocalizedString("Left Footer Text", comment: "")
        self.bottomCenterLabel.stringValue = NSLocalizedString("Center Footer Text", comment: "")
        self.bottomRightLabel.stringValue = NSLocalizedString("Right Footer Text", comment: "")

        self.topLeftLabel.toolTip = NSLocalizedString("Left Header Text", comment: "")
        self.topCenterLabel.toolTip = NSLocalizedString("Center Header Text", comment: "")
        self.topRightLabel.toolTip = NSLocalizedString("Right Header Text", comment: "")
        self.bottomLeftLabel.toolTip = NSLocalizedString("Left Footer Text", comment: "")
        self.bottomCenterLabel.toolTip = NSLocalizedString("Center Footer Text", comment: "")
        self.bottomRightLabel.toolTip = NSLocalizedString("Right Footer Text", comment: "")
        
        self.pageRangeLabel.stringValue = String(format: "%@:", NSLocalizedString("Page Range", comment: ""))

        self.saveToTemplateButton.title = NSLocalizedString("Add to Template", comment: "")
        
        self.templateNameLabel.stringValue = NSLocalizedString("Name:", comment: "")

        self.batchButton.title = NSLocalizedString("Batch", comment: "")
        self.cancelButton.title = NSLocalizedString("Cancel", comment: "")
        self.applyButton.title = NSLocalizedString("Apply", comment: "")

        self.batesBox.title = NSLocalizedString("Bates Settings", comment: "")
        self.prefixLabel.stringValue = String(format: "%@:", NSLocalizedString("Prefix", comment: ""))
        self.prefixLabel.toolTip = String(format: "%@:", NSLocalizedString("Prefix", comment: ""))
        self.prefixTextField.stringValue = self.headerFooterObj.batesPrefixString

        self.suffixLabel.stringValue = String(format: "%@:", NSLocalizedString("Suffix", comment: ""))
        self.suffixLabel.toolTip = String(format: "%@:", NSLocalizedString("Suffix", comment: ""))
        self.suffixTextField.stringValue = self.headerFooterObj.batesSuffixString

        self.batesNumberLabel.stringValue = String(format: "%@:", NSLocalizedString("Number of Digits", comment: ""))
        self.batesNumberStepper.integerValue = self.headerFooterObj.batesDigits
        self.batesNumberTextField.integerValue = self.headerFooterObj.batesDigits

    }
    
    override func addNotification() {
        NotificationCenter.default.addObserver(self, selector: #selector(pageChangeNotification), name: Notification.Name("PDFViewPageChangedNotification"), object: self.pdfView)
        NotificationCenter.default.addObserver(self, selector: #selector(themeChanged), name: Notification.Name("AppleInterfaceThemeChangedNotification"), object: nil)
    }
    
    func checkPageRangeValidate(pageRangeString: String) -> Bool {
        let fileAttribute = KMFileAttribute()
        fileAttribute.filePath = (self.pdfDocument?.documentURL!.path)!
        fileAttribute.bAllPage = false
        fileAttribute.pagesString = self.pageRangeCombobox.stringValue
        
        if fileAttribute.fetchSelectPages().count == 0 {
            return false
        }
        return true
    }

    func popMenu(sender: NSButton) {
        if self.isBates {
            let menu = NSMenu()
            let title = NSLocalizedString("Insert Bates", comment: "")
            let item = menu.addItem(withTitle: title, action: #selector(menuItem_InsertBates), keyEquivalent: "")
            item.target = self
            item.representedObject = sender
            
            menu.popUp(positioning: menu.item(at: 0), at: NSPoint(x: 0, y: 15), in: sender)
        } else {
            let dateMenu = NSMenu()
            var title = NSLocalizedString("Insert Page Number", comment: "")
            var item = dateMenu.addItem(withTitle: title, action: #selector(menuItem_InsertPage), keyEquivalent: "")
            item.target = self
            item.representedObject = sender
            title = NSLocalizedString("Insert Date", comment: "")
            item = dateMenu.addItem(withTitle: title, action: #selector(menuItem_InsertDate), keyEquivalent: "")
            item.target = self
            item.representedObject = sender
            
            dateMenu.popUp(positioning: dateMenu.item(at: 0), at: NSPoint(x: 0, y: 15), in: sender)
        }
    }

    @objc func menuItem_InsertBates(_ sender: NSMenuItem) {
        guard let button = sender.representedObject as? NSButton else { return }
        self.window?.makeFirstResponder(self)
        self.headerFooterObj.batesDigits = self.batesNumberTextField.integerValue
        var tString = "<<#\((self.batesNumberTextField.integerValue))#\(self.batesStartpageCombobox.stringValue)"
        if !self.prefixTextField.stringValue.isEmpty {
            tString += "#\(self.prefixTextField.stringValue)"
        } else if self.prefixTextField.stringValue.isEmpty && !self.suffixTextField.stringValue.isEmpty {
            tString += "#"
        }
        if !self.suffixTextField.stringValue.isEmpty {
            tString += "#\(self.suffixTextField.stringValue)"
        }
        tString += ">>"
        if let textView = self.fetchTargetTextView(tag: button.tag) {
            textView.string = textView.string + tString
            self.updateHeaderFooterString(newString: textView.string, tag: button.tag)
        }
        self.applyButton.isEnabled = true
        self.updatePDFView()
    }

    @objc func menuItem_InsertPage(_ sender: NSMenuItem) {
        guard let button = sender.representedObject as? NSButton else { return }
        let tString = "<<\(self.pageCombobox.stringValue)>>"
        if let textView = self.fetchTargetTextView(tag: button.tag) {
            textView.string = textView.string + tString
            self.updateHeaderFooterString(newString: textView.string, tag: button.tag)
        }
        self.applyButton.isEnabled = true
        self.updatePDFView()
    }

    @objc func menuItem_InsertDate(_ sender: NSMenuItem) {
        guard let button = sender.representedObject as? NSButton else { return }
        let tString = "<<\(self.headerFooterObj.dateFormatString)>>"
        if let textView = self.fetchTargetTextView(tag: button.tag) {
            textView.string = textView.string + tString
            self.updateHeaderFooterString(newString: textView.string, tag: button.tag)
        }
        self.applyButton.isEnabled = true
        self.updatePDFView()
    }

    func fetchTargetTextView(tag: Int) -> NSTextView? {
        var textView: NSTextView?
        switch tag {
        case 0:
            textView = self.topLeftTextView
        case 1:
            textView = self.topCenterTextView
        case 2:
            textView = self.topRightTextView
        case 3:
            textView = self.bottomLeftTextView
        case 4:
            textView = self.BottomCenterTextView
        case 5:
            textView = self.bottomRightTextView
        default:
            break
        }
        return textView
    }

    func updateHeaderFooterString(newString: String, tag: Int) {
        switch tag {
        case 0:
            self.headerFooterObj.topLeftString = newString
        case 1:
            self.headerFooterObj.topCenterString = newString
        case 2:
            self.headerFooterObj.topRightString = newString
        case 3:
            self.headerFooterObj.bottomLeftString = newString
        case 4:
            self.headerFooterObj.bottomCenterString = newString
        case 5:
            self.headerFooterObj.bottomRightString = newString
        default:
            break
        }
    }
    
    func textDidChange(_ notification: Notification) {
        guard let textView = notification.object as? NSTextView else {
            return
        }

        switch textView {
        case topLeftTextView:
            headerFooterObj.topLeftString = topLeftTextView.string
        case topCenterTextView:
            headerFooterObj.topCenterString = topCenterTextView.string
        case topRightTextView:
            headerFooterObj.topRightString = topRightTextView.string
        case bottomLeftTextView:
            headerFooterObj.bottomLeftString = bottomLeftTextView.string
        case BottomCenterTextView:
            headerFooterObj.bottomCenterString = BottomCenterTextView.string
        case bottomRightTextView:
            headerFooterObj.bottomRightString = bottomRightTextView.string
        default:
            break
        }

        applyButton.isEnabled = !(
            topLeftTextView.string.isEmpty &&
            topCenterTextView.string.isEmpty &&
            topRightTextView.string.isEmpty &&
            bottomLeftTextView.string.isEmpty &&
            BottomCenterTextView.string.isEmpty &&
            bottomRightTextView.string.isEmpty
        )

        self.updatePDFView()
    }

    func saveAsPDFToPath(_ path: String, autoOpen: Bool) {
        // self.loadingView.hidden = false
        // self.loadingView.progressIndicator.startAnimation(nil)
        DispatchQueue.global(qos: .default).async { [unowned self] in
            var filePath = self.pdfDocument?.documentURL?.path
            let password = self.password
            if filePath == nil {
                let writeSuccess = self.pdfDocument!.write(to: URL(fileURLWithPath: kNewDocumentTempSavePath(NSLocalizedString("Untitled", comment: ""))))
                if writeSuccess {
                    let newDocument = CPDFDocument(url: URL(fileURLWithPath: kNewDocumentTempSavePath(NSLocalizedString("Untitled", comment: ""))))
                    filePath = newDocument?.documentURL?.path
                } else {
                    return
                }
            }

            guard let filePath = filePath else { return }
            
            let document = CPDFDocument(url: NSURL.fileURL(withPath: filePath))
            document?.unlock(withPassword: password)
            
            let font = NSFont.boldSystemFont(ofSize: self.headerFooterObj.getTextFontSize())
            let style = NSMutableParagraphStyle()
            style.alignment = .center
            style.lineBreakMode = .byCharWrapping
            let dictionary: [NSAttributedString.Key: Any] = [.paragraphStyle: style, .font: font]
            let size = ("text" as NSString).boundingRect(with: CGSize(width: CGFloat.greatestFiniteMagnitude, height: CGFloat.greatestFiniteMagnitude),
                                                         options: [.usesLineFragmentOrigin, .usesFontLeading],
                                                         attributes: dictionary).size

            if self.isBates {
                if let bates = document?.bates() {
                    bates.margin = NSEdgeInsets(top: max(self.headerFooterObj.topMargin - size.height, 0),
                                                left: self.headerFooterObj.leftMargin,
                                                bottom: max(self.headerFooterObj.bottomMargin - size.height, 0),
                                                right: self.headerFooterObj.rightMargin)
                    let pagesString = self.headerFooterObj.pageRangeString
                    if  pagesString.count != 0 {
                        bates.pageString = pagesString
                    } else {
                        let pageString = "0-\(document!.pageCount - 1)"
                        bates.pageString = pageString
                    }

                    let items = [self.headerFooterObj.topLeftString,
                                 self.headerFooterObj.topCenterString,
                                 self.headerFooterObj.topRightString,
                                 self.headerFooterObj.bottomLeftString,
                                 self.headerFooterObj.bottomCenterString,
                                 self.headerFooterObj.bottomRightString]

                    for i in 0..<items.count {
                        let text = items[i]
                        bates.setText(text, at: UInt(i))
                        bates.setTextColor(self.headerFooterObj.getTextColor(), at: UInt(i))
                        bates.setFontSize(self.headerFooterObj.getTextFontSize(), at: UInt(i))
                    }

                    bates.update()
                }
            } else {
                if let headerFooter = document?.headerFooter() {
                    headerFooter.margin = NSEdgeInsets(top: max(self.headerFooterObj.topMargin - size.height, 0),
                                                       left: self.headerFooterObj.leftMargin,
                                                       bottom: max(self.headerFooterObj.bottomMargin - size.height, 0),
                                                       right: self.headerFooterObj.rightMargin)

                    let pagesString = self.headerFooterObj.pageRangeString
                    if  pagesString.count != 0 {
                        headerFooter.pageString = pagesString
                    } else {
                        let pageString = "0-\(document!.pageCount - 1)"
                        headerFooter.pageString = pageString
                    }

                    let pageCount = self.pdfView.document?.pageCount ?? UInt(0 + (self.headerFooterObj.startString as NSString).integerValue - 1)
                    let topLeftString = self.convertPageFormat(self.headerFooterObj.topLeftString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")
                    let topCenterString = self.convertPageFormat(self.headerFooterObj.topCenterString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")
                    let topRightString = self.convertPageFormat(self.headerFooterObj.topRightString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")
                    let bottomLeftString = self.convertPageFormat(self.headerFooterObj.bottomLeftString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")
                    let bottomCenterString = self.convertPageFormat(self.headerFooterObj.bottomCenterString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")
                    let bottomRightString = self.convertPageFormat(self.headerFooterObj.bottomRightString, startPage: self.headerFooterObj.startString, pageCount: "\(pageCount)")

                    let items = [topLeftString, topCenterString, topRightString, bottomLeftString, bottomCenterString, bottomRightString]

                    for i in 0..<items.count {
                        let text = items[i]
                        headerFooter.setText(text, at: UInt(i))
                        headerFooter.setTextColor(self.headerFooterObj.getTextColor(), at: UInt(i))
                        headerFooter.setFontSize(self.headerFooterObj.getTextFontSize(), at: UInt(i))
                    }

                    headerFooter.update()
                }
            }

            let documentPath = NSTemporaryDirectory()
            let tempPath = (documentPath as NSString).appendingPathComponent((filePath as NSString).lastPathComponent)
            if FileManager.default.fileExists(atPath: tempPath) {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            let result = document?.write(to: URL(fileURLWithPath: tempPath)) ?? false
            if result {
                if FileManager.default.fileExists(atPath: path) {
                    try? FileManager.default.removeItem(atPath: path)
                }
                try? FileManager.default.moveItem(atPath: tempPath, toPath: path)
            } else {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            if result {
                DispatchQueue.main.async {
                    // self.loadingView.hidden = true
                    // self.loadingView.progressIndicator.stopAnimation(nil)
                    self.cancelAction?(self)
                    if autoOpen {
                        NSDocumentController.shared.openDocument(withContentsOf: URL(fileURLWithPath: path), display: true) { _, _, _ in
                        }
                    } else {
                        NSWorkspace.shared.selectFile(path, inFileViewerRootedAtPath: "")
                    }
                }
            } else {
                DispatchQueue.main.async {
                    // self.loadingView.hidden = true
                    // self.loadingView.progressIndicator.stopAnimation(nil)
                }
            }
        }
    }
    
    func kNewDocumentTempSavePath(_ fileName: String) -> String {
        let searchPath = NSSearchPathForDirectoriesInDomains(.applicationSupportDirectory, .userDomainMask, true).last
        let append1 = searchPath?.stringByAppendingPathComponent(Bundle.main.bundleIdentifier!)
        let append2 = append1!.stringByAppendingPathComponent(String(format: "%@", fileName))
        return append2
    }

    static func saveAsPDFRemoveAllHeaderFooter(_ PDFDocument: CPDFDocument,
                                              password: String?,
                                              toPath path: String,
                                              completionHandler handler: ((Int) -> Void)?) {
        DispatchQueue.global(qos: .default).async {
            var filePath = PDFDocument.documentURL?.path
            let document = CPDFDocument(url: URL(fileURLWithPath: filePath ?? ""))
            if let password = password {
                document?.unlock(withPassword: password)
            }

            if let headerFooter = document?.headerFooter {
                headerFooter()?.clear()
            }

            let documentPath = NSTemporaryDirectory()
            let tempPath = (documentPath as NSString).appendingPathComponent((path as NSString).lastPathComponent)
            if FileManager.default.fileExists(atPath: tempPath) {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            let result = document?.write(to: URL(fileURLWithPath: tempPath)) ?? false
            if result {
                if FileManager.default.fileExists(atPath: path) {
                    try? FileManager.default.removeItem(atPath: path)
                }
                try? FileManager.default.moveItem(atPath: tempPath, toPath: path)
            } else {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            DispatchQueue.main.async {
                if result {
                    handler?(1)
                } else {
                    handler?(0)
                }
            }
        }
    }

    static func saveAsPDFRemoveAllHeaderFooterBates(_ PDFDocument: CPDFDocument,
                                                  password: String?,
                                                  toPath path: String,
                                                  completionHandler handler: ((Int) -> Void)?) {
        DispatchQueue.global(qos: .default).async {
            var filePath = PDFDocument.documentURL?.path
            let document = CPDFDocument(url: URL(fileURLWithPath: filePath ?? ""))
            if let password = password {
                document?.unlock(withPassword: password)
            }

            if let bates = document?.bates {
                bates()?.clear()
            }

            let documentPath = NSTemporaryDirectory()
            let tempPath = (documentPath as NSString).appendingPathComponent((path as NSString).lastPathComponent)
            if FileManager.default.fileExists(atPath: tempPath) {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            let result = document?.write(to: URL(fileURLWithPath: tempPath)) ?? false
            if result {
                if FileManager.default.fileExists(atPath: path) {
                    try? FileManager.default.removeItem(atPath: path)
                }
                try? FileManager.default.moveItem(atPath: tempPath, toPath: path)
            } else {
                try? FileManager.default.removeItem(atPath: tempPath)
            }

            DispatchQueue.main.async {
                if result {
                    handler?(1)
                } else {
                    handler?(0)
                }
            }
        }
    }


    @objc func controlTextDidEndEditing(_ obj: Notification) {
        guard let comboBox = obj.object as? NSComboBox else {
            return
        }

        if comboBox == pageRangeCombobox {
            if !checkPageRangeValidate(pageRangeString: comboBox.stringValue) {
                let alert = NSAlert()
                alert.alertStyle = .critical
                alert.messageText = "\(pdfDocument?.documentURL?.lastPathComponent ?? NSLocalizedString("Untitled", comment: "")) \(NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: ""))"
                alert.runModal()

                headerFooterObj.pageRangeType = .all
                pageRangeCombobox.selectItem(at: 0)
//                pageRangeCombobox.isEditable = false
                pageRangeCombobox.isSelectable = false
                return
            } else {
                headerFooterObj.pageRangeString = comboBox.stringValue
                self.updatePDFView()
            }
        }
    }

    @objc func comboBoxSelectionDidChange(_ notification: Notification) {
        guard let comboBox = notification.object as? NSComboBox else {
            return
        }

        if comboBox == pageRangeCombobox {
//            pageRangeCombobox.isEditable = false

            switch comboBox.indexOfSelectedItem {
            case 0:
                headerFooterObj.pageRangeType = .all
            case 1:
                headerFooterObj.pageRangeType = .odd
            case 2:
                headerFooterObj.pageRangeType = .even
            default:
                headerFooterObj.pageRangeType = .other
                pageRangeCombobox.stringValue = ""
                pageRangeCombobox.isEditable = true
                window?.makeFirstResponder(pageRangeCombobox)
            }
        }
    }

    @objc func pageChangeNotification(_ notification: Notification) {
        currentPageIndexTextField.stringValue = pdfView.currentPage().label ?? ""
    }

    @objc func themeChanged(_ notification: Notification) {
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.3) {
            self.updateViewColor()
        }
    }
    
    func updatePDFView() {
        self.pdfView.needsDisplay = true
    }

}

extension KMHeaderFooterView {
    @IBAction func previousPage(_ sender: Any) {
        if pdfView.canGoToPreviousPage() {
            pdfView.goToPreviousPage(nil)
        }
        let index = pdfDocument?.index(for: pdfView.currentPage()) ?? 0
        currentPageIndexTextField.stringValue = "\(index + 1)"
    }

    @IBAction func nextPage(_ sender: Any) {
        if pdfView.canGoToNextPage() {
            pdfView.goToNextPage(nil)
        }
        
        let index = pdfDocument?.index(for: pdfView.currentPage()) ?? 0
        currentPageIndexTextField.stringValue = "\(index + 1)"
    }

    @IBAction func pageIndexTextFieldEndEdit(_ sender: NSTextField) {
        pdfView.go(to: pdfDocument?.page(at: UInt(sender.integerValue - 1)))
    }

    @IBAction func chooseColor(_ sender: NSColorWell) {
        headerFooterObj.textColor =  KMWatermarkAdjectiveText.color(red: sender.color.redComponent, green: sender.color.greenComponent, blue: sender.color.blueComponent, alpha: sender.color.alphaComponent)
        self.updatePDFView()
    }

    @IBAction func chooseFont(_ sender: NSComboBox) {
        headerFooterObj.textFont = KMWatermarkAdjectiveText.font(name: "Helvetica", size: sender.stringValue.stringToCGFloat())
        self.updatePDFView()
    }

    @IBAction func topMarginTextFiedEndEdit(_ sender: NSTextField) {
        topMarginStepper.integerValue = sender.integerValue
        headerFooterObj.topMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func topMarginStepperAction(_ sender: NSStepper) {
        topMargintextField.integerValue = sender.integerValue
        headerFooterObj.topMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func bottomMarginTextFieldEndEdit(_ sender: NSTextField) {
        bottomMarginStepper.integerValue = sender.integerValue
        headerFooterObj.bottomMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func bottomMarginStepperAction(_ sender: NSStepper) {
        bottomMarginTextField.integerValue = sender.integerValue
        headerFooterObj.bottomMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func leftMarginTextFieldEndEdit(_ sender: NSTextField) {
        leftMarginStepper.integerValue = sender.integerValue
        headerFooterObj.leftMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func leftMarginStepperAction(_ sender: NSStepper) {
        leftMarginTextField.integerValue = sender.integerValue
        headerFooterObj.leftMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func rightMarginTextFieldEndEdit(_ sender: NSTextField) {
        rightMarginStepper.integerValue = sender.integerValue
        headerFooterObj.rightMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func rightMarginStepperAction(_ sender: NSStepper) {
        rightMarginTextField.integerValue = sender.integerValue
        headerFooterObj.rightMargin = CGFloat(sender.integerValue)
        self.updatePDFView()
    }

    @IBAction func chooseDateFormate(_ sender: NSComboBox) {
        headerFooterObj.dateFormatString = sender.stringValue
        self.updatePDFView()
    }

    @IBAction func choosePageFormate(_ sender: NSComboBox) {
        headerFooterObj.pageFormatString = sender.stringValue
        self.updatePDFView()
    }

    @IBAction func chooseStartPage(_ sender: NSComboBox) {
        headerFooterObj.startString = sender.stringValue
        self.updatePDFView()
    }

    @IBAction func buttonClicked_AddHeaderFooter(_ sender: NSButton) {
        popMenu(sender: sender)
    }

    @IBAction func saveTemplateOrNot(_ sender: Any) {
        
    }

    @IBAction func templateNameTextFieldEndEdit(_ sender: NSTextField) {
        headerFooterObj.id = sender.stringValue
    }

    @IBAction func batch(_ sender: Any) {
        if saveToTemplateButton.state == .on {
            if templateNameTextField.stringValue.count < 1 {
                return
            }
            
            headerFooterObj.id = templateNameTextField.stringValue
            KMHeaderFooterManager.defaultManager.addTemplate(headerFooterObj)
        }
        
        if let operateCallBack = operateCallBack {
            operateCallBack(headerFooterObj)
        }
    }

    @IBAction func cancel(_ sender: Any) {
        self.cancelAction?(self)
    }

    @IBAction func apply(_ sender: Any) {
        let needSave = saveToTemplateButton.state == .on
        
        var pages = [String]()
        switch pageRangeCombobox.indexOfSelectedItem {
        case 0:
            for i in 0..<pdfDocument!.pageCount {
                pages.append("\(i)")
            }
        case 1:
            for i in 0..<pdfDocument!.pageCount where i % 2 == 0 {
                pages.append("\(i)")
            }
        case 2:
            for i in 0..<pdfDocument!.pageCount where i % 2 != 0 {
                pages.append("\(i)")
            }
        default:
            let fileAttribute = KMFileAttribute()
            fileAttribute.filePath = pdfDocument?.documentURL?.path ?? ""
            fileAttribute.bAllPage = false
            fileAttribute.pagesString = pageRangeCombobox.stringValue
            
            if fileAttribute.fetchSelectPages().count == 0 {
                let alert = NSAlert()
                alert.alertStyle = .critical
                alert.messageText = "\(fileAttribute.filePath.lastPathComponent) \(NSLocalizedString("Invalid page range or the page number is out of range. Please try again.", comment: ""))"
                alert.runModal()
                return
            }
            for num in fileAttribute.fetchSelectPages() {
                pages.append("\(num)")
            }
        }
        
        if pages.count > 0 {
            headerFooterObj.pageRangeString = pages.joined(separator: ",")
        } else {
            headerFooterObj.pageRangeString = ""
        }
        
        if type == .Add {
            let object = headerFooterObj
            if object.topLeftString.isEmpty && object.topCenterString.isEmpty && object.topRightString.isEmpty &&
                object.bottomLeftString.isEmpty && object.bottomCenterString.isEmpty && object.bottomRightString.isEmpty {
                return
            }
            if needSave {
                if templateNameTextField.stringValue.isEmpty {
                    headerFooterObj.id = initialID
                } else {
                    headerFooterObj.id = templateNameTextField.stringValue
                }
                KMHeaderFooterManager.defaultManager.addTemplate(headerFooterObj)
            }
            if let operateCallBack = operateCallBack {
                operateCallBack(headerFooterObj)
            }
        } else if type == .Edit {
            if needSave {
                if templateNameTextField.stringValue.isEmpty {
                    headerFooterObj.id = initialID
                } else {
                    headerFooterObj.id = templateNameTextField.stringValue
                }
                
                originalHeaderFooterObj = headerFooterObj.copy() as! KMHeaderFooterModel
                KMHeaderFooterManager.defaultManager.store()
            }
            if let operateCallBack = operateCallBack {
                operateCallBack(originalHeaderFooterObj)
            }
        } else if type == .Use {
            
            var fileName: String?
            if isBates {
                fileName = "\(pdfDocument?.documentURL?.lastPathComponent.deletingPathExtension ?? NSLocalizedString("Untitled", comment: ""))_Bates"
            } else {
                fileName = "\(pdfDocument?.documentURL?.lastPathComponent.deletingPathExtension ?? NSLocalizedString("Untitled", comment: ""))_Header_Footer"
            }
            
            let savePanelAccessoryViewController = KMSavePanelAccessoryController()
            let savePanel = NSSavePanel()
            savePanel.nameFieldStringValue = fileName ?? ""
            savePanel.allowedFileTypes = ["pdf"]
            savePanel.accessoryView = savePanelAccessoryViewController.view
            savePanel.beginSheetModal(for: self.window!) { result in
                if result.rawValue == NSApplication.ModalResponse.OK.rawValue {
                    self.saveAsPDFToPath(savePanel.url!.path, autoOpen: savePanelAccessoryViewController.openAutomaticButton.state == .on ? true : false)
                    
                    if needSave {
                        if self.templateNameTextField.stringValue.isEmpty {
                            self.headerFooterObj.id = self.initialID
                        } else {
                            self.headerFooterObj.id = self.templateNameTextField.stringValue
                        }
                        KMHeaderFooterManager.defaultManager.addTemplate(self.headerFooterObj)
                        NotificationCenter.default.post(name: NSNotification.Name("KMBatchOperateHeaderFootersNotification"), object: self)
                    }
                }
            }
        }
    }

    @IBAction func batesprefixChanged(_ sender: NSTextField) {
        headerFooterObj.batesPrefixString = sender.stringValue
        self.updatePDFView()
    }

    @IBAction func batesSuffixChanged(_ sender: NSTextField) {
        headerFooterObj.batesSuffixString = sender.stringValue
        self.updatePDFView()
    }

    @IBAction func batesNumberChanged(_ sender: NSTextField) {
        batesNumberStepper.integerValue = sender.integerValue
        headerFooterObj.batesDigits = sender.integerValue
        self.updatePDFView()
    }

    @IBAction func bateNumberStepperAction(_ sender: NSStepper) {
        batesNumberTextField.integerValue = sender.integerValue
        headerFooterObj.batesDigits = sender.integerValue
        self.updatePDFView()
    }

    @IBAction func batesStartPageComboboxChanged(_ sender: NSComboBox) {
        headerFooterObj.startString = sender.stringValue
        self.updatePDFView()
    }
}

extension KMHeaderFooterView{
    func convertPageFormat(_ oldString: String, startPage: String, pageCount: String) -> String {
        var newString = oldString
        
        for pageFormat in pageFormatArray {
            let formattedPage = "<<\(pageFormat)>>"
            
            if newString.contains(formattedPage) {
                var tString: String?
                
                switch pageFormat {
                case "<<1>>":
                    tString = "<<\(startPage)>>"
                case "<<1 of n>>":
                    tString = "<<\(startPage)>> of \(pageCount)"
                case "<<1/n>>":
                    tString = "<<\(startPage)>>/\(pageCount)"
                case "<<Page 1>>":
                    tString = "Page \(startPage)"
                case "<<Page 1 of n>>":
                    tString = "Page \(startPage) of \(pageCount)"
                default:
                    break
                }
                
                if let tString = tString {
                    newString = newString.replacingOccurrences(of: formattedPage, with: tString)
                }
            }
        }
        
        newString = convertDateFormat(newString)
        return newString
    }
}