Draw Selections

Author: thecoolwinter

Sources/CodeEditSourceEditor/Minimap/MinimapLineFragmentView.swift

@@ -10,7 +10,7 @@ import CodeEditTextView
 
 /// A custom line fragment view for the minimap.
 ///
-/// Instead of drawing line contents, this view calculates a series of boxes or 'runs' to draw to represent the text
+/// Instead of drawing line contents, this view calculates a series of bubbles or 'runs' to draw to represent the text
 /// in the line fragment.
 ///
 /// Runs are calculated when the view's fragment is set, and cached until invalidated, and all whitespace
@@ -114,12 +114,12 @@ final class MinimapLineFragmentView: LineFragmentView {
         for run in drawingRuns {
             let rect = CGRect(
                 x: 8 + (CGFloat(run.range.location) * 1.5),
-                y: 0,
+                y: 0.25,
                 width: CGFloat(run.range.length) * 1.5,
                 height: 2.0
             )
             context.setFillColor(run.color.cgColor)
-            context.fill(rect)
+            context.fill(rect.pixelAligned)
         }
 
         context.restoreGState()

Sources/CodeEditSourceEditor/Minimap/MinimapLineRenderer.swift

@@ -59,4 +59,8 @@ final class MinimapLineRenderer: TextLayoutManagerRenderDelegate {
     func lineFragmentView(for lineFragment: LineFragment) -> LineFragmentView {
         MinimapLineFragmentView(textStorage: textView?.textStorage)
     }
+
+    func characterXPosition(in lineFragment: LineFragment, for offset: Int) -> CGFloat {
+        8 + (CGFloat(offset) * 1.5)
+    }
 }

Sources/CodeEditSourceEditor/Minimap/MinimapView+Draw.swift

@@ -0,0 +1,21 @@
+//
+//  MinimapView+Draw.swift
+//  CodeEditSourceEditor
+//
+//  Created by Khan Winter on 4/16/25.
+//
+
+import AppKit
+import CodeEditTextView
+
+public class MinimapContentView: FlippedNSView {
+    weak var textView: TextView?
+    weak var selectionManager: TextSelectionManager?
+
+    override public func draw(_ dirtyRect: NSRect) {
+        super.draw(dirtyRect)
+        if textView?.isSelectable ?? false {
+            selectionManager?.drawSelections(in: dirtyRect)
+        }
+    }
+}

Sources/CodeEditSourceEditor/Minimap/MinimapView+TextLayoutManagerDelegate.swift

@@ -10,7 +10,6 @@ import CodeEditTextView
 
 extension MinimapView: TextLayoutManagerDelegate {
     public func layoutManagerHeightDidUpdate(newHeight: CGFloat) {
-//        contentView.frame.size.height = newHeight
         updateContentViewHeight()
     }
 

Sources/CodeEditSourceEditor/Minimap/MinimapView+TextSelectionManagerDelegate.swift

@@ -0,0 +1,29 @@
+//
+//  MinimapView+TextSelectionManagerDelegate.swift
+//  CodeEditSourceEditor
+//
+//  Created by Khan Winter on 4/16/25.
+//
+
+import AppKit
+import CodeEditTextView
+
+extension MinimapView: TextSelectionManagerDelegate {
+    public var visibleTextRange: NSRange? {
+        let minY = max(visibleRect.minY, 0)
+        let maxY = min(visibleRect.maxY, layoutManager?.estimatedHeight() ?? 3.0)
+        guard let minYLine = layoutManager?.textLineForPosition(minY),
+              let maxYLine = layoutManager?.textLineForPosition(maxY) else {
+            return nil
+        }
+        return NSRange(start: minYLine.range.location, end: maxYLine.range.max)
+    }
+
+    public func setNeedsDisplay() {
+        contentView.needsDisplay = true
+    }
+
+    public func estimatedLineHeight() -> CGFloat {
+        layoutManager?.estimateLineHeight() ?? 3.0
+    }
+}

Sources/CodeEditSourceEditor/Minimap/MinimapView.swift

@@ -30,7 +30,7 @@ public class MinimapView: FlippedNSView {
     /// The container scrollview for the minimap contents.
     public let scrollView: ForwardingScrollView
     /// The view text lines are rendered into.
-    public let contentView: FlippedNSView
+    public let contentView: MinimapContentView
     /// The box displaying the visible region on the minimap.
     public let documentVisibleView: NSView
     /// A small gray line on the left of the minimap distinguishing it from the editor.
@@ -41,6 +41,7 @@ public class MinimapView: FlippedNSView {
 
     /// The layout manager that uses the ``lineRenderer`` to render and layout lines.
     var layoutManager: TextLayoutManager?
+    var selectionManager: TextSelectionManager?
     /// A custom line renderer that lays out lines of text as 2px tall and draws contents as small lines
     /// using ``MinimapLineFragmentView``
     let lineRenderer: MinimapLineRenderer
@@ -82,7 +83,7 @@ public class MinimapView: FlippedNSView {
         scrollView.verticalScrollElasticity = .none
         scrollView.receiver = textView.enclosingScrollView
 
-        self.contentView = FlippedNSView()
+        self.contentView = MinimapContentView()
         contentView.translatesAutoresizingMaskIntoConstraints = false
 
         self.documentVisibleView = NSView()
@@ -121,6 +122,15 @@ public class MinimapView: FlippedNSView {
         self.layoutManager = layoutManager
         (textView.textStorage.delegate as? MultiStorageDelegate)?.addDelegate(layoutManager)
 
+        self.selectionManager = TextSelectionManager(
+            layoutManager: layoutManager,
+            textStorage: textView.textStorage,
+            textView: textView,
+            delegate: self
+        )
+        contentView.textView = textView
+        contentView.selectionManager = selectionManager
+
         wantsLayer = true
         layer?.backgroundColor = theme.background.cgColor
 
@@ -180,6 +190,16 @@ public class MinimapView: FlippedNSView {
             self?.updateContentViewHeight()
             self?.updateDocumentVisibleViewPosition()
         }
+
+        NotificationCenter.default.addObserver(
+            forName: TextSelectionManager.selectionChangedNotification,
+            object: textView?.selectionManager,
+            queue: .main
+        ) { [weak self] _ in
+            self?.selectionManager?.setSelectedRanges(
+                self?.textView?.selectionManager.textSelections.map(\.range) ?? []
+            )
+        }
     }
 
     required init?(coder: NSCoder) {