fix:修复@if标签移除的问题

mini-software · Oct 30, 2024 · b37b543 · b37b543
1 parent a27b211
commit b37b543
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/MiniWord/MiniWord.Implment.cs b/src/MiniWord/MiniWord.Implment.cs
@@ -779,15 +779,17 @@ private static void ReplaceIfStatements(OpenXmlElement rootXmlElement, List<Open
 
                     for (int i = paragraphIfIndex + 1; i <= paragraphEndIfIndex - 1; i++)
                     {
-                        if(rootXmlElement.ChildElements.Any(c=>c == elementList[i])) rootXmlElement.RemoveChild(elementList[i]);
+                        elementList[i].Remove();
                     }
                 }
-                if(rootXmlElement.ChildElements.Any(c => c == ifP))
-                    rootXmlElement.RemoveChild(ifP);
-                if (rootXmlElement.ChildElements.Any(c => c == endIfP))
-                    rootXmlElement.RemoveChild(endIfP);
+                // 从paragraphs中移除，防止死循环
                 paragraphs.Remove(ifP);
                 paragraphs.Remove(endIfP);
+                // 从doc元素移除
+                if (ifP.Parent != null)
+                    ifP.Remove();
+                if (endIfP.Parent != null)
+                    endIfP.Remove();
             }
         }