Refine the DOM element iterator implementation
[WebKit-https.git] / Source / WebCore / xml / XPathExpression.cpp
1 /*
2  * Copyright (C) 2005 Frerich Raabe <raabe@kde.org>
3  * Copyright (C) 2006, 2009 Apple Inc. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 #include "config.h"
28 #include "XPathExpression.h"
29
30 #include "Document.h"
31 #include "XPathNSResolver.h"
32 #include "XPathParser.h"
33 #include "XPathResult.h"
34 #include "XPathUtil.h"
35
36 namespace WebCore {
37
38 using namespace XPath;
39     
40 inline XPathExpression::XPathExpression(std::unique_ptr<XPath::Expression> expression)
41     : m_topExpression(WTFMove(expression))
42 {
43 }
44
45 ExceptionOr<Ref<XPathExpression>> XPathExpression::createExpression(const String& expression, RefPtr<XPathNSResolver>&& resolver)
46 {
47     auto parseResult = Parser::parseStatement(expression, WTFMove(resolver));
48     if (parseResult.hasException())
49         return parseResult.releaseException();
50
51     return adoptRef(*new XPathExpression(parseResult.releaseReturnValue()));
52 }
53
54 XPathExpression::~XPathExpression() = default;
55
56 // FIXME: Why does this take an XPathResult that it ignores?
57 ExceptionOr<Ref<XPathResult>> XPathExpression::evaluate(Node& contextNode, unsigned short type, XPathResult*)
58 {
59     if (!isValidContextNode(contextNode))
60         return Exception { NotSupportedError };
61
62     EvaluationContext& evaluationContext = Expression::evaluationContext();
63     evaluationContext.node = &contextNode;
64     evaluationContext.size = 1;
65     evaluationContext.position = 1;
66     evaluationContext.hadTypeConversionError = false;
67     auto result = XPathResult::create(contextNode.document(), m_topExpression->evaluate());
68     evaluationContext.node = nullptr; // Do not hold a reference to the context node, as this may prevent the whole document from being destroyed in time.
69
70     if (evaluationContext.hadTypeConversionError)
71         return Exception { SyntaxError };
72
73     if (type != XPathResult::ANY_TYPE) {
74         auto convertToResult = result->convertTo(type);
75         if (convertToResult.hasException())
76             return convertToResult.releaseException();
77     }
78
79     return result;
80 }
81
82 }