iOS PDFKit: How to add a highlight annotation

PDFViewController.swift

override func viewDidLoad() {
    createMenu()
}

private func createMenu() {
    let highlightItem = UIMenuItem(title: "Highlight", action: #selector(highlight(_:)))
    UIMenuController.shared.menuItems = [highlightItem]
}
    
@objc private func highlight(_ sender: UIMenuController?) {
    guard let currentSelection = pdfView.currentSelection else { return }
    let selections = currentSelection.selectionsByLine()
    guard let page = selections.first?.pages.first else { return }

    selections.forEach { selection in
        let highlight = PDFAnnotation(bounds: selection.bounds(for: page), forType: .highlight, withProperties: nil)
        highlight.endLineStyle = .square
        page.addAnnotation(highlight)
    }

    pdfView.clearSelection()
}

I can't take credit for this code. I found it on this Chinese website and thought there should be more examples like this one out on the web. https://qiita.com/ngo275/items/bbdc16daa58d2926b77c

Thanks a lot for this piece of code buddy, it helped my like a life-saving-jacket 😄
Thanks for this contribution

actually it is a japanese Page.