Ticket #39862: 39862.4.diff

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

@@ -47 +47 @@
         public $has_children;
 
         /**
+         * Total children of current element.
+         *
+         * @since 4.7.2
+         * @var bool
+         */
+        public $total_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
@@ -140 +156 @@
                 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 ] );
+                }
+
+                $this->total_children = 0;
+                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);
@@ -402 +427 @@
         /**
          * Unset all the children for a given top level element.
          *
-         * @since 2.7.0
-         *
          * @param object $e
          * @param array $children_elements
          */