1/*
2 * Copyright (C) 2009, 2011, 2012 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "config.h"
27#include "HTMLAllCollection.h"
28
29#include "Element.h"
30#include "NodeRareData.h"
31#include <JavaScriptCore/Identifier.h>
32#include <wtf/IsoMallocInlines.h>
33#include <wtf/Optional.h>
34#include <wtf/Variant.h>
35
36namespace WebCore {
37
38WTF_MAKE_ISO_ALLOCATED_IMPL(HTMLAllNamedSubCollection);
39
40Ref<HTMLAllCollection> HTMLAllCollection::create(Document& document, CollectionType type)
41{
42 return adoptRef(*new HTMLAllCollection(document, type));
43}
44
45inline HTMLAllCollection::HTMLAllCollection(Document& document, CollectionType type)
46 : AllDescendantsCollection(document, type)
47{
48}
49
50// https://html.spec.whatwg.org/multipage/infrastructure.html#dom-htmlallcollection-item
51Optional<Variant<RefPtr<HTMLCollection>, RefPtr<Element>>> HTMLAllCollection::namedOrIndexedItemOrItems(const AtomicString& nameOrIndex) const
52{
53 if (nameOrIndex.isNull())
54 return WTF::nullopt;
55
56 if (auto index = JSC::parseIndex(*nameOrIndex.impl()))
57 return Variant<RefPtr<HTMLCollection>, RefPtr<Element>> { RefPtr<Element> { item(index.value()) } };
58
59 return namedItemOrItems(nameOrIndex);
60}
61
62// https://html.spec.whatwg.org/multipage/infrastructure.html#concept-get-all-named
63Optional<Variant<RefPtr<HTMLCollection>, RefPtr<Element>>> HTMLAllCollection::namedItemOrItems(const AtomicString& name) const
64{
65 auto namedItems = this->namedItems(name);
66
67 if (namedItems.isEmpty())
68 return WTF::nullopt;
69 if (namedItems.size() == 1)
70 return Variant<RefPtr<HTMLCollection>, RefPtr<Element>> { RefPtr<Element> { WTFMove(namedItems[0]) } };
71
72 return Variant<RefPtr<HTMLCollection>, RefPtr<Element>> { RefPtr<HTMLCollection> { downcast<Document>(ownerNode()).allFilteredByName(name) } };
73}
74
75HTMLAllNamedSubCollection::~HTMLAllNamedSubCollection()
76{
77 ownerNode().nodeLists()->removeCachedCollection(this, m_name);
78}
79
80bool HTMLAllNamedSubCollection::elementMatches(Element& element) const
81{
82 const auto& id = element.getIdAttribute();
83 if (id == m_name)
84 return true;
85
86 if (!nameShouldBeVisibleInDocumentAll(element))
87 return false;
88
89 const auto& name = element.getNameAttribute();
90 if (name == m_name)
91 return true;
92
93 return false;
94}
95
96} // namespace WebCore
97