Add support for ngrok

Author: bekarice

README.md

@@ -5,9 +5,9 @@ This is a simple and opinionated plugin for helping develop/debug WooCommerce &
 
 ### Use Forwarded URLs
 
-Once activated, this plugin will make WordPress use `HTTP_X_FORWARDED_HOST` for all your links instead of your site URL. No configuration needed.
+Once activated, this plugin will make WordPress use `HTTP_X_FORWARDED_HOST` or `HTTP_X_ORIGINAL_HOST` for all your links instead of your site URL. No configuration needed.
 
-We like [ForwardHQ](https://fwd.wf) for this, along with a `ForwardFile` in your WordPress root.
+We like to use [NGrok](http://ngrok.com/) for this.
 
 ### Subscriptions (v1.5.x and v2.x.x)
 
@@ -46,6 +46,7 @@ Download and install just like any other WordPress plugin. If you want to be rea
 
 ### 1.0.0 - 2018-11-06
  * Refactor - Use namespaces and rename classes
+ * Tweak - Add support for ngrok
 
 ### 0.8.1 - 2017.12.13
  * Fix - Remove WC 3.3+ "Connect to WooCommerce" notice when official plugins are active

includes/Forwarded_URLs.php

@@ -12,10 +12,12 @@
  *
  * @package   WC-Dev-Helper/Classes
  * @author    SkyVerge
- * @copyright Copyright (c) 2015-2017, SkyVerge, Inc.
+ * @copyright Copyright (c) 2015-2018, SkyVerge, Inc.
  * @license   http://www.gnu.org/licenses/gpl-3.0.html GNU General Public License v3.0
  */
 
+namespace SkyVerge\WooCommerce\DevHelper;
+
 defined( 'ABSPATH' ) or exit;
 
 /**
@@ -26,7 +28,7 @@
  *
  * @since 0.1.0
  */
-class WC_Dev_Helper_Use_Forwarded_URLs {
+class Forwarded_URLs {
 
 
 	/** @var string non-forwarded host as defined in the siteurl option */
@@ -116,7 +118,7 @@ public function __construct() {
 	 */
 	private function has_forwarded_host() {
 
-		return array_key_exists( 'HTTP_X_FORWARDED_HOST', $_SERVER );
+		return array_key_exists( 'HTTP_X_FORWARDED_HOST', $_SERVER ) || array_key_exists( 'HTTP_X_ORIGINAL_HOST', $_SERVER );
 	}
 
 
@@ -128,7 +130,13 @@ private function has_forwarded_host() {
 	 */
 	public function get_forwarded_host() {
 
-		return $_SERVER['HTTP_X_FORWARDED_HOST'];
+		// are we using Forward HQ?
+		$host = $_SERVER['HTTP_X_FORWARDED_HOST'] ?? false;
+
+		// if not, check if we're using ngrok
+		$host = ! $host && isset( $_SERVER['HTTP_X_ORIGINAL_HOST'] ) ? $_SERVER['HTTP_X_ORIGINAL_HOST'] : $_SERVER['HTTP_HOST'];
+
+		return $host;
 	}