Add script and style REST APIs

lib/class-wp-rest-scripts-controller.php

@@ -0,0 +1,64 @@
+<?php
+/**
+ * Scripts controller.
+ *
+ * @package gutenberg
+ */
+
+/**
+ * Class WP_REST_Scripts_Controller
+ */
+class WP_REST_Scripts_Controller extends WP_REST_Dependencies_Controller {
+
+	/**
+	 * WP_REST_Scripts_Controller constructor.
+	 */
+	public function __construct() {
+		$this->namespace               = '__experimental';
+		$this->rest_base               = 'scripts';
+		$this->editor_block_dependency = 'editor_script';
+		$this->block_dependency        = 'script';
+		$this->object                  = wp_scripts();
+	}
+
+	/**
+	 * Helper to get Script URL.
+	 *
+	 * @param string $src Script URL.
+	 * @param string $ver Version URL.
+	 * @param string $handle Handle name.
+	 *
+	 * @return string
+	 */
+	public function get_url( $src, $ver, $handle ) {
+		if ( ! is_bool( $src ) && ! preg_match( '|^(https?:)?//|', $src ) && ! ( $this->object->content_url && 0 === strpos( $src, $this->object->content_url ) ) ) {
+			$src = $this->object->base_url . $src;
+		}
+		if ( ! empty( $ver ) ) {
+			$src = add_query_arg( 'ver', $ver, $src );
+		}
+
+		/** This filter is documented in wp-includes/class.wp-scripts.php */
+		$src = esc_url( apply_filters( 'script_loader_src', $src, $handle ) );
+
+		return esc_url( $src );
+	}
+
+	/**
+	 * Get core assets.
+	 *
+	 * @return array
+	 */
+	public function get_core_assets() {
+		if ( null === $this->core_assets ) {
+			$wp_scripts = new WP_Scripts();
+			wp_default_scripts( $wp_scripts );
+			wp_default_packages_vendor( $wp_scripts );
+			wp_default_packages_scripts( $wp_scripts );
+			$handles           = wp_list_pluck( $wp_scripts->registered, 'handle' );
+			$this->core_assets = array_values( $handles );
+		}
+
+		return $this->core_assets;
+	}
+}

lib/class-wp-rest-styles-controller.php

@@ -0,0 +1,52 @@
+<?php
+/**
+ * Styles controller.
+ *
+ * @package gutenberg
+ */
+
+/**
+ * Class WP_REST_Styles_Controller
+ */
+class WP_REST_Styles_Controller extends WP_REST_Dependencies_Controller {
+
+	/**
+	 * WP_REST_Styles_Controller constructor.
+	 */
+	public function __construct() {
+		$this->namespace               = '__experimental';
+		$this->rest_base               = 'styles';
+		$this->editor_block_dependency = 'editor_style';
+		$this->block_dependency        = 'style';
+		$this->object                  = wp_styles();
+	}
+
+	/**
+	 * Helper to get Style URL.
+	 *
+	 * @param string $src Style URL.
+	 * @param string $ver Version URL.
+	 * @param string $handle Handle name.
+	 *
+	 * @return string
+	 */
+	public function get_url( $src, $ver, $handle ) {
+		return $this->object->_css_href( $src, $ver, $handle );
+	}
+
+	/**
+	 * Get core assets.
+	 *
+	 * @return array
+	 */
+	public function get_core_assets() {
+		if ( null === $this->core_assets ) {
+			$wp_styles = new WP_Styles();
+			wp_default_styles( $wp_styles );
+			$handles           = wp_list_pluck( $wp_styles->registered, 'handle' );
+			$this->core_assets = array_values( $handles );
+		}
+
+		return $this->core_assets;
+	}
+}

lib/load.php

@@ -53,6 +53,15 @@ function gutenberg_is_experiment_enabled( $name ) {
 	if ( ! class_exists( 'WP_REST_Menu_Locations_Controller' ) ) {
 		require_once dirname( __FILE__ ) . '/class-wp-rest-menu-locations-controller.php';
 	}
+	if ( ! class_exists( 'WP_REST_Dependencies_Controller' ) ) {
+		require_once dirname( __FILE__ ) . '/class-wp-rest-dependencies-controller.php';
+	}
+	if ( ! class_exists( 'WP_REST_Scripts_Controller' ) ) {
+		require_once dirname( __FILE__ ) . '/class-wp-rest-scripts-controller.php';
+	}
+	if ( ! class_exists( 'WP_REST_Styles_Controller' ) ) {
+		require_once dirname( __FILE__ ) . '/class-wp-rest-styles-controller.php';
+	}
 	if ( ! class_exists( 'WP_Rest_Customizer_Nonces' ) ) {
 		require_once dirname( __FILE__ ) . '/class-wp-rest-customizer-nonces.php';
 	}

lib/rest-api.php

@@ -235,6 +235,20 @@ function wp_api_nav_menus_taxonomy_args( $args, $taxonomy ) {
 }
 add_filter( 'register_taxonomy_args', 'wp_api_nav_menus_taxonomy_args', 10, 2 );
 
+/**
+ * Registers the scripts and styles area REST API routes.
+ */
+function gutenberg_register_script_style() {
+	// Scripts.
+	$controller = new WP_REST_Scripts_Controller();
+	$controller->register_routes();
+
+	// Styles.
+	$controller = new WP_REST_Styles_Controller();
+	$controller->register_routes();
+}
+add_action( 'rest_api_init', 'gutenberg_register_script_style' );
+
 /**
  * Shim for get_sample_permalink() to add support for auto-draft status.
  *
@@ -355,3 +369,49 @@ function gutenberg_term_search_handler( $search_handlers ) {
 	return $search_handlers;
 }
 add_filter( 'wp_rest_search_handlers', 'gutenberg_term_search_handler', 10, 5 );
+
+/**
+ * Adds style/script links to Block Types API endpoint response.
+ *
+ * @param \WP_REST_Response $response   API response.
+ * @param \WP_Block_Type    $block_type Block type.
+ *
+ * @return \WP_REST_Response Modified API response.
+ */
+function gutenberg_add_assets_links_to_block_type( $response, $block_type ) {
+	$links   = array();
+	$scripts = array( 'editor_script', 'script' );
+	foreach ( $scripts as $script ) {
+		if ( ! isset( $block_type->$script ) ) {
+			continue;
+		}
+		$expected_handle = $block_type->$script;
+		if ( wp_script_is( $expected_handle, 'registered' ) ) {
+			$links[ 'https://api.w.org/' . $script ] = array(
+				'href'       => rest_url( sprintf( '%s/%s/%s', '__experimental', 'scripts', $expected_handle ) ),
+				'embeddable' => true,
+			);
+		}
+	}
+
+	$styles = array( 'editor_style', 'style' );
+	foreach ( $styles as $style ) {
+		if ( ! isset( $block_type->$style ) ) {
+			continue;
+		}
+		$expected_handle = $block_type->$style;
+		if ( wp_style_is( $expected_handle, 'registered' ) ) {
+			$links[ 'https://api.w.org/' . $style ] = array(
+				'href'       => rest_url( sprintf( '%s/%s/%s', '__experimental', 'styles', $expected_handle ) ),
+				'embeddable' => true,
+			);
+		}
+	}
+
+	if ( ! empty( $links ) ) {
+		$response->add_links( $links );
+	}
+
+	return $response;
+}
+add_filter( 'rest_prepare_block_type', 'gutenberg_add_assets_links_to_block_type', 10, 2 );

phpunit/class-rest-block-types-controller-test.php

@@ -0,0 +1,150 @@
+<?php
+/**
+ * REST API: REST_Block_Types_Controller_Test class
+ *
+ * @package    WordPress
+ * @subpackage REST_API
+ */
+
+/**
+ * Tests for REST API for Widgets.
+ *
+ * @see WP_Test_REST_Controller_Testcase
+ */
+class REST_Block_Types_Controller_Test extends WP_Test_REST_TestCase {
+	/**
+	 * @var int
+	 */
+	protected static $admin_id;
+
+	/**
+	 * Create fake data before our tests run.
+	 *
+	 * @param WP_UnitTest_Factory $factory Helper that lets us create fake data.
+	 */
+	public static function wpSetUpBeforeClass( $factory ) {
+		self::$admin_id = $factory->user->create(
+			array(
+				'role' => 'administrator',
+			)
+		);
+		wp_register_script( 'core-assets-test-script', home_url( '/assets/test-script.js' ) );
+		wp_register_script( 'core-assets-test-editor-script', home_url( '/assets/test-editor-script.js' ) );
+		wp_register_style( 'core-assets-test-style', home_url( '/assets/test-style.css' ) );
+		wp_register_style( 'core-assets-test-editor-style', home_url( '/assets/test-editor-style.css' ) );
+	}
+
+	/**
+	 * Tear down after class tests are done.
+	 */
+	public static function wpTearDownAfterClass() {
+		self::delete_user( self::$admin_id );
+		unregister_block_type( 'core/assets-test' );
+	}
+
+	/**
+	 * Test whether test block type contains scripts/styles links.
+	 *
+	 * @dataProvider dataProviderTestCases
+	 */
+	public function test_links_in_block( $block_name, $block_parameters ) {
+		wp_set_current_user( self::$admin_id );
+		register_block_type( $block_name, $block_parameters );
+		$endpoint = sprintf( '/wp/v2/block-types/%s', $block_name );
+		$request  = new WP_REST_Request( 'GET', $endpoint );
+		$response = rest_get_server()->dispatch( $request );
+		$links    = $response->get_links();
+		$this->assertEquals( 200, $response->get_status() );
+		foreach ( $block_parameters as $block_parameter => $parameter_value ) {
+			$links_key = sprintf( 'https://api.w.org/%s', $block_parameter );
+			if ( empty( $parameter_value ) ) {
+				$this->assertArrayNotHasKey( $links_key, $links );
+			} else {
+				$this->assertArrayHasKey( $links_key, $links );
+				$this->assertNotEmpty( $links[ $links_key ] );
+			}
+		}
+	}
+
+	/**
+	 * Provide test cases.
+	 *
+	 * @return array[]
+	 */
+	public function dataProviderTestCases() {
+		return array(
+			array(
+				'core/assets-test-1',
+				array(
+					'script'        => '',
+					'style'         => '',
+					'editor_script' => '',
+					'editor_style'  => '',
+				),
+			),
+			array(
+				'core/assets-test-2',
+				array(
+					'script'        => 'core-assets-test-script',
+					'style'         => '',
+					'editor_script' => '',
+					'editor_style'  => '',
+				),
+			),
+			array(
+				'core/assets-test-3',
+				array(
+					'script'        => '',
+					'style'         => 'core-assets-test-style',
+					'editor_script' => '',
+					'editor_style'  => '',
+				),
+			),
+			array(
+				'core/assets-test-4',
+				array(
+					'script'        => '',
+					'style'         => '',
+					'editor_script' => 'core-assets-test-editor-script',
+					'editor_style'  => '',
+				),
+			),
+			array(
+				'core/assets-test-5',
+				array(
+					'script'        => '',
+					'style'         => '',
+					'editor_script' => '',
+					'editor_style'  => 'core-assets-test-editor-style',
+				),
+			),
+			array(
+				'core/assets-test-6',
+				array(
+					'script'        => '',
+					'style'         => 'core-assets-test-style',
+					'editor_script' => '',
+					'editor_style'  => 'core-assets-test-editor-style',
+				),
+			),
+			array(
+				'core/assets-test-7',
+				array(
+					'script'        => 'core-assets-test-script',
+					'style'         => 'core-assets-test-style',
+					'editor_script' => 'core-assets-test-editor-script',
+					'editor_style'  => 'core-assets-test-editor-style',
+				),
+			),
+			array(
+				'core/assets-test-8',
+				array(
+					'script'        => 'core-assets-test-script',
+					'style'         => '',
+					'editor_script' => 'core-assets-test-editor-script',
+					'editor_style'  => 'core-assets-test-editor-style',
+				),
+			),
+		);
+	}
+}