Mutate view hierarchy only on annotations count change

krzyzanowskim · Dec 27, 2023 · 89555f8 · 89555f8
1 parent 9e66e48
commit 89555f8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/AnnotationsPlugin/AnnotationsPlugin.swift b/Sources/AnnotationsPlugin/AnnotationsPlugin.swift
@@ -83,6 +83,11 @@ extension AnnotationsPlugin {
             guard let dataSource = parent.dataSource else {
                 return
             }
+
+            let annotations = dataSource.textViewAnnotations()
+            guard annotations.count != annotationsViews.count else {
+                return
+            }
 
             // Remove all views
             for view in annotationsViews {
@@ -92,7 +97,7 @@ extension AnnotationsPlugin {
 
             // Add views for annotations
             let textLayoutManager = context.textView.textLayoutManager
-            for annotation in dataSource.textViewAnnotations() {
+            for annotation in annotations {
                 textLayoutManager.ensureLayout(for: NSTextRange(location: annotation.location))
                 if let textLineFragment = textLayoutManager.textLineFragment(at: annotation.location) {
                     if let annotationView = dataSource.textView(context.textView, viewForLineAnnotation: annotation, textLineFragment: textLineFragment) {