WebCore:
[WebKit-https.git] / WebCore / editing / markup.cpp
1 /*
2  * Copyright (C) 2004, 2005, 2006, 2007 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 "markup.h"
28
29 #include "CDATASection.h"
30 #include "CSSComputedStyleDeclaration.h"
31 #include "CSSPropertyNames.h"
32 #include "CSSRule.h"
33 #include "CSSRuleList.h"
34 #include "CSSStyleRule.h"
35 #include "CSSStyleSelector.h"
36 #include "CSSValueKeywords.h"
37 #include "Comment.h"
38 #include "DeleteButtonController.h"
39 #include "DeprecatedStringList.h"
40 #include "Document.h"
41 #include "DocumentFragment.h"
42 #include "DocumentType.h"
43 #include "Editor.h"
44 #include "Frame.h"
45 #include "HTMLElement.h"
46 #include "HTMLNames.h"
47 #include "InlineTextBox.h"
48 #include "KURL.h"
49 #include "Logging.h"
50 #include "ProcessingInstruction.h"
51 #include "QualifiedName.h"
52 #include "Range.h"
53 #include "Selection.h"
54 #include "TextIterator.h"
55 #include "htmlediting.h"
56 #include "visible_units.h"
57
58 using namespace std;
59
60 namespace WebCore {
61
62 using namespace HTMLNames;
63
64 static inline bool shouldSelfClose(const Node *node);
65
66 class AttributeChange {
67 public:
68     AttributeChange()
69         : m_name(nullAtom, nullAtom, nullAtom)
70     {
71     }
72
73     AttributeChange(PassRefPtr<Element> element, const QualifiedName& name, const String& value)
74         : m_element(element), m_name(name), m_value(value)
75     {
76     }
77
78     void apply()
79     {
80         m_element->setAttribute(m_name, m_value);
81     }
82
83 private:
84     RefPtr<Element> m_element;
85     QualifiedName m_name;
86     String m_value;
87 };
88
89 static DeprecatedString escapeTextForMarkup(const String& in, bool isAttributeValue)
90 {
91     DeprecatedString s = "";
92
93     unsigned len = in.length();
94     for (unsigned i = 0; i < len; ++i) {
95         switch (in[i]) {
96             case '&':
97                 s += "&amp;";
98                 break;
99             case '<':
100                 s += "&lt;";
101                 break;
102             case '>':
103                 s += "&gt;";
104                 break;
105             case '"':
106                 if (isAttributeValue) {
107                     s += "&quot;";
108                     break;
109                 }
110                 // fall through
111             default:
112                 s += DeprecatedChar(in[i]);
113         }
114     }
115
116     return s;
117 }
118     
119 static String urlAttributeToQuotedString(String urlString)
120 {
121     UChar quoteChar = '"';
122     if (urlString.stripWhiteSpace().startsWith("javascript:", false)) {
123         // minimal escaping for javascript urls
124         if (urlString.contains('"')) {
125             if (urlString.contains('\''))
126                 urlString.replace('"', "&quot;");
127             else
128                 quoteChar = '\'';
129         }
130     } else
131         // FIXME this does not fully match other browsers. Firefox escapes spaces and other special characters.
132         urlString = escapeTextForMarkup(urlString.deprecatedString(), true);
133
134     String res;
135     res.append(quoteChar);
136     res.append(urlString);
137     res.append(quoteChar);
138     return res;
139 }
140
141 static String stringValueForRange(const Node *node, const Range *range)
142 {
143     String str = node->nodeValue().copy();
144     if (range) {
145         ExceptionCode ec;
146         if (node == range->endContainer(ec))
147             str.truncate(range->endOffset(ec));
148         if (node == range->startContainer(ec))
149             str.remove(0, range->startOffset(ec));
150     }
151     return str;
152 }
153
154 static String renderedText(const Node* node, const Range* range)
155 {
156     if (!node->isTextNode())
157         return String();
158
159     ExceptionCode ec;
160     const Text* textNode = static_cast<const Text*>(node);
161     unsigned startOffset = 0;
162     unsigned endOffset = textNode->length();
163
164     if (range && node == range->startContainer(ec))
165         startOffset = range->startOffset(ec);
166     if (range && node == range->endContainer(ec))
167         endOffset = range->endOffset(ec);
168     
169     Position start(const_cast<Node*>(node), startOffset);
170     Position end(const_cast<Node*>(node), endOffset);
171     Range r(node->document(), start, end);
172     return plainText(&r);
173 }
174
175 static PassRefPtr<CSSMutableStyleDeclaration> styleFromMatchedRulesForElement(Element* element, bool authorOnly = true)
176 {
177     RefPtr<CSSMutableStyleDeclaration> style = new CSSMutableStyleDeclaration();
178     RefPtr<CSSRuleList> matchedRules = element->document()->styleSelector()->styleRulesForElement(element, authorOnly);
179     if (matchedRules) {
180         for (unsigned i = 0; i < matchedRules->length(); i++) {
181             if (matchedRules->item(i)->type() == CSSRule::STYLE_RULE) {
182                 RefPtr<CSSMutableStyleDeclaration> s = static_cast<CSSStyleRule*>(matchedRules->item(i))->style();
183                 style->merge(s.get(), true);
184             }
185         }
186     }
187     
188     return style.release();
189 }
190
191 static void removeEnclosingMailBlockquoteStyle(CSSMutableStyleDeclaration* style, Node* node)
192 {
193     Node* blockquote = nearestMailBlockquote(node);
194     if (!blockquote || !blockquote->parentNode())
195         return;
196             
197     RefPtr<CSSMutableStyleDeclaration> parentStyle = Position(blockquote->parentNode(), 0).computedStyle()->copyInheritableProperties();
198     RefPtr<CSSMutableStyleDeclaration> blockquoteStyle = Position(blockquote, 0).computedStyle()->copyInheritableProperties();
199     parentStyle->diff(blockquoteStyle.get());
200     blockquoteStyle->diff(style);
201 }
202
203 static bool shouldAddNamespaceElem(const Element* elem)
204 {
205     // Don't add namespace attribute if it is already defined for this elem.
206     const AtomicString& prefix = elem->prefix();
207     AtomicString attr = !prefix.isEmpty() ? "xmlns:" + prefix : "xmlns";
208     return !elem->hasAttribute(attr);
209 }
210
211 static bool shouldAddNamespaceAttr(const Attribute* attr, HashMap<AtomicStringImpl*, AtomicStringImpl*>& namespaces)
212 {
213     // Don't add namespace attributes twice
214     static const AtomicString xmlnsURI = "http://www.w3.org/2000/xmlns/";
215     static const QualifiedName xmlnsAttr(nullAtom, "xmlns", xmlnsURI);
216     if (attr->name() == xmlnsAttr) {
217         namespaces.set(emptyAtom.impl(), attr->value().impl());
218         return false;
219     }
220     
221     QualifiedName xmlnsPrefixAttr("xmlns", attr->localName(), xmlnsURI);
222     if (attr->name() == xmlnsPrefixAttr) {
223         namespaces.set(attr->localName().impl(), attr->value().impl());
224         return false;
225     }
226     
227     return true;
228 }
229
230 static String addNamespace(const AtomicString& prefix, const AtomicString& ns, HashMap<AtomicStringImpl*, AtomicStringImpl*>& namespaces)
231 {
232     if (ns.isEmpty())
233         return "";
234     
235     // Use emptyAtoms's impl() for both null and empty strings since the HashMap can't handle 0 as a key
236     AtomicStringImpl* pre = prefix.isEmpty() ? emptyAtom.impl() : prefix.impl();
237     AtomicStringImpl* foundNS = namespaces.get(pre);
238     if (foundNS != ns.impl()) {
239         namespaces.set(pre, ns.impl());
240         return " xmlns" + (!prefix.isEmpty() ? ":" + prefix : "") + "=\"" + escapeTextForMarkup(ns, true) + "\"";
241     }
242     
243     return "";
244 }
245
246 static DeprecatedString startMarkup(const Node *node, const Range *range, EAnnotateForInterchange annotate, bool convertBlocksToInlines = false, HashMap<AtomicStringImpl*, AtomicStringImpl*>* namespaces = 0)
247 {
248     bool documentIsHTML = node->document()->isHTMLDocument();
249     switch (node->nodeType()) {
250         case Node::TEXT_NODE: {
251             if (Node* parent = node->parentNode()) {
252                 if (parent->hasTagName(listingTag)
253                         || parent->hasTagName(scriptTag)
254                         || parent->hasTagName(styleTag)
255                         || parent->hasTagName(textareaTag)
256                         || parent->hasTagName(xmpTag))
257                     return stringValueForRange(node, range).deprecatedString();
258             }
259             bool useRenderedText = annotate && !enclosingNodeWithTag(const_cast<Node*>(node), selectTag);
260             DeprecatedString markup = escapeTextForMarkup(useRenderedText ? renderedText(node, range) : stringValueForRange(node, range), false);
261             return annotate ? convertHTMLTextToInterchangeFormat(markup, static_cast<const Text*>(node)) : markup;
262         }
263         case Node::COMMENT_NODE:
264             return static_cast<const Comment*>(node)->toString().deprecatedString();
265         case Node::DOCUMENT_NODE: {
266             // Documents do not normally contain a docType as a child node, force it to print here instead.
267             const DocumentType* docType = static_cast<const Document*>(node)->doctype();
268             if (docType)
269                 return docType->toString().deprecatedString();
270             return "";
271         }
272         case Node::DOCUMENT_FRAGMENT_NODE:
273             return "";
274         case Node::DOCUMENT_TYPE_NODE:
275             return static_cast<const DocumentType*>(node)->toString().deprecatedString();
276         case Node::PROCESSING_INSTRUCTION_NODE:
277             return static_cast<const ProcessingInstruction*>(node)->toString().deprecatedString();
278         case Node::ELEMENT_NODE: {
279             DeprecatedString markup = DeprecatedChar('<');
280             const Element* el = static_cast<const Element*>(node);
281             convertBlocksToInlines &= isBlock(const_cast<Node*>(node));
282             markup += el->nodeNamePreservingCase().deprecatedString();
283             NamedAttrMap *attrs = el->attributes();
284             unsigned length = attrs->length();
285             if (!documentIsHTML && namespaces && shouldAddNamespaceElem(el))
286                 markup += addNamespace(el->prefix(), el->namespaceURI(), *namespaces).deprecatedString();
287
288             for (unsigned int i = 0; i < length; i++) {
289                 Attribute *attr = attrs->attributeItem(i);
290                 // We'll handle the style attribute separately, below.
291                 if (attr->name() == styleAttr && el->isHTMLElement() && (annotate || convertBlocksToInlines))
292                     continue;
293                 if (documentIsHTML)
294                     markup += " " + attr->name().localName().deprecatedString();
295                 else
296                     markup += " " + attr->name().toString().deprecatedString();
297                 if (el->isURLAttribute(attr))
298                     markup += "=" + urlAttributeToQuotedString(attr->value()).deprecatedString();
299                 else
300                     markup += "=\"" + escapeTextForMarkup(attr->value(), true) + "\"";
301                 if (!documentIsHTML && namespaces && shouldAddNamespaceAttr(attr, *namespaces))
302                     markup += addNamespace(attr->prefix(), attr->namespaceURI(), *namespaces).deprecatedString();
303             }
304             
305             if (el->isHTMLElement() && (annotate || convertBlocksToInlines)) {
306                 Element* element = const_cast<Element*>(el);
307                 RefPtr<CSSMutableStyleDeclaration> style = static_cast<HTMLElement*>(element)->getInlineStyleDecl()->copy();
308                 if (annotate) {
309                     RefPtr<CSSMutableStyleDeclaration> styleFromMatchedRules = styleFromMatchedRulesForElement(const_cast<Element*>(el));
310                     style->merge(styleFromMatchedRules.get());
311                 }
312                 if (convertBlocksToInlines)
313                     style->setProperty(CSS_PROP_DISPLAY, CSS_VAL_INLINE, true);
314                 if (style->length() > 0)
315                     markup += " style=\"" + escapeTextForMarkup(style->cssText(), true) + "\"";
316             }
317             
318             if (shouldSelfClose(el)) {
319                 if (el->isHTMLElement())
320                     markup += " "; // XHTML 1.0 <-> HTML compatibility.
321                 markup += "/>";
322             } else
323                 markup += ">";
324             
325             return markup;
326         }
327         case Node::CDATA_SECTION_NODE:
328             return static_cast<const CDATASection*>(node)->toString().deprecatedString();
329         case Node::ATTRIBUTE_NODE:
330         case Node::ENTITY_NODE:
331         case Node::ENTITY_REFERENCE_NODE:
332         case Node::NOTATION_NODE:
333         case Node::XPATH_NAMESPACE_NODE:
334             break;
335     }
336     return "";
337 }
338
339 static inline bool doesHTMLForbidEndTag(const Node *node)
340 {
341     if (node->isHTMLElement()) {
342         const HTMLElement* htmlElt = static_cast<const HTMLElement*>(node);
343         return (htmlElt->endTagRequirement() == TagStatusForbidden);
344     }
345     return false;
346 }
347
348 // Rules of self-closure
349 // 1. No elements in HTML documents use the self-closing syntax.
350 // 2. Elements w/ children never self-close because they use a separate end tag.
351 // 3. HTML elements which do not have a "forbidden" end tag will close with a separate end tag.
352 // 4. Other elements self-close.
353 static inline bool shouldSelfClose(const Node *node)
354 {
355     if (node->document()->isHTMLDocument())
356         return false;
357     if (node->hasChildNodes())
358         return false;
359     if (node->isHTMLElement() && !doesHTMLForbidEndTag(node))
360         return false;
361     return true;
362 }
363
364 static DeprecatedString endMarkup(const Node *node)
365 {
366     if (node->isElementNode() && !shouldSelfClose(node) && (node->hasChildNodes() || !doesHTMLForbidEndTag(node)))
367         return "</" + static_cast<const Element*>(node)->nodeNamePreservingCase().deprecatedString() + ">";
368     return "";
369 }
370
371 static DeprecatedString markup(Node* startNode, bool onlyIncludeChildren, Vector<Node*>* nodes, const HashMap<AtomicStringImpl*, AtomicStringImpl*>* namespaces = 0)
372 {
373     HashMap<AtomicStringImpl*, AtomicStringImpl*> namespaceHash;
374     if (namespaces)
375         namespaceHash = *namespaces;
376     
377     DeprecatedString me = "";
378     if (!onlyIncludeChildren) {
379         if (nodes)
380             nodes->append(startNode);
381         me += startMarkup(startNode, 0, DoNotAnnotateForInterchange, false, &namespaceHash);
382     }
383     // print children
384     if (!(startNode->document()->isHTMLDocument() && doesHTMLForbidEndTag(startNode)))
385         for (Node* current = startNode->firstChild(); current; current = current->nextSibling())
386             me += markup(current, false, nodes, &namespaceHash);
387     
388     // Print my ending tag
389     if (!onlyIncludeChildren)
390         me += endMarkup(startNode);
391     
392     return me;
393 }
394
395 static void completeURLs(Node* node, const String& baseURL)
396 {
397     Vector<AttributeChange> changes;
398
399     KURL baseURLAsKURL(baseURL.deprecatedString());
400
401     Node* end = node->traverseNextSibling();
402     for (Node* n = node; n != end; n = n->traverseNextNode()) {
403         if (n->isElementNode()) {
404             Element* e = static_cast<Element*>(n);
405             NamedAttrMap* attrs = e->attributes();
406             unsigned length = attrs->length();
407             for (unsigned i = 0; i < length; i++) {
408                 Attribute* attr = attrs->attributeItem(i);
409                 if (e->isURLAttribute(attr)) {
410                     String completedURL = KURL(baseURLAsKURL, attr->value().deprecatedString()).url();
411                     changes.append(AttributeChange(e, attr->name(), completedURL));
412                 }
413             }
414         }
415     }
416
417     size_t numChanges = changes.size();
418     for (size_t i = 0; i < numChanges; ++i)
419         changes[i].apply();
420 }
421
422 static bool needInterchangeNewlineAfter(const VisiblePosition& v)
423 {
424     VisiblePosition next = v.next();
425     Node* upstreamNode = next.deepEquivalent().upstream().node();
426     Node* downstreamNode = v.deepEquivalent().downstream().node();
427     // Add an interchange newline if a paragraph break is selected and a br won't already be added to the markup to represent it.
428     return isEndOfParagraph(v) && isStartOfParagraph(next) && !(upstreamNode->hasTagName(brTag) && upstreamNode == downstreamNode);
429 }
430
431 static PassRefPtr<CSSMutableStyleDeclaration> styleFromMatchedRulesAndInlineDecl(Node* node)
432 {
433     if (!node->isHTMLElement())
434         return 0;
435     
436     HTMLElement* element = static_cast<HTMLElement*>(node);
437     RefPtr<CSSMutableStyleDeclaration> style = styleFromMatchedRulesForElement(element);
438     RefPtr<CSSMutableStyleDeclaration> inlineStyleDecl = element->getInlineStyleDecl();
439     style->merge(inlineStyleDecl.get());
440     return style.release();
441 }
442
443 static bool propertyMissingOrEqualToNone(CSSMutableStyleDeclaration* style, int propertyID)
444 {
445     if (!style)
446         return false;
447     RefPtr<CSSValue> value = style->getPropertyCSSValue(propertyID);
448     if (!value)
449         return true;
450     if (!value->isPrimitiveValue())
451         return false;
452     return static_cast<CSSPrimitiveValue*>(value.get())->getIdent() == CSS_VAL_NONE;
453 }
454
455 static bool elementHasTextDecorationProperty(Node* node)
456 {
457     RefPtr<CSSMutableStyleDeclaration> style = styleFromMatchedRulesAndInlineDecl(node);
458     if (!style)
459         return false;
460     return !propertyMissingOrEqualToNone(style.get(), CSS_PROP_TEXT_DECORATION);
461 }
462
463 // FIXME: Shouldn't we omit style info when annotate == DoNotAnnotateForInterchange? 
464 // FIXME: At least, annotation and style info should probably not be included in range.markupString()
465 DeprecatedString createMarkup(const Range* range, Vector<Node*>* nodes, EAnnotateForInterchange annotate, bool convertBlocksToInlines)
466 {
467     static const DeprecatedString interchangeNewlineString = DeprecatedString("<br class=\"") + AppleInterchangeNewline + "\">";
468
469     if (!range || range->isDetached())
470         return "";
471
472     Document* document = range->ownerDocument();
473     if (!document)
474         return "";
475
476     // Disable the delete button so it's elements are not serialized into the markup,
477     // but make sure neither endpoint is inside the delete user interface.
478     Frame* frame = document->frame();
479     DeleteButtonController* deleteButton = frame ? frame->editor()->deleteButtonController() : 0;
480     RefPtr<Range> updatedRange = avoidIntersectionWithNode(range, deleteButton ? deleteButton->containerElement() : 0);
481     if (deleteButton)
482         deleteButton->disable();
483
484     ExceptionCode ec = 0;
485     bool collapsed = updatedRange->collapsed(ec);
486     ASSERT(ec == 0);
487     if (collapsed)
488         return "";
489     Node* commonAncestor = updatedRange->commonAncestorContainer(ec);
490     ASSERT(ec == 0);
491     if (!commonAncestor)
492         return "";
493
494     document->updateLayoutIgnorePendingStylesheets();
495
496     DeprecatedStringList markups;
497     Node* pastEnd = updatedRange->pastEndNode();
498     Node* lastClosed = 0;
499     Vector<Node*> ancestorsToClose;
500     
501     Node* startNode = updatedRange->startNode();
502     VisiblePosition visibleStart(updatedRange->startPosition(), VP_DEFAULT_AFFINITY);
503     VisiblePosition visibleEnd(updatedRange->endPosition(), VP_DEFAULT_AFFINITY);
504     if (annotate && needInterchangeNewlineAfter(visibleStart)) {
505         if (visibleStart == visibleEnd.previous()) {
506             if (deleteButton)
507                 deleteButton->enable();
508             return interchangeNewlineString;
509         }
510
511         markups.append(interchangeNewlineString);
512         startNode = visibleStart.next().deepEquivalent().node();
513     }
514
515     Node* next;
516     for (Node* n = startNode; n != pastEnd; n = next) {
517         next = n->traverseNextNode();
518         bool skipDescendants = false;
519         bool addMarkupForNode = true;
520         
521         if (!n->renderer() && !enclosingNodeWithTag(n, selectTag)) {
522             skipDescendants = true;
523             addMarkupForNode = false;
524             next = n->traverseNextSibling();
525             // Don't skip over pastEnd.
526             if (pastEnd && pastEnd->isDescendantOf(n))
527                 next = pastEnd;
528         }
529
530         if (isBlock(n) && canHaveChildrenForEditing(n) && next == pastEnd)
531             // Don't write out empty block containers that aren't fully selected.
532             continue;
533         
534         // Add the node to the markup.
535         if (addMarkupForNode) {
536             markups.append(startMarkup(n, updatedRange.get(), annotate));
537             if (nodes)
538                 nodes->append(n);
539         }
540         
541         if (n->firstChild() == 0 || skipDescendants) {
542             // Node has no children, or we are skipping it's descendants, add its close tag now.
543             if (addMarkupForNode) {
544                 markups.append(endMarkup(n));
545                 lastClosed = n;
546             }
547             
548             // Check if the node is the last leaf of a tree.
549             if (!n->nextSibling() || next == pastEnd) {
550                 if (!ancestorsToClose.isEmpty()) {
551                     // Close up the ancestors.
552                     do {
553                         Node *ancestor = ancestorsToClose.last();
554                         if (next != pastEnd && next->isDescendantOf(ancestor))
555                             break;
556                         // Not at the end of the range, close ancestors up to sibling of next node.
557                         markups.append(endMarkup(ancestor));
558                         lastClosed = ancestor;
559                         ancestorsToClose.removeLast();
560                     } while (!ancestorsToClose.isEmpty());
561                 }
562                 
563                 // Surround the currently accumulated markup with markup for ancestors we never opened as we leave the subtree(s) rooted at those ancestors.
564                 Node* nextParent = next ? next->parentNode() : 0;
565                 if (next != pastEnd && n != nextParent) {
566                     Node* lastAncestorClosedOrSelf = n->isDescendantOf(lastClosed) ? lastClosed : n;
567                     for (Node *parent = lastAncestorClosedOrSelf->parent(); parent != 0 && parent != nextParent; parent = parent->parentNode()) {
568                         // All ancestors that aren't in the ancestorsToClose list should either be a) unrendered:
569                         if (!parent->renderer())
570                             continue;
571                         // or b) ancestors that we never encountered during a pre-order traversal starting at startNode:
572                         ASSERT(startNode->isDescendantOf(parent));
573                         markups.prepend(startMarkup(parent, updatedRange.get(), annotate));
574                         markups.append(endMarkup(parent));
575                         if (nodes)
576                             nodes->append(parent);
577                         lastClosed = parent;
578                     }
579                 }
580             }
581         } else if (addMarkupForNode && !skipDescendants)
582             // We added markup for this node, and we're descending into it.  Set it to close eventually.
583             ancestorsToClose.append(n);
584     }
585     
586     // Include ancestors that aren't completely inside the range but are required to retain 
587     // the structure and appearance of the copied markup.
588     Node* specialCommonAncestor = 0;
589     Node* commonAncestorBlock = commonAncestor ? enclosingBlock(commonAncestor) : 0;
590     if (annotate && commonAncestorBlock) {
591         if (commonAncestorBlock->hasTagName(tbodyTag) || commonAncestorBlock->hasTagName(trTag)) {
592             Node* table = commonAncestorBlock->parentNode();
593             while (table && !table->hasTagName(tableTag))
594                 table = table->parentNode();
595             if (table)
596                 specialCommonAncestor = table;
597         } else if (commonAncestorBlock->hasTagName(listingTag)
598                     || commonAncestorBlock->hasTagName(olTag)
599                     || commonAncestorBlock->hasTagName(preTag)
600                     || commonAncestorBlock->hasTagName(tableTag)
601                     || commonAncestorBlock->hasTagName(ulTag)
602                     || commonAncestorBlock->hasTagName(xmpTag))
603             specialCommonAncestor = commonAncestorBlock;
604     }
605     
606     Node* checkAncestor = specialCommonAncestor ? specialCommonAncestor : commonAncestor;
607     if (checkAncestor->renderer()) {
608         RefPtr<CSSMutableStyleDeclaration> checkAncestorStyle = computedStyle(checkAncestor)->copyInheritableProperties();
609         if (!propertyMissingOrEqualToNone(checkAncestorStyle.get(), CSS_PROP__WEBKIT_TEXT_DECORATIONS_IN_EFFECT))
610             specialCommonAncestor = elementHasTextDecorationProperty(checkAncestor) ? checkAncestor : enclosingNodeOfType(checkAncestor, &elementHasTextDecorationProperty);
611     }
612     
613     if (Node *enclosingAnchor = enclosingNodeWithTag(specialCommonAncestor ? specialCommonAncestor : commonAncestor, aTag))
614         specialCommonAncestor = enclosingAnchor;
615     
616     Node* body = enclosingNodeWithTag(commonAncestor, bodyTag);
617     // FIXME: Only include markup for a fully selected root (and ancestors of lastClosed up to that root) if
618     // there are styles/attributes on those nodes that need to be included to preserve the appearance of the copied markup.
619     // FIXME: Do this for all fully selected blocks, not just the body.
620     Node* fullySelectedRoot = body && *Selection::selectionFromContentsOfNode(body).toRange() == *updatedRange ? body : 0;
621     if (annotate && fullySelectedRoot)
622         specialCommonAncestor = fullySelectedRoot;
623         
624     if (specialCommonAncestor) {
625         // Also include all of the ancestors of lastClosed up to this special ancestor.
626         for (Node* ancestor = lastClosed->parentNode(); ancestor; ancestor = ancestor->parentNode()) {
627             if (ancestor == fullySelectedRoot && !convertBlocksToInlines) {
628                 RefPtr<CSSMutableStyleDeclaration> style = styleFromMatchedRulesAndInlineDecl(fullySelectedRoot);
629                 
630                 // Bring the background attribute over, but not as an attribute because a background attribute on a div
631                 // appears to have no effect.
632                 if (!style->getPropertyCSSValue(CSS_PROP_BACKGROUND_IMAGE) && static_cast<Element*>(fullySelectedRoot)->hasAttribute(backgroundAttr))
633                     style->setProperty(CSS_PROP_BACKGROUND_IMAGE, "url('" + static_cast<Element*>(fullySelectedRoot)->getAttribute(backgroundAttr) + "')");
634                 
635                 if (style->length()) {
636                     markups.prepend("<div style=\"" + escapeTextForMarkup(style->cssText(), true) + "\">");
637                     markups.append("</div>");
638                 }
639             } else {
640                 markups.prepend(startMarkup(ancestor, updatedRange.get(), annotate, convertBlocksToInlines));
641                 markups.append(endMarkup(ancestor));
642             }
643             if (nodes)
644                 nodes->append(ancestor);
645             
646             lastClosed = ancestor;
647             
648             if (ancestor == specialCommonAncestor)
649                 break;
650         }
651     }
652     
653     // Add a wrapper span with the styles that all of the nodes in the markup inherit.
654     Node* parentOfLastClosed = lastClosed ? lastClosed->parentNode() : 0;
655     if (parentOfLastClosed && parentOfLastClosed->renderer()) {
656         RefPtr<CSSMutableStyleDeclaration> style = computedStyle(parentOfLastClosed)->copyInheritableProperties();
657         // Styles that Mail blockquotes contribute should only be placed on the Mail blockquote, to help
658         // us differentiate those styles from ones that the user has applied.  This helps us
659         // get the color of content pasted into blockquotes right.
660         removeEnclosingMailBlockquoteStyle(style.get(), parentOfLastClosed);
661         
662         if (style->length() > 0) {
663             DeprecatedString openTag = DeprecatedString("<span class=\"") + AppleStyleSpanClass + "\" style=\"" + escapeTextForMarkup(style->cssText(), true) + "\">";
664             markups.prepend(openTag);
665             markups.append("</span>");
666         }
667     }
668
669     // FIXME: The interchange newline should be placed in the block that it's in, not after all of the content, unconditionally.
670     if (annotate && needInterchangeNewlineAfter(visibleEnd.previous()))
671         markups.append(interchangeNewlineString);
672
673     bool selectedOneOrMoreParagraphs = startOfParagraph(visibleStart) != startOfParagraph(visibleEnd) ||
674                                        isStartOfParagraph(visibleStart) && isEndOfParagraph(visibleEnd);
675                                       
676     // Retain the Mail quote level by including all ancestor mail block quotes.
677     if (lastClosed && annotate && selectedOneOrMoreParagraphs) {
678         for (Node *ancestor = lastClosed->parentNode(); ancestor; ancestor = ancestor->parentNode()) {
679             if (isMailBlockquote(ancestor)) {
680                 markups.prepend(startMarkup(ancestor, updatedRange.get(), annotate));
681                 markups.append(endMarkup(ancestor));
682             }
683         }
684     }
685
686     if (deleteButton)
687         deleteButton->enable();
688
689     return markups.join("");
690 }
691
692 PassRefPtr<DocumentFragment> createFragmentFromMarkup(Document* document, const String& markup, const String& baseURL)
693 {
694     ASSERT(document->documentElement()->isHTMLElement());
695     // FIXME: What if the document element is not an HTML element?
696     HTMLElement *element = static_cast<HTMLElement*>(document->documentElement());
697
698     RefPtr<DocumentFragment> fragment = element->createContextualFragment(markup);
699
700     if (fragment && !baseURL.isEmpty() && baseURL != document->baseURL())
701         completeURLs(fragment.get(), baseURL);
702
703     return fragment.release();
704 }
705
706 DeprecatedString createMarkup(const Node* node, EChildrenOnly includeChildren,
707     Vector<Node*>* nodes, EAnnotateForInterchange annotate)
708 {
709     ASSERT(annotate == DoNotAnnotateForInterchange); // annotation not yet implemented for this code path
710
711     if (!node)
712         return "";
713
714     Document* document = node->document();
715     Frame* frame = document->frame();
716     DeleteButtonController* deleteButton = frame ? frame->editor()->deleteButtonController() : 0;
717
718     // disable the delete button so it's elements are not serialized into the markup
719     if (deleteButton) {
720         if (node->isDescendantOf(deleteButton->containerElement()))
721             return "";
722         deleteButton->disable();
723     }
724
725     document->updateLayoutIgnorePendingStylesheets();
726     DeprecatedString result(markup(const_cast<Node*>(node), includeChildren, nodes));
727
728     if (deleteButton)
729         deleteButton->enable();
730
731     return result;
732 }
733
734 static void fillContainerFromString(ContainerNode* paragraph, const DeprecatedString& string)
735 {
736     Document* document = paragraph->document();
737
738     ExceptionCode ec = 0;
739     if (string.isEmpty()) {
740         paragraph->appendChild(createBlockPlaceholderElement(document), ec);
741         ASSERT(ec == 0);
742         return;
743     }
744
745     ASSERT(string.find('\n') == -1);
746
747     DeprecatedStringList tabList = DeprecatedStringList::split('\t', string, true);
748     DeprecatedString tabText = "";
749     bool first = true;
750     while (!tabList.isEmpty()) {
751         DeprecatedString s = tabList.first();
752         tabList.pop_front();
753
754         // append the non-tab textual part
755         if (!s.isEmpty()) {
756             if (!tabText.isEmpty()) {
757                 paragraph->appendChild(createTabSpanElement(document, tabText), ec);
758                 ASSERT(ec == 0);
759                 tabText = "";
760             }
761             RefPtr<Node> textNode = document->createTextNode(stringWithRebalancedWhitespace(s, first, tabList.isEmpty()));
762             paragraph->appendChild(textNode.release(), ec);
763             ASSERT(ec == 0);
764         }
765
766         // there is a tab after every entry, except the last entry
767         // (if the last character is a tab, the list gets an extra empty entry)
768         if (!tabList.isEmpty())
769             tabText += '\t';
770         else if (!tabText.isEmpty()) {
771             paragraph->appendChild(createTabSpanElement(document, tabText), ec);
772             ASSERT(ec == 0);
773         }
774         
775         first = false;
776     }
777 }
778
779 PassRefPtr<DocumentFragment> createFragmentFromText(Range* context, const String& text)
780 {
781     if (!context)
782         return 0;
783
784     Node* styleNode = context->startNode();
785     if (!styleNode) {
786         styleNode = context->startPosition().node();
787         if (!styleNode)
788             return 0;
789     }
790
791     Document* document = styleNode->document();
792     RefPtr<DocumentFragment> fragment = document->createDocumentFragment();
793     
794     if (text.isEmpty())
795         return fragment.release();
796
797     DeprecatedString string = text.deprecatedString();
798     string.replace("\r\n", "\n");
799     string.replace('\r', '\n');
800
801     ExceptionCode ec = 0;
802     RenderObject* renderer = styleNode->renderer();
803     if (renderer && renderer->style()->preserveNewline()) {
804         fragment->appendChild(document->createTextNode(string), ec);
805         ASSERT(ec == 0);
806         if (string.endsWith("\n")) {
807             RefPtr<Element> element;
808             element = document->createElementNS(xhtmlNamespaceURI, "br", ec);
809             ASSERT(ec == 0);
810             element->setAttribute(classAttr, AppleInterchangeNewline);            
811             fragment->appendChild(element.release(), ec);
812             ASSERT(ec == 0);
813         }
814         return fragment.release();
815     }
816
817     // A string with no newlines gets added inline, rather than being put into a paragraph.
818     if (string.find('\n') == -1) {
819         fillContainerFromString(fragment.get(), string);
820         return fragment.release();
821     }
822
823     // Break string into paragraphs. Extra line breaks turn into empty paragraphs.
824     DeprecatedStringList list = DeprecatedStringList::split('\n', string, true); // true gets us empty strings in the list
825     while (!list.isEmpty()) {
826         DeprecatedString s = list.first();
827         list.pop_front();
828
829         RefPtr<Element> element;
830         if (s.isEmpty() && list.isEmpty()) {
831             // For last line, use the "magic BR" rather than a P.
832             element = document->createElementNS(xhtmlNamespaceURI, "br", ec);
833             ASSERT(ec == 0);
834             element->setAttribute(classAttr, AppleInterchangeNewline);            
835         } else {
836             element = createDefaultParagraphElement(document);
837             fillContainerFromString(element.get(), s);
838         }
839         fragment->appendChild(element.release(), ec);
840         ASSERT(ec == 0);
841     }
842     return fragment.release();
843 }
844
845 PassRefPtr<DocumentFragment> createFragmentFromNodes(Document *document, const Vector<Node*>& nodes)
846 {
847     if (!document)
848         return 0;
849
850     // disable the delete button so it's elements are not serialized into the markup
851     if (document->frame())
852         document->frame()->editor()->deleteButtonController()->disable();
853
854     RefPtr<DocumentFragment> fragment = document->createDocumentFragment();
855
856     ExceptionCode ec = 0;
857     size_t size = nodes.size();
858     for (size_t i = 0; i < size; ++i) {
859         RefPtr<Element> element = createDefaultParagraphElement(document);
860         element->appendChild(nodes[i], ec);
861         ASSERT(ec == 0);
862         fragment->appendChild(element.release(), ec);
863         ASSERT(ec == 0);
864     }
865
866     if (document->frame())
867         document->frame()->editor()->deleteButtonController()->enable();
868
869     return fragment.release();
870 }
871
872 }