133 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * WPSEO plugin file.
 | 
						|
 *
 | 
						|
 * @package WPSEO\XML_Sitemaps
 | 
						|
 */
 | 
						|
 | 
						|
use Yoast\WP\SEO\Conditionals\Deactivating_Yoast_Seo_Conditional;
 | 
						|
 | 
						|
/**
 | 
						|
 * Rewrite setup and handling for sitemaps functionality.
 | 
						|
 */
 | 
						|
class WPSEO_Sitemaps_Router {
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Sets up init logic.
 | 
						|
	 */
 | 
						|
	public function __construct() {
 | 
						|
		// If we add rewrite rules during the plugin's deactivation, the flush_rewrite_rules that we perform afterwards won't properly flush those new rules.
 | 
						|
		if ( YoastSEO()->classes->get( Deactivating_Yoast_Seo_Conditional::class )->is_met() ) {
 | 
						|
			return;
 | 
						|
		}
 | 
						|
 | 
						|
		add_action( 'init', [ $this, 'init' ], 1 );
 | 
						|
		add_filter( 'redirect_canonical', [ $this, 'redirect_canonical' ] );
 | 
						|
		add_action( 'template_redirect', [ $this, 'template_redirect' ], 0 );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Sets up rewrite rules.
 | 
						|
	 */
 | 
						|
	public function init() {
 | 
						|
		global $wp;
 | 
						|
 | 
						|
		$wp->add_query_var( 'sitemap' );
 | 
						|
		$wp->add_query_var( 'sitemap_n' );
 | 
						|
		$wp->add_query_var( 'yoast-sitemap-xsl' );
 | 
						|
 | 
						|
		add_rewrite_rule( 'sitemap_index\.xml$', 'index.php?sitemap=1', 'top' );
 | 
						|
		add_rewrite_rule( '([^/]+?)-sitemap([0-9]+)?\.xml$', 'index.php?sitemap=$matches[1]&sitemap_n=$matches[2]', 'top' );
 | 
						|
		add_rewrite_rule( '([a-z]+)?-?sitemap\.xsl$', 'index.php?yoast-sitemap-xsl=$matches[1]', 'top' );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Stop trailing slashes on sitemap.xml URLs.
 | 
						|
	 *
 | 
						|
	 * @param string $redirect The redirect URL currently determined.
 | 
						|
	 *
 | 
						|
	 * @return bool|string
 | 
						|
	 */
 | 
						|
	public function redirect_canonical( $redirect ) {
 | 
						|
 | 
						|
		if ( get_query_var( 'sitemap' ) || get_query_var( 'yoast-sitemap-xsl' ) ) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		return $redirect;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Redirects sitemap.xml to sitemap_index.xml.
 | 
						|
	 */
 | 
						|
	public function template_redirect() {
 | 
						|
		if ( ! $this->needs_sitemap_index_redirect() ) {
 | 
						|
			return;
 | 
						|
		}
 | 
						|
 | 
						|
		wp_safe_redirect( home_url( '/sitemap_index.xml' ), 301, 'Yoast SEO' );
 | 
						|
		exit;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Checks whether the current request needs to be redirected to sitemap_index.xml.
 | 
						|
	 *
 | 
						|
	 * @global WP_Query $wp_query Current query.
 | 
						|
	 *
 | 
						|
	 * @return bool True if redirect is needed, false otherwise.
 | 
						|
	 */
 | 
						|
	public function needs_sitemap_index_redirect() {
 | 
						|
		global $wp_query;
 | 
						|
 | 
						|
		$protocol = 'http://';
 | 
						|
		if ( ! empty( $_SERVER['HTTPS'] ) && $_SERVER['HTTPS'] === 'on' ) {
 | 
						|
			$protocol = 'https://';
 | 
						|
		}
 | 
						|
 | 
						|
		$domain = '';
 | 
						|
		if ( isset( $_SERVER['SERVER_NAME'] ) ) {
 | 
						|
			$domain = sanitize_text_field( wp_unslash( $_SERVER['SERVER_NAME'] ) );
 | 
						|
		}
 | 
						|
 | 
						|
		$path = '';
 | 
						|
		if ( isset( $_SERVER['REQUEST_URI'] ) ) {
 | 
						|
			$path = sanitize_text_field( wp_unslash( $_SERVER['REQUEST_URI'] ) );
 | 
						|
		}
 | 
						|
 | 
						|
		// Due to different environment configurations, we need to check both SERVER_NAME and HTTP_HOST.
 | 
						|
		$check_urls = [ $protocol . $domain . $path ];
 | 
						|
		if ( ! empty( $_SERVER['HTTP_HOST'] ) ) {
 | 
						|
			$check_urls[] = $protocol . sanitize_text_field( wp_unslash( $_SERVER['HTTP_HOST'] ) ) . $path;
 | 
						|
		}
 | 
						|
 | 
						|
		return $wp_query->is_404 && in_array( home_url( '/sitemap.xml' ), $check_urls, true );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Create base URL for the sitemap.
 | 
						|
	 *
 | 
						|
	 * @param string $page Page to append to the base URL.
 | 
						|
	 *
 | 
						|
	 * @return string base URL (incl page)
 | 
						|
	 */
 | 
						|
	public static function get_base_url( $page ) {
 | 
						|
 | 
						|
		global $wp_rewrite;
 | 
						|
 | 
						|
		$base = $wp_rewrite->using_index_permalinks() ? 'index.php/' : '/';
 | 
						|
 | 
						|
		/**
 | 
						|
		 * Filter the base URL of the sitemaps.
 | 
						|
		 *
 | 
						|
		 * @param string $base The string that should be added to home_url() to make the full base URL.
 | 
						|
		 */
 | 
						|
		$base = apply_filters( 'wpseo_sitemaps_base_url', $base );
 | 
						|
 | 
						|
		/*
 | 
						|
		 * Get the scheme from the configured home URL instead of letting WordPress
 | 
						|
		 * determine the scheme based on the requested URI.
 | 
						|
		 */
 | 
						|
		return home_url( $base . $page, wp_parse_url( get_option( 'home' ), PHP_URL_SCHEME ) );
 | 
						|
	}
 | 
						|
}
 |