Fix not removing associated data for type variant

elm-tooling · Feb 6, 2024 · 2d59d0d · 2d59d0d
1 parent b52d88e
commit 2d59d0d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/common/util/refactorEditUtils.ts b/src/common/util/refactorEditUtils.ts
@@ -476,14 +476,16 @@ export class RefactorEditUtils {
       return this.removeType(unionVariants[0].parent);
     }
 
-    let startPosition = nodeAtPosition.startPosition;
-    let endPosition = nodeAtPosition.endPosition;
-
     const unionVariant = unionVariants?.find(
       (a) =>
         a.text.startsWith(`${nodeAtPosition.text} `) ||
         a.text === nodeAtPosition.text,
     );
+
+    let startPosition =
+      unionVariant?.startPosition ?? nodeAtPosition.startPosition;
+    let endPosition = unionVariant?.endPosition ?? nodeAtPosition.endPosition;
+
     if (unionVariant?.previousSibling?.type == "eq") {
       startPosition = unionVariant.previousSibling?.endPosition;
       if (unionVariant.nextSibling?.type == "|") {