/*
    * Copyright (c) 2002-2025 Gargoyle Software Inc.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    * https://www.apache.org/licenses/LICENSE-2.0
    *
    * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.htmlunit.html.xpath;
  
  import java.util.ArrayList;
  import java.util.List;
  
  import javax.xml.transform.TransformerException;
  
  import org.htmlunit.html.DomNode;
  import org.htmlunit.xpath.XPathContext;
  import org.htmlunit.xpath.objects.XBoolean;
  import org.htmlunit.xpath.objects.XNodeSet;
  import org.htmlunit.xpath.objects.XNumber;
  import org.htmlunit.xpath.objects.XObject;
  import org.htmlunit.xpath.objects.XString;
  import org.htmlunit.xpath.xml.utils.PrefixResolver;
  import org.w3c.dom.Document;
  import org.w3c.dom.Node;
  import org.w3c.dom.NodeList;
  
  /**
   * Collection of XPath utility methods.
   *
   * @author Ahmed Ashour
   * @author Chuck Dumont
   * @author Ronald Brill
   */
  public final class XPathHelper {
  
      private static final ThreadLocal<Boolean> PROCESS_XPATH_ = new ThreadLocal<Boolean>() {
          @Override
          protected synchronized Boolean initialValue() {
              return Boolean.FALSE;
          }
      };
  
      /**
       * Private to avoid instantiation.
       */
      private XPathHelper() {
          // Empty.
      }
  
      /**
       * Evaluates an XPath expression from the specified node, returning the resultant nodes.
       *
       * @param <T> the type class
       * @param contextNode the node to start searching from
       * @param xpathExpr the XPath expression
       * @param prefixResolver the prefix resolver to use for resolving namespace prefixes, or null
       * @return the list of objects found
       */
      public static <T> List<T> getByXPath(final DomNode contextNode, final String xpathExpr,
              final PrefixResolver prefixResolver) {
          if (xpathExpr == null) {
              throw new IllegalArgumentException("Null is not a valid XPath expression");
          }
  
          PrefixResolver resolver = prefixResolver;
          if (resolver == null) {
              final Node xpathExpressionContext;
              if (contextNode.getNodeType() == Node.DOCUMENT_NODE) {
                  xpathExpressionContext = ((Document) contextNode).getDocumentElement();
              }
              else {
                  xpathExpressionContext = contextNode;
              }
  
              resolver = new HtmlUnitPrefixResolver(xpathExpressionContext);
          }
  
          try {
              final boolean caseSensitive = contextNode.getPage().hasCaseSensitiveTagNames();
              final XPathAdapter xpath = new XPathAdapter(xpathExpr, resolver, caseSensitive);
              return getByXPath(contextNode, xpath, prefixResolver);
          }
          catch (final Exception e) {
              throw new RuntimeException("Could not retrieve XPath >" + xpathExpr + "< on " + contextNode, e);
          }
      }
  
      /**
       * @param <T> the type of nodes expected
       * @param node the start node
       * @param xpath the {@link XPathAdapter} to search for
       * @param prefixResolver the {@link PrefixResolver} to be used
      * @return a list of nodes matching the given xpath
      * @throws TransformerException in case of error
      */
     public static <T> List<T> getByXPath(final Node node, final XPathAdapter xpath,
             final PrefixResolver prefixResolver) throws TransformerException {
         final List<T> list = new ArrayList<>();
 
         PROCESS_XPATH_.set(Boolean.TRUE);
         try {
             final XPathContext xpathSupport = new XPathContext();
             final int ctxtNode = xpathSupport.getDTMHandleFromNode(node);
             final XObject result = xpath.execute(xpathSupport, ctxtNode, prefixResolver);
 
             if (result instanceof XNodeSet) {
                 final NodeList nodelist = result.nodelist();
                 final int length = nodelist.getLength();
                 for (int i = 0; i < length; i++) {
                     list.add((T) nodelist.item(i));
                 }
             }
             else if (result instanceof XNumber) {
                 list.add((T) Double.valueOf(result.num()));
             }
             else if (result instanceof XBoolean) {
                 list.add((T) Boolean.valueOf(result.bool()));
             }
             else if (result instanceof XString) {
                 list.add((T) result.str());
             }
             else {
                 throw new RuntimeException("Unproccessed " + result.getClass().getName());
             }
         }
         finally {
             PROCESS_XPATH_.set(Boolean.FALSE);
         }
 
         return list;
     }
 
     /**
      * Returns whether the thread is currently evaluating XPath expression or no.
      * @return whether the thread is currently evaluating XPath expression or no
      */
     public static boolean isProcessingXPath() {
         return PROCESS_XPATH_.get().booleanValue();
     }
 
 }