[SuperEditor] Make selected text color strategy consider each colored span (Resolves #2093)

Author: angelosilvestre

super_editor/lib/src/default_editor/layout_single_column/_styler_user_selection.dart

@@ -132,25 +132,52 @@ class SingleColumnLayoutSelectionStyler extends SingleColumnLayoutStylePhase {
       editorStyleLog.finer('   - extent: ${textSelection?.extent}');
 
       if (viewModel is TextComponentViewModel) {
-        final componentTextColor = viewModel.textStyleBuilder({}).color;
-
-        final textWithSelectionAttributions =
-            textSelection != null && _selectedTextColorStrategy != null && componentTextColor != null
-                ? (viewModel.text.copyText(0)
-                  ..addAttribution(
-                    ColorAttribution(_selectedTextColorStrategy!(
-                      originalTextColor: componentTextColor,
-                      selectionHighlightColor: _selectionStyles.selectionColor,
-                    )),
-                    SpanRange(textSelection.start, textSelection.end - 1),
-                    // The selected range might already have a color attribution. We want to override it
-                    // with the selected text color.
-                    overwriteConflictingSpans: true,
-                  ))
-                : viewModel.text;
+        AttributedText? textWithSelectionAttributions;
+
+        if (textSelection != null && _selectedTextColorStrategy != null) {
+          final selectedRange = SpanRange(textSelection.start, textSelection.end - 1);
+
+          final componentTextColor = viewModel.textStyleBuilder({}).color;
+          if (componentTextColor != null) {
+            // Compute the selected text color for the default color of the node. If there is any
+            // text color attributions in the selected range, they will override this color.
+            textWithSelectionAttributions = viewModel.text.copyText(0)
+              ..addAttribution(
+                ColorAttribution(_selectedTextColorStrategy!(
+                  originalTextColor: componentTextColor,
+                  selectionHighlightColor: _selectionStyles.selectionColor,
+                )),
+                selectedRange,
+                // Override any existing color attributions.
+                overwriteConflictingSpans: true,
+              );
+          }
+
+          final coloredSpans = viewModel.text.getAttributionSpansInRange(
+            attributionFilter: (attr) => attr is ColorAttribution,
+            range: selectedRange,
+            // We should only change the selected portion of each span.
+            resizeSpansToFitInRange: true,
+          );
+          if (coloredSpans.isNotEmpty) {
+            // Compute and apply the selected text color for each span that has a color attribution.
+            textWithSelectionAttributions ??= viewModel.text.copyText(0);
+            for (final span in coloredSpans) {
+              textWithSelectionAttributions.addAttribution(
+                ColorAttribution(_selectedTextColorStrategy!(
+                  originalTextColor: (span.attribution as ColorAttribution).color,
+                  selectionHighlightColor: _selectionStyles.selectionColor,
+                )),
+                SpanRange(span.start, span.end),
+                // Override any existing color attributions.
+                overwriteConflictingSpans: true,
+              );
+            }
+          }
+        }
 
         viewModel
-          ..text = textWithSelectionAttributions
+          ..text = textWithSelectionAttributions ?? viewModel.text
           ..selection = textSelection
           ..selectionColor = _selectionStyles.selectionColor
           ..highlightWhenEmpty = highlightWhenEmpty;

super_editor/test/super_editor/supereditor_selection_test.dart

@@ -88,6 +88,79 @@ void main() {
         expect(richText.getSpanForPosition(const TextPosition(offset: 5))!.style!.color, Colors.green);
         expect(richText.getSpanForPosition(const TextPosition(offset: 16))!.style!.color, Colors.green);
       });
+
+      testWidgetsOnArbitraryDesktop("overrides each colored span", (tester) async {
+        final stylesheet = defaultStylesheet.copyWith(
+          selectedTextColorStrategy: ({required Color originalTextColor, required Color selectionHighlightColor}) {
+            if (originalTextColor == Colors.green) {
+              return Colors.red;
+            }
+
+            if (originalTextColor == Colors.yellow) {
+              return Colors.blue;
+            }
+
+            return Colors.white;
+          },
+        );
+
+        // Pump an editor with a paragraph with the following colors:
+        // Lorem ipsum dolor
+        // gggggg-----------
+        // ------yyyyyy-----
+        // ------------bbbbb (black, the default color)
+        await tester //
+            .createDocument()
+            .withCustomContent(
+              MutableDocument(
+                nodes: [
+                  ParagraphNode(
+                    id: '1',
+                    text: AttributedText(
+                      'Lorem ipsum dolor',
+                      AttributedSpans(
+                        attributions: [
+                          SpanMarker(
+                              attribution: const ColorAttribution(Colors.green),
+                              offset: 0,
+                              markerType: SpanMarkerType.start),
+                          SpanMarker(
+                              attribution: const ColorAttribution(Colors.green),
+                              offset: 5,
+                              markerType: SpanMarkerType.end),
+                          SpanMarker(
+                              attribution: const ColorAttribution(Colors.yellow),
+                              offset: 6,
+                              markerType: SpanMarkerType.start),
+                          SpanMarker(
+                              attribution: const ColorAttribution(Colors.yellow),
+                              offset: 11,
+                              markerType: SpanMarkerType.end),
+                        ],
+                      ),
+                    ),
+                  ),
+                ],
+              ),
+            )
+            .useStylesheet(stylesheet)
+            .pump();
+
+        // Triple tap to select the whole paragraph.
+        await tester.tripleTapInParagraph('1', 2);
+
+        // Ensure that all spans changed colors.
+        final richText = SuperEditorInspector.findRichTextInParagraph('1');
+
+        expect(richText.getSpanForPosition(const TextPosition(offset: 0))!.style!.color, Colors.red);
+        expect(richText.getSpanForPosition(const TextPosition(offset: 5))!.style!.color, Colors.red);
+
+        expect(richText.getSpanForPosition(const TextPosition(offset: 6))!.style!.color, Colors.blue);
+        expect(richText.getSpanForPosition(const TextPosition(offset: 11))!.style!.color, Colors.blue);
+
+        expect(richText.getSpanForPosition(const TextPosition(offset: 12))!.style!.color, Colors.white);
+        expect(richText.getSpanForPosition(const TextPosition(offset: 16))!.style!.color, Colors.white);
+      });
     });
 
     testWidgetsOnArbitraryDesktop("calculates upstream document selection within a single node", (tester) async {