Refactor logic in generation text edits for array element deletion

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/DataMapManager.java

@@ -1322,15 +1322,15 @@ private void genDeleteMappingSource(SemanticModel semanticModel, ExpressionNode
                         textEdits.add(new TextEdit(CommonUtils.toRange(expr.lineRange()), ""));
                     } else {
                         if (memberIdx + 1 == expressions.size()) {
-                            LinePosition startPos = expressions.get(memberIdx - 1).lineRange().startLine();
+                            LinePosition startPos = expressions.get(memberIdx - 1).lineRange().endLine();
                             LinePosition endPos = expr.lineRange().endLine();
                             textEdits.add(new TextEdit(CommonUtils.toRange(startPos, endPos), ""));
                         } else if (memberIdx == 0) {
                             LinePosition startPos = expr.lineRange().startLine();
                             LinePosition endPos = expressions.get(1).lineRange().startLine();
                             textEdits.add(new TextEdit(CommonUtils.toRange(startPos, endPos), ""));
                         } else {
-                            LinePosition startPos = expressions.get(memberIdx - 1).lineRange().startLine();
+                            LinePosition startPos = expressions.get(memberIdx - 1).lineRange().endLine();
                             LinePosition endPos = expr.lineRange().endLine();
                             textEdits.add(new TextEdit(CommonUtils.toRange(startPos, endPos), ""));
                         }

flow-model-generator/modules/flow-model-generator-ls-extension/src/test/java/io/ballerina/flowmodelgenerator/extension/DataMappingDeleteTest.java

@@ -61,6 +61,8 @@ protected Object[] getConfigsList() {
                 {Path.of("variable5.json")},
                 {Path.of("function_defn1.json")},
                 {Path.of("function_defn2.json")},
+                {Path.of("variable6.json")},
+                {Path.of("variable7.json")}
         };
     }
 

flow-model-generator/modules/flow-model-generator-ls-extension/src/test/resources/delete_mapping/config/variable6.json

@@ -0,0 +1,40 @@
+{
+  "source": "variable5.bal",
+  "description": "",
+  "codedata": {
+    "node": "VARIABLE",
+    "lineRange": {
+      "fileName": "variable5.bal",
+      "startLine": {
+        "line": 11,
+        "offset": 8
+      },
+      "endLine": {
+        "line": 13,
+        "offset": 10
+      }
+    },
+    "sourceCode": "ArrType arr2 = {\n            rec: [{i: 0}]\n        };"
+  },
+  "mapping": {
+    "output": "arr2.rec.1"
+  },
+  "targetField": "arr2",
+  "output": {
+    "variable5.bal": [
+      {
+        "range": {
+          "start": {
+            "line": 12,
+            "character": 24
+          },
+          "end": {
+            "line": 12,
+            "character": 32
+          }
+        },
+        "newText": ""
+      }
+    ]
+  }
+}

flow-model-generator/modules/flow-model-generator-ls-extension/src/test/resources/delete_mapping/config/variable7.json

@@ -0,0 +1,40 @@
+{
+  "source": "variable5.bal",
+  "description": "",
+  "codedata": {
+    "node": "VARIABLE",
+    "lineRange": {
+      "fileName": "variable5.bal",
+      "startLine": {
+        "line": 11,
+        "offset": 8
+      },
+      "endLine": {
+        "line": 13,
+        "offset": 10
+      }
+    },
+    "sourceCode": "ArrType arr2 = {\n            rec: [{i: 0}]\n        };"
+  },
+  "mapping": {
+    "output": "arr2.rec.2"
+  },
+  "targetField": "arr2",
+  "output": {
+    "variable5.bal": [
+      {
+        "range": {
+          "start": {
+            "line": 12,
+            "character": 32
+          },
+          "end": {
+            "line": 12,
+            "character": 40
+          }
+        },
+        "newText": ""
+      }
+    ]
+  }
+}

flow-model-generator/modules/flow-model-generator-ls-extension/src/test/resources/delete_mapping/source/variable5.bal

@@ -0,0 +1,18 @@
+type RecType record {|
+    int i;
+|};
+
+type ArrType record {|
+    RecType[] rec;
+|}
+
+public function main() returns error? {
+    do {
+        ArrType arr1 = {};
+        ArrType arr2 = {
+            rec: [{i: 0}, {i: 1}, {i: 2}]
+        };
+    } on fail error e {
+
+    }
+}