Add template_type guards (#36318)

props @hellofromtonya
WordPress · Nov 11, 2021 · 5a45613 · 5a45613
1 parent 7e0e9af
commit 5a45613
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/compat/wordpress-5.9/block-template-utils.php b/lib/compat/wordpress-5.9/block-template-utils.php
@@ -218,6 +218,10 @@ function _get_block_templates_paths( $base_directory ) {
 	 * @return array|null Template.
 	 */
 	function _get_block_template_file( $template_type, $slug ) {
+		if ( 'wp_template' !== $template_type && 'wp_template_part' !== $template_type ) {
+			return null;
+		}
+
 		$template_base_paths = array(
 			'wp_template'      => 'block-templates',
 			'wp_template_part' => 'block-template-parts',
@@ -264,6 +268,10 @@ function _get_block_template_file( $template_type, $slug ) {
 	 * @return array Template.
 	 */
 	function _get_block_templates_files( $template_type ) {
+		if ( 'wp_template' !== $template_type && 'wp_template_part' !== $template_type ) {
+			return null;
+		}
+
 		$template_base_paths = array(
 			'wp_template'      => 'block-templates',
 			'wp_template_part' => 'block-template-parts',