Ticket #39862: 39862.3.diff

src/wp-includes/class-wp-walker.php

@@ -48 +48 @@
         public $has_children;
 
         /**
+         * Total children of current element.
+         *
+         * @since 4.7.2
+         * @var int
+         */
+        public $total_children;
+
+        /**
          * Starts the list before the elements are added.
          *
          * The $args parameter holds additional values that may be used with the child
@@ -141 +149 @@
                 if ( isset( $args[0] ) && is_array( $args[0] ) ) {
                         $args[0]['has_children'] = $this->has_children; // Back-compat.
                 }
+                
+                if ( $this->has_children ) {
+                        $this->total_children = count( $children_elements[ $id ] );
+                }
 
                 $cb_args = array_merge( array(&$output, $element, $depth), $args);
                 call_user_func_array(array($this, 'start_el'), $cb_args);