Add cpt logic

WordPress · Apr 17, 2019 · 90d2a2a · 90d2a2a
1 parent 830973b
commit 90d2a2a
Show file tree

Hide file tree

Showing 4 changed files with 252 additions and 99 deletions.
diff --git a/lib/class-wp-rest-widget-areas-controller.php b/lib/class-wp-rest-widget-areas-controller.php
@@ -0,0 +1,246 @@
+<?php
+
+class WP_REST_Widget_Areas extends WP_REST_Controller {
+	public function __construct() {
+		$this->namespace = '__experimental';
+		$this->rest_base = 'widget-areas';
+	}
+
+	public function register_routes() {
+		register_rest_route(
+			$this->namespace,
+			'/' . $this->rest_base,
+			array(
+				array(
+					'methods'             => WP_REST_Server::READABLE,
+					'callback'            => array( $this, 'get_items' ),
+					'permission_callback' => array( $this, 'get_items_permissions_check' ),
+				),
+				'schema' => array( $this, 'get_public_item_schema' ),
+			)
+		);
+		register_rest_route(
+			$this->namespace,
+			'/' . $this->rest_base . '/(?P<id>.+)',
+			array(
+				'args' => array(
+					'id' => array(
+						'description' => __( 'The sidebar’s ID.', 'gutenberg' ),
+						'type'        => 'string',
+					),
+				),
+				array(
+					'methods'             => WP_REST_Server::READABLE,
+					'callback'            => array( $this, 'get_item' ),
+					'permission_callback' => array( $this, 'get_item_permissions_check' ),
+				),
+				array(
+					'methods'             => WP_REST_Server::EDITABLE,
+					'callback'            => array( $this, 'update_item' ),
+					'permission_callback' => array( $this, 'update_item_permissions_check' ),
+					'args'                => $this->get_endpoint_args_for_item_schema( WP_REST_Server::EDITABLE ),
+				),
+				'schema' => array( $this, 'get_public_item_schema' ),
+			)
+		);
+	}
+
+	public function get_items_permissions_check( $request ) {
+		if ( ! current_user_can( 'edit_theme_options' ) ) {
+			return new WP_Error(
+				'rest_user_cannot_edit',
+				__( 'Sorry, you are not allowed to edit sidebars.', 'gutenberg' )
+			);
+		}
+
+		return true;
+	}
+
+	public function get_items( $request ) {
+		global $wp_registered_sidebars;
+
+		$data = array();
+
+		foreach ( array_keys( $wp_registered_sidebars ) as $sidebar_id ) {
+			$data[ $sidebar_id ] = $this->get_sidebar_data( $sidebar_id );
+		}
+
+		return rest_ensure_response( $data );
+	}
+
+	public function get_item_permissions_check( $request ) {
+		if ( ! current_user_can( 'edit_theme_options' ) ) {
+			return new WP_Error(
+				'rest_user_cannot_edit',
+				__( 'Sorry, you are not allowed to edit sidebars.', 'gutenberg' )
+			);
+		}
+
+		return true;
+	}
+
+	public function get_item( $request ) {
+		return rest_ensure_response( $this->get_sidebar_data( $request['id'] ) );
+	}
+
+	public function update_item_permissions_check( $request ) {
+		if ( ! current_user_can( 'edit_theme_options' ) ) {
+			return new WP_Error(
+				'rest_user_cannot_edit',
+				__( 'Sorry, you are not allowed to edit sidebars.', 'gutenberg' )
+			);
+		}
+
+		return true;
+	}
+
+	public function update_item( $request ) {
+		$status = $this->update_sidebar_data( $request['id'], $request );
+		if ( is_wp_error( $status ) ) {
+			return $status;
+		}
+
+		return rest_ensure_response( $this->get_sidebar_data( $request['id'] ) );
+	}
+
+	// TODO: Add schema
+
+	protected function get_sidebar_data( $sidebar_id ) {
+		global $wp_registered_sidebars;
+
+		if ( ! isset( $wp_registered_sidebars[ $sidebar_id ] ) ) {
+			return new WP_Error(
+				'rest_sidebar_invalid_id',
+				__( 'Invalid sidebar ID.', 'gutenberg' ),
+				array( 'status' => 404 )
+			);
+		}
+
+		$sidebar = $wp_registered_sidebars[ $sidebar_id ];
+		$content = '';
+		$blocks  = array();
+
+		$sidebars_items = wp_get_sidebars_widgets();
+		if ( is_numeric( $sidebars_items[ $sidebar_id ] ) ) {
+			$post    = get_post( $sidebars_items[ $sidebar_id ] );
+			$content = apply_filters( 'the_content', $post->post_content );
+		} elseif ( ! empty( $sidebars_items[ $sidebar_id ] ) ) {
+			foreach ( $sidebars_items[ $sidebar_id ] as $item ) {
+				if ( is_array( $item ) && isset( $item['blockName'] ) ) {
+					$blocks[] = $item;
+				} else {
+					$blocks[] = array(
+						'blockName' => 'core/legacy-widget',
+						'attrs'     => array(
+							'identifier' => $item,
+							'instance'   => $this->get_sidebars_widget_instance( $sidebar, $item ),
+						),
+						'innerHTML' => '',
+					);
+				}
+			}
+			$content = serialize_blocks( $blocks );
+		}
+
+		return array_merge(
+			$sidebar,
+			array( 'content' => $content )
+		);
+	}
+
+	protected function update_sidebar_data( $sidebar_id, $request ) {
+		global $wp_registered_sidebars;
+
+		if ( ! isset( $wp_registered_sidebars[ $sidebar_id ] ) ) {
+			return new WP_Error(
+				'rest_sidebar_invalid_id',
+				__( 'Invalid sidebar ID.', 'gutenberg' ),
+				array( 'status' => 404 )
+			);
+		}
+		if ( isset( $request['content'] ) && is_string( $request['content'] ) ) {
+			$sidebars = wp_get_sidebars_widgets();
+			$sidebar = $sidebars_items[ $sidebar_id ];
+			$items = array();
+			$post_id = wp_insert_post(
+				array(
+					'ID'           => is_numeric( $sidebar ) ? $sidebar : 0,
+					'post_content' => $request['content'],
+				)
+			);
+			if( ! is_numeric( $sidebar ) ) {
+				wp_set_sidebars_widgets(
+					array_merge(
+						$sidebars,
+						array(
+							$sidebar_id           => $post_id,
+							'wp_inactive_widgets' => array_merge(
+								$sidebars['wp_inactive_widgets'],
+								$sidebar
+							),
+						)
+					)
+				);
+			}
+		}
+
+		return true;
+	}
+
+	private function get_sidebars_widget_instance( $sidebar, $id ) {
+		list( $object, $number, $name ) = $this->get_widget_info( $id );
+		if ( ! $object ) {
+			return array();
+		}
+
+		$object->_set( $number );
+
+		$instances = $object->get_settings();
+		$instance  = $instances[ $number ];
+
+		$args = array_merge(
+			$sidebar,
+			array(
+				'widget_id'   => $id,
+				'widget_name' => $name,
+			)
+		);
+
+		/**
+		 * Filters the settings for a particular widget instance.
+		 *
+		 * Returning false will effectively short-circuit display of the widget.
+		 *
+		 * @since 2.8.0
+		 *
+		 * @param array     $instance The current widget instance's settings.
+		 * @param WP_Widget $this     The current widget instance.
+		 * @param array     $args     An array of default widget arguments.
+		 */
+		$instance = apply_filters( 'widget_display_callback', $instance, $object, $args );
+
+		if ( false === $instance ) {
+			return array();
+		}
+
+		return $instance;
+	}
+
+	private function get_widget_info( $id ) {
+		global $wp_registered_widgets;
+
+		if (
+			! isset( $wp_registered_widgets[ $id ]['callback'][0] ) ||
+			! isset( $wp_registered_widgets[ $id ]['params'][0]['number'] ) ||
+			! isset( $wp_registered_widgets[ $id ]['name'] ) ||
+			! ( $wp_registered_widgets[ $id ]['callback'][0] instanceof WP_Widget )
+		) {
+			return array( null, null, null );
+		}
+
+		$object = $wp_registered_widgets[ $id ]['callback'][0];
+		$number = $wp_registered_widgets[ $id ]['params'][0]['number'];
+		$name   = $wp_registered_widgets[ $id ]['name'];
+		return array( $object, $number, $name );
+	}
+}
diff --git a/lib/load.php b/lib/load.php
@@ -18,8 +18,8 @@
 	if ( ! class_exists( 'WP_REST_Widget_Updater_Controller' ) ) {
 		require dirname( __FILE__ ) . '/class-wp-rest-widget-updater-controller.php';
 	}
-	if ( ! class_exists( 'WP_REST_Sidebars_Controller' ) ) {
-		require dirname( __FILE__ ) . '/class-wp-rest-sidebars-controller.php';
+	if ( ! class_exists( 'WP_REST_Widget_Areas' ) ) {
+		require dirname( __FILE__ ) . '/class-wp-rest-widget-areas-controller.php';
 	}
 	/**
 	* End: Include for phase 2

diff --git a/lib/register.php b/lib/register.php
@@ -1,93 +0,0 @@
-<?php
-/**
- * Initialization and wp-admin integration for the Gutenberg editor plugin.
- *
- * @package gutenberg
- */
-
-if ( ! defined( 'ABSPATH' ) ) {
-	die( 'Silence is golden.' );
-}
-
-function gutenberg_output_block_widget( $options, $block ) {
-	echo $options['before_widget'];
-	echo render_block( $block );
-	echo $options['after_widget'];
-}
-
-function gutenberg_swap_out_sidebars_blocks_for_block_widgets( $sidebars_items ) {
-	global $wp_registered_widgets;
-
-	foreach ( $sidebars_items as $sidebar_id => $items ) {
-		foreach ( $items as $index => $item ) {
-			if ( ! is_array( $item ) || ! isset( $item['blockName'] ) ) {
-				continue;
-			}
-
-			$widget_id = 'block-widget-' . md5( serialize( $item ) );
-
-			$sidebars_items[ $sidebar_id ][ $index ] = $widget_id;
-
-			if ( isset( $wp_registered_widgets[ $widget_id ] ) ) {
-				continue;
-			}
-
-			wp_register_sidebar_widget(
-				$widget_id,
-				// TODO: Can we get the block's title somehow?
-				/* translators: %s: Name of the block */
-				sprintf( __( 'Block: %s', 'gutenberg' ), $item['blockName'] ),
-				'gutenberg_output_block_widget',
-				array(
-					'classname'   => 'block-widget',
-					'description' => sprintf(
-						/* translators: %s: Name of the block */
-						__( 'Displays a ‘%s’ block.', 'gutenberg' ),
-						$item['blockName']
-					),
-				),
-				$item
-			);
-		}
-	}
-
-	return $sidebars_items;
-}
-add_filter( 'sidebars_widgets', 'gutenberg_swap_out_sidebars_blocks_for_block_widgets' );
-
-function gutenberg_swap_out_sidebars_block_widgets_for_blocks( $sidebars_widgets ) {
-	global $wp_registered_widgets;
-
-	foreach ( $sidebars_widgets as $sidebar_id => $widgets ) {
-		foreach ( $widgets as $index => $widget_id ) {
-			if ( 0 !== strpos( $widget_id, 'block-widget-' ) ) {
-				continue;
-			}
-
-			if ( ! isset( $wp_registered_widgets[ $widget_id ] ) ) {
-				unset( $sidebars_widgets[ $sidebar_id ][ $index ] );
-				continue;
-			}
-
-			$block = $wp_registered_widgets[ $widget_id ]['params'][0];
-
-			$sidebars_widgets[ $sidebar_id ][ $index ] = $block;
-		}
-	}
-
-	return $sidebars_widgets;
-}
-add_filter( 'pre_update_option_sidebars_widgets', 'gutenberg_swap_out_sidebars_block_widgets_for_blocks' );
-
-function gutenberg_get_sidebars_items() {
-	remove_filter( 'sidebars_widgets', 'gutenberg_swap_out_sidebars_blocks_for_block_widgets' );
-	$sidebars_widgets = wp_get_sidebars_widgets();
-	add_filter( 'sidebars_widgets', 'gutenberg_swap_out_sidebars_blocks_for_block_widgets' );
-	return $sidebars_widgets;
-}
-
-function gutenberg_set_sidebars_items( $sidebars_items ) {
-	remove_filter( 'pre_update_option_sidebars_widgets', 'gutenberg_swap_out_sidebars_block_widgets_for_blocks' );
-	wp_set_sidebars_widgets( $sidebars_items );
-	add_filter( 'pre_update_option_sidebars_widgets', 'gutenberg_swap_out_sidebars_block_widgets_for_blocks' );
-}

diff --git a/lib/rest-api.php b/lib/rest-api.php
@@ -68,11 +68,11 @@ function gutenberg_register_rest_widget_updater_routes() {
 }
 add_action( 'rest_api_init', 'gutenberg_register_rest_widget_updater_routes' );
 
-function gutenberg_register_rest_sidebars_routes() {
-	$sidebar_controller = new WP_REST_Sidebars_Controller();
-	$sidebar_controller->register_routes();
+function gutenberg_register_rest_widget_areas() {
+	$widget_areas_controller = new WP_REST_Widget_Areas();
+	$widget_areas_controller->register_routes();
 }
-add_action( 'rest_api_init', 'gutenberg_register_rest_sidebars_routes' );
+add_action( 'rest_api_init', 'gutenberg_register_rest_widget_areas' );
 /**
  * End: Include for phase 2
  */