Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

/**

[2] Fix | Delete

* @author OnTheGo Systems

[3] Fix | Delete

[4] Fix | Delete

class WPML_Canonicals {

[5] Fix | Delete

const CANONICAL_FOR_DUPLICATED_POST = 'duplicate';

[6] Fix | Delete

const CANONICAL_FOR_NON_TRANSLATABLE_POST = 'non-translatable';

[7] Fix | Delete

/** @var SitePress */

[8] Fix | Delete

private $sitepress;

[9] Fix | Delete

/** @var WPML_Translations */

[10] Fix | Delete

private $wpml_translations;

[11] Fix | Delete

/** @var WPML_Translation_Element_Factory */

[12] Fix | Delete

private $translation_element_factory;

[13] Fix | Delete

[14] Fix | Delete

/**

[15] Fix | Delete

* WPML_Canonicals constructor.

[16] Fix | Delete

[17] Fix | Delete

* @param SitePress $sitepress

[18] Fix | Delete

* @param WPML_Translation_Element_Factory $translation_element_factory

[19] Fix | Delete

* @param WPML_Translations $wpml_translations

[20] Fix | Delete

[21] Fix | Delete

public function __construct(

[22] Fix | Delete

SitePress $sitepress,

[23] Fix | Delete

WPML_Translation_Element_Factory $translation_element_factory,

[24] Fix | Delete

WPML_Translations $wpml_translations = null

[25] Fix | Delete

) {

[26] Fix | Delete

$this->sitepress = $sitepress;

[27] Fix | Delete

$this->translation_element_factory = $translation_element_factory;

[28] Fix | Delete

$this->wpml_translations = $wpml_translations;

[29] Fix | Delete

}

[30] Fix | Delete

[31] Fix | Delete

/**

[32] Fix | Delete

* @param int $post_id

[33] Fix | Delete

[34] Fix | Delete

* @return bool|string

[35] Fix | Delete

* @throws \InvalidArgumentException

[36] Fix | Delete

[37] Fix | Delete

private function must_filter_permalink( $post_id ) {

[38] Fix | Delete

$this->init_wpml_translations();

[39] Fix | Delete

$post_element = $this->translation_element_factory->create( $post_id, 'post' );

[40] Fix | Delete

$must_handle_canonicals = $this->must_handle_a_canonical_url();

[41] Fix | Delete

[42] Fix | Delete

if ( $post_element->is_translatable() ) {

[43] Fix | Delete

if ( $must_handle_canonicals && $this->wpml_translations->is_a_duplicate_of( $post_element ) && $this->is_permalink_filter_from_rel_canonical() ) {

[44] Fix | Delete

return self::CANONICAL_FOR_DUPLICATED_POST;

[45] Fix | Delete

}

[46] Fix | Delete

} elseif ( $must_handle_canonicals ) {

[47] Fix | Delete

return self::CANONICAL_FOR_NON_TRANSLATABLE_POST;

[48] Fix | Delete

}

[49] Fix | Delete

[50] Fix | Delete

return false;

[51] Fix | Delete

}

[52] Fix | Delete

[53] Fix | Delete

/**

[54] Fix | Delete

* @param string $link

[55] Fix | Delete

* @param int $post_id

[56] Fix | Delete

[57] Fix | Delete

* @return null|string

[58] Fix | Delete

* @throws \InvalidArgumentException

[59] Fix | Delete

[60] Fix | Delete

public function permalink_filter( $link, $post_id ) {

[61] Fix | Delete

switch ( $this->must_filter_permalink( $post_id ) ) {

[62] Fix | Delete

case self::CANONICAL_FOR_DUPLICATED_POST:

[63] Fix | Delete

$post_element = $this->translation_element_factory->create( $post_id, 'post' );

[64] Fix | Delete

[65] Fix | Delete

return $this->get_canonical_of_duplicate( $post_element );

[66] Fix | Delete

[67] Fix | Delete

case self::CANONICAL_FOR_NON_TRANSLATABLE_POST:

[68] Fix | Delete

return $this->get_url_in_default_language_if_rel_canonical( $link );

[69] Fix | Delete

[70] Fix | Delete

default:

[71] Fix | Delete

return null;

[72] Fix | Delete

}

[73] Fix | Delete

}

[74] Fix | Delete

[75] Fix | Delete

/**

[76] Fix | Delete

* @param string $canonical_url

[77] Fix | Delete

* @param WP_Post $post

[78] Fix | Delete

[79] Fix | Delete

* @return string|bool

[80] Fix | Delete

[81] Fix | Delete

public function get_canonical_url( $canonical_url, $post, $request_language ) {

[82] Fix | Delete

if ( $post && $this->sitepress->get_wp_api()->is_front_end() ) {

[83] Fix | Delete

try {

[84] Fix | Delete

/** @var WPML_Post_Element $post_element */

[85] Fix | Delete

$post_element = $this->translation_element_factory->create( $post->ID, 'post' );

[86] Fix | Delete

[87] Fix | Delete

$should_translate_canonical_url = apply_filters(

[88] Fix | Delete

'wpml_must_translate_canonical_url', true, $post_element

[89] Fix | Delete

);

[90] Fix | Delete

[91] Fix | Delete

if ( ! $should_translate_canonical_url ) {

[92] Fix | Delete

return $canonical_url;

[93] Fix | Delete

}

[94] Fix | Delete

[95] Fix | Delete

if ( ! $post_element->is_translatable() ) {

[96] Fix | Delete

global $wpml_url_filters;

[97] Fix | Delete

$wpml_url_filters->remove_global_hooks();

[98] Fix | Delete

$canonical_url = $this->sitepress->convert_url_string( $canonical_url, $this->sitepress->get_default_language() );

[99] Fix | Delete

$wpml_url_filters->add_global_hooks();

[100] Fix | Delete

} else {

[101] Fix | Delete

$this->init_wpml_translations();

[102] Fix | Delete

if ( $this->wpml_translations->is_a_duplicate_of( $post_element ) ) {

[103] Fix | Delete

$canonical_url = (string) $this->get_canonical_of_duplicate( $post_element );

[104] Fix | Delete

} elseif ( $post_element->get_language_code() != $request_language ) {

[105] Fix | Delete

$canonical_url = $this->sitepress->convert_url_string( $canonical_url, $post_element->get_language_code() );

[106] Fix | Delete

}

[107] Fix | Delete

}

[108] Fix | Delete

} catch ( InvalidArgumentException $e ) {

[109] Fix | Delete

}

[110] Fix | Delete

}

[111] Fix | Delete

[112] Fix | Delete

return $canonical_url;

[113] Fix | Delete

}

[114] Fix | Delete

[115] Fix | Delete

/**

[116] Fix | Delete

* @param $url

[117] Fix | Delete

[118] Fix | Delete

* @return string

[119] Fix | Delete

[120] Fix | Delete

public function get_general_canonical_url( $url ) {

[121] Fix | Delete

global $wpml_url_filters;

[122] Fix | Delete

$wpml_url_filters->remove_global_hooks();

[123] Fix | Delete

$canonical_url = $this->sitepress->convert_url_string( $url, $this->sitepress->get_current_language() );

[124] Fix | Delete

$wpml_url_filters->add_global_hooks();

[125] Fix | Delete

[126] Fix | Delete

return $canonical_url;

[127] Fix | Delete

}

[128] Fix | Delete

[129] Fix | Delete

private function has_wp_get_canonical_url() {

[130] Fix | Delete

return $this->sitepress->get_wp_api()->function_exists( 'wp_get_canonical_url' );

[131] Fix | Delete

}

[132] Fix | Delete

[133] Fix | Delete

/**

[134] Fix | Delete

* @return bool

[135] Fix | Delete

[136] Fix | Delete

private function is_permalink_filter_from_rel_canonical() {

[137] Fix | Delete

$back_trace_stack = $this->sitepress->get_wp_api()->get_backtrace( 20 );

[138] Fix | Delete

$keywords = array( 'rel_canonical', 'canonical', 'generate_canonical' );

[139] Fix | Delete

[140] Fix | Delete

$result = false;

[141] Fix | Delete

if ( $back_trace_stack ) {

[142] Fix | Delete

foreach ( $back_trace_stack as $key => $value ) {

[143] Fix | Delete

foreach ( $keywords as $keyword ) {

[144] Fix | Delete

if ( 'function' === $key && $keyword === $value ) {

[145] Fix | Delete

$result = true;

[146] Fix | Delete

break;

[147] Fix | Delete

}

[148] Fix | Delete

}

[149] Fix | Delete

}

[150] Fix | Delete

}

[151] Fix | Delete

[152] Fix | Delete

return $result;

[153] Fix | Delete

}

[154] Fix | Delete

[155] Fix | Delete

/**

[156] Fix | Delete

* @param string $link

[157] Fix | Delete

[158] Fix | Delete

* @return bool|string

[159] Fix | Delete

[160] Fix | Delete

private function get_url_in_default_language_if_rel_canonical( $link ) {

[161] Fix | Delete

if ( $this->is_permalink_filter_from_rel_canonical() ) {

[162] Fix | Delete

$default_language = $this->sitepress->get_default_language();

[163] Fix | Delete

$link = (string) $this->sitepress->convert_url( $link, $default_language );

[164] Fix | Delete

}

[165] Fix | Delete

[166] Fix | Delete

return $link;

[167] Fix | Delete

}

[168] Fix | Delete

[169] Fix | Delete

/**

[170] Fix | Delete

* @param WPML_Post_Element $post_element

[171] Fix | Delete

[172] Fix | Delete

* @return false|string

[173] Fix | Delete

[174] Fix | Delete

private function get_canonical_of_duplicate( $post_element ) {

[175] Fix | Delete

$source_element = $post_element->get_source_element();

[176] Fix | Delete

if ( $source_element ) {

[177] Fix | Delete

$source_element_id = $source_element->get_id();

[178] Fix | Delete

$source_language_code = $source_element->get_language_code();

[179] Fix | Delete

$current_language = $this->sitepress->get_current_language();

[180] Fix | Delete

$this->sitepress->switch_lang( $source_language_code );

[181] Fix | Delete

$new_link = get_permalink( $source_element_id );

[182] Fix | Delete

$this->sitepress->switch_lang( $current_language );

[183] Fix | Delete

} else {

[184] Fix | Delete

$new_link = get_permalink( $post_element->get_id() );

[185] Fix | Delete

}

[186] Fix | Delete

[187] Fix | Delete

return $new_link;

[188] Fix | Delete

}

[189] Fix | Delete

[190] Fix | Delete

/**

[191] Fix | Delete

* @return bool

[192] Fix | Delete

[193] Fix | Delete

private function must_handle_a_canonical_url() {

[194] Fix | Delete

return ! $this->has_wp_get_canonical_url() && $this->sitepress->get_wp_api()->is_front_end();

[195] Fix | Delete

}

[196] Fix | Delete

[197] Fix | Delete

private function init_wpml_translations() {

[198] Fix | Delete

if ( ! $this->wpml_translations ) {

[199] Fix | Delete

$this->wpml_translations = new WPML_Translations( $this->sitepress );

[200] Fix | Delete

}

[201] Fix | Delete

}

[202] Fix | Delete

}

[203] Fix | Delete

[204] Fix | Delete