056ab705f32c913bc2b4a983d657b770e7d81f9f
[WebKit-https.git] / WebCore / editing / ReplaceSelectionCommand.cpp
1 /*
2  * Copyright (C) 2005, 2006, 2008 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 COMPUTER, 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 COMPUTER, 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 "ReplaceSelectionCommand.h"
28
29 #include "ApplyStyleCommand.h"
30 #include "BeforeTextInsertedEvent.h"
31 #include "BreakBlockquoteCommand.h"
32 #include "CSSComputedStyleDeclaration.h"
33 #include "CSSMutableStyleDeclaration.h"
34 #include "CSSProperty.h"
35 #include "CSSPropertyNames.h"
36 #include "CSSValueKeywords.h"
37 #include "Document.h"
38 #include "DocumentFragment.h"
39 #include "EditingText.h"
40 #include "Element.h"
41 #include "EventNames.h"
42 #include "Frame.h"
43 #include "HTMLElement.h"
44 #include "HTMLInputElement.h"
45 #include "HTMLInterchange.h"
46 #include "HTMLNames.h"
47 #include "SelectionController.h"
48 #include "SmartReplace.h"
49 #include "TextIterator.h"
50 #include "htmlediting.h"
51 #include "markup.h"
52 #include "visible_units.h"
53 #include <wtf/StdLibExtras.h>
54
55 namespace WebCore {
56
57 using namespace HTMLNames;
58
59 enum EFragmentType { EmptyFragment, SingleTextNodeFragment, TreeFragment };
60
61 // --- ReplacementFragment helper class
62
63 class ReplacementFragment : public Noncopyable {
64 public:
65     ReplacementFragment(Document*, DocumentFragment*, bool matchStyle, const VisibleSelection&);
66
67     Node* firstChild() const;
68     Node* lastChild() const;
69
70     bool isEmpty() const;
71     
72     bool hasInterchangeNewlineAtStart() const { return m_hasInterchangeNewlineAtStart; }
73     bool hasInterchangeNewlineAtEnd() const { return m_hasInterchangeNewlineAtEnd; }
74     
75     void removeNode(PassRefPtr<Node>);
76     void removeNodePreservingChildren(Node*);
77
78 private:
79     PassRefPtr<StyledElement> insertFragmentForTestRendering(Node* context);
80     void removeUnrenderedNodes(Node*);
81     void restoreTestRenderingNodesToFragment(StyledElement*);
82     void removeInterchangeNodes(Node*);
83     
84     void insertNodeBefore(PassRefPtr<Node> node, Node* refNode);
85
86     RefPtr<Document> m_document;
87     RefPtr<DocumentFragment> m_fragment;
88     bool m_matchStyle;
89     bool m_hasInterchangeNewlineAtStart;
90     bool m_hasInterchangeNewlineAtEnd;
91 };
92
93 static bool isInterchangeNewlineNode(const Node *node)
94 {
95     DEFINE_STATIC_LOCAL(String, interchangeNewlineClassString, (AppleInterchangeNewline));
96     return node && node->hasTagName(brTag) && 
97            static_cast<const Element *>(node)->getAttribute(classAttr) == interchangeNewlineClassString;
98 }
99
100 static bool isInterchangeConvertedSpaceSpan(const Node *node)
101 {
102     DEFINE_STATIC_LOCAL(String, convertedSpaceSpanClassString, (AppleConvertedSpace));
103     return node->isHTMLElement() && 
104            static_cast<const HTMLElement *>(node)->getAttribute(classAttr) == convertedSpaceSpanClassString;
105 }
106
107 static Position positionAvoidingPrecedingNodes(Position pos)
108 {
109     // If we're already on a break, it's probably a placeholder and we shouldn't change our position.
110     if (pos.node()->hasTagName(brTag))
111         return pos;
112
113     // We also stop when changing block flow elements because even though the visual position is the
114     // same.  E.g.,
115     //   <div>foo^</div>^
116     // The two positions above are the same visual position, but we want to stay in the same block.
117     Node* stopNode = pos.node()->enclosingBlockFlowElement();
118     while (stopNode != pos.node() && VisiblePosition(pos) == VisiblePosition(pos.next()))
119         pos = pos.next();
120     return pos;
121 }
122
123 ReplacementFragment::ReplacementFragment(Document* document, DocumentFragment* fragment, bool matchStyle, const VisibleSelection& selection)
124     : m_document(document),
125       m_fragment(fragment),
126       m_matchStyle(matchStyle), 
127       m_hasInterchangeNewlineAtStart(false), 
128       m_hasInterchangeNewlineAtEnd(false)
129 {
130     if (!m_document)
131         return;
132     if (!m_fragment)
133         return;
134     if (!m_fragment->firstChild())
135         return;
136     
137     Element* editableRoot = selection.rootEditableElement();
138     ASSERT(editableRoot);
139     if (!editableRoot)
140         return;
141     
142     Node* shadowAncestorNode = editableRoot->shadowAncestorNode();
143     
144     if (!editableRoot->getAttributeEventListener(eventNames().webkitBeforeTextInsertedEvent) &&
145         // FIXME: Remove these checks once textareas and textfields actually register an event handler.
146         !(shadowAncestorNode && shadowAncestorNode->renderer() && shadowAncestorNode->renderer()->isTextControl()) &&
147         editableRoot->isContentRichlyEditable()) {
148         removeInterchangeNodes(m_fragment.get());
149         return;
150     }
151
152     Node* styleNode = selection.base().node();
153     RefPtr<StyledElement> holder = insertFragmentForTestRendering(styleNode);
154     
155     RefPtr<Range> range = VisibleSelection::selectionFromContentsOfNode(holder.get()).toNormalizedRange();
156     String text = plainText(range.get());
157     // Give the root a chance to change the text.
158     RefPtr<BeforeTextInsertedEvent> evt = BeforeTextInsertedEvent::create(text);
159     ExceptionCode ec = 0;
160     editableRoot->dispatchEvent(evt, ec);
161     ASSERT(ec == 0);
162     if (text != evt->text() || !editableRoot->isContentRichlyEditable()) {
163         restoreTestRenderingNodesToFragment(holder.get());
164         removeNode(holder);
165
166         m_fragment = createFragmentFromText(selection.toNormalizedRange().get(), evt->text());
167         if (!m_fragment->firstChild())
168             return;
169         holder = insertFragmentForTestRendering(styleNode);
170     }
171     
172     removeInterchangeNodes(holder.get());
173     
174     removeUnrenderedNodes(holder.get());
175     restoreTestRenderingNodesToFragment(holder.get());
176     removeNode(holder);
177 }
178
179 bool ReplacementFragment::isEmpty() const
180 {
181     return (!m_fragment || !m_fragment->firstChild()) && !m_hasInterchangeNewlineAtStart && !m_hasInterchangeNewlineAtEnd;
182 }
183
184 Node *ReplacementFragment::firstChild() const 
185
186     return m_fragment ? m_fragment->firstChild() : 0; 
187 }
188
189 Node *ReplacementFragment::lastChild() const 
190
191     return m_fragment ? m_fragment->lastChild() : 0; 
192 }
193
194 void ReplacementFragment::removeNodePreservingChildren(Node *node)
195 {
196     if (!node)
197         return;
198
199     while (RefPtr<Node> n = node->firstChild()) {
200         removeNode(n);
201         insertNodeBefore(n.release(), node);
202     }
203     removeNode(node);
204 }
205
206 void ReplacementFragment::removeNode(PassRefPtr<Node> node)
207 {
208     if (!node)
209         return;
210     
211     ContainerNode* parent = node->parentNode();
212     if (!parent)
213         return;
214     
215     ExceptionCode ec = 0;
216     parent->removeChild(node.get(), ec);
217     ASSERT(ec == 0);
218 }
219
220 void ReplacementFragment::insertNodeBefore(PassRefPtr<Node> node, Node* refNode)
221 {
222     if (!node || !refNode)
223         return;
224         
225     ContainerNode* parent = refNode->parentNode();
226     if (!parent)
227         return;
228         
229     ExceptionCode ec = 0;
230     parent->insertBefore(node, refNode, ec);
231     ASSERT(ec == 0);
232 }
233
234 PassRefPtr<StyledElement> ReplacementFragment::insertFragmentForTestRendering(Node* context)
235 {
236     HTMLElement* body = m_document->body();
237     if (!body)
238         return 0;
239
240     RefPtr<StyledElement> holder = createDefaultParagraphElement(m_document.get());
241     
242     ExceptionCode ec = 0;
243
244     // Copy the whitespace and user-select style from the context onto this element.
245     // FIXME: We should examine other style properties to see if they would be appropriate to consider during the test rendering.
246     Node* n = context;
247     while (n && !n->isElementNode())
248         n = n->parentNode();
249     if (n) {
250         RefPtr<CSSComputedStyleDeclaration> conFontStyle = computedStyle(n);
251         CSSStyleDeclaration* style = holder->style();
252         style->setProperty(CSSPropertyWhiteSpace, conFontStyle->getPropertyValue(CSSPropertyWhiteSpace), false, ec);
253         ASSERT(ec == 0);
254         style->setProperty(CSSPropertyWebkitUserSelect, conFontStyle->getPropertyValue(CSSPropertyWebkitUserSelect), false, ec);
255         ASSERT(ec == 0);
256     }
257     
258     holder->appendChild(m_fragment, ec);
259     ASSERT(ec == 0);
260     
261     body->appendChild(holder.get(), ec);
262     ASSERT(ec == 0);
263     
264     m_document->updateLayoutIgnorePendingStylesheets();
265     
266     return holder.release();
267 }
268
269 void ReplacementFragment::restoreTestRenderingNodesToFragment(StyledElement* holder)
270 {
271     if (!holder)
272         return;
273     
274     ExceptionCode ec = 0;
275     while (RefPtr<Node> node = holder->firstChild()) {
276         holder->removeChild(node.get(), ec);
277         ASSERT(ec == 0);
278         m_fragment->appendChild(node.get(), ec);
279         ASSERT(ec == 0);
280     }
281 }
282
283 void ReplacementFragment::removeUnrenderedNodes(Node* holder)
284 {
285     Vector<Node*> unrendered;
286
287     for (Node* node = holder->firstChild(); node; node = node->traverseNextNode(holder))
288         if (!isNodeRendered(node) && !isTableStructureNode(node))
289             unrendered.append(node);
290
291     size_t n = unrendered.size();
292     for (size_t i = 0; i < n; ++i)
293         removeNode(unrendered[i]);
294 }
295
296 void ReplacementFragment::removeInterchangeNodes(Node* container)
297 {
298     // Interchange newlines at the "start" of the incoming fragment must be
299     // either the first node in the fragment or the first leaf in the fragment.
300     Node* node = container->firstChild();
301     while (node) {
302         if (isInterchangeNewlineNode(node)) {
303             m_hasInterchangeNewlineAtStart = true;
304             removeNode(node);
305             break;
306         }
307         node = node->firstChild();
308     }
309     if (!container->hasChildNodes())
310         return;
311     // Interchange newlines at the "end" of the incoming fragment must be
312     // either the last node in the fragment or the last leaf in the fragment.
313     node = container->lastChild();
314     while (node) {
315         if (isInterchangeNewlineNode(node)) {
316             m_hasInterchangeNewlineAtEnd = true;
317             removeNode(node);
318             break;
319         }
320         node = node->lastChild();
321     }
322     
323     node = container->firstChild();
324     while (node) {
325         Node *next = node->traverseNextNode();
326         if (isInterchangeConvertedSpaceSpan(node)) {
327             RefPtr<Node> n = 0;
328             while ((n = node->firstChild())) {
329                 removeNode(n);
330                 insertNodeBefore(n, node);
331             }
332             removeNode(node);
333             if (n)
334                 next = n->traverseNextNode();
335         }
336         node = next;
337     }
338 }
339
340 ReplaceSelectionCommand::ReplaceSelectionCommand(Document* document, PassRefPtr<DocumentFragment> fragment,
341         bool selectReplacement, bool smartReplace, bool matchStyle, bool preventNesting, bool movingParagraph,
342         EditAction editAction) 
343     : CompositeEditCommand(document),
344       m_selectReplacement(selectReplacement), 
345       m_smartReplace(smartReplace),
346       m_matchStyle(matchStyle),
347       m_documentFragment(fragment),
348       m_preventNesting(preventNesting),
349       m_movingParagraph(movingParagraph),
350       m_editAction(editAction),
351       m_shouldMergeEnd(false)
352 {
353 }
354
355 static bool hasMatchingQuoteLevel(VisiblePosition endOfExistingContent, VisiblePosition endOfInsertedContent)
356 {
357     Position existing = endOfExistingContent.deepEquivalent();
358     Position inserted = endOfInsertedContent.deepEquivalent();
359     bool isInsideMailBlockquote = nearestMailBlockquote(inserted.node());
360     return isInsideMailBlockquote && (numEnclosingMailBlockquotes(existing) == numEnclosingMailBlockquotes(inserted));
361 }
362
363 bool ReplaceSelectionCommand::shouldMergeStart(bool selectionStartWasStartOfParagraph, bool fragmentHasInterchangeNewlineAtStart, bool selectionStartWasInsideMailBlockquote)
364 {
365     if (m_movingParagraph)
366         return false;
367     
368     VisiblePosition startOfInsertedContent(positionAtStartOfInsertedContent());
369     VisiblePosition prev = startOfInsertedContent.previous(true);
370     if (prev.isNull())
371         return false;
372     
373     // When we have matching quote levels, its ok to merge more frequently.
374     // For a successful merge, we still need to make sure that the inserted content starts with the beginning of a paragraph.
375     // And we should only merge here if the selection start was inside a mail blockquote.  This prevents against removing a 
376     // blockquote from newly pasted quoted content that was pasted into an unquoted position.  If that unquoted position happens 
377     // to be right after another blockquote, we don't want to merge and risk stripping a valid block (and newline) from the pasted content.
378     if (isStartOfParagraph(startOfInsertedContent) && selectionStartWasInsideMailBlockquote && hasMatchingQuoteLevel(prev, positionAtEndOfInsertedContent()))
379         return true;
380
381     return !selectionStartWasStartOfParagraph && 
382            !fragmentHasInterchangeNewlineAtStart &&
383            isStartOfParagraph(startOfInsertedContent) && 
384            !startOfInsertedContent.deepEquivalent().node()->hasTagName(brTag) &&
385            shouldMerge(startOfInsertedContent, prev);
386 }
387
388 bool ReplaceSelectionCommand::shouldMergeEnd(bool selectionEndWasEndOfParagraph)
389 {
390     VisiblePosition endOfInsertedContent(positionAtEndOfInsertedContent());
391     VisiblePosition next = endOfInsertedContent.next(true);
392     if (next.isNull())
393         return false;
394
395     return !selectionEndWasEndOfParagraph &&
396            isEndOfParagraph(endOfInsertedContent) && 
397            !endOfInsertedContent.deepEquivalent().node()->hasTagName(brTag) &&
398            shouldMerge(endOfInsertedContent, next);
399 }
400
401 static bool isMailPasteAsQuotationNode(const Node* node)
402 {
403     return node && node->hasTagName(blockquoteTag) && node->isElementNode() && static_cast<const Element*>(node)->getAttribute(classAttr) == ApplePasteAsQuotation;
404 }
405
406 // Wrap CompositeEditCommand::removeNodePreservingChildren() so we can update the nodes we track
407 void ReplaceSelectionCommand::removeNodePreservingChildren(Node* node)
408 {
409     if (m_firstNodeInserted == node)
410         m_firstNodeInserted = node->traverseNextNode();
411     if (m_lastLeafInserted == node)
412         m_lastLeafInserted = node->lastChild() ? node->lastChild() : node->traverseNextSibling();
413     CompositeEditCommand::removeNodePreservingChildren(node);
414 }
415
416 // Wrap CompositeEditCommand::removeNodeAndPruneAncestors() so we can update the nodes we track
417 void ReplaceSelectionCommand::removeNodeAndPruneAncestors(Node* node)
418 {
419     // prepare in case m_firstNodeInserted and/or m_lastLeafInserted get removed
420     // FIXME: shouldn't m_lastLeafInserted be adjusted using traversePreviousNode()?
421     Node* afterFirst = m_firstNodeInserted ? m_firstNodeInserted->traverseNextSibling() : 0;
422     Node* afterLast = m_lastLeafInserted ? m_lastLeafInserted->traverseNextSibling() : 0;
423     
424     CompositeEditCommand::removeNodeAndPruneAncestors(node);
425     
426     // adjust m_firstNodeInserted and m_lastLeafInserted since either or both may have been removed
427     if (m_lastLeafInserted && !m_lastLeafInserted->inDocument())
428         m_lastLeafInserted = afterLast;
429     if (m_firstNodeInserted && !m_firstNodeInserted->inDocument())
430         m_firstNodeInserted = m_lastLeafInserted && m_lastLeafInserted->inDocument() ? afterFirst : 0;
431 }
432
433 static bool isHeaderElement(Node* a)
434 {
435     if (!a)
436         return false;
437         
438     return a->hasTagName(h1Tag) ||
439            a->hasTagName(h2Tag) ||
440            a->hasTagName(h3Tag) ||
441            a->hasTagName(h4Tag) ||
442            a->hasTagName(h5Tag);
443 }
444
445 static bool haveSameTagName(Node* a, Node* b)
446 {
447     return a && b && a->isElementNode() && b->isElementNode() && static_cast<Element*>(a)->tagName() == static_cast<Element*>(b)->tagName();
448 }
449
450 bool ReplaceSelectionCommand::shouldMerge(const VisiblePosition& source, const VisiblePosition& destination)
451 {
452     if (source.isNull() || destination.isNull())
453         return false;
454         
455     Node* sourceNode = source.deepEquivalent().node();
456     Node* destinationNode = destination.deepEquivalent().node();
457     Node* sourceBlock = enclosingBlock(sourceNode);
458     Node* destinationBlock = enclosingBlock(destinationNode);
459     return !enclosingNodeOfType(source.deepEquivalent(), &isMailPasteAsQuotationNode) &&
460            sourceBlock && (!sourceBlock->hasTagName(blockquoteTag) || isMailBlockquote(sourceBlock))  &&
461            enclosingListChild(sourceBlock) == enclosingListChild(destinationNode) &&
462            enclosingTableCell(source.deepEquivalent()) == enclosingTableCell(destination.deepEquivalent()) &&
463            (!isHeaderElement(sourceBlock) || haveSameTagName(sourceBlock, destinationBlock)) &&
464            // Don't merge to or from a position before or after a block because it would
465            // be a no-op and cause infinite recursion.
466            !isBlock(sourceNode) && !isBlock(destinationNode);
467 }
468
469 // Style rules that match just inserted elements could change their appearance, like
470 // a div inserted into a document with div { display:inline; }.
471 void ReplaceSelectionCommand::negateStyleRulesThatAffectAppearance()
472 {
473     for (RefPtr<Node> node = m_firstNodeInserted.get(); node; node = node->traverseNextNode()) {
474         // FIXME: <rdar://problem/5371536> Style rules that match pasted content can change it's appearance
475         if (isStyleSpan(node.get())) {
476             HTMLElement* e = static_cast<HTMLElement*>(node.get());
477             // There are other styles that style rules can give to style spans,
478             // but these are the two important ones because they'll prevent
479             // inserted content from appearing in the right paragraph.
480             // FIXME: Hyatt is concerned that selectively using display:inline will give inconsistent
481             // results. We already know one issue because td elements ignore their display property
482             // in quirks mode (which Mail.app is always in). We should look for an alternative.
483             if (isBlock(e))
484                 e->getInlineStyleDecl()->setProperty(CSSPropertyDisplay, CSSValueInline);
485             if (e->renderer() && e->renderer()->style()->floating() != FNONE)
486                 e->getInlineStyleDecl()->setProperty(CSSPropertyFloat, CSSValueNone);
487
488             // Undo the effects of page zoom if we have an absolute font size.  When we copy, we
489             // compute the new font size as an absolute size so pasting will cause the zoom to be
490             // applied twice.
491             if (e->renderer() && e->renderer()->style() && e->renderer()->style()->effectiveZoom() != 1.0
492                 && e->renderer()->style()->fontDescription().isAbsoluteSize()) {
493                 float newSize = e->renderer()->style()->fontDescription().specifiedSize() / e->renderer()->style()->effectiveZoom();
494                 ExceptionCode ec = 0;
495                 e->style()->setProperty(CSSPropertyFontSize, String::number(newSize), false, ec);
496                 ASSERT(!ec);
497             }
498         }
499         if (node == m_lastLeafInserted)
500             break;
501     }
502 }
503
504 void ReplaceSelectionCommand::removeUnrenderedTextNodesAtEnds()
505 {
506     document()->updateLayoutIgnorePendingStylesheets();
507     if (!m_lastLeafInserted->renderer() && 
508         m_lastLeafInserted->isTextNode() && 
509         !enclosingNodeWithTag(Position(m_lastLeafInserted.get(), 0), selectTag) && 
510         !enclosingNodeWithTag(Position(m_lastLeafInserted.get(), 0), scriptTag)) {
511         if (m_firstNodeInserted == m_lastLeafInserted) {
512             removeNode(m_lastLeafInserted.get());
513             m_lastLeafInserted = 0;
514             m_firstNodeInserted = 0;
515             return;
516         }
517         RefPtr<Node> previous = m_lastLeafInserted->traversePreviousNode();
518         removeNode(m_lastLeafInserted.get());
519         m_lastLeafInserted = previous;
520     }
521     
522     // We don't have to make sure that m_firstNodeInserted isn't inside a select or script element, because
523     // it is a top level node in the fragment and the user can't insert into those elements.
524     if (!m_firstNodeInserted->renderer() && 
525         m_firstNodeInserted->isTextNode()) {
526         if (m_firstNodeInserted == m_lastLeafInserted) {
527             removeNode(m_firstNodeInserted.get());
528             m_firstNodeInserted = 0;
529             m_lastLeafInserted = 0;
530             return;
531         }
532         RefPtr<Node> next = m_firstNodeInserted->traverseNextSibling();
533         removeNode(m_firstNodeInserted.get());
534         m_firstNodeInserted = next;
535     }
536 }
537
538 void ReplaceSelectionCommand::handlePasteAsQuotationNode()
539 {
540     Node* node = m_firstNodeInserted.get();
541     if (isMailPasteAsQuotationNode(node))
542         removeNodeAttribute(static_cast<Element*>(node), classAttr);
543 }
544
545 VisiblePosition ReplaceSelectionCommand::positionAtEndOfInsertedContent()
546 {
547     Node* lastNode = m_lastLeafInserted.get();
548     // FIXME: Why is this hack here?  What's special about <select> tags?
549     Node* enclosingSelect = enclosingNodeWithTag(firstDeepEditingPositionForNode(lastNode), selectTag);
550     if (enclosingSelect)
551         lastNode = enclosingSelect;
552     return lastDeepEditingPositionForNode(lastNode);
553 }
554
555 VisiblePosition ReplaceSelectionCommand::positionAtStartOfInsertedContent()
556 {
557     // Return the inserted content's first VisiblePosition.
558     return VisiblePosition(nextCandidate(positionInParentBeforeNode(m_firstNodeInserted.get())));
559 }
560
561 // Remove style spans before insertion if they are unnecessary.  It's faster because we'll 
562 // avoid doing a layout.
563 static bool handleStyleSpansBeforeInsertion(ReplacementFragment& fragment, const Position& insertionPos)
564 {
565     Node* topNode = fragment.firstChild();
566     
567     // Handling the case where we are doing Paste as Quotation or pasting into quoted content is more complicated (see handleStyleSpans)
568     // and doesn't receive the optimization.
569     if (isMailPasteAsQuotationNode(topNode) || nearestMailBlockquote(topNode))
570         return false;
571     
572     // Either there are no style spans in the fragment or a WebKit client has added content to the fragment
573     // before inserting it.  Look for and handle style spans after insertion.
574     if (!isStyleSpan(topNode))
575         return false;
576     
577     Node* sourceDocumentStyleSpan = topNode;
578     RefPtr<Node> copiedRangeStyleSpan = sourceDocumentStyleSpan->firstChild();
579
580     RefPtr<CSSMutableStyleDeclaration> styleAtInsertionPos = ApplyStyleCommand::editingStyleAtPosition(rangeCompliantEquivalent(insertionPos));
581
582     String styleText = styleAtInsertionPos->cssText();
583     
584     if (styleText == static_cast<Element*>(sourceDocumentStyleSpan)->getAttribute(styleAttr)) {
585         fragment.removeNodePreservingChildren(sourceDocumentStyleSpan);
586         if (!isStyleSpan(copiedRangeStyleSpan.get()))
587             return true;
588     }
589         
590     if (isStyleSpan(copiedRangeStyleSpan.get()) && styleText == static_cast<Element*>(copiedRangeStyleSpan.get())->getAttribute(styleAttr)) {
591         fragment.removeNodePreservingChildren(copiedRangeStyleSpan.get());
592         return true;
593     }
594     
595     return false;
596 }
597
598 // At copy time, WebKit wraps copied content in a span that contains the source document's 
599 // default styles.  If the copied Range inherits any other styles from its ancestors, we put 
600 // those styles on a second span.
601 // This function removes redundant styles from those spans, and removes the spans if all their 
602 // styles are redundant. 
603 // We should remove the Apple-style-span class when we're done, see <rdar://problem/5685600>.
604 // We should remove styles from spans that are overridden by all of their children, either here
605 // or at copy time.
606 void ReplaceSelectionCommand::handleStyleSpans()
607 {
608     Node* sourceDocumentStyleSpan = 0;
609     Node* copiedRangeStyleSpan = 0;
610     // The style span that contains the source document's default style should be at
611     // the top of the fragment, but Mail sometimes adds a wrapper (for Paste As Quotation),
612     // so search for the top level style span instead of assuming it's at the top.
613     for (Node* node = m_firstNodeInserted.get(); node; node = node->traverseNextNode()) {
614         if (isStyleSpan(node)) {
615             sourceDocumentStyleSpan = node;
616             // If the copied Range's common ancestor had user applied inheritable styles
617             // on it, they'll be on a second style span, just below the one that holds the 
618             // document defaults.
619             if (isStyleSpan(node->firstChild()))
620                 copiedRangeStyleSpan = node->firstChild();
621             break;
622         }
623     }
624     
625     // There might not be any style spans if we're pasting from another application or if 
626     // we are here because of a document.execCommand("InsertHTML", ...) call.
627     if (!sourceDocumentStyleSpan)
628         return;
629         
630     RefPtr<CSSMutableStyleDeclaration> sourceDocumentStyle = static_cast<HTMLElement*>(sourceDocumentStyleSpan)->getInlineStyleDecl()->copy();
631     ContainerNode* context = sourceDocumentStyleSpan->parentNode();
632     
633     // If Mail wraps the fragment with a Paste as Quotation blockquote, or if you're pasting into a quoted region,
634     // styles from blockquoteNode are allowed to override those from the source document, see <rdar://problem/4930986> and <rdar://problem/5089327>.
635     Node* blockquoteNode = isMailPasteAsQuotationNode(context) ? context : nearestMailBlockquote(context);
636     if (blockquoteNode) {
637         RefPtr<CSSMutableStyleDeclaration> blockquoteStyle = ApplyStyleCommand::editingStyleAtPosition(Position(blockquoteNode, 0));
638         RefPtr<CSSMutableStyleDeclaration> parentStyle = ApplyStyleCommand::editingStyleAtPosition(Position(blockquoteNode->parentNode(), 0));
639         parentStyle->diff(blockquoteStyle.get());
640
641         CSSMutableStyleDeclaration::const_iterator end = blockquoteStyle->end();
642         for (CSSMutableStyleDeclaration::const_iterator it = blockquoteStyle->begin(); it != end; ++it) {
643             const CSSProperty& property = *it;
644             sourceDocumentStyle->removeProperty(property.id());
645         }        
646
647         context = blockquoteNode->parentNode();
648     }
649
650     // This operation requires that only editing styles to be removed from sourceDocumentStyle.
651     prepareEditingStyleToApplyAt(sourceDocumentStyle.get(), Position(context, 0));
652
653     // Remove block properties in the span's style. This prevents properties that probably have no effect 
654     // currently from affecting blocks later if the style is cloned for a new block element during a future 
655     // editing operation.
656     // FIXME: They *can* have an effect currently if blocks beneath the style span aren't individually marked
657     // with block styles by the editing engine used to style them.  WebKit doesn't do this, but others might.
658     sourceDocumentStyle->removeBlockProperties();
659     
660     // The styles on sourceDocumentStyleSpan are all redundant, and there is no copiedRangeStyleSpan
661     // to consider.  We're finished.
662     if (sourceDocumentStyle->length() == 0 && !copiedRangeStyleSpan) {
663         removeNodePreservingChildren(sourceDocumentStyleSpan);
664         return;
665     }
666     
667     // There are non-redundant styles on sourceDocumentStyleSpan, but there is no
668     // copiedRangeStyleSpan.  Remove the span, because it could be surrounding block elements,
669     // and apply the styles to its children.
670     if (sourceDocumentStyle->length() > 0 && !copiedRangeStyleSpan) {
671         copyStyleToChildren(sourceDocumentStyleSpan, sourceDocumentStyle.get()); 
672         removeNodePreservingChildren(sourceDocumentStyleSpan);
673         return;
674     }
675     
676     RefPtr<CSSMutableStyleDeclaration> copiedRangeStyle = static_cast<HTMLElement*>(copiedRangeStyleSpan)->getInlineStyleDecl()->copy();
677     
678     // We're going to put sourceDocumentStyleSpan's non-redundant styles onto copiedRangeStyleSpan,
679     // as long as they aren't overridden by ones on copiedRangeStyleSpan.
680     sourceDocumentStyle->merge(copiedRangeStyle.get(), true);
681     copiedRangeStyle = sourceDocumentStyle;
682     
683     removeNodePreservingChildren(sourceDocumentStyleSpan);
684     
685     // Remove redundant styles.
686     context = copiedRangeStyleSpan->parentNode();
687     prepareEditingStyleToApplyAt(copiedRangeStyle.get(), Position(context, 0));
688
689     // See the comments above about removing block properties.
690     copiedRangeStyle->removeBlockProperties();
691
692     // All the styles on copiedRangeStyleSpan are redundant, remove it.
693     if (copiedRangeStyle->length() == 0) {
694         removeNodePreservingChildren(copiedRangeStyleSpan);
695         return;
696     }
697     
698     // Clear the redundant styles from the span's style attribute.
699     // FIXME: If font-family:-webkit-monospace is non-redundant, then the font-size should stay, even if it
700     // appears redundant.
701     setNodeAttribute(static_cast<Element*>(copiedRangeStyleSpan), styleAttr, copiedRangeStyle->cssText());
702 }
703
704 // Take the style attribute of a span and apply it to it's children instead.  This allows us to
705 // convert invalid HTML where a span contains block elements into valid HTML while preserving
706 // styles.
707 void ReplaceSelectionCommand::copyStyleToChildren(Node* parentNode, const CSSMutableStyleDeclaration* parentStyle)
708 {
709     ASSERT(parentNode->hasTagName(spanTag));
710     for (Node* childNode = parentNode->firstChild(); childNode; childNode = childNode->nextSibling()) {
711         if (childNode->isTextNode() || !isBlock(childNode) || childNode->hasTagName(preTag)) {
712             // In this case, put a span tag around the child node.
713             RefPtr<Node> newNode = parentNode->cloneNode(false);
714             ASSERT(newNode->hasTagName(spanTag));
715             HTMLElement* newSpan = static_cast<HTMLElement*>(newNode.get());
716             setNodeAttribute(newSpan, styleAttr, parentStyle->cssText());
717             insertNodeAfter(newSpan, childNode);
718             ExceptionCode ec = 0;
719             newSpan->appendChild(childNode, ec);
720             ASSERT(!ec);
721             childNode = newSpan;
722         } else if (childNode->isHTMLElement()) {
723             // Copy the style attribute and merge them into the child node.  We don't want to override
724             // existing styles, so don't clobber on merge.
725             RefPtr<CSSMutableStyleDeclaration> newStyle = parentStyle->copy();
726             HTMLElement* childElement = static_cast<HTMLElement*>(childNode);
727             RefPtr<CSSMutableStyleDeclaration> existingStyles = childElement->getInlineStyleDecl()->copy();
728             existingStyles->merge(newStyle.get(), false);
729             setNodeAttribute(childElement, styleAttr, existingStyles->cssText());
730         }
731     }
732 }
733
734 void ReplaceSelectionCommand::mergeEndIfNeeded()
735 {
736     if (!m_shouldMergeEnd)
737         return;
738
739     VisiblePosition startOfInsertedContent(positionAtStartOfInsertedContent());
740     VisiblePosition endOfInsertedContent(positionAtEndOfInsertedContent());
741     
742     // Bail to avoid infinite recursion.
743     if (m_movingParagraph) {
744         ASSERT_NOT_REACHED();
745         return;
746     }
747     
748     // Merging two paragraphs will destroy the moved one's block styles.  Always move the end of inserted forward 
749     // to preserve the block style of the paragraph already in the document, unless the paragraph to move would 
750     // include the what was the start of the selection that was pasted into, so that we preserve that paragraph's
751     // block styles.
752     bool mergeForward = !(inSameParagraph(startOfInsertedContent, endOfInsertedContent) && !isStartOfParagraph(startOfInsertedContent));
753     
754     VisiblePosition destination = mergeForward ? endOfInsertedContent.next() : endOfInsertedContent;
755     VisiblePosition startOfParagraphToMove = mergeForward ? startOfParagraph(endOfInsertedContent) : endOfInsertedContent.next();
756    
757     // Merging forward could result in deleting the destination anchor node.
758     // To avoid this, we add a placeholder node before the start of the paragraph.
759     if (endOfParagraph(startOfParagraphToMove) == destination) {
760         RefPtr<Node> placeholder = createBreakElement(document());
761         insertNodeBefore(placeholder, startOfParagraphToMove.deepEquivalent().node());
762         destination = VisiblePosition(Position(placeholder.get(), 0));
763     }
764
765     moveParagraph(startOfParagraphToMove, endOfParagraph(startOfParagraphToMove), destination);
766     
767     // Merging forward will remove m_lastLeafInserted from the document.
768     // FIXME: Maintain positions for the start and end of inserted content instead of keeping nodes.  The nodes are
769     // only ever used to create positions where inserted content starts/ends.  Also, we sometimes insert content
770     // directly into text nodes already in the document, in which case tracking inserted nodes is inadequate.
771     if (mergeForward) {
772         m_lastLeafInserted = destination.previous().deepEquivalent().node();
773         if (!m_firstNodeInserted->inDocument())
774             m_firstNodeInserted = endingSelection().visibleStart().deepEquivalent().node();
775         // If we merged text nodes, m_lastLeafInserted could be null. If this is the case,
776         // we use m_firstNodeInserted.
777         if (!m_lastLeafInserted)
778             m_lastLeafInserted = m_firstNodeInserted;
779     }
780 }
781
782 static Node* enclosingInline(Node* node)
783 {
784     while (ContainerNode* parent = node->parentNode()) {
785         if (parent->isBlockFlow() || parent->hasTagName(bodyTag))
786             return node;
787         // Stop if any previous sibling is a block.
788         for (Node* sibling = node->previousSibling(); sibling; sibling = sibling->previousSibling()) {
789             if (sibling->isBlockFlow())
790                 return node;
791         }
792         node = parent;
793     }
794     return node;
795 }
796
797 void ReplaceSelectionCommand::doApply()
798 {
799     VisibleSelection selection = endingSelection();
800     ASSERT(selection.isCaretOrRange());
801     ASSERT(selection.start().node());
802     if (!selection.isNonOrphanedCaretOrRange() || !selection.start().node())
803         return;
804     
805     bool selectionIsPlainText = !selection.isContentRichlyEditable();
806     
807     Element* currentRoot = selection.rootEditableElement();
808     ReplacementFragment fragment(document(), m_documentFragment.get(), m_matchStyle, selection);
809     
810     if (performTrivialReplace(fragment))
811         return;
812     
813     if (m_matchStyle)
814         m_insertionStyle = ApplyStyleCommand::editingStyleAtPosition(selection.start(), IncludeTypingStyle);
815     
816     VisiblePosition visibleStart = selection.visibleStart();
817     VisiblePosition visibleEnd = selection.visibleEnd();
818     
819     bool selectionEndWasEndOfParagraph = isEndOfParagraph(visibleEnd);
820     bool selectionStartWasStartOfParagraph = isStartOfParagraph(visibleStart);
821     
822     Node* startBlock = enclosingBlock(visibleStart.deepEquivalent().node());
823     
824     Position insertionPos = selection.start();
825     bool startIsInsideMailBlockquote = nearestMailBlockquote(insertionPos.node());
826     
827     if ((selectionStartWasStartOfParagraph && selectionEndWasEndOfParagraph && !startIsInsideMailBlockquote) ||
828         startBlock == currentRoot || isListItem(startBlock) || selectionIsPlainText)
829         m_preventNesting = false;
830     
831     if (selection.isRange()) {
832         // When the end of the selection being pasted into is at the end of a paragraph, and that selection
833         // spans multiple blocks, not merging may leave an empty line.
834         // When the start of the selection being pasted into is at the start of a block, not merging 
835         // will leave hanging block(s).
836         // Merge blocks if the start of the selection was in a Mail blockquote, since we handle  
837         // that case specially to prevent nesting. 
838         bool mergeBlocksAfterDelete = startIsInsideMailBlockquote || isEndOfParagraph(visibleEnd) || isStartOfBlock(visibleStart);
839         // FIXME: We should only expand to include fully selected special elements if we are copying a 
840         // selection and pasting it on top of itself.
841         deleteSelection(false, mergeBlocksAfterDelete, true, false);
842         visibleStart = endingSelection().visibleStart();
843         if (fragment.hasInterchangeNewlineAtStart()) {
844             if (isEndOfParagraph(visibleStart) && !isStartOfParagraph(visibleStart)) {
845                 if (!isEndOfDocument(visibleStart))
846                     setEndingSelection(visibleStart.next());
847             } else
848                 insertParagraphSeparator();
849         }
850         insertionPos = endingSelection().start();
851     } else {
852         ASSERT(selection.isCaret());
853         if (fragment.hasInterchangeNewlineAtStart()) {
854             VisiblePosition next = visibleStart.next(true);
855             if (isEndOfParagraph(visibleStart) && !isStartOfParagraph(visibleStart) && next.isNotNull())
856                 setEndingSelection(next);
857             else 
858                 insertParagraphSeparator();
859         }
860         // We split the current paragraph in two to avoid nesting the blocks from the fragment inside the current block.
861         // For example paste <div>foo</div><div>bar</div><div>baz</div> into <div>x^x</div>, where ^ is the caret.  
862         // As long as the  div styles are the same, visually you'd expect: <div>xbar</div><div>bar</div><div>bazx</div>, 
863         // not <div>xbar<div>bar</div><div>bazx</div></div>.
864         // Don't do this if the selection started in a Mail blockquote.
865         if (m_preventNesting && !startIsInsideMailBlockquote && !isEndOfParagraph(visibleStart) && !isStartOfParagraph(visibleStart)) {
866             insertParagraphSeparator();
867             setEndingSelection(endingSelection().visibleStart().previous());
868         }
869         insertionPos = endingSelection().start();
870     }
871     
872     // We don't want any of the pasted content to end up nested in a Mail blockquote, so first break 
873     // out of any surrounding Mail blockquotes. Unless we're inserting in a table, in which case
874     // breaking the blockquote will prevent the content from actually being inserted in the table.
875     if (startIsInsideMailBlockquote && m_preventNesting && !(enclosingNodeOfType(insertionPos, &isTableStructureNode))) { 
876         applyCommandToComposite(BreakBlockquoteCommand::create(document())); 
877         // This will leave a br between the split. 
878         Node* br = endingSelection().start().node(); 
879         ASSERT(br->hasTagName(brTag)); 
880         // Insert content between the two blockquotes, but remove the br (since it was just a placeholder). 
881         insertionPos = positionInParentBeforeNode(br);
882         removeNode(br);
883     }
884     
885     // Inserting content could cause whitespace to collapse, e.g. inserting <div>foo</div> into hello^ world.
886     prepareWhitespaceAtPositionForSplit(insertionPos);
887     
888     // NOTE: This would be an incorrect usage of downstream() if downstream() were changed to mean the last position after 
889     // p that maps to the same visible position as p (since in the case where a br is at the end of a block and collapsed 
890     // away, there are positions after the br which map to the same visible position as [br, 0]).  
891     Node* endBR = insertionPos.downstream().node()->hasTagName(brTag) ? insertionPos.downstream().node() : 0;
892     VisiblePosition originalVisPosBeforeEndBR;
893     if (endBR)
894         originalVisPosBeforeEndBR = VisiblePosition(endBR, 0, DOWNSTREAM).previous();
895     
896     startBlock = enclosingBlock(insertionPos.node());
897     
898     // Adjust insertionPos to prevent nesting.
899     // If the start was in a Mail blockquote, we will have already handled adjusting insertionPos above.
900     if (m_preventNesting && startBlock && !startIsInsideMailBlockquote) {
901         ASSERT(startBlock != currentRoot);
902         VisiblePosition visibleInsertionPos(insertionPos);
903         if (isEndOfBlock(visibleInsertionPos) && !(isStartOfBlock(visibleInsertionPos) && fragment.hasInterchangeNewlineAtEnd()))
904             insertionPos = positionInParentAfterNode(startBlock);
905         else if (isStartOfBlock(visibleInsertionPos))
906             insertionPos = positionInParentBeforeNode(startBlock);
907     }
908
909     // Paste into run of tabs splits the tab span.
910     insertionPos = positionOutsideTabSpan(insertionPos);
911     
912     // Paste at start or end of link goes outside of link.
913     insertionPos = positionAvoidingSpecialElementBoundary(insertionPos);
914     
915     // FIXME: Can this wait until after the operation has been performed?  There doesn't seem to be
916     // any work performed after this that queries or uses the typing style.
917     if (Frame* frame = document()->frame())
918         frame->selection()->clearTypingStyle();
919     
920     bool handledStyleSpans = handleStyleSpansBeforeInsertion(fragment, insertionPos);
921
922     // We don't want the destination to end up inside nodes that weren't selected.  To avoid that, we move the
923     // position forward without changing the visible position so we're still at the same visible location, but
924     // outside of preceding tags.
925     insertionPos = positionAvoidingPrecedingNodes(insertionPos);
926
927     // FIXME: When pasting rich content we're often prevented from heading down the fast path by style spans.  Try
928     // again here if they've been removed.
929     
930     // We're finished if there is nothing to add.
931     if (fragment.isEmpty() || !fragment.firstChild())
932         return;
933     
934     // 1) Insert the content.
935     // 2) Remove redundant styles and style tags, this inner <b> for example: <b>foo <b>bar</b> baz</b>.
936     // 3) Merge the start of the added content with the content before the position being pasted into.
937     // 4) Do one of the following: a) expand the last br if the fragment ends with one and it collapsed,
938     // b) merge the last paragraph of the incoming fragment with the paragraph that contained the 
939     // end of the selection that was pasted into, or c) handle an interchange newline at the end of the 
940     // incoming fragment.
941     // 5) Add spaces for smart replace.
942     // 6) Select the replacement if requested, and match style if requested.
943     
944     VisiblePosition startOfInsertedContent, endOfInsertedContent;
945     
946     RefPtr<Node> refNode = fragment.firstChild();
947     RefPtr<Node> node = refNode->nextSibling();
948     
949     fragment.removeNode(refNode);
950
951     Node* blockStart = enclosingBlock(insertionPos.node());
952     if ((isListElement(refNode.get()) || (isStyleSpan(refNode.get()) && isListElement(refNode->firstChild())))
953         && blockStart->renderer()->isListItem())
954         refNode = insertAsListItems(refNode, blockStart, insertionPos);
955     else
956         insertNodeAtAndUpdateNodesInserted(refNode, insertionPos);
957
958     // Mutation events (bug 22634) may have already removed the inserted content
959     if (!refNode->inDocument())
960         return;
961
962     bool plainTextFragment = isPlainTextMarkup(refNode.get());
963
964     while (node) {
965         Node* next = node->nextSibling();
966         fragment.removeNode(node);
967         insertNodeAfterAndUpdateNodesInserted(node, refNode.get());
968
969         // Mutation events (bug 22634) may have already removed the inserted content
970         if (!node->inDocument())
971             return;
972
973         refNode = node;
974         if (node && plainTextFragment)
975             plainTextFragment = isPlainTextMarkup(node.get());
976         node = next;
977     }
978     
979     removeUnrenderedTextNodesAtEnds();
980     
981     negateStyleRulesThatAffectAppearance();
982     
983     if (!handledStyleSpans)
984         handleStyleSpans();
985     
986     // Mutation events (bug 20161) may have already removed the inserted content
987     if (!m_firstNodeInserted || !m_firstNodeInserted->inDocument())
988         return;
989     
990     endOfInsertedContent = positionAtEndOfInsertedContent();
991     startOfInsertedContent = positionAtStartOfInsertedContent();
992     
993     // We inserted before the startBlock to prevent nesting, and the content before the startBlock wasn't in its own block and
994     // didn't have a br after it, so the inserted content ended up in the same paragraph.
995     if (startBlock && insertionPos.node() == startBlock->parentNode() && (unsigned)insertionPos.deprecatedEditingOffset() < startBlock->nodeIndex() && !isStartOfParagraph(startOfInsertedContent))
996         insertNodeAt(createBreakElement(document()).get(), startOfInsertedContent.deepEquivalent());
997     
998     Position lastPositionToSelect;
999     
1000     bool interchangeNewlineAtEnd = fragment.hasInterchangeNewlineAtEnd();
1001
1002     if (endBR && (plainTextFragment || shouldRemoveEndBR(endBR, originalVisPosBeforeEndBR)))
1003         removeNodeAndPruneAncestors(endBR);
1004     
1005     // Determine whether or not we should merge the end of inserted content with what's after it before we do
1006     // the start merge so that the start merge doesn't effect our decision.
1007     m_shouldMergeEnd = shouldMergeEnd(selectionEndWasEndOfParagraph);
1008     
1009     if (shouldMergeStart(selectionStartWasStartOfParagraph, fragment.hasInterchangeNewlineAtStart(), startIsInsideMailBlockquote)) {
1010         VisiblePosition destination = startOfInsertedContent.previous();
1011         VisiblePosition startOfParagraphToMove = startOfInsertedContent;
1012         // We need to handle the case where we need to merge the end
1013         // but our destination node is inside an inline that is the last in the block.
1014         // We insert a placeholder before the newly inserted content to avoid being merged into the inline.
1015         Node* destinationNode = destination.deepEquivalent().node();
1016         if (m_shouldMergeEnd && destinationNode != enclosingInline(destinationNode) && enclosingInline(destinationNode)->nextSibling())
1017             insertNodeBefore(createBreakElement(document()), refNode.get());
1018         
1019         // Merging the the first paragraph of inserted content with the content that came
1020         // before the selection that was pasted into would also move content after 
1021         // the selection that was pasted into if: only one paragraph was being pasted, 
1022         // and it was not wrapped in a block, the selection that was pasted into ended 
1023         // at the end of a block and the next paragraph didn't start at the start of a block.
1024         // Insert a line break just after the inserted content to separate it from what 
1025         // comes after and prevent that from happening.
1026         VisiblePosition endOfInsertedContent = positionAtEndOfInsertedContent();
1027         if (startOfParagraph(endOfInsertedContent) == startOfParagraphToMove) {
1028             insertNodeAt(createBreakElement(document()).get(), endOfInsertedContent.deepEquivalent());
1029             // Mutation events (bug 22634) triggered by inserting the <br> might have removed the content we're about to move
1030             if (!startOfParagraphToMove.deepEquivalent().node()->inDocument())
1031                 return;
1032         }
1033
1034         // FIXME: Maintain positions for the start and end of inserted content instead of keeping nodes.  The nodes are
1035         // only ever used to create positions where inserted content starts/ends.
1036         moveParagraph(startOfParagraphToMove, endOfParagraph(startOfParagraphToMove), destination);
1037         m_firstNodeInserted = endingSelection().visibleStart().deepEquivalent().downstream().node();
1038         if (!m_lastLeafInserted->inDocument())
1039             m_lastLeafInserted = endingSelection().visibleEnd().deepEquivalent().upstream().node();
1040     }
1041             
1042     endOfInsertedContent = positionAtEndOfInsertedContent();
1043     startOfInsertedContent = positionAtStartOfInsertedContent();
1044     
1045     if (interchangeNewlineAtEnd) {
1046         VisiblePosition next = endOfInsertedContent.next(true);
1047
1048         if (selectionEndWasEndOfParagraph || !isEndOfParagraph(endOfInsertedContent) || next.isNull()) {
1049             if (!isStartOfParagraph(endOfInsertedContent)) {
1050                 setEndingSelection(endOfInsertedContent);
1051                 Node* enclosingNode = enclosingBlock(endOfInsertedContent.deepEquivalent().node());
1052                 if (isListItem(enclosingNode)) {
1053                     RefPtr<Node> newListItem = createListItemElement(document());
1054                     insertNodeAfter(newListItem, enclosingNode);
1055                     setEndingSelection(VisiblePosition(Position(newListItem, 0)));
1056                 } else
1057                     // Use a default paragraph element (a plain div) for the empty paragraph, using the last paragraph
1058                     // block's style seems to annoy users.
1059                     insertParagraphSeparator(true);
1060
1061                 // Select up to the paragraph separator that was added.
1062                 lastPositionToSelect = endingSelection().visibleStart().deepEquivalent();
1063                 updateNodesInserted(lastPositionToSelect.node());
1064             }
1065         } else {
1066             // Select up to the beginning of the next paragraph.
1067             lastPositionToSelect = next.deepEquivalent().downstream();
1068         }
1069         
1070     } else
1071         mergeEndIfNeeded();
1072     
1073     handlePasteAsQuotationNode();
1074     
1075     endOfInsertedContent = positionAtEndOfInsertedContent();
1076     startOfInsertedContent = positionAtStartOfInsertedContent();
1077     
1078     // Add spaces for smart replace.
1079     if (m_smartReplace && currentRoot) {
1080         // Disable smart replace for password fields.
1081         Node* start = currentRoot->shadowAncestorNode();
1082         if (start->hasTagName(inputTag) && static_cast<HTMLInputElement*>(start)->isPasswordField())
1083             m_smartReplace = false;
1084     }
1085     if (m_smartReplace) {
1086         bool needsTrailingSpace = !isEndOfParagraph(endOfInsertedContent) &&
1087                                   !isCharacterSmartReplaceExempt(endOfInsertedContent.characterAfter(), false);
1088         if (needsTrailingSpace) {
1089             RenderObject* renderer = m_lastLeafInserted->renderer();
1090             bool collapseWhiteSpace = !renderer || renderer->style()->collapseWhiteSpace();
1091             Node* endNode = positionAtEndOfInsertedContent().deepEquivalent().upstream().node();
1092             if (endNode->isTextNode()) {
1093                 Text* text = static_cast<Text*>(endNode);
1094                 insertTextIntoNode(text, text->length(), collapseWhiteSpace ? nonBreakingSpaceString() : " ");
1095             } else {
1096                 RefPtr<Node> node = document()->createEditingTextNode(collapseWhiteSpace ? nonBreakingSpaceString() : " ");
1097                 insertNodeAfterAndUpdateNodesInserted(node, endNode);
1098             }
1099         }
1100     
1101         bool needsLeadingSpace = !isStartOfParagraph(startOfInsertedContent) &&
1102                                  !isCharacterSmartReplaceExempt(startOfInsertedContent.previous().characterAfter(), true);
1103         if (needsLeadingSpace) {
1104             RenderObject* renderer = m_lastLeafInserted->renderer();
1105             bool collapseWhiteSpace = !renderer || renderer->style()->collapseWhiteSpace();
1106             Node* startNode = positionAtStartOfInsertedContent().deepEquivalent().downstream().node();
1107             if (startNode->isTextNode()) {
1108                 Text* text = static_cast<Text*>(startNode);
1109                 insertTextIntoNode(text, 0, collapseWhiteSpace ? nonBreakingSpaceString() : " ");
1110             } else {
1111                 RefPtr<Node> node = document()->createEditingTextNode(collapseWhiteSpace ? nonBreakingSpaceString() : " ");
1112                 // Don't updateNodesInserted.  Doing so would set m_lastLeafInserted to be the node containing the 
1113                 // leading space, but m_lastLeafInserted is supposed to mark the end of pasted content.
1114                 insertNodeBefore(node, startNode);
1115                 // FIXME: Use positions to track the start/end of inserted content.
1116                 m_firstNodeInserted = node;
1117             }
1118         }
1119     }
1120     
1121     // If we are dealing with a fragment created from plain text
1122     // no style matching is necessary.
1123     if (plainTextFragment)
1124         m_matchStyle = false;
1125         
1126     completeHTMLReplacement(lastPositionToSelect);
1127 }
1128
1129 bool ReplaceSelectionCommand::shouldRemoveEndBR(Node* endBR, const VisiblePosition& originalVisPosBeforeEndBR)
1130 {
1131     if (!endBR || !endBR->inDocument())
1132         return false;
1133         
1134     VisiblePosition visiblePos(Position(endBR, 0));
1135     
1136     // Don't remove the br if nothing was inserted.
1137     if (visiblePos.previous() == originalVisPosBeforeEndBR)
1138         return false;
1139     
1140     // Remove the br if it is collapsed away and so is unnecessary.
1141     if (!document()->inNoQuirksMode() && isEndOfBlock(visiblePos) && !isStartOfParagraph(visiblePos))
1142         return true;
1143         
1144     // A br that was originally holding a line open should be displaced by inserted content or turned into a line break.
1145     // A br that was originally acting as a line break should still be acting as a line break, not as a placeholder.
1146     return isStartOfParagraph(visiblePos) && isEndOfParagraph(visiblePos);
1147 }
1148
1149 void ReplaceSelectionCommand::completeHTMLReplacement(const Position &lastPositionToSelect)
1150 {
1151     Position start;
1152     Position end;
1153
1154     // FIXME: This should never not be the case.
1155     if (m_firstNodeInserted && m_firstNodeInserted->inDocument() && m_lastLeafInserted && m_lastLeafInserted->inDocument()) {
1156         
1157         start = positionAtStartOfInsertedContent().deepEquivalent();
1158         end = positionAtEndOfInsertedContent().deepEquivalent();
1159         
1160         // FIXME (11475): Remove this and require that the creator of the fragment to use nbsps.
1161         rebalanceWhitespaceAt(start);
1162         rebalanceWhitespaceAt(end);
1163
1164         if (m_matchStyle) {
1165             ASSERT(m_insertionStyle);
1166             applyStyle(m_insertionStyle.get(), start, end);
1167         }    
1168         
1169         if (lastPositionToSelect.isNotNull())
1170             end = lastPositionToSelect;
1171     } else if (lastPositionToSelect.isNotNull())
1172         start = end = lastPositionToSelect;
1173     else
1174         return;
1175     
1176     if (m_selectReplacement)
1177         setEndingSelection(VisibleSelection(start, end, SEL_DEFAULT_AFFINITY));
1178     else
1179         setEndingSelection(VisibleSelection(end, SEL_DEFAULT_AFFINITY));
1180 }
1181
1182 EditAction ReplaceSelectionCommand::editingAction() const
1183 {
1184     return m_editAction;
1185 }
1186
1187 void ReplaceSelectionCommand::insertNodeAfterAndUpdateNodesInserted(PassRefPtr<Node> insertChild, Node* refChild)
1188 {
1189     Node* nodeToUpdate = insertChild.get(); // insertChild will be cleared when passed
1190     insertNodeAfter(insertChild, refChild);
1191     updateNodesInserted(nodeToUpdate);
1192 }
1193
1194 void ReplaceSelectionCommand::insertNodeAtAndUpdateNodesInserted(PassRefPtr<Node> insertChild, const Position& p)
1195 {
1196     Node* nodeToUpdate = insertChild.get(); // insertChild will be cleared when passed
1197     insertNodeAt(insertChild, p);
1198     updateNodesInserted(nodeToUpdate);
1199 }
1200
1201 void ReplaceSelectionCommand::insertNodeBeforeAndUpdateNodesInserted(PassRefPtr<Node> insertChild, Node* refChild)
1202 {
1203     Node* nodeToUpdate = insertChild.get(); // insertChild will be cleared when passed
1204     insertNodeBefore(insertChild, refChild);
1205     updateNodesInserted(nodeToUpdate);
1206 }
1207
1208 // If the user is inserting a list into an existing list, instead of nesting the list,
1209 // we put the list items into the existing list.
1210 Node* ReplaceSelectionCommand::insertAsListItems(PassRefPtr<Node> listElement, Node* insertionBlock, const Position& insertPos)
1211 {
1212     while (listElement->hasChildNodes() && isListElement(listElement->firstChild()) && listElement->childNodeCount() == 1)
1213         listElement = listElement->firstChild();
1214
1215     bool isStart = isStartOfParagraph(insertPos);
1216     bool isEnd = isEndOfParagraph(insertPos);
1217     bool isMiddle = !isStart && !isEnd;
1218     Node* lastNode = insertionBlock;
1219
1220     // If we're in the middle of a list item, we should split it into two separate
1221     // list items and insert these nodes between them.
1222     if (isMiddle) {
1223         int textNodeOffset = insertPos.offsetInContainerNode();
1224         if (insertPos.node()->isTextNode() && textNodeOffset > 0)
1225             splitTextNode(static_cast<Text*>(insertPos.node()), textNodeOffset);
1226         splitTreeToNode(insertPos.node(), lastNode, true);
1227     }
1228
1229     while (RefPtr<Node> listItem = listElement->firstChild()) {
1230         ExceptionCode ec = 0;
1231         toContainerNode(listElement.get())->removeChild(listItem.get(), ec);
1232         ASSERT(!ec);
1233         if (isStart || isMiddle)
1234             insertNodeBefore(listItem, lastNode);
1235         else if (isEnd) {
1236             insertNodeAfter(listItem, lastNode);
1237             lastNode = listItem.get();
1238         } else
1239             ASSERT_NOT_REACHED();
1240     }
1241     if (isStart || isMiddle)
1242         lastNode = lastNode->previousSibling();
1243     if (isMiddle)
1244         insertNodeAfter(createListItemElement(document()), lastNode);
1245     updateNodesInserted(lastNode);
1246     return lastNode;
1247 }
1248
1249 void ReplaceSelectionCommand::updateNodesInserted(Node *node)
1250 {
1251     if (!node)
1252         return;
1253
1254     if (!m_firstNodeInserted)
1255         m_firstNodeInserted = node;
1256     
1257     if (node == m_lastLeafInserted)
1258         return;
1259     
1260     m_lastLeafInserted = node->lastDescendant();
1261 }
1262
1263 // During simple pastes, where we're just pasting a text node into a run of text, we insert the text node
1264 // directly into the text node that holds the selection.  This is much faster than the generalized code in
1265 // ReplaceSelectionCommand, and works around <https://bugs.webkit.org/show_bug.cgi?id=6148> since we don't 
1266 // split text nodes.
1267 bool ReplaceSelectionCommand::performTrivialReplace(const ReplacementFragment& fragment)
1268 {
1269     if (!fragment.firstChild() || fragment.firstChild() != fragment.lastChild() || !fragment.firstChild()->isTextNode())
1270         return false;
1271         
1272     // FIXME: Would be nice to handle smart replace in the fast path.
1273     if (m_smartReplace || fragment.hasInterchangeNewlineAtStart() || fragment.hasInterchangeNewlineAtEnd())
1274         return false;
1275     
1276     Text* textNode = static_cast<Text*>(fragment.firstChild());
1277     // Our fragment creation code handles tabs, spaces, and newlines, so we don't have to worry about those here.
1278     String text(textNode->data());
1279     
1280     Position start = endingSelection().start();
1281     Position end = endingSelection().end();
1282     
1283     if (start.anchorNode() != end.anchorNode() || !start.anchorNode()->isTextNode())
1284         return false;
1285         
1286     replaceTextInNode(static_cast<Text*>(start.anchorNode()), start.offsetInContainerNode(), end.offsetInContainerNode() - start.offsetInContainerNode(), text);
1287     
1288     end = Position(start.anchorNode(), start.offsetInContainerNode() + text.length());
1289     
1290     VisibleSelection selectionAfterReplace(m_selectReplacement ? start : end, end);
1291     
1292     setEndingSelection(selectionAfterReplace);
1293     
1294     return true;
1295 }
1296
1297 } // namespace WebCore