From f02fcda8c04448720d3d23bac6db1ca44cb0831e Mon Sep 17 00:00:00 2001
From: squidrye <mihirs167@gmail.com>
Date: Fri, 24 Mar 2023 00:12:42 +0530
Subject: [PATCH] fix: open-ai replace does not work on certain use-cases

---
 .../lib/src/core/transform/transaction.dart   | 30 ++++++++++++++-----
 1 file changed, 23 insertions(+), 7 deletions(-)

diff --git a/frontend/appflowy_flutter/packages/appflowy_editor/lib/src/core/transform/transaction.dart b/frontend/appflowy_flutter/packages/appflowy_editor/lib/src/core/transform/transaction.dart
index 15bce2a6b7..5aba71b243 100644
--- a/frontend/appflowy_flutter/packages/appflowy_editor/lib/src/core/transform/transaction.dart
+++ b/frontend/appflowy_flutter/packages/appflowy_editor/lib/src/core/transform/transaction.dart
@@ -371,6 +371,14 @@ extension TextTransaction on Transaction {
 
     if (textNodes.length < texts.length) {
       final length = texts.length;
+      var path = textNodes.first.path;
+
+      //FIND OFFSET CHARACTERS AFTER SELECTION
+      String offSetChar =
+      (document.nodeAtPath(selection.end.path) as TextNode)
+          .toPlainText()
+          .substring(selection.end.offset);
+
       for (var i = 0; i < texts.length; i++) {
         final text = texts[i];
         if (i == 0) {
@@ -380,7 +388,7 @@ extension TextTransaction on Transaction {
             textNodes.first.toPlainText().length,
             text,
           );
-        } else if (i == length - 1) {
+        } else if (i == length - 1 && textNodes.length >= 2) {
           replaceText(
             textNodes.last,
             0,
@@ -396,15 +404,23 @@ extension TextTransaction on Transaction {
               text,
             );
           } else {
-            var path = textNodes.first.path;
-            var j = i - textNodes.length + length - 1;
-            while (j > 0) {
-              path = path.next;
-              j--;
+            if (i == texts.length - 1) {
+              //ADD OFFSET CHARACTER TO END OF LAST TEXT-NODE TO AVOID DATA LOSS
+              document.insert(path, [
+                TextNode(
+                  delta: Delta()..insert("${text}${offSetChar}"),
+                ),
+              ]);
+            } else {
+              document.insert(path, [
+                TextNode(
+                  delta: Delta()..insert(text),
+                ),
+              ]);
             }
-            insertNode(path, TextNode(delta: Delta()..insert(text)));
           }
         }
+        path = path.next;
       }
       afterSelection = null;
       return;