Add Find feature. Setup NSTextFinderClient.

Sources/STTextView/NSTextContentManager+Helpers.swift

@@ -0,0 +1,22 @@
+//  Created by Marcin Krzyzanowski
+//  https://github.com/krzyzanowskim/STTextView/blob/main/LICENSE.md
+
+import Cocoa
+
+extension NSTextContentManager {
+
+    var documentString: String {
+        var result: String = ""
+        result.reserveCapacity(1024 * 4)
+
+        enumerateTextElements(from: nil, options: []) { textElement in
+            if let textParagraph = textElement as? NSTextParagraph {
+                result += textParagraph.attributedString.string
+            }
+
+            return true
+        }
+        return result
+    }
+
+}

Sources/STTextView/NSTextLayoutManager+Helpers.swift

@@ -1,7 +1,6 @@
 //  Created by Marcin Krzyzanowski
 //  https://github.com/krzyzanowskim/STTextView/blob/main/LICENSE.md
 
-
 import Cocoa
 
 extension NSTextLayoutManager {

Sources/STTextView/STTextFinderClient.swift

@@ -0,0 +1,124 @@
+//  Created by Marcin Krzyzanowski
+//  https://github.com/krzyzanowskim/STTextView/blob/main/LICENSE.md
+
+import Cocoa
+
+final class STTextFinderClient: NSObject, NSTextFinderClient {
+
+    weak var textView: STTextView?
+
+    private var textContentManager: NSTextContentManager? {
+        textView?.textContentStorage
+    }
+
+    var string: String {
+        textView?.string ?? ""
+    }
+
+    func stringLength() -> Int {
+        string.count
+    }
+
+    var isSelectable: Bool {
+        textView?.isSelectable ?? false
+    }
+
+    public var allowsMultipleSelection: Bool {
+        false
+    }
+
+    public var firstSelectedRange: NSRange {
+        guard let firstTextSelectionRange = textView?.textLayoutManager.textSelections.first?.textRanges.first,
+              let textContentManager = textContentManager else {
+            return NSRange()
+        }
+
+        return NSRange(firstTextSelectionRange, in: textContentManager)
+    }
+
+    public var selectedRanges: [NSValue] {
+        set {
+            guard let textContentManager = textContentManager,
+                  let textLayoutManager = textView?.textLayoutManager as? STTextLayoutManager else {
+                assertionFailure()
+                return
+            }
+
+            let textRanges = newValue.map(\.rangeValue).compactMap {
+                NSTextRange($0, in: textContentManager)
+            }
+
+            textLayoutManager.textSelections = [NSTextSelection(textRanges, affinity: .downstream, granularity: .character)]
+            textView?.updateSelectionHighlights()
+        }
+
+        get {
+            guard let textContentManager = textContentManager,
+                  let textLayoutManager = textView?.textLayoutManager,
+                  !textLayoutManager.textSelections.isEmpty
+            else {
+                return []
+            }
+
+            return textLayoutManager.textSelections.flatMap(\.textRanges).compactMap({ NSRange($0, in: textContentManager) }).map({ NSValue(range: $0) })
+        }
+    }
+
+    var isEditable: Bool {
+        textView?.isEditable ?? false
+    }
+
+    func scrollRangeToVisible(_ range: NSRange) {
+        guard let textContentManager = textContentManager,
+              let textRange = NSTextRange(range, in: textContentManager)
+        else {
+            return
+        }
+        textView?.scrollToSelection(NSTextSelection(range: textRange, affinity: .downstream, granularity: .character))
+    }
+
+    var visibleCharacterRanges: [NSValue] {
+        guard let viewportTextRange = textView?.textLayoutManager.textViewportLayoutController.viewportRange,
+              let textContentManager = textContentManager else {
+            return []
+        }
+
+        return [NSRange(viewportTextRange, in: textContentManager)].map({ NSValue(range: $0) })
+    }
+
+    func rects(forCharacterRange range: NSRange) -> [NSValue]? {
+        guard let textContentManager = textContentManager,
+              let textRange = NSTextRange(range, in: textContentManager)
+        else {
+            return nil
+        }
+
+        var rangeRects: [CGRect] = []
+        textView?.textLayoutManager.enumerateTextSegments(in: textRange, type: .standard, options: .rangeNotRequired, using: { _, rect, _, _ in
+            rangeRects.append(rect.pixelAligned)
+            return true
+        })
+
+        return rangeRects.map({ NSValue(rect: $0) })
+    }
+
+    func contentView(at index: Int, effectiveCharacterRange outRange: NSRangePointer) -> NSView {
+        outRange.pointee = NSRange(location: 0, length: stringLength())
+        return textView!
+    }
+
+    func drawCharacters(in range: NSRange, forContentView view: NSView) {
+        guard let textView = view as? STTextView,
+              let textRange = NSTextRange(range, in: textView.textContentStorage),
+              let context = NSGraphicsContext.current?.cgContext
+        else {
+            assertionFailure()
+            return
+        }
+
+        if let layoutFragment = textView.textLayoutManager.textLayoutFragment(for: textRange.location) {
+            layoutFragment.draw(at: layoutFragment.layoutFragmentFrame.origin, in: context)
+        }
+    }
+
+}

Sources/STTextView/STTextLayoutManager.swift

@@ -3,15 +3,12 @@
 
 import Cocoa
 
-final class STTextLayoutManager: NSTextLayoutManager {
+public final class STTextLayoutManager: NSTextLayoutManager {
 
-    static let didChangeSelectionNotification = STTextView.didChangeSelectionNotification
-
-    override var textSelections: [NSTextSelection] {
+    public override var textSelections: [NSTextSelection] {
         didSet {
-            let notification = Notification(name: STTextLayoutManager.didChangeSelectionNotification, object: self, userInfo: nil)
+            let notification = Notification(name: STTextView.didChangeSelectionNotification, object: self, userInfo: nil)
             NotificationCenter.default.post(notification)
         }
     }
-
 }

Sources/STTextView/STTextView+Delete.swift

@@ -75,6 +75,10 @@ extension STTextView {
         var didChange = false
         textContentStorage.performEditingTransaction {
             for textRange in textRanges where shouldChangeText(in: textRange, replacementString: nil) {
+                if didChange == false {
+                    willChangeText()
+                }
+
                 didChange = true
                 let nsrange = NSRange(textRange, in: textContentStorage)
                 textContentStorage.textStorage?.deleteCharacters(in: nsrange)

Sources/STTextView/STTextView+Find.swift

@@ -0,0 +1,22 @@
+//  Created by Marcin Krzyzanowski
+//  https://github.com/krzyzanowskim/STTextView/blob/main/LICENSE.md
+
+import Cocoa
+
+extension STTextView {
+
+    @objc func performFindPanelAction(_ sender: Any?) {
+        performTextFinderAction(sender)
+    }
+
+    @objc open override func performTextFinderAction(_ sender: Any?) {
+        guard let menuItem = sender as? NSMenuItem else {
+            assertionFailure("Unexpected caller")
+            return
+        }
+
+
+        textFinder.performAction(NSTextFinder.Action(rawValue: menuItem.tag)!)
+    }
+
+}

Sources/STTextView/STTextView+NSUserInterfaceValidations.swift

@@ -0,0 +1,42 @@
+//  Created by Marcin Krzyzanowski
+//  https://github.com/krzyzanowskim/STTextView/blob/main/LICENSE.md
+
+import Cocoa
+
+extension STTextView: NSMenuItemValidation {
+
+    public func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
+        validateUserInterfaceItem(menuItem)
+    }
+
+}
+
+extension STTextView: NSUserInterfaceValidations {
+
+    public func validateUserInterfaceItem(_ item: NSValidatedUserInterfaceItem) -> Bool {
+        switch item.action {
+        case #selector(undo(_:)):
+            let result = allowsUndo ? undoManager?.canUndo ?? false : false
+
+            // NSWindow does that like this, here (as debugged)
+            if let undoManager = undoManager {
+                (item as? NSMenuItem)?.title = undoManager.undoMenuItemTitle
+            }
+
+            return result
+        case #selector(redo(_:)):
+            let result = allowsUndo ? undoManager?.canRedo ?? false : false
+
+            // NSWindow does that like this, here (as debugged)
+            if let undoManager = undoManager {
+                (item as? NSMenuItem)?.title = undoManager.redoMenuItemTitle
+            }
+            return result
+        case #selector(performFindPanelAction(_:)), #selector(performTextFinderAction(_:)):
+            return textFinder.validateAction(NSTextFinder.Action(rawValue: item.tag)!)
+        default:
+            return true
+        }
+    }
+
+}

Sources/STTextView/STTextView+TextInputClient.swift

@@ -84,8 +84,6 @@ extension STTextView: NSTextInputClient {
 
     open func insertText(_ string: Any, replacementRange: NSRange) {
         guard isEditable else { return }
-        var didChange = false 
-
         textContentStorage.performEditingTransaction {
             switch string {
             case is String:
@@ -94,14 +92,16 @@ extension STTextView: NSTextInputClient {
                 }
                 if let textRange = NSTextRange(replacementRange, in: textContentStorage) {
                     if shouldChangeText(in: textRange, replacementString: string) {
+                        willChangeText()
                         replaceCharacters(in: textRange, with: string)
-                        didChange = true
+                        didChangeText()
                     }
                 } else if !textLayoutManager.textSelections.isEmpty {
                     for textRange in textLayoutManager.textSelections.flatMap(\.textRanges) {
                         if shouldChangeText(in: textRange, replacementString: string) {
+                            willChangeText()
                             replaceCharacters(in: textRange, with: string)
-                            didChange = true
+                            didChangeText()
                         }
                     }
                 }
@@ -111,14 +111,16 @@ extension STTextView: NSTextInputClient {
                 }
                 if let textRange = NSTextRange(replacementRange, in: textContentStorage) {
                     if shouldChangeText(in: textRange, replacementString: string.string) {
+                        willChangeText()
                         replaceCharacters(in: textRange, with: string)
-                        didChange = true
+                        didChangeText()
                     }
                 } else if !textLayoutManager.textSelections.isEmpty {
                     for textRange in textLayoutManager.textSelections.flatMap(\.textRanges) {
                         if shouldChangeText(in: textRange, replacementString: string.string) {
+                            willChangeText()
                             replaceCharacters(in: textRange, with: string)
-                            didChange = true
+                            didChangeText()
                         }
                     }
                 }
@@ -127,10 +129,6 @@ extension STTextView: NSTextInputClient {
             }
 
         }
-
-        if didChange {
-            didChangeText()
-        }
     }
 
 

Sources/STTextView/STTextView+Undo.swift

@@ -22,38 +22,3 @@ extension STTextView {
     }
 
 }
-
-extension STTextView: NSMenuItemValidation {
-
-    public func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
-        validateUserInterfaceItem(menuItem)
-    }
-
-}
-
-extension STTextView: NSUserInterfaceValidations {
-
-    public func validateUserInterfaceItem(_ item: NSValidatedUserInterfaceItem) -> Bool {
-        if item.action == #selector(undo(_:)) {
-            let result = allowsUndo ? undoManager?.canUndo ?? false : false
-
-            // NSWindow does that like this, here (as debugged)
-            if let undoManager = undoManager {
-                (item as? NSMenuItem)?.title = undoManager.undoMenuItemTitle
-            }
-
-            return result
-        } else if item.action == #selector(redo(_:)) {
-            let result = allowsUndo ? undoManager?.canRedo ?? false : false
-
-            // NSWindow does that like this, here (as debugged)
-            if let undoManager = undoManager {
-                (item as? NSMenuItem)?.title = undoManager.redoMenuItemTitle
-            }
-            return result
-        }
-
-        return true
-    }
-
-}