Files
wp-multisite-waas/inc/class-domain-mapping.php
2024-11-30 18:24:12 -07:00

610 lines
14 KiB
PHP

<?php
/**
* Handles Domain Mapping in WP Ultimo.
*
* @package WP_Ultimo
* @subpackage Domain_Mapping
* @since 2.0.0
*/
namespace WP_Ultimo;
use WP_Ultimo\Models\Domain;
// Exit if accessed directly
defined('ABSPATH') || exit;
/**
* Handles Domain Mapping in WP Ultimo.
*
* @since 2.0.0
*/
class Domain_Mapping {
use \WP_Ultimo\Traits\Singleton;
/**
* Keeps a copy of the current mapping.
*
* @since 2.0.0
* @var \WP_Ultimo\Models\Domain
*/
public $current_mapping = null;
/**
* Keeps a copy of the original URL.
*
* @since 2.0.0
* @var string
*/
public $original_url = null;
/**
* Runs on singleton instantiation.
*
* @since 2.0.0
* @return void
*/
public function init() {
if ($this->should_skip_checks()) {
$this->startup();
} else {
$this->maybe_startup();
} // end if;
} // end init;
/**
* Check if we should skip checks before running mapping functions.
*
* @since 2.0.0
* @return boolean
*/
public static function should_skip_checks() {
return defined('WP_ULTIMO_DOMAIN_MAPPING_SKIP_CHECKS') && WP_ULTIMO_DOMAIN_MAPPING_SKIP_CHECKS;
} // end should_skip_checks;
/**
* Run the checks to make sure the requirements for Domain mapping are in place and execute it.
*
* @since 2.0.0
* @return void
*/
public function maybe_startup() {
/*
* Don't run during installation...
*/
if (defined('WP_INSTALLING') && $_SERVER['SCRIPT_NAME'] !== '/wp-activate.php') {
return;
} // end if;
/*
* Make sure we got loaded in the sunrise stage.
*/
if (did_action('muplugins_loaded')) {
return;
} // end if;
$is_enabled = (bool) wu_get_setting_early('enable_domain_mapping');
if ($is_enabled === false) {
return;
} // end if;
/*
* Start the engines!
*/
$this->startup();
} // end maybe_startup;
/**
* Actual handles domain mapping functionality.
*
* @since 2.0.0
* @return void
*/
public function startup() {
/*
* Adds the necessary tables to the $wpdb global.
*/
if (empty($GLOBALS['wpdb']->wu_dmtable)) {
$GLOBALS['wpdb']->wu_dmtable = $GLOBALS['wpdb']->base_prefix . 'wu_domain_mappings';
$GLOBALS['wpdb']->ms_global_tables[] = 'wu_domain_mappings';
} // end if;
// Ensure cache is shared
wp_cache_add_global_groups(array('domain_mappings', 'network_mappings'));
/*
* Check if the URL being accessed right now is a mapped domain
*/
add_filter('pre_get_site_by_path', array($this, 'check_domain_mapping'), 10, 2);
/*
* When a site gets delete, clean up the mapped domains
*/
add_action('wp_delete_site', array($this, 'clear_mappings_on_delete'));
/*
* Adds the filters that will change the URLs when a mapped domains is in use
*/
add_action('ms_loaded', array($this, 'register_mapped_filters'), 11);
// add_action('allowed_http_origin', array($this, 'add_mapped_domains_as_allowed_origins'));
/**
* On WP Ultimo 1.X builds we used Mercator. The Mercator actions and filters are now deprecated.
*/
if (has_action('mercator_load')) {
do_action_deprecated('mercator_load', array(), '2.0.0', 'wu_domain_mapping_load');
} // end if;
add_action('wu_sso_site_allowed_domains', function($list, $site_id): array {
$domains = wu_get_domains(array(
'active' => true,
'blog_id' => $site_id,
'stage__not_in' => \WP_Ultimo\Models\Domain::INACTIVE_STAGES,
'fields' => 'domain',
));
return array_merge($list, $domains);
}, 10, 2);
/**
* Fired after our core Domain Mapping has been loaded
*
* Hook into this to handle any add-on functionality.
*/
do_action('wu_domain_mapping_load');
} // end startup;
/**
* Checks if an origin is a mapped domain.
*
* If that's the case, we should always allow that origin.
*
* @since 2.0.0
*
* @param string $origin The origin passed.
* @return string
*/
public function add_mapped_domains_as_allowed_origins($origin) {
if (!function_exists('wu_get_domain_by_domain')) {
return;
} // end if;
if (empty($origin) && wp_doing_ajax()) {
$origin = wu_get_current_url();
} // end if;
$the_domain = wp_parse_url($origin, PHP_URL_HOST);
$domain = wu_get_domain_by_domain($the_domain);
if ($domain) {
return $domain->get_domain();
} // end if;
return $origin;
} // end add_mapped_domains_as_allowed_origins;
/**
* Fixes the SSO target site in cases of domain mapping.
*
* @since 2.0.0
*
* @param \WP_Site $target_site The current target site.
* @param string $domain The domain being searched.
* @return \WP_Site
*/
public function fix_sso_target_site($target_site, $domain) {
if (!$target_site || !$target_site->blog_id) {
$mapping = \WP_Ultimo\Models\Domain::get_by_domain($domain);
if ($mapping) {
$target_site = get_site($mapping->get_site_id());
} // end if;
} // end if;
return $target_site;
} // end fix_sso_target_site;
/**
* Returns both the naked and www. version of the given domain
*
* @since 2.0.0
*
* @param string $domain Domain to get the naked and www. versions to.
* @return array
*/
public function get_www_and_nowww_versions($domain) {
if (strncmp($domain, 'www.', strlen('www.')) === 0) {
$www = $domain;
$nowww = substr($domain, 4);
} else {
$nowww = $domain;
$www = 'www.' . $domain;
} // end if;
return array($nowww, $www);
} // end get_www_and_nowww_versions;
/**
* Checks if we have a site associated with the domain being accessed
*
* This method tries to find a site on the network that has a mapping related to the current
* domain being accessed. This uses the default WordPress mapping functionality, added on 4.5.
*
* @since 2.0.0
*
* @param null|false|\WP_Site $site Site object being searched by path.
* @param string $domain Domain to search for.
* @return null|false|\WP_Site
*/
public function check_domain_mapping($site, $domain) {
// Have we already matched? (Allows other plugins to match first)
if (!empty($site)) {
return $site;
} // end if;
$domains = $this->get_www_and_nowww_versions($domain);
$mapping = Domain::get_by_domain($domains);
if (empty($mapping) || is_wp_error($mapping)) {
return $site;
} // end if;
if (has_filter('mercator.use_mapping')) {
$deprecated_args = array(
$mapping->is_active(),
$mapping,
$domain
);
$is_active = apply_filters_deprecated('mercator.use_mapping', $deprecated_args, '2.0.0', 'wu_use_domain_mapping');
} // end if;
/**
* Determine whether a mapping should be used
*
* Typically, you'll want to only allow active mappings to be used. However,
* if you want to use more advanced logic, or allow non-active domains to
* be mapped too, simply filter here.
*
* @param boolean $is_active Should the mapping be treated as active?
* @param WP_Ultimo\Models\Domain $mapping Mapping that we're inspecting
* @param string $domain
*/
$is_active = apply_filters('wu_use_domain_mapping', $mapping->is_active(), $mapping, $domain);
// Ignore non-active mappings
if (!$is_active) {
return $site;
} // end if;
// Fetch the actual data for the site
$mapped_site = $mapping->get_site();
if (empty($mapped_site)) {
return $site;
} // end if;
/*
* Note: This is only for backwards compatibility with WPMU Domain Mapping,
* do not rely on this constant in new code.
*/
defined('DOMAIN_MAPPING') or define('DOMAIN_MAPPING', 1); // phpcs:ignore
/*
* Decide if we use SSL
*/
if ($mapping->is_secure()) {
force_ssl_admin(true);
} // end if;
$_site = $site;
if (is_a($mapped_site, '\WP_Site')) {
$this->original_url = $mapped_site->domain . $mapped_site->path;
$_site = $mapped_site;
} elseif (is_a($mapped_site, '\WP_Ultimo\Models\Site')) {
$this->original_url = $mapped_site->get_domain() . $mapped_site->get_path();
$_site = $mapped_site->to_wp_site();
} // end if;
/*
* We found a site based on the mapped domain =)
*/
return $_site;
} // end check_domain_mapping;
/**
* Clear mappings for a site when it's deleted
*
* @param WP_Site $site Site being deleted.
*/
public function clear_mappings_on_delete($site) {
$mappings = Domain::get_by_site($site->blog_id);
if (empty($mappings)) {
return;
} // end if;
foreach ($mappings as $mapping) {
$error = $mapping->delete();
if (is_wp_error($error)) {
// translators: First placeholder is the mapping ID, second is the site ID.
$message = sprintf(__('Unable to delete mapping %1$d for site %2$d', 'wp-ultimo'), $mapping->get_id(), $site->blog_id);
trigger_error($message, E_USER_WARNING);
} // end if;
} // end foreach;
} // end clear_mappings_on_delete;
/**
* Register filters for URLs, if we've mapped
*
* @since 2.0.0
* @return void
*/
public function register_mapped_filters() {
$current_site = $GLOBALS['current_blog'];
if (!$current_site) {
return;
} // end if;
$real_domain = $current_site->domain;
$domain = $_SERVER['HTTP_HOST'];
if ($domain === $real_domain) {
// Domain hasn't been mapped
return;
} // end if;
$domains = $this->get_www_and_nowww_versions($domain);
$mapping = Domain::get_by_domain($domains);
if (empty($mapping) || is_wp_error($mapping)) {
return;
} // end if;
$this->current_mapping = $mapping;
add_filter('site_url', array($this, 'mangle_url'), -10, 4);
add_filter('home_url', array($this, 'mangle_url'), -10, 4);
add_filter('theme_file_uri', array($this, 'mangle_url'));
add_filter('stylesheet_directory_uri', array($this, 'mangle_url'));
add_filter('template_directory_uri', array($this, 'mangle_url'));
add_filter('plugins_url', array($this, 'mangle_url'), -10, 3);
add_filter('autoptimize_filter_base_replace_cdn', array($this, 'mangle_url'), 8); // @since 1.8.2 - Fix for Autoptimizer
// Fix srcset
add_filter('wp_calculate_image_srcset', array($this, 'fix_srcset')); // @since 1.5.5
// If on network site, also filter network urls
if (is_main_site()) {
add_filter('network_site_url', array($this, 'mangle_url'), -10, 3);
add_filter('network_home_url', array($this, 'mangle_url'), -10, 3);
} // end if;
add_filter('jetpack_sync_home_url', array($this, 'mangle_url'));
add_filter('jetpack_sync_site_url', array($this, 'mangle_url'));
/**
* Some plugins will save URL before the mapping was active
* or will build URLs in a different manner that is not included on
* the above filters.
*
* In cases like that, we want to add additional filters.
* The second parameter passed is the mangle_url callback.
*
* We recommend against using this filter directly.
* Instead, use the Domain_Mapping::apply_mapping_to_url method.
*
* @since 2.0.0
* @param array The mangle callable.
* @param self This object.
* @return void
*/
do_action('wu_domain_mapping_register_filters', array($this, 'mangle_url'), $this);
} // end register_mapped_filters;
/**
* Apply the replace URL to URL filters provided by other plugins.
*
* @since 2.0.0
*
* @param string|array $hooks List of hooks to apply the callback to.
* @return void
*/
public static function apply_mapping_to_url($hooks) {
add_action('wu_domain_mapping_register_filters', function($callback) use ($hooks) {
$hooks = (array) $hooks;
foreach ($hooks as $hook) {
add_filter($hook, $callback);
} // end foreach;
});
} // end apply_mapping_to_url;
/**
* Replaces the URL.
*
* @since 2.0.0
*
* @param string $url URL to replace.
* @param null|\WP_Ultimo\Models\Domain $current_mapping The current mapping.
* @return string
*/
public function replace_url($url, $current_mapping = null) {
if ($current_mapping === null) {
$current_mapping = $this->current_mapping;
} // end if;
// Replace the domain
$domain_base = parse_url($url, PHP_URL_HOST);
$domain = rtrim($domain_base . '/' . $current_mapping->get_site()->get_path(), '/');
$regex = '#^(\w+://)' . preg_quote($domain, '#') . '#i';
$mangled = preg_replace($regex, '${1}' . $current_mapping->get_domain(), $url);
/*
* Another try if we don't need to deal with subdirectory.
*/
if ($mangled === $url && $this->current_mapping !== $current_mapping) {
$domain = rtrim($domain_base, '/');
$regex = '#^(\w+://)' . preg_quote($domain, '#') . '#i';
$mangled = preg_replace($regex, '${1}' . $current_mapping->get_domain(), $url);
} // end if;
$mangled = wu_replace_scheme($mangled, $current_mapping->is_secure() ? 'https://' : 'http://');
return $mangled;
} // end replace_url;
/**
* Mangle the home URL to give our primary domain
*
* @param string $url The complete home URL including scheme and path.
* @param string $path Path relative to the home URL. Blank string if no path is specified.
* @param string|null $orig_scheme Scheme to give the home URL context. Accepts 'http', 'https', 'relative' or null.
* @param int|null $site_id Blog ID, or null for the current blog.
* @return string Mangled URL
*/
public function mangle_url($url, $path = '/', $orig_scheme = '', $site_id = 0) {
if (empty($site_id)) {
$site_id = get_current_blog_id();
} // end if;
$current_mapping = $this->current_mapping;
if (empty($current_mapping) || $site_id !== $current_mapping->get_site_id()) {
return $url;
} // end if;
return $this->replace_url($url);
} // end mangle_url;
/**
* Adds a fix to the srcset URLs when we need that domain mapped
*
* @since 1.5.5
* @param array $sources Image source URLs.
* @return array
*/
public function fix_srcset($sources) {
foreach ($sources as &$source) {
$sources[$source['value']]['url'] = $this->replace_url($sources[$source['value']]['url']);
} // end foreach;
return $sources;
} // end fix_srcset;
} // end class Domain_Mapping;