From cd035d008ee94b2b88617b5b9d7ff3c68c55e955 Mon Sep 17 00:00:00 2001
From: li3317 <li3317@purdue.edu>
Date: Sun, 23 Jan 2022 22:51:33 -0500
Subject: [PATCH] add collectAllIndividualStyle

---
 lib/src/models/documents/document.dart   |  6 +++++
 lib/src/models/documents/nodes/line.dart | 31 ++++++++++++++++++++++++
 lib/src/widgets/controller.dart          |  7 ++++++
 3 files changed, 44 insertions(+)

diff --git a/lib/src/models/documents/document.dart b/lib/src/models/documents/document.dart
index 62a3efc4..0bc42c4f 100644
--- a/lib/src/models/documents/document.dart
+++ b/lib/src/models/documents/document.dart
@@ -157,6 +157,12 @@ class Document {
     return (res.node as Line).collectStyle(res.offset, len);
   }
 
+  /// Returns all styles for each node within selection
+  List<Tuple2<int, Style>> collectAllIndividualStyles(int index, int len) {
+    final res = queryChild(index);
+    return (res.node as Line).collectAllIndividualStyles(res.offset, len);
+  }
+
   /// Returns all styles for any character within the specified text range.
   List<Style> collectAllStyles(int index, int len) {
     final res = queryChild(index);
diff --git a/lib/src/models/documents/nodes/line.dart b/lib/src/models/documents/nodes/line.dart
index f1f5c1f5..b8d25edc 100644
--- a/lib/src/models/documents/nodes/line.dart
+++ b/lib/src/models/documents/nodes/line.dart
@@ -1,6 +1,7 @@
 import 'dart:math' as math;
 
 import 'package:collection/collection.dart';
+import 'package:tuple/tuple.dart';
 
 import '../../quill_delta.dart';
 import '../attribute.dart';
@@ -382,6 +383,36 @@ class Line extends Container<Leaf?> {
     return result;
   }
 
+  /// Returns each offset in selection with its corresponding styles as a list
+  List<Tuple2<int, Style>> collectAllIndividualStyles(int offset, int len) {
+    final local = math.min(length - offset, len);
+    final result = <Tuple2<int, Style>>[];
+
+    final data = queryChild(offset, true);
+    var node = data.node as Leaf?;
+    if (node != null) {
+      result.add(Tuple2(0, node.style));
+      var pos = node.length - data.offset; // 2
+      while (!node!.isLast && pos < local) {
+        node = node.next as Leaf;
+        result.add(Tuple2(pos, node.style)); // 2
+        pos += node.length; // 2 + 1 = 3
+      }
+    }
+
+    // Add line style
+    // result.add(style);
+    // TODO: add block style as well
+
+    final remaining = len - local;
+    if (remaining > 0) {
+      final rest = nextLine!.collectAllIndividualStyles(0, remaining);
+      result.addAll(rest);
+    }
+
+    return result;
+  }
+
   /// Returns all styles for any character within the specified text range.
   /// In essence, it is UNION of each individual segment's styles
   List<Style> collectAllStyles(int offset, int len) {
diff --git a/lib/src/widgets/controller.dart b/lib/src/widgets/controller.dart
index 8df60cb5..58343eb1 100644
--- a/lib/src/widgets/controller.dart
+++ b/lib/src/widgets/controller.dart
@@ -84,6 +84,13 @@ class QuillController extends ChangeNotifier {
         .mergeAll(toggledStyle);
   }
 
+  /// Returns all styles for each node within selection
+  List<Tuple2<int, Style>> getAllIndividualSelectionStyles() {
+    final styles = document.collectAllIndividualStyles(
+        selection.start, selection.end - selection.start);
+    return styles;
+  }
+
   /// Returns all styles for any character within the specified text range.
   List<Style> getAllSelectionStyles() {
     final styles = document.collectAllStyles(