Add debug constant check

Author: DavidCramer

php/class-string-replace.php

@@ -34,13 +34,64 @@ class String_Replace {
 	public function __construct( Plugin $plugin ) {
 		$this->plugin = $plugin;
 		add_action( 'template_redirect', array( $this, 'init' ), 1 );
+		add_action( 'template_include', array( $this, 'init_debug' ), 1 );
+		$types = get_post_types_by_support( 'editor' );
+		foreach ( $types as $type ) {
+			$post_type = get_post_type_object( $type );
+			// Check if this is a rest supported type.
+			if ( true === $post_type->show_in_rest ) {
+				// Add filter only to rest supported types.
+				add_filter( 'rest_prepare_' . $type, array( $this, 'pre_filter_rest_content' ), 10, 3 );
+			}
+		}
+	}
+
+	/**
+	 * Filter out local urls in an 'edit' context rest request ( i.e for Gutenburg ).
+	 *
+	 * @param \WP_REST_Response $response The post data array to save.
+	 * @param \WP_Post          $post     The current post.
+	 * @param \WP_REST_Request  $request  The request object.
+	 *
+	 * @return \WP_REST_Response
+	 */
+	public function pre_filter_rest_content( $response, $post, $request ) {
+		$context = $request->get_param( 'context' );
+		if ( 'view' === $context ) {
+			$data                        = $response->get_data();
+			$data['content']['rendered'] = $this->replace_strings( $data['content']['rendered'] );
+			$response->set_data( $data );
+		}
+
+		return $response;
 	}
 
 	/**
 	 * Init the buffer capture and set the output callback.
 	 */
 	public function init() {
-		ob_start( array( $this, 'replace_strings' ) );
+		if ( ! defined( 'WP_DEBUG' ) || false === WP_DEBUG ) {
+			ob_start( array( $this, 'replace_strings' ) );
+		}
+	}
+
+	/**
+	 * Init the buffer capture in debug mode.
+	 *
+	 * @param string $template The template being loaded.
+	 *
+	 * @return null|string
+	 */
+	public function init_debug( $template ) {
+		if ( defined( 'WP_DEBUG' ) && true === WP_DEBUG ) {
+			ob_start();
+			include $template;
+			$html = ob_get_clean();
+			echo $this->replace_strings( $html ); // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped
+			$template = null;
+		}
+
+		return $template;
 	}
 
 	/**
@@ -50,10 +101,21 @@ public function init() {
 	 *
 	 * @return bool
 	 */
-	public function string_set( $string ) {
+	public static function string_set( $string ) {
 		return isset( self::$replacements[ $string ] );
 	}
 
+	/**
+	 * Check if a string is not set for replacement.
+	 *
+	 * @param string $string String to check.
+	 *
+	 * @return bool
+	 */
+	public static function string_not_set( $string ) {
+		return ! self::string_set( $string );
+	}
+
 	/**
 	 * Replace a string.
 	 *