wordpress/php-fpm/wordpress_files/plugins/polylang/frontend/frontend-auto-translate.php
2020-05-22 01:40:23 +00:00

290 lines
8.4 KiB
PHP

<?php
/**
* Auto translates the posts and terms ids
* Useful for example for themes querying a specific cat
*
* @since 1.1
*/
class PLL_Frontend_Auto_Translate {
public $model, $curlang;
/**
* Constructor
*
* @since 1.1
*
* @param object $polylang
*/
public function __construct( &$polylang ) {
$this->model = &$polylang->model;
$this->curlang = &$polylang->curlang;
add_action( 'pre_get_posts', array( $this, 'pre_get_posts' ) ); // after main Polylang filter
add_filter( 'get_terms_args', array( $this, 'get_terms_args' ), 20, 2 );
}
/**
* Helper function to get the translated post in the current language
*
* @since 1.8
*
* @param int $post_id
* @return int
*/
protected function get_post( $post_id ) {
return $this->model->post->get( $post_id, $this->curlang );
}
/**
* Helper function to get the translated term in the current language
*
* @since 1.8
*
* @param int $term_id
* @return int
*/
protected function get_term( $term_id ) {
return $this->model->term->get( $term_id, $this->curlang );
}
/**
* Filters posts query to automatically translate included ids
*
* @since 1.1
*
* @param object $query WP_Query object
*/
public function pre_get_posts( $query ) {
global $wpdb;
$qv = &$query->query_vars;
if ( $query->is_main_query() || isset( $qv['lang'] ) || ( ! empty( $qv['post_type'] ) && ! $this->model->is_translated_post_type( $qv['post_type'] ) ) ) {
return;
}
// /!\ always keep untranslated as is
// Term ids separated by a comma
$arr = array();
if ( ! empty( $qv['cat'] ) ) {
foreach ( explode( ',', $qv['cat'] ) as $cat ) {
$tr = $this->get_term( abs( $cat ) );
$arr[] = $cat < 0 ? -$tr : $tr;
}
$qv['cat'] = implode( ',', $arr );
}
// Category_name
$arr = array();
if ( ! empty( $qv['category_name'] ) ) {
foreach ( explode( ',', $qv['category_name'] ) as $slug ) {
$arr[] = $this->get_translated_term_by( 'slug', $slug, 'category' );
}
$qv['category_name'] = implode( ',', $arr );
}
// Array of term ids
foreach ( array( 'category__and', 'category__in', 'category__not_in', 'tag__and', 'tag__in', 'tag__not_in' ) as $key ) {
$arr = array();
if ( ! empty( $qv[ $key ] ) ) {
foreach ( $qv[ $key ] as $cat ) {
$arr[] = ( $tr = $this->get_term( $cat ) ) ? $tr : $cat;
}
$qv[ $key ] = $arr;
}
}
// Tag
$arr = array();
if ( ! empty( $qv['tag'] ) ) {
$sep = strpos( $qv['tag'], ',' ) !== false ? ',' : '+'; // Two possible separators for tag slugs
foreach ( explode( $sep, $qv['tag'] ) as $slug ) {
$arr[] = $this->get_translated_term_by( 'slug', $slug, 'post_tag' );
}
$qv['tag'] = implode( $sep, $arr );
}
// tag_id can only take one id
if ( ! empty( $qv['tag_id'] ) && $tr_id = $this->get_term( $qv['tag_id'] ) ) {
$qv['tag_id'] = $tr_id;
}
// Array of tag slugs
foreach ( array( 'tag_slug__and', 'tag_slug__in' ) as $key ) {
$arr = array();
if ( ! empty( $qv[ $key ] ) ) {
foreach ( $qv[ $key ] as $slug ) {
$arr[] = $this->get_translated_term_by( 'slug', $slug, 'post_tag' );
}
$qv[ $key ] = $arr;
}
}
// Custom taxonomies
// According to the codex, this type of query is deprecated as of WP 3.1 but it does not appear in WP 3.5 source code
foreach ( array_intersect( $this->model->get_translated_taxonomies(), get_taxonomies( array( '_builtin' => false ) ) ) as $taxonomy ) {
$tax = get_taxonomy( $taxonomy );
$arr = array();
if ( ! empty( $qv[ $tax->query_var ] ) ) {
$sep = strpos( $qv[ $tax->query_var ], ',' ) !== false ? ',' : '+'; // Two possible separators
foreach ( explode( $sep, $qv[ $tax->query_var ] ) as $slug ) {
$arr[] = $this->get_translated_term_by( 'slug', $slug, $taxonomy );
}
$qv[ $tax->query_var ] = implode( $sep, $arr );
}
}
// Tax_query since WP 3.1
if ( ! empty( $qv['tax_query'] ) && is_array( $qv['tax_query'] ) ) {
$qv['tax_query'] = $this->translate_tax_query_recursive( $qv['tax_query'] );
}
// p, page_id, post_parent can only take one id
foreach ( array( 'p', 'page_id', 'post_parent' ) as $key ) {
if ( ! empty( $qv[ $key ] ) && $tr_id = $this->get_post( $qv[ $key ] ) ) {
$qv[ $key ] = $tr_id;
}
}
// name, can only take one slug
if ( ! empty( $qv['name'] ) ) {
if ( empty( $qv['post_type'] ) ) {
$post_types = array( 'post' );
} elseif ( 'any' === $qv['post_type'] ) {
$post_types = get_post_types( array( 'exclude_from_search' => false ) ); // May return a empty array
} else {
$post_types = (array) $qv['post_type'];
}
if ( ! empty( $post_types ) ) {
// No function to get post by name except get_posts itself
$id = $wpdb->get_var(
sprintf(
"SELECT ID from {$wpdb->posts}
WHERE {$wpdb->posts}.post_type IN ( '%s' )
AND post_name='%s'",
implode( "', '", esc_sql( $post_types ) ),
esc_sql( $qv['name'] )
)
);
$qv['name'] = ( $id && ( $tr_id = $this->get_post( $id ) ) && $tr = get_post( $tr_id ) ) ? $tr->post_name : $qv['name'];
}
}
// pagename, the page id is already available in queried_object_id
if ( ! empty( $qv['pagename'] ) && ! empty( $query->queried_object_id ) && $tr_id = $this->get_post( $query->queried_object_id ) ) {
$query->queried_object_id = $tr_id;
$qv['pagename'] = get_page_uri( $tr_id );
}
// Array of post ids
// post_parent__in & post_parent__not_in since WP 3.6
foreach ( array( 'post__in', 'post__not_in', 'post_parent__in', 'post_parent__not_in' ) as $key ) { // phpcs:ignore WordPressVIPMinimum.Performance.WPQueryParams.PostNotIn
$arr = array();
if ( ! empty( $qv[ $key ] ) ) {
// post__in used by the 2 functions below
// Useless to filter them as output is already in the right language and would result in performance loss
foreach ( debug_backtrace( DEBUG_BACKTRACE_IGNORE_ARGS ) as $trace ) { // phpcs:ignore WordPress.PHP.DevelopmentFunctions
if ( in_array( $trace['function'], array( 'wp_nav_menu', 'gallery_shortcode' ) ) ) {
return;
}
}
foreach ( $qv[ $key ] as $p ) {
$arr[] = ( $tr = $this->get_post( $p ) ) ? $tr : $p;
}
$qv[ $key ] = $arr;
}
}
}
/**
* Filters terms query to automatically translate included ids
*
* @since 1.1.1
*
* @param array $args
* @param array $taxonomies
* @return array modified $args
*/
public function get_terms_args( $args, $taxonomies ) {
if ( ! isset( $args['lang'] ) && ! empty( $args['include'] ) && ( empty( $taxonomies ) || $this->model->is_translated_taxonomy( $taxonomies ) ) ) {
$arr = array();
foreach ( wp_parse_id_list( $args['include'] ) as $id ) {
$arr[] = ( $tr = $this->get_term( $id ) ) ? $tr : $id;
}
$args['include'] = $arr;
}
return $args;
}
/**
* Translates tax queries
* Compatible with nested tax queries introduced in WP 4.1
*
* @since 1.7
*
* @param array $tax_queries
* @return array translated tax queries
*/
protected function translate_tax_query_recursive( $tax_queries ) {
foreach ( $tax_queries as $key => $q ) {
if ( isset( $q['taxonomy'], $q['terms'] ) && $this->model->is_translated_taxonomy( $q['taxonomy'] ) ) {
$arr = array();
$field = isset( $q['field'] ) && in_array( $q['field'], array( 'slug', 'name' ) ) ? $q['field'] : 'term_id';
foreach ( (array) $q['terms'] as $t ) {
$arr[] = $this->get_translated_term_by( $field, $t, $q['taxonomy'] );
}
$tax_queries[ $key ]['terms'] = $arr;
}
// Nested queries
elseif ( is_array( $q ) ) {
$tax_queries[ $key ] = $this->translate_tax_query_recursive( $q );
}
}
return $tax_queries;
}
/**
* Translates a term given one field.
*
* @since 2.3.3
*
* @param string $field Either 'slug', 'name', 'term_id', or 'term_taxonomy_id'
* @param string|int $term Search for this term value
* @param string $taxonomy Taxonomy name.
* @return string|int Translated term slug, name, term_id or term_taxonomy_id
*/
protected function get_translated_term_by( $field, $term, $taxonomy ) {
if ( 'term_id' === $field ) {
if ( $tr_id = $this->get_term( $term ) ) {
return $tr_id;
}
} else {
$terms = get_terms( $taxonomy, array( $field => $term, 'lang' => '' ) );
if ( ! empty( $terms ) && ! is_wp_error( $terms ) ) {
$t = reset( $terms );
$tr_id = $this->get_term( $t->term_id );
if ( ! is_wp_error( $tr = get_term( $tr_id, $taxonomy ) ) ) {
return $tr->$field;
}
}
}
return $term;
}
}