_ Git - physioassist-wordpress.git/blob

   1 <?php
   2 /**
   3  * Created by PhpStorm.
   4  * User: bruce
   5  * Date: 17/10/17
   6  * Time: 10:56 PM
   7  */
   8
   9 class WPML_LS_Display_As_Translated_Link {
  10
  11         /** @var SitePress $sitepress */
  12         private $sitepress;
  13         /** @var IWPML_URL_Converter_Strategy $url_converter */
  14         private $url_converter;
  15         /** @var WP_Query $wp_query */
  16         private $wp_query;
  17         /** @var WPML_Translation_Element_Factory $element_factory */
  18         private $element_factory;
  19         /** @var string $default_language */
  20         private $default_language;
  21         /** @var string $processed_language */
  22         private $processed_language;
  23
  24         public function __construct(
  25                 SitePress $sitepress,
  26                 IWPML_URL_Converter_Strategy $url_converter,
  27                 WP_Query $wp_query,
  28                 WPML_Translation_Element_Factory $element_factory
  29         ) {
  30                 $this->sitepress        = $sitepress;
  31                 $this->url_converter    = $url_converter;
  32                 $this->wp_query         = $wp_query;
  33                 $this->element_factory  = $element_factory;
  34                 $this->default_language = $sitepress->get_default_language();
  35         }
  36
  37         public function get_url( $translations, $lang ) {
  38                 $queried_object = $this->wp_query->get_queried_object();
  39                 if ( $queried_object instanceof WP_Post ) {
  40                         return $this->get_post_url( $translations, $lang, $queried_object );
  41                 } elseif ( $queried_object instanceof WP_Term ) {
  42                         return $this->get_term_url( $translations, $lang, $queried_object );
  43                 } else {
  44                         return null;
  45                 }
  46         }
  47
  48         private function get_post_url( $translations, $lang, $queried_object ) {
  49                 $url = null;
  50
  51                 if ( $this->sitepress->is_display_as_translated_post_type( $queried_object->post_type ) &&
  52                          isset( $translations[ $this->default_language ] ) ) {
  53
  54                         $this->sitepress->switch_lang( $this->default_language );
  55                         $this->processed_language = $lang;
  56                         add_filter( 'post_link_category', array( $this, 'adjust_category_in_post_permalink' ) );
  57                         $url = get_permalink( $translations[ $this->default_language ]->element_id );
  58                         remove_filter( 'post_link_category', array( $this, 'adjust_category_in_post_permalink' ) );
  59                         $this->sitepress->switch_lang();
  60                         $url = $this->url_converter->convert_url_string( $url, $lang );
  61                 }
  62
  63                 return $url;
  64         }
  65
  66         private function get_term_url( $translations, $lang, $queried_object ) {
  67                 $url = null;
  68
  69                 if ( $this->sitepress->is_display_as_translated_taxonomy( $queried_object->taxonomy ) &&
  70                          isset( $translations[ $this->default_language ] ) ) {
  71
  72                         $url = get_term_link( (int) $translations[ $this->default_language ]->term_id, $queried_object->taxonomy );
  73                         $url = $this->url_converter->convert_url_string( $url, $lang );
  74                 }
  75
  76                 return $url;
  77         }
  78
  79         /**
  80          * The permalink needs to be adjusted when the URL structure contains the category tag (%category%).
  81          *
  82          * @param WP_Term $cat
  83          *
  84          * @return WP_Term
  85          */
  86         public function adjust_category_in_post_permalink( $cat ) {
  87                 $cat_element = $this->element_factory->create( $cat->term_id, 'term' );
  88                 $translation = $cat_element->get_translation( $this->processed_language );
  89
  90                 if ( $translation ) {
  91                         $cat = $translation->get_wp_object() ?: $cat;
  92                 }
  93
  94                 return $cat;
  95         }
  96
  97 }