1 /*
2 * Copyright (C) 2010 Google Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
15 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
16 * DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY
17 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
18 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
19 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
20 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
22 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25 #include "config.h"
26 #include "ClassList.h"
27
28 #include "Element.h"
29 #include "HTMLNames.h"
30 #include "HTMLParserIdioms.h"
31 #include "SpaceSplitString.h"
32 #include <wtf/text/StringBuilder.h>
33
34 namespace WebCore {
35
36 using namespace HTMLNames;
37
ClassList(Element * element)38 ClassList::ClassList(Element* element)
39 : m_element(element)
40 {
41 if (m_element->document()->inQuirksMode())
42 m_classNamesForQuirksMode.set(m_element->fastGetAttribute(classAttr), false);
43 }
44
ref()45 void ClassList::ref()
46 {
47 m_element->ref();
48 }
49
deref()50 void ClassList::deref()
51 {
52 m_element->deref();
53 }
54
length() const55 unsigned ClassList::length() const
56 {
57 return m_element->hasClass() ? classNames().size() : 0;
58 }
59
item(unsigned index) const60 const AtomicString ClassList::item(unsigned index) const
61 {
62 if (index >= length())
63 return AtomicString();
64 return classNames()[index];
65 }
66
contains(const AtomicString & token,ExceptionCode & ec) const67 bool ClassList::contains(const AtomicString& token, ExceptionCode& ec) const
68 {
69 if (!validateToken(token, ec))
70 return false;
71 return containsInternal(token);
72 }
73
containsInternal(const AtomicString & token) const74 bool ClassList::containsInternal(const AtomicString& token) const
75 {
76 return m_element->hasClass() && classNames().contains(token);
77 }
78
add(const AtomicString & token,ExceptionCode & ec)79 void ClassList::add(const AtomicString& token, ExceptionCode& ec)
80 {
81 if (!validateToken(token, ec))
82 return;
83 addInternal(token);
84 }
85
addInternal(const AtomicString & token)86 void ClassList::addInternal(const AtomicString& token)
87 {
88 const AtomicString& oldClassName(m_element->fastGetAttribute(classAttr));
89 if (oldClassName.isEmpty())
90 m_element->setAttribute(classAttr, token);
91 else if (!containsInternal(token)) {
92 const AtomicString& newClassName(addToken(oldClassName, token));
93 m_element->setAttribute(classAttr, newClassName);
94 }
95 }
96
remove(const AtomicString & token,ExceptionCode & ec)97 void ClassList::remove(const AtomicString& token, ExceptionCode& ec)
98 {
99 if (!validateToken(token, ec))
100 return;
101 removeInternal(token);
102 }
103
removeInternal(const AtomicString & token)104 void ClassList::removeInternal(const AtomicString& token)
105 {
106 // Check using contains first since it uses AtomicString comparisons instead
107 // of character by character testing.
108 if (!containsInternal(token))
109 return;
110 const AtomicString& newClassName(removeToken(m_element->fastGetAttribute(classAttr), token));
111 m_element->setAttribute(classAttr, newClassName);
112 }
113
toggle(const AtomicString & token,ExceptionCode & ec)114 bool ClassList::toggle(const AtomicString& token, ExceptionCode& ec)
115 {
116 if (!validateToken(token, ec))
117 return false;
118
119 if (containsInternal(token)) {
120 removeInternal(token);
121 return false;
122 }
123 addInternal(token);
124 return true;
125 }
126
toString() const127 String ClassList::toString() const
128 {
129 return m_element->fastGetAttribute(classAttr);
130 }
131
reset(const String & newClassName)132 void ClassList::reset(const String& newClassName)
133 {
134 if (!m_classNamesForQuirksMode.isNull())
135 m_classNamesForQuirksMode.set(newClassName, false);
136 }
137
classNames() const138 const SpaceSplitString& ClassList::classNames() const
139 {
140 ASSERT(m_element->hasClass());
141 if (!m_classNamesForQuirksMode.isNull())
142 return m_classNamesForQuirksMode;
143 return m_element->attributeMap()->classNames();
144 }
145
146 } // namespace WebCore
147