class-translator.php

Source code

<?php
/**
 * Translator component.
 *
 * @package HivePress\Components
 */

namespace HivePress\Components;

use HivePress\Helpers as hp;

// Exit if accessed directly.
defined( 'ABSPATH' ) || exit;

/**
 * Handles translations.
 */
final class Translator extends Component {

	/**
	 * Class constructor.
	 *
	 * @param array $args Component arguments.
	 */
	public function __construct( $args = [] ) {
		if ( $this->is_multilingual() ) {

			// Register options.
			add_filter( 'alloptions', [ $this, 'register_options' ] );

			// Registers comments.
			add_filter( 'comments_clauses', [ $this, 'register_comments' ], 1, 2 );

			// Register post.
			add_action( 'hivepress/v1/models/post/create', [ $this, 'register_post' ], 10, 2 );
		}

		parent::__construct( $args );
	}

	/**
	 * Gets language code.
	 *
	 * @return string
	 */
	public function get_language() {
		$language = hp\get_first_array_value( explode( '_', get_locale() ) );

		if ( $this->is_multilingual() ) {
			$language = apply_filters( 'wpml_current_language', $language );
		}

		return $language;
	}

	/**
	 * Gets region code.
	 *
	 * @return string
	 */
	public function get_region() {
		$parts = explode( '_', get_locale() );

		if ( count( $parts ) > 1 ) {
			return hp\get_last_array_value( $parts );
		}
	}

	/**
	 * Gets translation string.
	 *
	 * @param string $key String key.
	 * @return string
	 */
	public function get_string( $key ) {
		return hp\get_array_value( hivepress()->get_config( 'strings' ), $key );
	}

	/**
	 * Gets translated object ID.
	 *
	 * @param int    $id Object ID.
	 * @param string $type Object type.
	 * @return int
	 */
	public function get_object_id( $id, $type = 'page' ) {
		return apply_filters( 'wpml_object_id', $id, $type );
	}

	/**
	 * Checks multilingual status.
	 *
	 * @return bool
	 */
	public function is_multilingual() {
		return defined( 'ICL_SITEPRESS_VERSION' );
	}

	/**
	 * Registers options.
	 *
	 * @param array $options Options.
	 * @return array
	 */
	public function register_options( $options ) {
		remove_filter( 'alloptions', [ $this, 'register_options' ] );

		foreach ( $options as $name => $value ) {
			if ( strpos( $name, 'hp_page_' ) === 0 && absint( $value ) > 1 ) {
				add_filter( 'option_' . $name, [ $this, 'get_object_id' ] );
			}
		}

		return $options;
	}

	/**
	 * Registers comments.
	 *
	 * @param array  $clauses Query clauses.
	 * @param object $query Query object.
	 * @return array
	 */
	public function register_comments( $clauses, $query ) {
		global $sitepress;

		if ( strpos( hp\get_array_value( $query->query_vars, 'type', '' ), 'hp_' ) === 0 ) {
			remove_filter( 'comments_clauses', [ $sitepress, 'comments_clauses' ] );
		}

		return $clauses;
	}

	/**
	 * Registers post.
	 *
	 * @param int    $post_id Post ID.
	 * @param string $post_type Post type.
	 */
	public function register_post( $post_id, $post_type ) {

		// Check request.
		if ( is_admin() ) {
			return;
		}

		// Check status.
		if ( get_post_status( $post_id ) !== 'auto-draft' ) {
			return;
		}

		// Set translation.
		do_action(
			'wpml_set_element_language_details',
			[
				'element_id'    => $post_id,
				'element_type'  => 'post_' . $post_type,
				'trid'          => false,
				'language_code' => $this->get_language(),
			]
		);
	}
}