Add block data to REST API

lib/blocks.php

@@ -93,3 +93,104 @@ function do_blocks( $content ) {
 	return $content_after_blocks;
 }
 add_filter( 'the_content', 'do_blocks', 9 ); // BEFORE do_shortcode() and wpautop().
+
+/**
+ * Extract the blocks from post content for the REST API post response.
+ *
+ * @since 1.1.0
+ *
+ * @param string $content The post content.
+ *
+ * @return array Array of block data.
+ */
+function gutenberg_add_blocks_to_post_resource( $content ) {
+	$registry = WP_Block_Type_Registry::get_instance();
+	$blocks   = gutenberg_parse_blocks( $content );
+	$data     = array();
+
+	// Loop thru the blocks, adding rendered content when available.
+	foreach ( $blocks as $block ) {
+		$block_name  = isset( $block['blockName'] ) ? $block['blockName'] : null;
+		$attributes  = is_array( $block['attrs'] ) ? $block['attrs'] : null;
+		$raw_content = isset( $block['rawContent'] ) ? $block['rawContent'] : null;
+
+		// Skip block if we didn’t get a valid block name.
+		if ( null === $block_name ) {
+			continue;
+		}
+
+		// Set up rendered content, if available.
+		$block['renderedContent'] = null;
+		$block_type = $registry->get_registered( $block_name );
+		if ( null !== $block_type ) {
+			$block['renderedContent'] = $block_type->render( $attributes, $raw_content );
+		}
+
+		// Add the item data.
+		$data[] = array(
+			'type'       => $block_name,
+			'attributes' => $attributes,
+			'content'    => $block['rawContent'],
+			'rendered'   => $block['renderedContent'],
+		);
+	}
+
+	return $data;
+}
+
+/**
+ * Attach a post's block data callback to the REST API response.
+ *
+ * @since 1.1.0
+ *
+ * @param string | array $post_types Post type, or array of post types.
+ */
+function gutenberg_attach_block_response_callback( $post_types ) {
+	if ( empty( $post_types ) ) {
+		$post_types = 'post';
+	}
+	if ( ! is_array( $post_types ) ) {
+		$post_types = array( $post_types );
+	}
+	foreach ( $post_types as $post_type ) {
+
+		/**
+		 * Filter whether a post type has its blocks data added the REST API response content.
+		 *
+		 * @since 1.1.0
+		 *
+		 * @param bool   $blocks_show_in_rest Whether to show blocks in the REST API response.
+		 * @param string $post_type           The post type.
+		 */
+		if ( apply_filters( 'gutenberg_add_blocks_to_rest_for_post_type', true, $post_type ) ) {
+			add_filter( 'rest_prepare_' . $post_type, 'gutenberg_extract_blocks_from_post_content', 10, 3 );
+		}
+	}
+}
+gutenberg_attach_block_response_callback( array( 'post', 'page' ) );
+
+/**
+ * Attach a post's block data to the REST API response.
+ *
+ * @since 1.1.0
+ *
+ * @param WP_REST_Response $response The response object.
+ * @param WP_Post          $post     The Post object.
+ *
+ * @return WP_REST_Response $response The filtered response object.
+ */
+function gutenberg_extract_blocks_from_post_content( $response, $post ) {
+	if ( ! $post ) {
+		return $response;
+	}
+
+	// Extract the block data from the post content.
+	$blocks = gutenberg_add_blocks_to_post_resource( $post->post_content );
+
+	// Add block data to the response as part of 'content'.
+	$content           = $response->get_data( 'content' );
+	$content['blocks'] = $blocks;
+	$response->set_data( array( 'content' => $content ) );
+
+	return $response;
+}

phpunit/class-rest-blocks-api-test.php

@@ -0,0 +1,95 @@
+<?php
+/**
+ * WP_REST_BlockAPI tests.
+ *
+ * @package gutenberg
+ */
+
+/**
+ * Tests for WP_REST_BlockAPI.
+ */
+class WP_REST_BlockAPI_Test extends WP_UnitTestCase {
+
+	/**
+	 * Our fake reusable block's post ID.
+	 *
+	 * @var int
+	 */
+	protected static $test_block_post_id;
+
+	/**
+	 * Our fake reusable block's post ID.
+	 *
+	 * @var int
+	 */
+	protected static $demo_post_content;
+
+	/**
+	 * Our fake editor's user ID.
+	 *
+	 * @var int
+	 */
+	protected static $editor_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 ) {
+		global $wp_rest_server;
+		$wp_rest_server = new WP_REST_Server;
+		do_action( 'rest_api_init' );
+
+		self::$demo_post_content = file_get_contents(
+			dirname( __FILE__ ) . '/fixtures/long-content.html'
+		);
+
+		self::$test_block_post_id = wp_insert_post( array(
+			'post_type' => 'post',
+			'post_status' => 'publish',
+			'post_name' => 'gutenberg-block-test',
+			'post_title' => 'My cool block',
+			'post_content' => self::$demo_post_content,
+		) );
+
+		self::$editor_id = $factory->user->create( array(
+			'role' => 'editor',
+		) );
+	}
+
+	/**
+	 * Delete our fake data after our tests run.
+	 */
+	public static function wpTearDownAfterClass() {
+		wp_delete_post( self::$test_block_post_id );
+
+		self::delete_user( self::$editor_id );
+	}
+
+	/**
+	 * Check that we can GET blocks from a post request.
+	 */
+	public function test_get_items() {
+		global $wp_rest_server;
+		wp_set_current_user( self::$editor_id );
+
+		$request = new WP_REST_Request( 'GET', '/wp/v2/posts/' . self::$test_block_post_id );
+		$this->assertNotEquals( null, $request );
+
+		$response = $wp_rest_server->dispatch( $request );
+
+		$this->assertEquals( 200, $response->get_status() );
+		$response->get_data();
+
+		$blocks = isset( $response->data['content']['blocks'] ) ? $response->data['content']['blocks'] : false;
+
+		$first_block_url_attribute = wp_unslash( $blocks[0]['attributes']['url'] );
+		$first_block_type = wp_unslash( $blocks[0]['type'] );
+
+		$this->assertEquals( count( $blocks ), 43, 'The demo post content api call should contain 43 blocks.' );
+		$this->assertEquals( 'https://cldup.com/GCwahb3aOb.jpg', $first_block_url_attribute, 'Block attributes should be returned correctly.' );
+		$this->assertEquals( 'core/cover-image', $first_block_type, 'Block type should be returned correctly.' );
+	}
+
+}