feat: add support for previewing drafts from WP (note: Gutenberg not …

…yet supported) WordPress/gutenberg#4555 WordPress/gutenberg#13998
wpengine · Feb 4, 2021 · 76223e6 · 76223e6
1 parent cd42993
commit 76223e6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/plugins/wpe-headless/includes/replacement/callbacks.php b/plugins/wpe-headless/includes/replacement/callbacks.php
@@ -130,7 +130,13 @@ function wpe_headless_post_preview_link( $link, $post ) {
 		/**
 		 * Add preview and preview ID back to path to support Next.js preview mode
 		 */
-		$path = trailingslashit( $path ) . 'preview/' . $preview_id;
+		if ($post->post_status !== 'publish') {
+			$path = 'draft/';
+		} else {
+			$path = $path ? trailingslashit( $path ) : '';
+		}
+
+		$path .= 'preview/' . $preview_id;
 
 		$link = add_query_arg(
 			array(