From f93008fbaf0bf1973f02c1ce3dee71db65880705 Mon Sep 17 00:00:00 2001
From: wpalani <abdulrahmanalani95@gmail.com>
Date: Sun, 3 Nov 2024 16:33:52 -0700
Subject: [PATCH] Redirect interceptor

---
 includes/Application.php                 |  1 +
 includes/ExternalRedirectInterceptor.php | 38 ++++++++++++++++++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 includes/ExternalRedirectInterceptor.php

diff --git a/includes/Application.php b/includes/Application.php
index 47754709d..05aecab97 100644
--- a/includes/Application.php
+++ b/includes/Application.php
@@ -71,6 +71,7 @@ public function __construct( Container $container ) {
 		if ( Permissions::is_authorized_admin() || Permissions::rest_is_authorized_admin() ) {
 			new RestAPI();
 			new WP_Admin();
+			new ExternalRedirectInterceptor();
 		}
 
 		if ( Permissions::is_authorized_admin() ) {
diff --git a/includes/ExternalRedirectInterceptor.php b/includes/ExternalRedirectInterceptor.php
new file mode 100644
index 000000000..10c052ff9
--- /dev/null
+++ b/includes/ExternalRedirectInterceptor.php
@@ -0,0 +1,38 @@
+<?php
+namespace NewfoldLabs\WP\Module\Onboarding;
+
+use NewfoldLabs\WP\Module\Onboarding\Data\Data;
+
+/**
+ * Class to intercept redirect calls and filter them.
+ *
+ * The purpose of this class is to prevent any redirects while the user is on the onboarding page.
+ * The only allowed redirect is to the brand plugin page.
+ */
+class ExternalRedirectInterceptor {
+	public function __construct() {
+		if ( ! isset( $_GET['page'] ) || WP_Admin::$slug !== \sanitize_text_field( $_GET['page'] ) ) {
+			return;
+		}
+
+		\add_filter( 'wp_redirect', array( $this, 'wp_redirect' ), 10, 1 );
+	}
+
+	/**
+	 * Intercept wp_redirect calls and filter them.
+	 *
+	 * @param string $location The location to redirect to.
+	 */
+	public function wp_redirect($location): string {
+		$runtime_data = Data::runtime();
+		$brand_plugin_url  = $runtime_data['currentBrand']['pluginDashboardPage'];
+
+		// Intercept if the redirect is going anywhere other than the brand plugin page.
+		if ( strpos( $location, $brand_plugin_url ) !== 0 ) {
+			return '';
+		}
+
+		// Allow the redirect if it's going to the brand plugin page.
+		return $location;
+	}
+}