From 954d5b9444a5dfa8f1c1fdefd9666860566e3e03 Mon Sep 17 00:00:00 2001
From: hehong <619215399@qq.com>
Date: Sat, 15 Jul 2023 11:31:38 +0800
Subject: [PATCH] copy/cut select image and text together.

---
 lib/src/models/documents/document.dart        |  6 +++
 lib/src/models/documents/nodes/line.dart      | 38 +++++++++++++++++++
 lib/src/widgets/controller.dart               |  6 +++
 lib/src/widgets/editor.dart                   |  2 +-
 lib/src/widgets/raw_editor.dart               |  8 ++--
 ...editor_state_selection_delegate_mixin.dart | 38 ++++++++++++++-----
 6 files changed, 84 insertions(+), 14 deletions(-)

diff --git a/lib/src/models/documents/document.dart b/lib/src/models/documents/document.dart
index 122c9a9c..13c98017 100644
--- a/lib/src/models/documents/document.dart
+++ b/lib/src/models/documents/document.dart
@@ -165,6 +165,12 @@ class Document {
     return (res.node as Line).collectAllIndividualStyles(res.offset, len);
   }
 
+  List<OffsetValue> collectAllIndividualStyleAndEmbed(int index, int len) {
+    final res = queryChild(index);
+    return (res.node as Line)
+        .collectAllIndividualStylesAndEmbed(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 fdd342ac..653be164 100644
--- a/lib/src/models/documents/nodes/line.dart
+++ b/lib/src/models/documents/nodes/line.dart
@@ -430,6 +430,44 @@ class Line extends Container<Leaf?> {
     return result;
   }
 
+  List<OffsetValue> collectAllIndividualStylesAndEmbed(int offset, int len,
+      {int beg = 0}) {
+    final local = math.min(length - offset, len);
+    final result = <OffsetValue>[];
+
+    final data = queryChild(offset, true);
+    var node = data.node as Leaf?;
+    if (node != null) {
+      var pos = 0;
+      if (node is Text) {
+        pos = node.length - data.offset;
+        result.add(OffsetValue(beg, node.style));
+      } else if (node.value is Embeddable) {
+        pos = node.length - data.offset;
+        result.add(OffsetValue(beg, node.value as Embeddable));
+      }
+      while (!node!.isLast && pos < local) {
+        node = node.next as Leaf;
+        if (node is Text) {
+          result.add(OffsetValue(pos + beg, node.style));
+          pos += node.length;
+        } else if (node.value is Embeddable) {
+          result.add(OffsetValue(pos + beg, node.value as Embeddable));
+          pos += node.length;
+        }
+      }
+    }
+
+    final remaining = len - local;
+    if (remaining > 0 && nextLine != null) {
+      final rest =
+      nextLine!.collectAllIndividualStylesAndEmbed(0, remaining, beg: local);
+      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 aedd7015..91932a0c 100644
--- a/lib/src/widgets/controller.dart
+++ b/lib/src/widgets/controller.dart
@@ -166,6 +166,12 @@ class QuillController extends ChangeNotifier {
     return styles;
   }
 
+  List<OffsetValue> getAllIndividualSelectionStylesAndEmbed() {
+    final stylesAndEmbed = document.collectAllIndividualStyleAndEmbed(
+        selection.start, selection.end - selection.start);
+    return stylesAndEmbed;
+  }
+
   /// Returns plain text for each node within selection
   String getPlainText() {
     final text =
diff --git a/lib/src/widgets/editor.dart b/lib/src/widgets/editor.dart
index 554c1239..5fe89f8f 100644
--- a/lib/src/widgets/editor.dart
+++ b/lib/src/widgets/editor.dart
@@ -38,7 +38,7 @@ abstract class EditorState extends State<RawEditor>
 
   EditorTextSelectionOverlay? get selectionOverlay;
 
-  List<OffsetValue<Style>> get pasteStyle;
+  List<OffsetValue> get pasteStyleAndEmbed;
 
   String get pastePlainText;
 
diff --git a/lib/src/widgets/raw_editor.dart b/lib/src/widgets/raw_editor.dart
index 48b65df0..ff6a4140 100644
--- a/lib/src/widgets/raw_editor.dart
+++ b/lib/src/widgets/raw_editor.dart
@@ -318,8 +318,8 @@ class RawEditorState extends EditorState
 
   // for pasting style
   @override
-  List<OffsetValue<Style>> get pasteStyle => _pasteStyle;
-  List<OffsetValue<Style>> _pasteStyle = <OffsetValue<Style>>[];
+  List<OffsetValue> get pasteStyleAndEmbed => _pasteStyleAndEmbed;
+  List<OffsetValue> _pasteStyleAndEmbed = <OffsetValue>[];
 
   @override
   String get pastePlainText => _pastePlainText;
@@ -1435,7 +1435,7 @@ class RawEditorState extends EditorState
   void copySelection(SelectionChangedCause cause) {
     controller.copiedImageUrl = null;
     _pastePlainText = controller.getPlainText();
-    _pasteStyle = controller.getAllIndividualSelectionStyles();
+    _pasteStyleAndEmbed = controller.getAllIndividualSelectionStylesAndEmbed();
 
     final selection = textEditingValue.selection;
     final text = textEditingValue.text;
@@ -1464,7 +1464,7 @@ class RawEditorState extends EditorState
   void cutSelection(SelectionChangedCause cause) {
     controller.copiedImageUrl = null;
     _pastePlainText = controller.getPlainText();
-    _pasteStyle = controller.getAllIndividualSelectionStyles();
+    _pasteStyleAndEmbed = controller.getAllIndividualSelectionStylesAndEmbed();
 
     if (widget.readOnly) {
       return;
diff --git a/lib/src/widgets/raw_editor/raw_editor_state_selection_delegate_mixin.dart b/lib/src/widgets/raw_editor/raw_editor_state_selection_delegate_mixin.dart
index fdd9244b..aefa0b85 100644
--- a/lib/src/widgets/raw_editor/raw_editor_state_selection_delegate_mixin.dart
+++ b/lib/src/widgets/raw_editor/raw_editor_state_selection_delegate_mixin.dart
@@ -3,6 +3,7 @@ import 'dart:math' as math;
 import 'package:flutter/rendering.dart';
 import 'package:flutter/widgets.dart';
 
+import '../../../flutter_quill.dart';
 import '../../models/documents/document.dart';
 import '../../models/documents/nodes/leaf.dart';
 import '../../utils/delta.dart';
@@ -10,6 +11,9 @@ import '../editor.dart';
 
 mixin RawEditorStateSelectionDelegateMixin on EditorState
     implements TextSelectionDelegate {
+
+  bool isContainEmbed = false;
+
   @override
   TextEditingValue get textEditingValue {
     return widget.controller.plainTextEditingValue;
@@ -26,6 +30,7 @@ mixin RawEditorStateSelectionDelegateMixin on EditorState
       return;
     }
 
+    isContainEmbed = false;
     final insertedText = _adjustInsertedText(diff.inserted);
 
     widget.controller.replaceText(
@@ -37,15 +42,29 @@ mixin RawEditorStateSelectionDelegateMixin on EditorState
   void _applyPasteStyle(String insertedText, int start) {
     if (insertedText == pastePlainText && pastePlainText != '') {
       final pos = start;
-      for (var i = 0; i < pasteStyle.length; i++) {
-        final offset = pasteStyle[i].offset;
-        final style = pasteStyle[i].value;
-        widget.controller.formatTextStyle(
-            pos + offset,
-            i == pasteStyle.length - 1
-                ? pastePlainText.length - offset
-                : pasteStyle[i + 1].offset,
-            style);
+      for (var i = 0; i < pasteStyleAndEmbed.length; i++) {
+        final offset = pasteStyleAndEmbed[i].offset;
+        final styleAndEmbed = pasteStyleAndEmbed[i].value;
+
+        if (styleAndEmbed is Embeddable) {
+          widget.controller.replaceText(pos + offset, 0, styleAndEmbed, null);
+        } else {
+          widget.controller.formatTextStyle(
+              pos + offset,
+              i == pasteStyleAndEmbed.length - 1
+                  ? pastePlainText.length - offset
+                  : pasteStyleAndEmbed[i + 1].offset,
+              styleAndEmbed);
+        }
+      }
+    }else if(isContainEmbed){
+      final pos = start;
+      for (var i = 0; i < pasteStyleAndEmbed.length; i++) {
+        final offset = pasteStyleAndEmbed[i].offset;
+        final style = pasteStyleAndEmbed[i].value;
+        if (style is Embeddable) {
+          widget.controller.replaceText(pos + offset, 0, style, null);
+        }
       }
     }
   }
@@ -61,6 +80,7 @@ mixin RawEditorStateSelectionDelegateMixin on EditorState
     final sb = StringBuffer();
     for (var i = 0; i < text.length; i++) {
       if (text.codeUnitAt(i) == Embed.kObjectReplacementInt) {
+        isContainEmbed = true;
         continue;
       }
       sb.write(text[i]);