Protect functions using two container node function, each of which can fire mutation...
[WebKit-https.git] / Source / WebCore / html / HTMLTableElement.cpp
1 /*
2  * Copyright (C) 1997 Martin Jones (mjones@kde.org)
3  *           (C) 1997 Torben Weis (weis@kde.org)
4  *           (C) 1998 Waldo Bastian (bastian@kde.org)
5  *           (C) 1999 Lars Knoll (knoll@kde.org)
6  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
7  * Copyright (C) 2003, 2004, 2005, 2006, 2008, 2010, 2011 Apple Inc. All rights reserved.
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Library General Public
11  * License as published by the Free Software Foundation; either
12  * version 2 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Library General Public License for more details.
18  *
19  * You should have received a copy of the GNU Library General Public License
20  * along with this library; see the file COPYING.LIB.  If not, write to
21  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
22  * Boston, MA 02110-1301, USA.
23  */
24
25 #include "config.h"
26 #include "HTMLTableElement.h"
27
28 #include "Attribute.h"
29 #include "CSSImageValue.h"
30 #include "CSSPropertyNames.h"
31 #include "CSSStyleSheet.h"
32 #include "CSSValueKeywords.h"
33 #include "CSSValuePool.h"
34 #include "ExceptionCode.h"
35 #include "HTMLNames.h"
36 #include "HTMLParserIdioms.h"
37 #include "HTMLTableCaptionElement.h"
38 #include "HTMLTableRowElement.h"
39 #include "HTMLTableRowsCollection.h"
40 #include "HTMLTableSectionElement.h"
41 #include "RenderTable.h"
42 #include "Text.h"
43
44 namespace WebCore {
45
46 using namespace HTMLNames;
47
48 HTMLTableElement::HTMLTableElement(const QualifiedName& tagName, Document* document)
49     : HTMLElement(tagName, document)
50     , m_borderAttr(false)
51     , m_borderColorAttr(false)
52     , m_frameAttr(false)
53     , m_rulesAttr(UnsetRules)
54     , m_padding(1)
55 {
56     ASSERT(hasTagName(tableTag));
57 }
58
59 PassRefPtr<HTMLTableElement> HTMLTableElement::create(Document* document)
60 {
61     return adoptRef(new HTMLTableElement(tableTag, document));
62 }
63
64 PassRefPtr<HTMLTableElement> HTMLTableElement::create(const QualifiedName& tagName, Document* document)
65 {
66     return adoptRef(new HTMLTableElement(tagName, document));
67 }
68
69 HTMLTableCaptionElement* HTMLTableElement::caption() const
70 {
71     for (Node* child = firstChild(); child; child = child->nextSibling()) {
72         if (child->hasTagName(captionTag))
73             return static_cast<HTMLTableCaptionElement*>(child);
74     }
75     return 0;
76 }
77
78 void HTMLTableElement::setCaption(PassRefPtr<HTMLTableCaptionElement> newCaption, ExceptionCode& ec)
79 {
80     deleteCaption();
81     insertBefore(newCaption, firstChild(), ec);
82 }
83
84 HTMLTableSectionElement* HTMLTableElement::tHead() const
85 {
86     for (Node* child = firstChild(); child; child = child->nextSibling()) {
87         if (child->hasTagName(theadTag))
88             return static_cast<HTMLTableSectionElement*>(child);
89     }
90     return 0;
91 }
92
93 void HTMLTableElement::setTHead(PassRefPtr<HTMLTableSectionElement> newHead, ExceptionCode& ec)
94 {
95     deleteTHead();
96
97     Node* child;
98     for (child = firstChild(); child; child = child->nextSibling())
99         if (child->isElementNode() && !child->hasTagName(captionTag) && !child->hasTagName(colgroupTag))
100             break;
101
102     insertBefore(newHead, child, ec);
103 }
104
105 HTMLTableSectionElement* HTMLTableElement::tFoot() const
106 {
107     for (Node* child = firstChild(); child; child = child->nextSibling()) {
108         if (child->hasTagName(tfootTag))
109             return static_cast<HTMLTableSectionElement*>(child);
110     }
111     return 0;
112 }
113
114 void HTMLTableElement::setTFoot(PassRefPtr<HTMLTableSectionElement> newFoot, ExceptionCode& ec)
115 {
116     deleteTFoot();
117
118     Node* child;
119     for (child = firstChild(); child; child = child->nextSibling())
120         if (child->isElementNode() && !child->hasTagName(captionTag) && !child->hasTagName(colgroupTag) && !child->hasTagName(theadTag))
121             break;
122
123     insertBefore(newFoot, child, ec);
124 }
125
126 PassRefPtr<HTMLElement> HTMLTableElement::createTHead()
127 {
128     if (HTMLTableSectionElement* existingHead = tHead())
129         return existingHead;
130     RefPtr<HTMLTableSectionElement> head = HTMLTableSectionElement::create(theadTag, document());
131     ExceptionCode ec;
132     setTHead(head, ec);
133     return head.release();
134 }
135
136 void HTMLTableElement::deleteTHead()
137 {
138     ExceptionCode ec;
139     removeChild(tHead(), ec);
140 }
141
142 PassRefPtr<HTMLElement> HTMLTableElement::createTFoot()
143 {
144     if (HTMLTableSectionElement* existingFoot = tFoot())
145         return existingFoot;
146     RefPtr<HTMLTableSectionElement> foot = HTMLTableSectionElement::create(tfootTag, document());
147     ExceptionCode ec;
148     setTFoot(foot, ec);
149     return foot.release();
150 }
151
152 void HTMLTableElement::deleteTFoot()
153 {
154     ExceptionCode ec;
155     removeChild(tFoot(), ec);
156 }
157
158 PassRefPtr<HTMLElement> HTMLTableElement::createCaption()
159 {
160     if (HTMLTableCaptionElement* existingCaption = caption())
161         return existingCaption;
162     RefPtr<HTMLTableCaptionElement> caption = HTMLTableCaptionElement::create(captionTag, document());
163     ExceptionCode ec;
164     setCaption(caption, ec);
165     return caption.release();
166 }
167
168 void HTMLTableElement::deleteCaption()
169 {
170     ExceptionCode ec;
171     removeChild(caption(), ec);
172 }
173
174 HTMLTableSectionElement* HTMLTableElement::lastBody() const
175 {
176     for (Node* child = lastChild(); child; child = child->previousSibling()) {
177         if (child->hasTagName(tbodyTag))
178             return static_cast<HTMLTableSectionElement*>(child);
179     }
180     return 0;
181 }
182
183 PassRefPtr<HTMLElement> HTMLTableElement::insertRow(int index, ExceptionCode& ec)
184 {
185     if (index < -1) {
186         ec = INDEX_SIZE_ERR;
187         return 0;
188     }
189
190     RefPtr<Node> protectFromMutationEvents(this);
191
192     RefPtr<HTMLTableRowElement> lastRow = 0;
193     RefPtr<HTMLTableRowElement> row = 0;
194     if (index == -1)
195         lastRow = HTMLTableRowsCollection::lastRow(this);
196     else {
197         for (int i = 0; i <= index; ++i) {
198             row = HTMLTableRowsCollection::rowAfter(this, lastRow.get());
199             if (!row) {
200                 if (i != index) {
201                     ec = INDEX_SIZE_ERR;
202                     return 0;
203                 }
204                 break;
205             }
206             lastRow = row;
207         }
208     }
209
210     RefPtr<ContainerNode> parent;
211     if (lastRow)
212         parent = row ? row->parentNode() : lastRow->parentNode();
213     else {
214         parent = lastBody();
215         if (!parent) {
216             RefPtr<HTMLTableSectionElement> newBody = HTMLTableSectionElement::create(tbodyTag, document());
217             RefPtr<HTMLTableRowElement> newRow = HTMLTableRowElement::create(document());
218             newBody->appendChild(newRow, ec);
219             appendChild(newBody.release(), ec);
220             return newRow.release();
221         }
222     }
223
224     RefPtr<HTMLTableRowElement> newRow = HTMLTableRowElement::create(document());
225     parent->insertBefore(newRow, row.get(), ec);
226     return newRow.release();
227 }
228
229 void HTMLTableElement::deleteRow(int index, ExceptionCode& ec)
230 {
231     HTMLTableRowElement* row = 0;
232     if (index == -1)
233         row = HTMLTableRowsCollection::lastRow(this);
234     else {
235         for (int i = 0; i <= index; ++i) {
236             row = HTMLTableRowsCollection::rowAfter(this, row);
237             if (!row)
238                 break;
239         }
240     }
241     if (!row) {
242         ec = INDEX_SIZE_ERR;
243         return;
244     }
245     row->remove(ec);
246 }
247
248 static inline bool isTableCellAncestor(Node* n)
249 {
250     return n->hasTagName(theadTag) || n->hasTagName(tbodyTag) ||
251            n->hasTagName(tfootTag) || n->hasTagName(trTag) ||
252            n->hasTagName(thTag);
253 }
254
255 static bool setTableCellsChanged(Node* n)
256 {
257     ASSERT(n);
258     bool cellChanged = false;
259
260     if (n->hasTagName(tdTag))
261         cellChanged = true;
262     else if (isTableCellAncestor(n)) {
263         for (Node* child = n->firstChild(); child; child = child->nextSibling())
264             cellChanged |= setTableCellsChanged(child);
265     }
266
267     if (cellChanged)
268        n->setNeedsStyleRecalc();
269
270     return cellChanged;
271 }
272
273 static bool getBordersFromFrameAttributeValue(const AtomicString& value, bool& borderTop, bool& borderRight, bool& borderBottom, bool& borderLeft)
274 {
275     borderTop = false;
276     borderRight = false;
277     borderBottom = false;
278     borderLeft = false;
279
280     if (equalIgnoringCase(value, "above"))
281         borderTop = true;
282     else if (equalIgnoringCase(value, "below"))
283         borderBottom = true;
284     else if (equalIgnoringCase(value, "hsides"))
285         borderTop = borderBottom = true;
286     else if (equalIgnoringCase(value, "vsides"))
287         borderLeft = borderRight = true;
288     else if (equalIgnoringCase(value, "lhs"))
289         borderLeft = true;
290     else if (equalIgnoringCase(value, "rhs"))
291         borderRight = true;
292     else if (equalIgnoringCase(value, "box") || equalIgnoringCase(value, "border"))
293         borderTop = borderBottom = borderLeft = borderRight = true;
294     else if (!equalIgnoringCase(value, "void"))
295         return false;
296     return true;
297 }
298
299 void HTMLTableElement::collectStyleForAttribute(Attribute* attr, StylePropertySet* style)
300 {
301     if (attr->name() == widthAttr)
302         addHTMLLengthToStyle(style, CSSPropertyWidth, attr->value());
303     else if (attr->name() == heightAttr)
304         addHTMLLengthToStyle(style, CSSPropertyHeight, attr->value());
305     else if (attr->name() == borderAttr) {
306         int borderWidth = attr->isEmpty() ? 1 : attr->value().toInt();
307         addPropertyToAttributeStyle(style, CSSPropertyBorderWidth, borderWidth, CSSPrimitiveValue::CSS_PX);
308     } else if (attr->name() == bordercolorAttr) {
309         if (!attr->isEmpty())
310             addHTMLColorToStyle(style, CSSPropertyBorderColor, attr->value());
311     } else if (attr->name() == bgcolorAttr)
312         addHTMLColorToStyle(style, CSSPropertyBackgroundColor, attr->value());
313     else if (attr->name() == backgroundAttr) {
314         String url = stripLeadingAndTrailingHTMLSpaces(attr->value());
315         if (!url.isEmpty())
316             style->setProperty(CSSProperty(CSSPropertyBackgroundImage, CSSImageValue::create(document()->completeURL(url).string())));
317     } else if (attr->name() == valignAttr) {
318         if (!attr->isEmpty())
319             addPropertyToAttributeStyle(style, CSSPropertyVerticalAlign, attr->value());
320     } else if (attr->name() == cellspacingAttr) {
321         if (!attr->isEmpty())
322             addHTMLLengthToStyle(style, CSSPropertyBorderSpacing, attr->value());
323     } else if (attr->name() == vspaceAttr) {
324         addHTMLLengthToStyle(style, CSSPropertyMarginTop, attr->value());
325         addHTMLLengthToStyle(style, CSSPropertyMarginBottom, attr->value());
326     } else if (attr->name() == hspaceAttr) {
327         addHTMLLengthToStyle(style, CSSPropertyMarginLeft, attr->value());
328         addHTMLLengthToStyle(style, CSSPropertyMarginRight, attr->value());
329     } else if (attr->name() == alignAttr) {
330         if (!attr->value().isEmpty()) {
331             if (equalIgnoringCase(attr->value(), "center")) {
332                 addPropertyToAttributeStyle(style, CSSPropertyWebkitMarginStart, CSSValueAuto);
333                 addPropertyToAttributeStyle(style, CSSPropertyWebkitMarginEnd, CSSValueAuto);
334             } else
335                 addPropertyToAttributeStyle(style, CSSPropertyFloat, attr->value());
336         }
337     } else if (attr->name() == rulesAttr) {
338         // The presence of a valid rules attribute causes border collapsing to be enabled.
339         if (m_rulesAttr != UnsetRules)
340             addPropertyToAttributeStyle(style, CSSPropertyBorderCollapse, CSSValueCollapse);
341     } else if (attr->name() == frameAttr) {
342         bool borderTop;
343         bool borderRight;
344         bool borderBottom;
345         bool borderLeft;
346         if (getBordersFromFrameAttributeValue(attr->value(), borderTop, borderRight, borderBottom, borderLeft)) {
347             addPropertyToAttributeStyle(style, CSSPropertyBorderWidth, CSSValueThin);
348             addPropertyToAttributeStyle(style, CSSPropertyBorderTopStyle, borderTop ? CSSValueSolid : CSSValueHidden);
349             addPropertyToAttributeStyle(style, CSSPropertyBorderBottomStyle, borderBottom ? CSSValueSolid : CSSValueHidden);
350             addPropertyToAttributeStyle(style, CSSPropertyBorderLeftStyle, borderLeft ? CSSValueSolid : CSSValueHidden);
351             addPropertyToAttributeStyle(style, CSSPropertyBorderRightStyle, borderRight ? CSSValueSolid : CSSValueHidden);
352         }
353     } else
354         HTMLElement::collectStyleForAttribute(attr, style);
355 }
356
357 bool HTMLTableElement::isPresentationAttribute(const QualifiedName& name) const
358 {
359     if (name == widthAttr || name == heightAttr || name == bgcolorAttr || name == backgroundAttr || name == valignAttr || name == vspaceAttr || name == hspaceAttr || name == alignAttr || name == cellspacingAttr || name == borderAttr || name == bordercolorAttr || name == frameAttr || name == rulesAttr)
360         return true;
361     return HTMLElement::isPresentationAttribute(name);
362 }
363
364 void HTMLTableElement::parseAttribute(Attribute* attr)
365 {
366     CellBorders bordersBefore = cellBorders();
367     unsigned short oldPadding = m_padding;
368
369     if (attr->name() == borderAttr)  {
370         // FIXME: This attribute is a mess.
371         m_borderAttr = true;
372         if (!attr->isNull()) {
373             int border = attr->isEmpty() ? 1 : attr->value().toInt();
374             m_borderAttr = border;
375         }
376     } else if (attr->name() == bordercolorAttr) {
377         m_borderColorAttr = !attr->isEmpty();
378     } else if (attr->name() == frameAttr) {
379         // FIXME: This attribute is a mess.
380         bool borderTop;
381         bool borderRight;
382         bool borderBottom;
383         bool borderLeft;
384         m_frameAttr = getBordersFromFrameAttributeValue(attr->value(), borderTop, borderRight, borderBottom, borderLeft);
385     } else if (attr->name() == rulesAttr) {
386         m_rulesAttr = UnsetRules;
387         if (equalIgnoringCase(attr->value(), "none"))
388             m_rulesAttr = NoneRules;
389         else if (equalIgnoringCase(attr->value(), "groups"))
390             m_rulesAttr = GroupsRules;
391         else if (equalIgnoringCase(attr->value(), "rows"))
392             m_rulesAttr = RowsRules;
393         if (equalIgnoringCase(attr->value(), "cols"))
394             m_rulesAttr = ColsRules;
395         if (equalIgnoringCase(attr->value(), "all"))
396             m_rulesAttr = AllRules;
397     } else if (attr->name() == cellpaddingAttr) {
398         if (!attr->value().isEmpty())
399             m_padding = max(0, attr->value().toInt());
400         else
401             m_padding = 1;
402     } else if (attr->name() == colsAttr) {
403         // ###
404     } else
405         HTMLElement::parseAttribute(attr);
406
407     if (bordersBefore != cellBorders() || oldPadding != m_padding) {
408         m_sharedCellStyle = 0;
409         bool cellChanged = false;
410         for (Node* child = firstChild(); child; child = child->nextSibling())
411             cellChanged |= setTableCellsChanged(child);
412         if (cellChanged)
413             setNeedsStyleRecalc();
414     }
415 }
416
417 static StylePropertySet* leakBorderStyle(int value)
418 {
419     RefPtr<StylePropertySet> style = StylePropertySet::create();
420     style->setProperty(CSSPropertyBorderTopStyle, value);
421     style->setProperty(CSSPropertyBorderBottomStyle, value);
422     style->setProperty(CSSPropertyBorderLeftStyle, value);
423     style->setProperty(CSSPropertyBorderRightStyle, value);
424     return style.release().leakRef();
425 }
426
427 StylePropertySet* HTMLTableElement::additionalAttributeStyle()
428 {
429     if ((!m_borderAttr && !m_borderColorAttr) || m_frameAttr)
430         return 0;
431
432     if (m_borderColorAttr) {
433         static StylePropertySet* solidBorderStyle = leakBorderStyle(CSSValueSolid);
434         return solidBorderStyle;
435     }
436     static StylePropertySet* outsetBorderStyle = leakBorderStyle(CSSValueOutset);
437     return outsetBorderStyle;
438 }
439
440 HTMLTableElement::CellBorders HTMLTableElement::cellBorders() const
441 {
442     switch (m_rulesAttr) {
443         case NoneRules:
444         case GroupsRules:
445             return NoBorders;
446         case AllRules:
447             return SolidBorders;
448         case ColsRules:
449             return SolidBordersColsOnly;
450         case RowsRules:
451             return SolidBordersRowsOnly;
452         case UnsetRules:
453             if (!m_borderAttr)
454                 return NoBorders;
455             if (m_borderColorAttr)
456                 return SolidBorders;
457             return InsetBorders;
458     }
459     ASSERT_NOT_REACHED();
460     return NoBorders;
461 }
462
463 PassRefPtr<StylePropertySet> HTMLTableElement::createSharedCellStyle()
464 {
465     RefPtr<StylePropertySet> style = StylePropertySet::create();
466
467     CSSValuePool* cssValuePool = document()->cssValuePool().get();
468
469     switch (cellBorders()) {
470     case SolidBordersColsOnly:
471         style->setProperty(CSSPropertyBorderLeftWidth, CSSValueThin);
472         style->setProperty(CSSPropertyBorderRightWidth, CSSValueThin);
473         style->setProperty(CSSPropertyBorderLeftStyle, CSSValueSolid);
474         style->setProperty(CSSPropertyBorderRightStyle, CSSValueSolid);
475         style->setProperty(CSSPropertyBorderColor, cssValuePool->createInheritedValue());
476         break;
477     case SolidBordersRowsOnly:
478         style->setProperty(CSSPropertyBorderTopWidth, CSSValueThin);
479         style->setProperty(CSSPropertyBorderBottomWidth, CSSValueThin);
480         style->setProperty(CSSPropertyBorderTopStyle, CSSValueSolid);
481         style->setProperty(CSSPropertyBorderBottomStyle, CSSValueSolid);
482         style->setProperty(CSSPropertyBorderColor, cssValuePool->createInheritedValue());
483         break;
484     case SolidBorders:
485         style->setProperty(CSSPropertyBorderWidth, cssValuePool->createValue(1, CSSPrimitiveValue::CSS_PX));
486         style->setProperty(CSSPropertyBorderStyle, cssValuePool->createIdentifierValue(CSSValueSolid));
487         style->setProperty(CSSPropertyBorderColor, cssValuePool->createInheritedValue());
488         break;
489     case InsetBorders:
490         style->setProperty(CSSPropertyBorderWidth, cssValuePool->createValue(1, CSSPrimitiveValue::CSS_PX));
491         style->setProperty(CSSPropertyBorderStyle, cssValuePool->createIdentifierValue(CSSValueInset));
492         style->setProperty(CSSPropertyBorderColor, cssValuePool->createInheritedValue());
493         break;
494     case NoBorders:
495         style->setProperty(CSSPropertyBorderWidth, cssValuePool->createValue(0, CSSPrimitiveValue::CSS_PX));
496         break;
497     }
498
499     if (m_padding)
500         style->setProperty(CSSPropertyPadding, cssValuePool->createValue(m_padding, CSSPrimitiveValue::CSS_PX));
501
502     return style.release();
503 }
504
505 StylePropertySet* HTMLTableElement::additionalCellStyle()
506 {
507     if (!m_sharedCellStyle)
508         m_sharedCellStyle = createSharedCellStyle();
509     return m_sharedCellStyle.get();
510 }
511
512 static StylePropertySet* leakGroupBorderStyle(int rows)
513 {
514     RefPtr<StylePropertySet> style = StylePropertySet::create();
515     if (rows) {
516         style->setProperty(CSSPropertyBorderTopWidth, CSSValueThin);
517         style->setProperty(CSSPropertyBorderBottomWidth, CSSValueThin);
518         style->setProperty(CSSPropertyBorderTopStyle, CSSValueSolid);
519         style->setProperty(CSSPropertyBorderBottomStyle, CSSValueSolid);
520     } else {
521         style->setProperty(CSSPropertyBorderLeftWidth, CSSValueThin);
522         style->setProperty(CSSPropertyBorderRightWidth, CSSValueThin);
523         style->setProperty(CSSPropertyBorderLeftStyle, CSSValueSolid);
524         style->setProperty(CSSPropertyBorderRightStyle, CSSValueSolid);
525     }
526     return style.release().leakRef();
527 }
528
529 StylePropertySet* HTMLTableElement::additionalGroupStyle(bool rows)
530 {
531     if (m_rulesAttr != GroupsRules)
532         return 0;
533
534     if (rows) {
535         static StylePropertySet* rowBorderStyle = leakGroupBorderStyle(true);
536         return rowBorderStyle;
537     }
538     static StylePropertySet* columnBorderStyle = leakGroupBorderStyle(false);
539     return columnBorderStyle;
540 }
541
542 bool HTMLTableElement::isURLAttribute(Attribute *attr) const
543 {
544     return attr->name() == backgroundAttr || HTMLElement::isURLAttribute(attr);
545 }
546
547 HTMLCollection* HTMLTableElement::rows()
548 {
549     if (!m_rowsCollection)
550         m_rowsCollection = HTMLTableRowsCollection::create(this);
551     return m_rowsCollection.get();
552 }
553
554 HTMLCollection* HTMLTableElement::tBodies()
555 {
556     return ensureCachedHTMLCollection(TableTBodies);
557 }
558
559 String HTMLTableElement::rules() const
560 {
561     return getAttribute(rulesAttr);
562 }
563
564 String HTMLTableElement::summary() const
565 {
566     return getAttribute(summaryAttr);
567 }
568
569 void HTMLTableElement::addSubresourceAttributeURLs(ListHashSet<KURL>& urls) const
570 {
571     HTMLElement::addSubresourceAttributeURLs(urls);
572
573     addSubresourceURL(urls, document()->completeURL(getAttribute(backgroundAttr)));
574 }
575
576 }