Ticket #6531: 6531.5b.diff

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

@@ -976 +976 @@
          *                   being absolute paths.
          */
         public function get_files( $type = null, $depth = 0, $search_parent = false ) {
-                $files = (array) self::scandir( $this->get_stylesheet_directory(), $type, $depth );
+                // get and cache all theme files to start with.
+                $label = 'list_files_cache_' . $this->get('Name') . '-' . $this->get('Version');
+                $all_files = get_transient( $label );
+                if ( empty( $all_files ) ) {
+                        $all_files = (array) self::scandir( $this->get_stylesheet_directory(), null, -1 );
 
-                if ( $search_parent && $this->parent() )
-                        $files += (array) self::scandir( $this->get_template_directory(), $type, $depth );
+                        if ( $search_parent && $this->parent() )
+                                $all_files += (array) self::scandir( $this->get_template_directory(), null, -1 );
 
+                        set_transient( $label, $all_files, HOUR_IN_SECONDS );
+                }
+
+                // Filter $all_files by $type & $depth
+                $files = array();
+                if ( $type ) {
+                        $type = (array) $type;
+                        $_extensions = implode( '|', $type );
+                }
+                foreach ($all_files as $key => $file) {
+                        if ( -1 != $depth && substr_count($key,'/') > $depth ) continue; // Filter by depth.
+                        if ( ! $type || preg_match( '~\.(' . $_extensions . ')$~', $file ) ) { // Filter by type.
+                                $files[ $key ] = $file;
+                        }
+                }
+
                 return $files;
         }