Only display featured image UI when theme supports it too (#6541)

* Only display featured image UI when theme supports it too

* JSX formatting style

editor/components/post-featured-image/check.js

@@ -2,9 +2,14 @@
  * Internal dependencies
  */
 import PostTypeSupportCheck from '../post-type-support-check';
+import ThemeSupportCheck from '../theme-support-check';
 
 function PostFeaturedImageCheck( props ) {
-	return <PostTypeSupportCheck { ...props } supportKeys="thumbnail" />;
+	return (
+		<ThemeSupportCheck supportKeys="post-thumbnails">
+			<PostTypeSupportCheck { ...props } supportKeys="thumbnail" />
+		</ThemeSupportCheck>
+	);
 }
 
 export default PostFeaturedImageCheck;

editor/components/theme-support-check/index.js

@@ -0,0 +1,28 @@
+/**
+ * External dependencies
+ */
+import { get, some, castArray } from 'lodash';
+
+/**
+ * WordPress dependencies
+ */
+import { withSelect } from '@wordpress/data';
+
+export function ThemeSupportCheck( { themeSupports, children, supportKeys } ) {
+	const isSupported = some(
+		castArray( supportKeys ), ( key ) => get( themeSupports, [ key ], false )
+	);
+
+	if ( ! isSupported ) {
+		return null;
+	}
+
+	return children;
+}
+
+export default withSelect( ( select ) => {
+	const { getThemeSupports } = select( 'core' );
+	return {
+		themeSupports: getThemeSupports(),
+	};
+} )( ThemeSupportCheck );

editor/components/theme-support-check/test/index.js

@@ -0,0 +1,38 @@
+/**
+ * External dependencies
+ */
+import { shallow } from 'enzyme';
+
+/**
+ * Internal dependencies
+ */
+import { ThemeSupportCheck } from '../index';
+
+describe( 'ThemeSupportCheck', () => {
+	it( 'should not render if there\'s no support check provided', () => {
+		const wrapper = shallow( <ThemeSupportCheck>foobar</ThemeSupportCheck> );
+		expect( wrapper.type() ).toBe( null );
+	} );
+
+	it( 'should render if post-thumbnails are supported', () => {
+		const themeSupports = {
+			'post-thumbnails': true,
+		};
+		const supportKeys = 'post-thumbnails';
+		const wrapper = shallow( <ThemeSupportCheck
+			supportKeys={ supportKeys }
+			themeSupports={ themeSupports }>foobar</ThemeSupportCheck> );
+		expect( wrapper.type() ).not.toBe( null );
+	} );
+
+	it( 'should not render if theme doesn\'t support post-thumbnails', () => {
+		const themeSupports = {
+			'post-thumbnails': false,
+		};
+		const supportKeys = 'post-thumbnails';
+		const wrapper = shallow( <ThemeSupportCheck
+			supportKeys={ supportKeys }
+			themeSupports={ themeSupports }>foobar</ThemeSupportCheck> );
+		expect( wrapper.type() ).toBe( null );
+	} );
+} );

lib/rest-api.php

@@ -370,6 +370,11 @@ function gutenberg_ensure_wp_json_has_theme_supports( $response ) {
 
 		$site_info['theme_supports']['formats'] = $formats;
 	}
+	if ( ! array_key_exists( 'post-thumbnails', $site_info['theme_supports'] ) ) {
+		if ( get_theme_support( 'post-thumbnails' ) ) {
+			$site_info['theme_supports']['post-thumbnails'] = true;
+		}
+	}
 	$response->set_data( $site_info );
 	return $response;
 }