Strip unwanted classNames from all nodes

java/org/chromium/distiller/DomUtil.java

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.distiller;
 
 import com.google.gwt.core.client.JsArray;
 import com.google.gwt.core.client.JsArrayString;
 import com.google.gwt.dom.client.AnchorElement;
 import com.google.gwt.dom.client.Document;
@@ skipping 456 matching lines @@
     }
 
     /**
      * Strips all "target" attributes from anchor nodes in the tree rooted at |rootNode|
      */
     public static void stripTargetAttributes(Node rootNode) {
         stripAttributeFromTags(rootNode, "TARGET", new String[]{"A"});
     }
 
     /**
+     * Strips unwanted classNames from all nodes in the tree rooted at |root|.

[mdjones, 2016/10/07 23:03:37]

Can you mention here and in the CL description what you mean by "unwanted class
names"? This function does something very specific.

[wychen, 2016/10/07 23:11:52]

Will do.

[wychen, 2016/10/07 23:32:20]

Done.

+     */
+    public static void stripUnwantedClassNames(Node root) {
+        if (root.getNodeType() == Node.ELEMENT_NODE) {
+            Element element = Element.as(root);
+            if (element.hasAttribute("class")) {
+                stripUnwantedClassName(element);
+            }
+        }
+        NodeList<Element> elems = DomUtil.querySelectorAll(root, "[class]");

[mdjones, 2016/10/07 23:03:37]

Why not select all elements and iterate over the resulting list?

[wychen, 2016/10/07 23:11:53]

Not all elements have class attribute, so skipping them using querySelectorAll
should be much faster.

+        for (int i = 0; i < elems.getLength(); i++) {
+            stripUnwantedClassName(elems.getItem(i));
+        }
+    }
+
+    public static void stripUnwantedClassName(Element elem) {
+        if (elem.getClassName().contains("caption")) {
+            elem.setClassName("caption");
+        } else {
+            elem.removeAttribute("class");
+        }
+    }
+
+    /**
      * Get a list of relevant nodes from a subtree.
      * @param root The root of the subtree.
      * @return A list of relevant nodes.
      */
     public static List<Node> getOutputNodes(Node root) {
         final List<Node> nodes = new ArrayList<>();
         new DomWalker(new DomWalker.Visitor() {
             @Override
             public boolean visit(Node n) {
                 switch (n.getNodeType()) {
@@ skipping 49 matching lines @@
     }-*/;
 
     public static native Document createHTMLDocument(Document doc) /*-{
         return doc.implementation.createHTMLDocument();
     }-*/;
 
     public static native Element getFirstElementChild(Document document) /*-{
         return document.firstElementChild;
     }-*/;
 }