class-wpml-tm-rest-apply-tp-translation.php 1.69 KB
<?php

use WPML\LIB\WP\User;

class WPML_TM_REST_Apply_TP_Translation extends WPML_REST_Base {
	/** @var WPML_TP_Apply_Translations */
	private $apply_translations;

	public function __construct( WPML_TP_Apply_Translations $apply_translations ) {
		parent::__construct( 'wpml/tm/v1' );

		$this->apply_translations = $apply_translations;
	}

	public function add_hooks() {
		$this->register_routes();
	}

	public function register_routes() {
		parent::register_route(
			'/tp/apply-translations',
			array(
				'methods'  => WP_REST_Server::CREATABLE,
				'callback' => array( $this, 'apply_translations' ),
			)
		);
	}

	/**
	 * @return WP_Error|int|array
	 */
	public function apply_translations( WP_REST_Request $request ) {
		try {
			$params = $request->get_json_params();

			if ( $params ) {
				if ( ! isset( $params['original_element_id'] ) ) {
					$params = array_filter( $params, array( $this, 'validate_job' ) );
					if ( ! $params ) {
						return array();
					}
				}
			}

			return $this->apply_translations->apply( $params )->map( array( $this, 'map_jobs_to_array' ) );
		} catch ( Exception $e ) {
			return new WP_Error( 400, $e->getMessage() );
		}
	}

	public function get_allowed_capabilities( WP_REST_Request $request ) {
		return [ User::CAP_ADMINISTRATOR, User::CAP_MANAGE_TRANSLATIONS, User::CAP_TRANSLATE ];
	}

	public function map_jobs_to_array( WPML_TM_Job_Entity $job ) {
		return [
			'id'     => $job->get_id(),
			'type'   => $job->get_type(),
			'status' => $job->get_status(),
		];
	}

	/**
	 * @param array $job
	 *
	 * @return bool
	 */
	private function validate_job( array $job ) {
		return isset( $job['id'], $job['type'] ) && \WPML_TM_Job_Entity::is_type_valid( $job['type'] );
	}
}