Updated FaustWP to create a preview link for all post types. Removed two filters and added a new one to catch all post types.

Removed filters for rest_prepare_post and rest_prepare_page

Author: colinmurphy

plugins/faustwp/composer.json

@@ -34,7 +34,7 @@
     "lint": "parallel-lint -e php --no-colors --exclude vendor .",
     "phpcs": "phpcs",
     "phpcs:fix": "phpcbf",
-    "phpstan": "phpstan analyze --ansi --memory-limit=1G",
+    "phpstan": "phpstan --ansi --memory-limit=1G",
     "suite": [
       "@lint",
       "@phpcs",

plugins/faustwp/includes/replacement/callbacks.php

@@ -321,8 +321,43 @@ function enqueue_preview_scripts() {
 	);
 }
 
-add_filter( 'rest_prepare_post', __NAMESPACE__ . '\\preview_link_in_rest_response', 10, 2 );
-add_filter( 'rest_prepare_page', __NAMESPACE__ . '\\preview_link_in_rest_response', 10, 2 );
+add_filter( 'rest_api_init', __NAMESPACE__ . '\\register_preview_link_hooks_for_all_draft_post_types' );
+
+/**
+ * Registers the preview link hooks for all post types.
+ */
+function register_preview_link_hooks_for_all_draft_post_types() {
+	$post_types = get_post_types(
+		array(
+			'public' => true,
+		)
+	);
+
+	foreach ( $post_types as $post_type ) {
+		add_filter( 'rest_prepare_' . $post_type, __NAMESPACE__ . '\\preview_link_in_rest_response', 10, 2 );
+	}
+}
+
+add_filter( 'rest_post_dispatch', __NAMESPACE__ . '\\rest_post_dispatch', 10, 3 );
+
+/**
+ * Adds the preview link to rest responses.
+ *
+ * @param WP_REST_Response $response The rest response object.
+ * @param WP_Post          $post Post object.
+ *
+ * @return WP_REST_Response The rest response object.
+ */
+function rest_post_dispatch( $response, $post ) {
+
+	if ( isset( $post->post_status ) && 'draft' === $post->post_status ) {
+		$response->data['link'] = get_preview_post_link( $post->ID );
+	}
+
+	return $response;
+}
+
+
 /**
  * Adds the preview link to rest responses.
  *