1   ////////////////////////////////////////////////////////////////////////////////
2   // checkstyle: Checks Java source code for adherence to a set of rules.
3   // Copyright (C) 2001-2019 the original author or authors.
4   //
5   // This library is free software; you can redistribute it and/or
6   // modify it under the terms of the GNU Lesser General Public
7   // License as published by the Free Software Foundation; either
8   // version 2.1 of the License, or (at your option) any later version.
9   //
10  // This library is distributed in the hope that it will be useful,
11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  // Lesser General Public License for more details.
14  //
15  // You should have received a copy of the GNU Lesser General Public
16  // License along with this library; if not, write to the Free Software
17  // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18  ////////////////////////////////////////////////////////////////////////////////
19  
20  package com.puppycrawl.tools.checkstyle.internal.utils;
21  
22  import java.util.List;
23  import java.util.stream.Collectors;
24  
25  import com.puppycrawl.tools.checkstyle.xpath.AbstractNode;
26  import net.sf.saxon.om.Item;
27  import net.sf.saxon.om.NodeInfo;
28  import net.sf.saxon.sxpath.XPathDynamicContext;
29  import net.sf.saxon.sxpath.XPathEvaluator;
30  import net.sf.saxon.sxpath.XPathExpression;
31  import net.sf.saxon.trans.XPathException;
32  
33  /**
34   * XpathUtil.
35   * @noinspection ClassOnlyUsedInOnePackage
36   */
37  public final class XpathUtil {
38  
39      private XpathUtil() {
40      }
41  
42      /**
43       * Returns list of nodes matching xpath expression given node context.
44       * @param xpath Xpath expression
45       * @param rootNode {@code NodeInfo} node context
46       * @return list of nodes matching xpath expression given node context
47       */
48      public static List<NodeInfo> getXpathItems(String xpath, AbstractNode rootNode)
49              throws XPathException {
50          final XPathEvaluator xpathEvaluator = new XPathEvaluator();
51          final XPathExpression xpathExpression = xpathEvaluator.createExpression(xpath);
52          final XPathDynamicContext xpathDynamicContext = xpathExpression
53                  .createDynamicContext(rootNode);
54          final List<Item<?>> items = xpathExpression.evaluate(xpathDynamicContext);
55          return items.stream().map(item -> (NodeInfo) item).collect(Collectors.toList());
56      }
57  
58  }