Ticket #22924: 22924.2.diff

src/wp-admin/theme-editor.php

@@ -61 +61 @@
         wp_die( __( 'The requested theme does not exist.' ) . ' ' . $theme->errors()->get_error_message() );
 }
 
-$allowed_files = $theme->get_files( 'php', 1 );
-$has_templates = ! empty( $allowed_files );
-$style_files = $theme->get_files( 'css' );
-$allowed_files['style.css'] = $style_files['style.css'];
+$allowed_files = $style_files = array();
+$has_templates = false;
+
 /**
- * Filter the allowed files.
+ * Filter the list of file types allowed for editing in the Theme editor.
  *
  * @since 4.4.0
  *
- * @param array  $style_files List of style files.
+ * @param array  $style_files List of file types. Default types include 'php' and 'css'.
  * @param object $theme       The current Theme object.
  */
-$allowed_files += apply_filters( 'wp_theme_editor_filetypes', $style_files, $theme );
+$file_types = apply_filters( 'wp_theme_editor_filetypes', array( 'php', 'css' ), $theme );
 
+$file_types = array_merge( $file_types, array( 'php', 'css' ) );
+
+foreach ( $file_types as $type ) {
+        switch ( $type ) {
+                case 'php':
+                        $allowed_files += $theme->get_files( 'php', 1 );
+                        $has_templates = ! empty( $allowed_files );
+                        break;
+                case 'css':
+                        $style_files = $theme->get_files( 'css' );
+                        $allowed_files['style.css'] = $style_files['style.css'];
+                        $allowed_files += $style_files;
+                        break;
+                default:
+                        $allowed_files += $theme->get_files( $type );
+                        break;
+        }
+}
+
 if ( empty( $file ) ) {
         $relative_file = 'style.css';
         $file = $allowed_files['style.css'];