class-rewrite.php

<?php 

class Sublanguage_rewrite extends Sublanguage_current {
	
	/**
	 * @from 2.0
	 */
	var $rewritable_post_types = array();
	
	/**
	 * @from 2.0
	 */
	var $rewritable_taxonomies = array();
	
	/**
	 * @from 1.4.7
	 */
	public function load() {
		
		parent::load();
		
		add_filter('register_post_type_args', array($this, 'register_post_type_args'), 10, 2);
		add_action('registered_post_type', array($this, 'registered_post_type'), 10, 2);
		
		add_filter('register_taxonomy_args', array($this, 'register_taxonomy_args'), 10, 2);
		add_action('registered_taxonomy', array($this, 'registered_taxonomy'), 10, 3);
		
		// rewrite pages
		add_filter('page_rewrite_rules', array($this, 'page_rewrite_rules'));
		
		// Append language slugs to every rules
		add_filter('rewrite_rules_array', array($this, 'append_language_slugs'), 12);
		
	}
	
	/**
	 * Shortcut cpt rule generation
	 *
	 * @filter 'register_post_type_args'
	 * @from 2.0
	 */
	public function register_post_type_args($args, $post_type) {
		
		if ($this->is_post_type_translatable($post_type) && isset($args['rewrite']) && $args['rewrite'] !== false && get_option('permalink_structure') != '') {
			
			// -> WP_Post_Type::set_props()
			if ( ! is_array( $args['rewrite'] ) ) {
				$args['rewrite'] = array();
			}
			if ( empty( $args['rewrite']['slug'] ) ) {
				$args['rewrite']['slug'] = $post_type;
			}
			if ( ! isset( $args['rewrite']['with_front'] ) ) {
				$args['rewrite']['with_front'] = true;
			}
			if ( ! isset( $args['rewrite']['pages'] ) ) {
				$args['rewrite']['pages'] = true;
			}
			if ( ! isset( $args['rewrite']['feeds'] ) || ! $args['has_archive'] ) {
				$args['rewrite']['feeds'] = (bool) $args['has_archive'];
			}
			if ( ! isset( $args['rewrite']['ep_mask'] ) ) {
				if ( isset( $args['permalink_epmask'] ) ) {
					$args['rewrite']['ep_mask'] = $args['permalink_epmask'];
				} else {
					$args['rewrite']['ep_mask'] = EP_PERMALINK;
				}
			}
			
			$this->rewritable_post_types[$post_type] = $args['rewrite'];
			
			$args['rewrite'] = false; // fake it to skip normal rules generation
				
		}
		
		return $args;
		
	}
	
	/**
	 * Translate custom post type rewrite rules.
	 * See WP_Post_Type::add_rewrite_rules()
	 *
	 * @hook 'registered_post_type'
	 * @from 2.0
	 */
	public function registered_post_type($post_type, $post_type_obj) {
		global $wp_rewrite;
		
		if (isset($this->rewritable_post_types[$post_type])) {
			
			$post_type_obj->rewrite = $this->rewritable_post_types[$post_type];
			$post_type_obj->rewrite['walk_dirs'] = false;
			
			$translation_slugs = array();
			
			foreach ($this->get_languages() as $language) {
			
				$translation_slugs[] = $this->translate_cpt($post_type, $language, $post_type);
			
			}
		
			$translation_slugs = array_unique($translation_slugs);
			
			$translation_slug = '(' . implode('|', $translation_slugs) . ')';
			
			add_rewrite_tag( "%$post_type-slug%", $translation_slug, "sublanguage_slug=" );
			
			if ( $post_type_obj->hierarchical ) {
				add_rewrite_tag( "%$post_type%", '(.+?)', $post_type_obj->query_var ? "{$post_type_obj->query_var}=" : "post_type=$post_type&pagename=" );
			} else {
				add_rewrite_tag( "%$post_type%", '([^/]+)', $post_type_obj->query_var ? "{$post_type_obj->query_var}=" : "post_type=$post_type&name=" );
			}
			
			if ( $post_type_obj->has_archive ) {
				
				foreach ($this->get_languages() as $language) {
			
					$archive_slugs[] = $this->translate_cpt_archive($post_type, $language);
			
				}
				
				$archive_slugs = array_unique($archive_slugs);
				
				$archive_slug = '(' . implode('|', $archive_slugs) . ')';
				
				if ( $post_type_obj->rewrite['with_front'] ) {
					$archive_slug = substr( $wp_rewrite->front, 1 ) . $archive_slug;
				} else {
					$archive_slug = $wp_rewrite->root . $archive_slug;
				}

				add_rewrite_rule( "{$archive_slug}/?$", 'index.php?post_type='.$post_type.'&sublanguage_slug=$matches[1]', 'top' );
				if ( $post_type_obj->rewrite['feeds'] && $wp_rewrite->feeds ) {
					$feeds = '(' . trim( implode( '|', $wp_rewrite->feeds ) ) . ')';
					add_rewrite_rule( "{$archive_slug}/feed/$feeds/?$", 'index.php?post_type='.$post_type.'&sublanguage_slug=$matches[1]&feed=$matches[2]', 'top' );
					add_rewrite_rule( "{$archive_slug}/$feeds/?$", 'index.php?post_type='.$post_type.'&sublanguage_slug=$matches[1]&feed=$matches[2]', 'top' );
				}
				if ( $post_type_obj->rewrite['pages'] ) {
					add_rewrite_rule( "{$archive_slug}/{$wp_rewrite->pagination_base}/([0-9]{1,})/?$", 'index.php?post_type='.$post_type.'&sublanguage_slug=$matches[1]&paged=$matches[2]', 'top' );
				}
			}

			$permastruct_args = $post_type_obj->rewrite;
			$permastruct_args['feed'] = $permastruct_args['feeds'];
			
			add_permastruct($post_type, "%$post_type-slug%/%$post_type%", $permastruct_args);
			
			// -> Get ride of attachment rules
			add_filter($post_type.'_rewrite_rules', array($this, 'drop_cpt_attachment_rules'));
			
		}
	
	}
	
	/**
	 * Get ride of attachment rules: buggy and useless!
	 *
	 * @filter "{$permastructname}_rewrite_rules"
	 * @from 2.0
	 */
	public function drop_cpt_attachment_rules($rules) {
		
		$new_rules = array();
		
		foreach ($rules as $match => $rewrite) {
		
			if (!preg_match('/(?:\?|&)attachment=/', $rewrite)) {
			
				$new_rules[$match] = $rewrite;
				
			}
			
		}
		
		return $new_rules;
	}
	
	/**
	 * Shortcut taxonomy rule generation
	 *
	 * @filter 'register_taxonomy_args'
	 * @from 2.0
	 */
	public function register_taxonomy_args($args, $taxonomy) {
		
		if ($this->is_taxonomy_translatable($taxonomy) && (!isset($args['rewrite']) || $args['rewrite'] !== false) && get_option('permalink_structure') != '') {
			
			if (!isset($args['rewrite'])) {
				
				$args['rewrite'] = $taxonomy;
				
			}
			
			$args['rewrite'] = wp_parse_args( $args['rewrite'], array(
				'with_front'   => true,
				'hierarchical' => false,
				'ep_mask'      => EP_NONE,
			) );

			if ( empty( $args['rewrite']['slug'] ) ) {
				$args['rewrite']['slug'] = sanitize_title_with_dashes($taxonomy);
			}
			
			$this->rewritable_taxonomies[$taxonomy] = $args['rewrite'];
			
			$args['rewrite'] = false; // fake it to skip normal rules generation
			
		}
		
		return $args;
	}
	
	/**
	 * Translate taxonomy rewrite rules.
	 * Bypass: WP_Taxonomy::add_rewrite_rules()
	 *
	 * @hook 'registered_taxonomy'
	 * @from 2.0
	 */
	public function registered_taxonomy($taxonomy, $object_type, $taxonomy_obj) {
		global $wp, $wp_taxonomies;
		
		if (isset($this->rewritable_taxonomies[$taxonomy])) {
			
			$taxonomy_obj['rewrite'] = $this->rewritable_taxonomies[$taxonomy]; // why is $taxonomy_obj an array!?
			$taxonomy_obj['rewrite']['walk_dirs'] = false;
			
			if ($taxonomy_obj['hierarchical'] && $taxonomy_obj['rewrite']['hierarchical']) {
				$tag = '(.+?)'; // -> not supported yet!!
			} else {
				$tag = '([^/]+)';
			}
			
			$translation_slugs = array();
			
			foreach ($this->get_languages() as $language) {
			
				$translation_slugs[] = $this->translate_taxonomy($taxonomy, $language, $taxonomy);
			
			}
			
			$translation_slugs = array_unique($translation_slugs);
			
			$translation_slug = '(' . implode('|', $translation_slugs) . ')';
			
			add_rewrite_tag( "%$taxonomy-slug%", $translation_slug, 'sublanguage_slug=' );
			add_rewrite_tag( "%$taxonomy%", $tag, $taxonomy_obj['query_var'] ? $taxonomy_obj['query_var'].'=' : "taxonomy=$taxonomy&term=" );
			
			add_permastruct( $taxonomy, "%$taxonomy-slug%/%$taxonomy%", $taxonomy_obj['rewrite'] );
			
			// set back the original rewrite properties
			$wp_taxonomies[$taxonomy]->rewrite = $taxonomy_obj['rewrite'];
			
		}
		
	}
	
	/**
	 * Overwrite page rules to bypass WP "verbose page rules", because get_page_by_path() is impossible to hook into.
	 *
	 * @filter 'page_rewrite_rules'
	 * @from 2.0
	 * @from 2.3 Use root page names
	 */
	public function page_rewrite_rules($rules) {
		global $wp_rewrite, $wpdb;
		
		if ($this->is_post_type_translatable('page')) {
			
			$page_query = new WP_Query(array(
				'post_type' => 'page',
				'post_status' => 'publish',
				'post_parent' => 0,
				'posts_per_page' => -1,
				'post__not_in' => array_filter(array_map('intval', array(
					get_option('page_for_posts'),
					get_option('page_on_front')
				))),
				$this->language_query_var => false
			));
			
			$duplicate_rules = array();
		
			foreach ($page_query->posts as $page) {
			
				$slugs = array();
			
				foreach ($this->get_languages() as $language) {
				
					$translated_slug = $this->translate_post_field($page, 'post_name', $language);
				
					if (!in_array($translated_slug, $slugs)) {
					
						$slugs[] = $translated_slug;
				
					}
			
				}
				
				if ($slugs) {
					
					$regex_base = '((?:'.implode('|', $slugs).').*?)';
					
					if (!empty($wp_rewrite->endpoints)) {
					
						foreach ($wp_rewrite->endpoints as $ep) {
						
							if ($ep[0] & EP_PAGES) {
							
								$epregex = $regex_base . '/' . $ep[1] . '(/(.*))?/?$';
								$duplicate_rules[$epregex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&'.$ep[2].'=$matches[3]';
						
							}
						
						}
					
					}
					
					if (apply_filters('sublanguage_page_use_feeds', false)) {
												
						$feedregex = '(' . implode('|', (array) $wp_rewrite->feeds) . ')/?$';
						$feedregex1 = $regex_base . '/' . $wp_rewrite->feed_base . '/' . $feedregex;
						$feedregex2 = $regex_base . '/' . $feedregex;
						
						$duplicate_rules[$feedregex1] = $wp_rewrite->index . '?sublanguage_page=$matches[1]&feed=$matches[2]';
						$duplicate_rules[$feedregex2] = $wp_rewrite->index . '?sublanguage_page=$matches[1]&feed=$matches[2]';
						
					}
					
					if (apply_filters('sublanguage_page_use_comment_pagination', false)) {
						
						$commentregex = $regex_base . '/' . $wp_rewrite->comments_pagination_base . '-([0-9]{1,})/?$';
						$duplicate_rules[$commentregex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&cpage=$matches[2]';
					
					}
					
					if (apply_filters('sublanguage_page_use_trackback', false)) {
						
						$trackbackregex = $regex_base . '/trackback/?$';
						$duplicate_rules[$trackbackregex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&tb=1';
					
					}
					
					if (apply_filters('sublanguage_page_use_embed', false)) {
						
						$embedregex = $regex_base . '/embed/?$';
						$duplicate_rules[$embedregex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&embed=true'; // sic!
					
					}
					
					if (apply_filters('sublanguage_page_use_pagination', false)) {
						
						$pageregex = $regex_base . '/' . $wp_rewrite->pagination_base . '/?([0-9]{1,})/?$';
						$duplicate_rules[$pageregex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&page=$matches[2]';
						
						$regex = $regex_base . '(?:/([0-9]+))?/?$';
						$duplicate_rules[$regex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]&page=$matches[2]';
						
					} else {
						
						$regex = $regex_base . '/?$';
						$duplicate_rules[$regex] =  $wp_rewrite->index . '?sublanguage_page=$matches[1]';
					
					}
					
				}
		
			}
		
			$rules = array_merge($duplicate_rules, $rules);
		
		}
		
		return $rules;
	}
	
	/**
	 * Append language slugs to every rules
	 *
	 * @filter 'rewrite_rules_array'
	 * @from 2.0
	 */
	public function append_language_slugs($rules) {
		
		$language_slugs = array();
			
		foreach ($this->get_languages() as $language) {
		
			$language_slugs[] = $language->post_name;
		
		}
		
		$new_rules = array();
		
		$new_rules['(?:' . implode('|', $language_slugs) . ')/?$'] = 'index.php'; // -> rule for home
		
		$languages_slug = '(?:' . implode('/|', $language_slugs) . '/)?';
		
		$black_list = array(
// 			'^wp-json/?$',
// 			'^wp-json/(.*)?',
			'^index.php/wp-json/?$',
			'^index.php/wp-json/(.*)?'
		);
		
		foreach ($rules as $key => $rule) {
			
			if (in_array($key, $black_list)) { // -> REST API?
			
				$new_rules[$key] = $rule;
			
			} else if (substr($key, 0, 1) === '^') { // -> get ride of the leading ^
				
				$new_rules[$languages_slug . substr($key, 1)] = $rule;
			
			} else {
			
				$new_rules[$languages_slug . $key] = $rule;
			
			}
							
		}
		
		return $new_rules;
	}
	

	
}