Replicate API changes for template author support from WP core

WordPress · Nov 24, 2021 · a815bbc · a815bbc
1 parent 60f8a87
commit a815bbc
Show file tree

Hide file tree

Showing 5 changed files with 34 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
@@ -544,6 +544,7 @@ function _build_block_template_result_from_post( $post ) {
 		$template->status         = $post->post_status;
 		$template->has_theme_file = $has_theme_file;
 		$template->is_custom      = true;
+		$template->post_author    = $post->post_author;
 
 		if ( 'wp_template' === $post->post_type && isset( $default_template_types[ $template->slug ] ) ) {
 			$template->is_custom = false;

diff --git a/lib/compat/wordpress-5.9/class-gutenberg-rest-templates-controller.php b/lib/compat/wordpress-5.9/class-gutenberg-rest-templates-controller.php
@@ -385,6 +385,24 @@ protected function prepare_item_for_database( $request ) {
 			}
 		}
 
+		if ( ! empty( $request['author'] ) ) {
+			$post_author = (int) $request['author'];
+
+			if ( get_current_user_id() !== $post_author ) {
+				$user_obj = get_userdata( $post_author );
+
+				if ( ! $user_obj ) {
+					return new WP_Error(
+						'rest_invalid_author',
+						__( 'Invalid author ID.' ),
+						array( 'status' => 400 )
+					);
+				}
+			}
+
+			$changes->post_author = $post_author;
+		}
+
 		return $changes;
 	}
 
@@ -412,6 +430,7 @@ public function prepare_item_for_response( $template, $request ) { // phpcs:igno
 			'status'         => $template->status,
 			'wp_id'          => $template->wp_id,
 			'has_theme_file' => $template->has_theme_file,
+			'author'         => (int) $template->post_author,
 		);
 
 		if ( 'wp_template_part' === $template->type ) {
@@ -579,6 +598,11 @@ public function get_item_schema() {
 					'context'     => array( 'embed', 'view', 'edit' ),
 					'readonly'    => true,
 				),
+				'author'         => array(
+					'description' => __( 'The ID for the author of the template.' ),
+					'type'        => 'integer',
+					'context'     => array( 'view', 'edit', 'embed' ),
+				),
 			),
 		);
 

diff --git a/lib/full-site-editing/class-wp-block-template.php b/lib/full-site-editing/class-wp-block-template.php
@@ -94,4 +94,11 @@ class WP_Block_Template {
 	 * @var bool
 	 */
 	public $is_custom = true;
+
+	/**
+	 * Author.
+	 *
+	 * @var int
+	 */
+	public $post_author = 0;
 }
diff --git a/lib/full-site-editing/template-parts.php b/lib/full-site-editing/template-parts.php
@@ -54,6 +54,7 @@ function gutenberg_register_template_part_post_type() {
 			'excerpt',
 			'editor',
 			'revisions',
+			'author',
 		),
 	);
 

diff --git a/lib/full-site-editing/templates.php b/lib/full-site-editing/templates.php
@@ -55,6 +55,7 @@ function gutenberg_register_template_post_type() {
 			'excerpt',
 			'editor',
 			'revisions',
+			'author',
 		),
 	);