Mail hangs when removing multiple rows from large table.
[WebKit.git] / Source / WebCore / editing / DeleteSelectionCommand.cpp
1 /*
2  * Copyright (C) 2005 Apple Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "config.h"
27 #include "DeleteSelectionCommand.h"
28
29 #include "Document.h"
30 #include "DocumentMarkerController.h"
31 #include "Editor.h"
32 #include "EditorClient.h"
33 #include "ElementIterator.h"
34 #include "Frame.h"
35 #include "HTMLBRElement.h"
36 #include "HTMLLinkElement.h"
37 #include "HTMLNames.h"
38 #include "HTMLStyleElement.h"
39 #include "HTMLTableElement.h"
40 #include "NodeTraversal.h"
41 #include "RenderTableCell.h"
42 #include "RenderText.h"
43 #include "RenderedDocumentMarker.h"
44 #include "Text.h"
45 #include "VisibleUnits.h"
46 #include "htmlediting.h"
47
48 namespace WebCore {
49
50 using namespace HTMLNames;
51
52 static bool isTableRow(const Node* node)
53 {
54     return node && node->hasTagName(trTag);
55 }
56
57 static bool isTableCellEmpty(Node* cell)
58 {
59     ASSERT(isTableCell(cell));
60     return VisiblePosition(firstPositionInNode(cell)) == VisiblePosition(lastPositionInNode(cell));
61 }
62
63 static bool isTableRowEmpty(Node* row)
64 {
65     if (!isTableRow(row))
66         return false;
67         
68     for (Node* child = row->firstChild(); child; child = child->nextSibling())
69         if (isTableCell(child) && !isTableCellEmpty(child))
70             return false;
71     
72     return true;
73 }
74
75 DeleteSelectionCommand::DeleteSelectionCommand(Document& document, bool smartDelete, bool mergeBlocksAfterDelete, bool replace, bool expandForSpecialElements, bool sanitizeMarkup, EditAction editingAction)
76     : CompositeEditCommand(document, editingAction)
77     , m_hasSelectionToDelete(false)
78     , m_smartDelete(smartDelete)
79     , m_mergeBlocksAfterDelete(mergeBlocksAfterDelete)
80     , m_needPlaceholder(false)
81     , m_replace(replace)
82     , m_expandForSpecialElements(expandForSpecialElements)
83     , m_pruneStartBlockIfNecessary(false)
84     , m_startsAtEmptyLine(false)
85     , m_sanitizeMarkup(sanitizeMarkup)
86 {
87 }
88
89 DeleteSelectionCommand::DeleteSelectionCommand(const VisibleSelection& selection, bool smartDelete, bool mergeBlocksAfterDelete, bool replace, bool expandForSpecialElements, bool sanitizeMarkup, EditAction editingAction)
90     : CompositeEditCommand(selection.start().anchorNode()->document(), editingAction)
91     , m_hasSelectionToDelete(true)
92     , m_smartDelete(smartDelete)
93     , m_mergeBlocksAfterDelete(mergeBlocksAfterDelete)
94     , m_needPlaceholder(false)
95     , m_replace(replace)
96     , m_expandForSpecialElements(expandForSpecialElements)
97     , m_pruneStartBlockIfNecessary(false)
98     , m_startsAtEmptyLine(false)
99     , m_sanitizeMarkup(sanitizeMarkup)
100     , m_selectionToDelete(selection)
101 {
102 }
103
104 void DeleteSelectionCommand::initializeStartEnd(Position& start, Position& end)
105 {
106     HTMLElement* startSpecialContainer = nullptr;
107     HTMLElement* endSpecialContainer = nullptr;
108  
109     start = m_selectionToDelete.start();
110     end = m_selectionToDelete.end();
111  
112     // For HRs, we'll get a position at (HR,1) when hitting delete from the beginning of the previous line, or (HR,0) when forward deleting,
113     // but in these cases, we want to delete it, so manually expand the selection
114     if (start.deprecatedNode()->hasTagName(hrTag))
115         start = positionBeforeNode(start.deprecatedNode());
116     else if (end.deprecatedNode()->hasTagName(hrTag))
117         end = positionAfterNode(end.deprecatedNode());
118     
119     // FIXME: This is only used so that moveParagraphs can avoid the bugs in special element expansion.
120     if (!m_expandForSpecialElements)
121         return;
122     
123     while (1) {
124         startSpecialContainer = nullptr;
125         endSpecialContainer = nullptr;
126     
127         Position s = positionBeforeContainingSpecialElement(start, &startSpecialContainer);
128         Position e = positionAfterContainingSpecialElement(end, &endSpecialContainer);
129         
130         if (!startSpecialContainer && !endSpecialContainer)
131             break;
132
133         m_mergeBlocksAfterDelete = false;
134
135         if (VisiblePosition(start) != m_selectionToDelete.visibleStart() || VisiblePosition(end) != m_selectionToDelete.visibleEnd())
136             break;
137
138         // If we're going to expand to include the startSpecialContainer, it must be fully selected.
139         if (startSpecialContainer && !endSpecialContainer && comparePositions(positionInParentAfterNode(startSpecialContainer), end) > -1)
140             break;
141
142         // If we're going to expand to include the endSpecialContainer, it must be fully selected.
143         if (endSpecialContainer && !startSpecialContainer && comparePositions(start, positionInParentBeforeNode(endSpecialContainer)) > -1)
144             break;
145
146         if (startSpecialContainer && startSpecialContainer->isDescendantOf(endSpecialContainer))
147             // Don't adjust the end yet, it is the end of a special element that contains the start
148             // special element (which may or may not be fully selected).
149             start = s;
150         else if (endSpecialContainer && endSpecialContainer->isDescendantOf(startSpecialContainer))
151             // Don't adjust the start yet, it is the start of a special element that contains the end
152             // special element (which may or may not be fully selected).
153             end = e;
154         else {
155             start = s;
156             end = e;
157         }
158     }
159 }
160
161 void DeleteSelectionCommand::setStartingSelectionOnSmartDelete(const Position& start, const Position& end)
162 {
163     VisiblePosition newBase;
164     VisiblePosition newExtent;
165     if (startingSelection().isBaseFirst()) {
166         newBase = start;
167         newExtent = end;
168     } else {
169         newBase = end;
170         newExtent = start;        
171     }
172     setStartingSelection(VisibleSelection(newBase, newExtent, startingSelection().isDirectional())); 
173 }
174     
175 void DeleteSelectionCommand::initializePositionData()
176 {
177     Position start, end;
178     initializeStartEnd(start, end);
179     
180     if (!isEditablePosition(start, ContentIsEditable))
181         start = firstEditablePositionAfterPositionInRoot(start, highestEditableRoot(start));
182     if (!isEditablePosition(end, ContentIsEditable))
183         end = lastEditablePositionBeforePositionInRoot(end, highestEditableRoot(start));
184
185     m_upstreamStart = start.upstream();
186     m_downstreamStart = start.downstream();
187     m_upstreamEnd = end.upstream();
188     m_downstreamEnd = end.downstream();
189     
190     m_startRoot = editableRootForPosition(start);
191     m_endRoot = editableRootForPosition(end);
192     
193     m_startTableRow = enclosingNodeOfType(start, &isTableRow);
194     m_endTableRow = enclosingNodeOfType(end, &isTableRow);
195     
196     // Don't move content out of a table cell.
197     // If the cell is non-editable, enclosingNodeOfType won't return it by default, so
198     // tell that function that we don't care if it returns non-editable nodes.
199     Node* startCell = enclosingNodeOfType(m_upstreamStart, &isTableCell, CanCrossEditingBoundary);
200     Node* endCell = enclosingNodeOfType(m_downstreamEnd, &isTableCell, CanCrossEditingBoundary);
201     // FIXME: This isn't right.  A borderless table with two rows and a single column would appear as two paragraphs.
202     if (endCell && endCell != startCell)
203         m_mergeBlocksAfterDelete = false;
204     
205     // Usually the start and the end of the selection to delete are pulled together as a result of the deletion.
206     // Sometimes they aren't (like when no merge is requested), so we must choose one position to hold the caret 
207     // and receive the placeholder after deletion.
208     VisiblePosition visibleEnd(m_downstreamEnd);
209     if (m_mergeBlocksAfterDelete && !isEndOfParagraph(visibleEnd))
210         m_endingPosition = m_downstreamEnd;
211     else
212         m_endingPosition = m_downstreamStart;
213     
214     // We don't want to merge into a block if it will mean changing the quote level of content after deleting 
215     // selections that contain a whole number paragraphs plus a line break, since it is unclear to most users 
216     // that such a selection actually ends at the start of the next paragraph. This matches TextEdit behavior 
217     // for indented paragraphs.
218     // Only apply this rule if the endingSelection is a range selection.  If it is a caret, then other operations have created
219     // the selection we're deleting (like the process of creating a selection to delete during a backspace), and the user isn't in the situation described above.
220     if (numEnclosingMailBlockquotes(start) != numEnclosingMailBlockquotes(end) 
221             && isStartOfParagraph(visibleEnd) && isStartOfParagraph(VisiblePosition(start)) 
222             && endingSelection().isRange()) {
223         m_mergeBlocksAfterDelete = false;
224         m_pruneStartBlockIfNecessary = true;
225     }
226
227     // Handle leading and trailing whitespace, as well as smart delete adjustments to the selection
228     m_leadingWhitespace = m_upstreamStart.leadingWhitespacePosition(m_selectionToDelete.affinity());
229     m_trailingWhitespace = m_downstreamEnd.trailingWhitespacePosition(VP_DEFAULT_AFFINITY);
230
231     if (m_smartDelete) {
232     
233         // skip smart delete if the selection to delete already starts or ends with whitespace
234         Position pos = VisiblePosition(m_upstreamStart, m_selectionToDelete.affinity()).deepEquivalent();
235         bool skipSmartDelete = pos.trailingWhitespacePosition(VP_DEFAULT_AFFINITY, true).isNotNull();
236         if (!skipSmartDelete)
237             skipSmartDelete = m_downstreamEnd.leadingWhitespacePosition(VP_DEFAULT_AFFINITY, true).isNotNull();
238
239         // extend selection upstream if there is whitespace there
240         bool hasLeadingWhitespaceBeforeAdjustment = m_upstreamStart.leadingWhitespacePosition(m_selectionToDelete.affinity(), true).isNotNull();
241         if (!skipSmartDelete && hasLeadingWhitespaceBeforeAdjustment) {
242             VisiblePosition visiblePos = VisiblePosition(m_upstreamStart, VP_DEFAULT_AFFINITY).previous();
243             pos = visiblePos.deepEquivalent();
244             // Expand out one character upstream for smart delete and recalculate
245             // positions based on this change.
246             m_upstreamStart = pos.upstream();
247             m_downstreamStart = pos.downstream();
248             m_leadingWhitespace = m_upstreamStart.leadingWhitespacePosition(visiblePos.affinity());
249
250             setStartingSelectionOnSmartDelete(m_upstreamStart, m_upstreamEnd);
251         }
252         
253         // trailing whitespace is only considered for smart delete if there is no leading
254         // whitespace, as in the case where you double-click the first word of a paragraph.
255         if (!skipSmartDelete && !hasLeadingWhitespaceBeforeAdjustment && m_downstreamEnd.trailingWhitespacePosition(VP_DEFAULT_AFFINITY, true).isNotNull()) {
256             // Expand out one character downstream for smart delete and recalculate
257             // positions based on this change.
258             pos = VisiblePosition(m_downstreamEnd, VP_DEFAULT_AFFINITY).next().deepEquivalent();
259             m_upstreamEnd = pos.upstream();
260             m_downstreamEnd = pos.downstream();
261             m_trailingWhitespace = m_downstreamEnd.trailingWhitespacePosition(VP_DEFAULT_AFFINITY);
262
263             setStartingSelectionOnSmartDelete(m_downstreamStart, m_downstreamEnd);
264         }
265     }
266     
267     // We must pass call parentAnchoredEquivalent on the positions since some editing positions
268     // that appear inside their nodes aren't really inside them.  [hr, 0] is one example.
269     // FIXME: parentAnchoredEquivalent should eventually be moved into enclosing element getters
270     // like the one below, since editing functions should obviously accept editing positions.
271     // FIXME: Passing false to enclosingNodeOfType tells it that it's OK to return a non-editable
272     // node.  This was done to match existing behavior, but it seems wrong.
273     m_startBlock = enclosingNodeOfType(m_downstreamStart.parentAnchoredEquivalent(), &isBlock, CanCrossEditingBoundary);
274     m_endBlock = enclosingNodeOfType(m_upstreamEnd.parentAnchoredEquivalent(), &isBlock, CanCrossEditingBoundary);
275 }
276
277 void DeleteSelectionCommand::saveTypingStyleState()
278 {
279     // A common case is deleting characters that are all from the same text node. In 
280     // that case, the style at the start of the selection before deletion will be the 
281     // same as the style at the start of the selection after deletion (since those
282     // two positions will be identical). Therefore there is no need to save the
283     // typing style at the start of the selection, nor is there a reason to 
284     // compute the style at the start of the selection after deletion (see the 
285     // early return in calculateTypingStyleAfterDelete).
286     // However, if typing style was previously set from another text node at the previous
287     // position (now deleted), we need to clear that style as well.
288     if (m_upstreamStart.deprecatedNode() == m_downstreamEnd.deprecatedNode() && m_upstreamStart.deprecatedNode()->isTextNode()) {
289         frame().selection().clearTypingStyle();
290         return;
291     }
292
293     // Figure out the typing style in effect before the delete is done.
294     m_typingStyle = EditingStyle::create(m_selectionToDelete.start(), EditingStyle::EditingPropertiesInEffect);
295     m_typingStyle->removeStyleAddedByNode(enclosingAnchorElement(m_selectionToDelete.start()));
296
297     // If we're deleting into a Mail blockquote, save the style at end() instead of start()
298     // We'll use this later in computeTypingStyleAfterDelete if we end up outside of a Mail blockquote
299     if (enclosingNodeOfType(m_selectionToDelete.start(), isMailBlockquote))
300         m_deleteIntoBlockquoteStyle = EditingStyle::create(m_selectionToDelete.end());
301     else
302         m_deleteIntoBlockquoteStyle = nullptr;
303 }
304
305 bool DeleteSelectionCommand::handleSpecialCaseBRDelete()
306 {
307     Node* nodeAfterUpstreamStart = m_upstreamStart.computeNodeAfterPosition();
308     Node* nodeAfterDownstreamStart = m_downstreamStart.computeNodeAfterPosition();
309     // Upstream end will appear before BR due to canonicalization
310     Node* nodeAfterUpstreamEnd = m_upstreamEnd.computeNodeAfterPosition();
311
312     if (!nodeAfterUpstreamStart || !nodeAfterDownstreamStart)
313         return false;
314
315     // Check for special-case where the selection contains only a BR on a line by itself after another BR.
316     bool upstreamStartIsBR = nodeAfterUpstreamStart->hasTagName(brTag);
317     bool downstreamStartIsBR = nodeAfterDownstreamStart->hasTagName(brTag);
318     // We should consider that the BR is on a line by itself also when we have <br><br>. This test should be true only
319     // when the two elements are siblings and should be false in a case like <div><br></div><br>.
320     bool isBROnLineByItself = upstreamStartIsBR && downstreamStartIsBR && ((nodeAfterDownstreamStart == nodeAfterUpstreamEnd) || (nodeAfterUpstreamEnd && nodeAfterUpstreamEnd->hasTagName(brTag) && nodeAfterUpstreamStart->nextSibling() == nodeAfterUpstreamEnd));
321
322     if (isBROnLineByItself) {
323         removeNode(nodeAfterDownstreamStart);
324         return true;
325     }
326
327     // FIXME: This code doesn't belong in here.
328     // We detect the case where the start is an empty line consisting of BR not wrapped in a block element.
329     if (upstreamStartIsBR && downstreamStartIsBR
330         && !(isStartOfBlock(positionBeforeNode(nodeAfterUpstreamStart)) && isEndOfBlock(positionAfterNode(nodeAfterDownstreamStart)))
331         && (!nodeAfterUpstreamEnd || nodeAfterUpstreamEnd->hasTagName(brTag) || nodeAfterUpstreamEnd->previousSibling() != nodeAfterUpstreamStart)) {
332         m_startsAtEmptyLine = true;
333         m_endingPosition = m_downstreamEnd;
334     }
335     
336     return false;
337 }
338
339 static Position firstEditablePositionInNode(Node* node)
340 {
341     ASSERT(node);
342     Node* next = node;
343     while (next && !next->hasEditableStyle())
344         next = NodeTraversal::next(*next, node);
345     return next ? firstPositionInOrBeforeNode(next) : Position();
346 }
347
348 void DeleteSelectionCommand::insertBlockPlaceholderForTableCellIfNeeded(Element& element)
349 {
350     // Make sure empty cell has some height.
351     auto* renderer = element.renderer();
352     if (!is<RenderTableCell>(renderer))
353         return;
354     if (downcast<RenderTableCell>(*renderer).contentHeight() > 0)
355         return;
356     insertBlockPlaceholder(firstEditablePositionInNode(&element));
357 }
358     
359 void DeleteSelectionCommand::removeNodeUpdatingStates(Node& node, ShouldAssumeContentIsAlwaysEditable shouldAssumeContentIsAlwaysEditable)
360 {
361     if (&node == m_startBlock && !isEndOfBlock(VisiblePosition(firstPositionInNode(m_startBlock.get())).previous()))
362         m_needPlaceholder = true;
363     else if (&node == m_endBlock && !isStartOfBlock(VisiblePosition(lastPositionInNode(m_startBlock.get())).next()))
364         m_needPlaceholder = true;
365     
366     // FIXME: Update the endpoints of the range being deleted.
367     updatePositionForNodeRemoval(m_endingPosition, node);
368     updatePositionForNodeRemoval(m_leadingWhitespace, node);
369     updatePositionForNodeRemoval(m_trailingWhitespace, node);
370     
371     CompositeEditCommand::removeNode(&node, shouldAssumeContentIsAlwaysEditable);
372 }
373     
374 static inline bool shouldRemoveContentOnly(const Node& node)
375 {
376     return isTableStructureNode(&node) || node.isRootEditableElement();
377 }
378
379 void DeleteSelectionCommand::removeNode(PassRefPtr<Node> node, ShouldAssumeContentIsAlwaysEditable shouldAssumeContentIsAlwaysEditable)
380 {
381     if (!node)
382         return;
383         
384     if (m_startRoot != m_endRoot && !(node->isDescendantOf(m_startRoot.get()) && node->isDescendantOf(m_endRoot.get()))) {
385         // If a node is not in both the start and end editable roots, remove it only if its inside an editable region.
386         if (!node->parentNode()->hasEditableStyle()) {
387             // Don't remove non-editable atomic nodes.
388             if (!node->firstChild())
389                 return;
390             // Search this non-editable region for editable regions to empty.
391             RefPtr<Node> child = node->firstChild();
392             while (child) {
393                 RefPtr<Node> nextChild = child->nextSibling();
394                 removeNode(child.get(), shouldAssumeContentIsAlwaysEditable);
395                 // Bail if nextChild is no longer node's child.
396                 if (nextChild && nextChild->parentNode() != node)
397                     return;
398                 child = nextChild;
399             }
400             
401             // Don't remove editable regions that are inside non-editable ones, just clear them.
402             return;
403         }
404     }
405     
406     if (shouldRemoveContentOnly(*node)) {
407         // Do not remove an element of table structure; remove its contents.
408         // Likewise for the root editable element.
409         auto* child = NodeTraversal::next(*node, node.get());
410         while (child) {
411             if (shouldRemoveContentOnly(*child)) {
412                 child = NodeTraversal::next(*child, node.get());
413                 continue;
414             }
415             auto* remove = child;
416             child = NodeTraversal::nextSkippingChildren(*child, node.get());
417             removeNodeUpdatingStates(*remove, shouldAssumeContentIsAlwaysEditable);
418         }
419         
420         ASSERT(is<Element>(*node));
421         auto& element = downcast<Element>(*node);
422         document().updateLayoutIgnorePendingStylesheets();
423         // Check if we need to insert a placeholder for descendant table cells.
424         auto* descendant = ElementTraversal::next(element, &element);
425         while (descendant) {
426             auto* placeholderCandidate = descendant;
427             descendant = ElementTraversal::next(*descendant, &element);
428             insertBlockPlaceholderForTableCellIfNeeded(*placeholderCandidate);
429         }
430         insertBlockPlaceholderForTableCellIfNeeded(element);
431         return;
432     }
433     removeNodeUpdatingStates(*node, shouldAssumeContentIsAlwaysEditable);
434 }
435
436 static void updatePositionForTextRemoval(Node* node, int offset, int count, Position& position)
437 {
438     if (position.anchorType() != Position::PositionIsOffsetInAnchor || position.containerNode() != node)
439         return;
440
441     if (position.offsetInContainerNode() > offset + count)
442         position.moveToOffset(position.offsetInContainerNode() - count);
443     else if (position.offsetInContainerNode() > offset)
444         position.moveToOffset(offset);
445 }
446
447 void DeleteSelectionCommand::deleteTextFromNode(PassRefPtr<Text> node, unsigned offset, unsigned count)
448 {
449     // FIXME: Update the endpoints of the range being deleted.
450     updatePositionForTextRemoval(node.get(), offset, count, m_endingPosition);
451     updatePositionForTextRemoval(node.get(), offset, count, m_leadingWhitespace);
452     updatePositionForTextRemoval(node.get(), offset, count, m_trailingWhitespace);
453     updatePositionForTextRemoval(node.get(), offset, count, m_downstreamEnd);
454     
455     CompositeEditCommand::deleteTextFromNode(node, offset, count);
456 }
457
458 void DeleteSelectionCommand::makeStylingElementsDirectChildrenOfEditableRootToPreventStyleLoss()
459 {
460     RefPtr<Range> range = m_selectionToDelete.toNormalizedRange();
461     RefPtr<Node> node = range->firstNode();
462     while (node && node != range->pastLastNode()) {
463         RefPtr<Node> nextNode = NodeTraversal::next(*node);
464         if ((is<HTMLStyleElement>(*node) && !downcast<HTMLStyleElement>(*node).hasAttributeWithoutSynchronization(scopedAttr)) || is<HTMLLinkElement>(*node)) {
465             nextNode = NodeTraversal::nextSkippingChildren(*node);
466             RefPtr<ContainerNode> rootEditableElement = node->rootEditableElement();
467             if (rootEditableElement) {
468                 removeNode(node);
469                 appendNode(node, rootEditableElement);
470             }
471         }
472         node = nextNode;
473     }
474 }
475
476 void DeleteSelectionCommand::handleGeneralDelete()
477 {
478     if (m_upstreamStart.isNull())
479         return;
480
481     int startOffset = m_upstreamStart.deprecatedEditingOffset();
482     Node* startNode = m_upstreamStart.deprecatedNode();
483     
484     makeStylingElementsDirectChildrenOfEditableRootToPreventStyleLoss();
485
486     // Never remove the start block unless it's a table, in which case we won't merge content in.
487     if (startNode == m_startBlock && !startOffset && canHaveChildrenForEditing(*startNode) && !is<HTMLTableElement>(*startNode)) {
488         startOffset = 0;
489         startNode = NodeTraversal::next(*startNode);
490         if (!startNode)
491             return;
492     }
493
494     int startNodeCaretMaxOffset = caretMaxOffset(*startNode);
495     if (startOffset >= startNodeCaretMaxOffset && is<Text>(*startNode)) {
496         Text& text = downcast<Text>(*startNode);
497         if (text.length() > static_cast<unsigned>(startNodeCaretMaxOffset))
498             deleteTextFromNode(&text, startNodeCaretMaxOffset, text.length() - startNodeCaretMaxOffset);
499     }
500
501     if (startOffset >= lastOffsetForEditing(*startNode)) {
502         startNode = NodeTraversal::nextSkippingChildren(*startNode);
503         startOffset = 0;
504     }
505
506     // Done adjusting the start.  See if we're all done.
507     if (!startNode)
508         return;
509
510     if (startNode == m_downstreamEnd.deprecatedNode()) {
511         if (m_downstreamEnd.deprecatedEditingOffset() - startOffset > 0) {
512             if (is<Text>(*startNode)) {
513                 // in a text node that needs to be trimmed
514                 Text& text = downcast<Text>(*startNode);
515                 deleteTextFromNode(&text, startOffset, m_downstreamEnd.deprecatedEditingOffset() - startOffset);
516             } else {
517                 removeChildrenInRange(startNode, startOffset, m_downstreamEnd.deprecatedEditingOffset());
518                 m_endingPosition = m_upstreamStart;
519             }
520         }
521
522         // The selection to delete is all in one node.
523         if (!startNode->renderer() || (!startOffset && m_downstreamEnd.atLastEditingPositionForNode()))
524             removeNode(startNode);
525     }
526     else {
527         bool startNodeWasDescendantOfEndNode = m_upstreamStart.deprecatedNode()->isDescendantOf(m_downstreamEnd.deprecatedNode());
528         // The selection to delete spans more than one node.
529         RefPtr<Node> node(startNode);
530         
531         if (startOffset > 0) {
532             if (is<Text>(*startNode)) {
533                 // in a text node that needs to be trimmed
534                 Text& text = downcast<Text>(*node);
535                 deleteTextFromNode(&text, startOffset, text.length() - startOffset);
536                 node = NodeTraversal::next(*node);
537             } else {
538                 node = startNode->traverseToChildAt(startOffset);
539             }
540         } else if (startNode == m_upstreamEnd.deprecatedNode() && is<Text>(*startNode)) {
541             Text& text = downcast<Text>(*m_upstreamEnd.deprecatedNode());
542             deleteTextFromNode(&text, 0, m_upstreamEnd.deprecatedEditingOffset());
543         }
544         
545         // handle deleting all nodes that are completely selected
546         while (node && node != m_downstreamEnd.deprecatedNode()) {
547             if (comparePositions(firstPositionInOrBeforeNode(node.get()), m_downstreamEnd) >= 0) {
548                 // NodeTraversal::nextSkippingChildren just blew past the end position, so stop deleting
549                 node = nullptr;
550             } else if (!m_downstreamEnd.deprecatedNode()->isDescendantOf(*node)) {
551                 RefPtr<Node> nextNode = NodeTraversal::nextSkippingChildren(*node);
552                 // if we just removed a node from the end container, update end position so the
553                 // check above will work
554                 updatePositionForNodeRemoval(m_downstreamEnd, *node);
555                 removeNode(node.get());
556                 node = nextNode.get();
557             } else {
558                 Node* n = node->lastDescendant();
559                 if (m_downstreamEnd.deprecatedNode() == n && m_downstreamEnd.deprecatedEditingOffset() >= caretMaxOffset(*n)) {
560                     removeNode(node.get());
561                     node = nullptr;
562                 } else
563                     node = NodeTraversal::next(*node);
564             }
565         }
566         
567         if (m_downstreamEnd.deprecatedNode() != startNode && !m_upstreamStart.deprecatedNode()->isDescendantOf(m_downstreamEnd.deprecatedNode()) && m_downstreamEnd.anchorNode()->isConnected() && m_downstreamEnd.deprecatedEditingOffset() >= caretMinOffset(*m_downstreamEnd.deprecatedNode())) {
568             if (m_downstreamEnd.atLastEditingPositionForNode() && !canHaveChildrenForEditing(*m_downstreamEnd.deprecatedNode())) {
569                 // The node itself is fully selected, not just its contents.  Delete it.
570                 removeNode(m_downstreamEnd.deprecatedNode());
571             } else {
572                 if (is<Text>(*m_downstreamEnd.deprecatedNode())) {
573                     // in a text node that needs to be trimmed
574                     Text& text = downcast<Text>(*m_downstreamEnd.deprecatedNode());
575                     if (m_downstreamEnd.deprecatedEditingOffset() > 0) {
576                         deleteTextFromNode(&text, 0, m_downstreamEnd.deprecatedEditingOffset());
577                     }
578                 // Remove children of m_downstreamEnd.deprecatedNode() that come after m_upstreamStart.
579                 // Don't try to remove children if m_upstreamStart was inside m_downstreamEnd.deprecatedNode()
580                 // and m_upstreamStart has been removed from the document, because then we don't 
581                 // know how many children to remove.
582                 // FIXME: Make m_upstreamStart a position we update as we remove content, then we can
583                 // always know which children to remove.
584                 } else if (!(startNodeWasDescendantOfEndNode && !m_upstreamStart.anchorNode()->isConnected())) {
585                     unsigned offset = 0;
586                     if (m_upstreamStart.deprecatedNode()->isDescendantOf(m_downstreamEnd.deprecatedNode())) {
587                         Node* n = m_upstreamStart.deprecatedNode();
588                         while (n && n->parentNode() != m_downstreamEnd.deprecatedNode())
589                             n = n->parentNode();
590                         if (n)
591                             offset = n->computeNodeIndex() + 1;
592                     }
593                     removeChildrenInRange(m_downstreamEnd.deprecatedNode(), offset, m_downstreamEnd.deprecatedEditingOffset());
594                     m_downstreamEnd = createLegacyEditingPosition(m_downstreamEnd.deprecatedNode(), offset);
595                 }
596             }
597         }
598     }
599 }
600
601 void DeleteSelectionCommand::fixupWhitespace()
602 {
603     document().updateLayoutIgnorePendingStylesheets();
604     // FIXME: isRenderedCharacter should be removed, and we should use VisiblePosition::characterAfter and VisiblePosition::characterBefore
605     if (m_leadingWhitespace.isNotNull() && !m_leadingWhitespace.isRenderedCharacter() && is<Text>(*m_leadingWhitespace.deprecatedNode())) {
606         Text& textNode = downcast<Text>(*m_leadingWhitespace.deprecatedNode());
607         ASSERT(!textNode.renderer() || textNode.renderer()->style().collapseWhiteSpace());
608         replaceTextInNodePreservingMarkers(&textNode, m_leadingWhitespace.deprecatedEditingOffset(), 1, nonBreakingSpaceString());
609     }
610     if (m_trailingWhitespace.isNotNull() && !m_trailingWhitespace.isRenderedCharacter() && is<Text>(*m_trailingWhitespace.deprecatedNode())) {
611         Text& textNode = downcast<Text>(*m_trailingWhitespace.deprecatedNode());
612         ASSERT(!textNode.renderer() || textNode.renderer()->style().collapseWhiteSpace());
613         replaceTextInNodePreservingMarkers(&textNode, m_trailingWhitespace.deprecatedEditingOffset(), 1, nonBreakingSpaceString());
614     }
615 }
616
617 // If a selection starts in one block and ends in another, we have to merge to bring content before the
618 // start together with content after the end.
619 void DeleteSelectionCommand::mergeParagraphs()
620 {
621     if (!m_mergeBlocksAfterDelete) {
622         if (m_pruneStartBlockIfNecessary) {
623             // We aren't going to merge into the start block, so remove it if it's empty.
624             prune(m_startBlock);
625             // Removing the start block during a deletion is usually an indication that we need
626             // a placeholder, but not in this case.
627             m_needPlaceholder = false;
628         }
629         return;
630     }
631     
632     // It shouldn't have been asked to both try and merge content into the start block and prune it.
633     ASSERT(!m_pruneStartBlockIfNecessary);
634
635     // FIXME: Deletion should adjust selection endpoints as it removes nodes so that we never get into this state (4099839).
636     if (!m_downstreamEnd.anchorNode()->isConnected() || !m_upstreamStart.anchorNode()->isConnected())
637          return;
638
639     // FIXME: The deletion algorithm shouldn't let this happen.
640     if (comparePositions(m_upstreamStart, m_downstreamEnd) > 0)
641         return;
642         
643     // There's nothing to merge.
644     if (m_upstreamStart == m_downstreamEnd)
645         return;
646         
647     VisiblePosition startOfParagraphToMove(m_downstreamEnd);
648     VisiblePosition mergeDestination(m_upstreamStart);
649     
650     // m_downstreamEnd's block has been emptied out by deletion.  There is no content inside of it to
651     // move, so just remove it.
652     Element* endBlock = enclosingBlock(m_downstreamEnd.deprecatedNode());
653     if (!endBlock || !endBlock->contains(startOfParagraphToMove.deepEquivalent().deprecatedNode()) || !startOfParagraphToMove.deepEquivalent().deprecatedNode()) {
654         removeNode(enclosingBlock(m_downstreamEnd.deprecatedNode()));
655         return;
656     }
657     
658     // We need to merge into m_upstreamStart's block, but it's been emptied out and collapsed by deletion.
659     if (!mergeDestination.deepEquivalent().deprecatedNode() || !mergeDestination.deepEquivalent().deprecatedNode()->isDescendantOf(enclosingBlock(m_upstreamStart.containerNode())) || m_startsAtEmptyLine) {
660         insertNodeAt(HTMLBRElement::create(document()).ptr(), m_upstreamStart);
661         mergeDestination = VisiblePosition(m_upstreamStart);
662     }
663     
664     if (mergeDestination == startOfParagraphToMove)
665         return;
666         
667     VisiblePosition endOfParagraphToMove = endOfParagraph(startOfParagraphToMove, CanSkipOverEditingBoundary);
668     
669     if (mergeDestination == endOfParagraphToMove)
670         return;
671     
672     // The rule for merging into an empty block is: only do so if its farther to the right.
673     // FIXME: Consider RTL.
674     if (!m_startsAtEmptyLine && isStartOfParagraph(mergeDestination) && startOfParagraphToMove.absoluteCaretBounds().x() > mergeDestination.absoluteCaretBounds().x()) {
675         if (mergeDestination.deepEquivalent().downstream().deprecatedNode()->hasTagName(brTag)) {
676             removeNodeAndPruneAncestors(mergeDestination.deepEquivalent().downstream().deprecatedNode());
677             m_endingPosition = startOfParagraphToMove.deepEquivalent();
678             return;
679         }
680     }
681     
682     // Block images, tables and horizontal rules cannot be made inline with content at mergeDestination.  If there is 
683     // any (!isStartOfParagraph(mergeDestination)), don't merge, just move the caret to just before the selection we deleted.
684     // See https://bugs.webkit.org/show_bug.cgi?id=25439
685     if (isRenderedAsNonInlineTableImageOrHR(startOfParagraphToMove.deepEquivalent().deprecatedNode()) && !isStartOfParagraph(mergeDestination)) {
686         m_endingPosition = m_upstreamStart;
687         return;
688     }
689     
690     RefPtr<Range> range = Range::create(document(), startOfParagraphToMove.deepEquivalent().parentAnchoredEquivalent(), endOfParagraphToMove.deepEquivalent().parentAnchoredEquivalent());
691     RefPtr<Range> rangeToBeReplaced = Range::create(document(), mergeDestination.deepEquivalent().parentAnchoredEquivalent(), mergeDestination.deepEquivalent().parentAnchoredEquivalent());
692     if (!frame().editor().client()->shouldMoveRangeAfterDelete(range.get(), rangeToBeReplaced.get()))
693         return;
694     
695     // moveParagraphs will insert placeholders if it removes blocks that would require their use, don't let block
696     // removals that it does cause the insertion of *another* placeholder.
697     bool needPlaceholder = m_needPlaceholder;
698     bool paragraphToMergeIsEmpty = (startOfParagraphToMove == endOfParagraphToMove);
699     moveParagraph(startOfParagraphToMove, endOfParagraphToMove, mergeDestination, false, !paragraphToMergeIsEmpty);
700     m_needPlaceholder = needPlaceholder;
701     // The endingPosition was likely clobbered by the move, so recompute it (moveParagraph selects the moved paragraph).
702     m_endingPosition = endingSelection().start();
703 }
704
705 void DeleteSelectionCommand::removePreviouslySelectedEmptyTableRows()
706 {
707     if (m_endTableRow && m_endTableRow->isConnected() && m_endTableRow != m_startTableRow) {
708         Node* row = m_endTableRow->previousSibling();
709         while (row && row != m_startTableRow) {
710             RefPtr<Node> previousRow = row->previousSibling();
711             if (isTableRowEmpty(row))
712                 // Use a raw removeNode, instead of DeleteSelectionCommand's, because
713                 // that won't remove rows, it only empties them in preparation for this function.
714                 CompositeEditCommand::removeNode(row);
715             row = previousRow.get();
716         }
717     }
718     
719     // Remove empty rows after the start row.
720     if (m_startTableRow && m_startTableRow->isConnected() && m_startTableRow != m_endTableRow) {
721         Node* row = m_startTableRow->nextSibling();
722         while (row && row != m_endTableRow) {
723             RefPtr<Node> nextRow = row->nextSibling();
724             if (isTableRowEmpty(row))
725                 CompositeEditCommand::removeNode(row);
726             row = nextRow.get();
727         }
728     }
729
730     if (m_endTableRow && m_endTableRow->isConnected() && m_endTableRow != m_startTableRow) {
731         if (isTableRowEmpty(m_endTableRow.get())) {
732             // Don't remove m_endTableRow if it's where we're putting the ending selection.
733             if (!m_endingPosition.deprecatedNode()->isDescendantOf(*m_endTableRow)) {
734                 // FIXME: We probably shouldn't remove m_endTableRow unless it's fully selected, even if it is empty.
735                 // We'll need to start adjusting the selection endpoints during deletion to know whether or not m_endTableRow
736                 // was fully selected here.
737                 CompositeEditCommand::removeNode(m_endTableRow.get());
738             }
739         }
740     }
741 }
742
743 void DeleteSelectionCommand::calculateTypingStyleAfterDelete()
744 {
745     if (!m_typingStyle)
746         return;
747         
748     // Compute the difference between the style before the delete and the style now
749     // after the delete has been done. Set this style on the frame, so other editing
750     // commands being composed with this one will work, and also cache it on the command,
751     // so the Frame::appliedEditing can set it after the whole composite command 
752     // has completed.
753     
754     // If we deleted into a blockquote, but are now no longer in a blockquote, use the alternate typing style
755     if (m_deleteIntoBlockquoteStyle && !enclosingNodeOfType(m_endingPosition, isMailBlockquote, CanCrossEditingBoundary))
756         m_typingStyle = m_deleteIntoBlockquoteStyle;
757     m_deleteIntoBlockquoteStyle = nullptr;
758
759     m_typingStyle->prepareToApplyAt(m_endingPosition);
760     if (m_typingStyle->isEmpty())
761         m_typingStyle = nullptr;
762     // This is where we've deleted all traces of a style but not a whole paragraph (that's handled above).
763     // In this case if we start typing, the new characters should have the same style as the just deleted ones,
764     // but, if we change the selection, come back and start typing that style should be lost.  Also see 
765     // preserveTypingStyle() below.
766     frame().selection().setTypingStyle(m_typingStyle);
767 }
768
769 void DeleteSelectionCommand::clearTransientState()
770 {
771     m_selectionToDelete = VisibleSelection();
772     m_upstreamStart.clear();
773     m_downstreamStart.clear();
774     m_upstreamEnd.clear();
775     m_downstreamEnd.clear();
776     m_endingPosition.clear();
777     m_leadingWhitespace.clear();
778     m_trailingWhitespace.clear();
779 }
780     
781 String DeleteSelectionCommand::originalStringForAutocorrectionAtBeginningOfSelection()
782 {
783     if (!m_selectionToDelete.isRange())
784         return String();
785
786     VisiblePosition startOfSelection = m_selectionToDelete.start();
787     if (!isStartOfWord(startOfSelection))
788         return String();
789
790     VisiblePosition nextPosition = startOfSelection.next();
791     if (nextPosition.isNull())
792         return String();
793
794     RefPtr<Range> rangeOfFirstCharacter = Range::create(document(), startOfSelection.deepEquivalent(), nextPosition.deepEquivalent());
795     Vector<RenderedDocumentMarker*> markers = document().markers().markersInRange(rangeOfFirstCharacter.get(), DocumentMarker::Autocorrected);
796     for (auto* marker : markers) {
797         int startOffset = marker->startOffset();
798         if (startOffset == startOfSelection.deepEquivalent().offsetInContainerNode())
799             return marker->description();
800     }
801     return String();
802 }
803
804 // This method removes div elements with no attributes that have only one child or no children at all.
805 void DeleteSelectionCommand::removeRedundantBlocks()
806 {
807     Node* node = m_endingPosition.containerNode();
808     Node* rootNode = node->rootEditableElement();
809    
810     while (node != rootNode) {
811         if (isRemovableBlock(node)) {
812             if (node == m_endingPosition.anchorNode())
813                 updatePositionForNodeRemovalPreservingChildren(m_endingPosition, *node);
814             
815             CompositeEditCommand::removeNodePreservingChildren(node);
816             node = m_endingPosition.anchorNode();
817         } else
818             node = node->parentNode();
819     }
820 }
821
822 void DeleteSelectionCommand::doApply()
823 {
824     // If selection has not been set to a custom selection when the command was created,
825     // use the current ending selection.
826     if (!m_hasSelectionToDelete)
827         m_selectionToDelete = endingSelection();
828
829     if (!m_selectionToDelete.isNonOrphanedRange())
830         return;
831
832     String originalString = originalStringForAutocorrectionAtBeginningOfSelection();
833
834     // If the deletion is occurring in a text field, and we're not deleting to replace the selection, then let the frame call across the bridge to notify the form delegate. 
835     if (!m_replace) {
836         Element* textControl = enclosingTextFormControl(m_selectionToDelete.start());
837         if (textControl && textControl->focused())
838             frame().editor().textWillBeDeletedInTextField(textControl);
839     }
840
841     // save this to later make the selection with
842     EAffinity affinity = m_selectionToDelete.affinity();
843     
844     Position downstreamEnd = m_selectionToDelete.end().downstream();
845     m_needPlaceholder = isStartOfParagraph(m_selectionToDelete.visibleStart(), CanCrossEditingBoundary)
846             && isEndOfParagraph(m_selectionToDelete.visibleEnd(), CanCrossEditingBoundary)
847             && !lineBreakExistsAtVisiblePosition(m_selectionToDelete.visibleEnd());
848     if (m_needPlaceholder) {
849         // Don't need a placeholder when deleting a selection that starts just before a table
850         // and ends inside it (we do need placeholders to hold open empty cells, but that's
851         // handled elsewhere).
852         if (auto* table = isLastPositionBeforeTable(m_selectionToDelete.visibleStart())) {
853             if (m_selectionToDelete.end().deprecatedNode()->isDescendantOf(*table))
854                 m_needPlaceholder = false;
855         }
856     }
857         
858     
859     // set up our state
860     initializePositionData();
861
862     // Delete any text that may hinder our ability to fixup whitespace after the delete
863     deleteInsignificantTextDownstream(m_trailingWhitespace);    
864
865     saveTypingStyleState();
866     
867     // deleting just a BR is handled specially, at least because we do not
868     // want to replace it with a placeholder BR!
869     if (handleSpecialCaseBRDelete()) {
870         calculateTypingStyleAfterDelete();
871         setEndingSelection(VisibleSelection(m_endingPosition, affinity, endingSelection().isDirectional()));
872         clearTransientState();
873         rebalanceWhitespace();
874         return;
875     }
876     
877     handleGeneralDelete();
878     
879     fixupWhitespace();
880     
881     mergeParagraphs();
882     
883     removePreviouslySelectedEmptyTableRows();
884     
885     if (m_needPlaceholder) {
886         if (m_sanitizeMarkup)
887             removeRedundantBlocks();
888         insertNodeAt(HTMLBRElement::create(document()), m_endingPosition);
889     }
890
891     bool shouldRebalaceWhiteSpace = true;
892     if (!frame().editor().behavior().shouldRebalanceWhiteSpacesInSecureField()) {
893         Node* node = m_endingPosition.deprecatedNode();
894         if (is<Text>(node)) {
895             Text& textNode = downcast<Text>(*node);
896             if (textNode.length() && textNode.renderer())
897                 shouldRebalaceWhiteSpace = textNode.renderer()->style().textSecurity() == TSNONE;
898         }        
899     }
900     if (shouldRebalaceWhiteSpace)
901         rebalanceWhitespaceAt(m_endingPosition);
902
903     calculateTypingStyleAfterDelete();
904
905     if (!originalString.isEmpty())
906         frame().editor().deletedAutocorrectionAtPosition(m_endingPosition, originalString);
907
908     setEndingSelection(VisibleSelection(m_endingPosition, affinity, endingSelection().isDirectional()));
909     clearTransientState();
910 }
911
912 // Normally deletion doesn't preserve the typing style that was present before it.  For example,
913 // type a character, Bold, then delete the character and start typing.  The Bold typing style shouldn't
914 // stick around.  Deletion should preserve a typing style that *it* sets, however.
915 bool DeleteSelectionCommand::preservesTypingStyle() const
916 {
917     return m_typingStyle;
918 }
919
920 } // namespace WebCore