first commit

wp-content/plugins/sitepress-multilingual-cms/classes/icl-to-ate-migration/endpoints/Translators/GetFromICL.php

@@ -0,0 +1,34 @@
+<?php
+
+namespace WPML\ICLToATEMigration\Endpoints\Translators;
+
+use WPML\Ajax\IHandler;
+use WPML\Collect\Support\Collection;
+use WPML\FP\Either;
+use WPML\FP\Fns;
+use WPML_TM_ATE_API;
+
+class GetFromICL implements IHandler {
+
+	/**
+	 * @var WPML_TM_ATE_API
+	 */
+	private $apiClient;
+
+	public function __construct( WPML_TM_ATE_API $apiClient ) {
+		$this->apiClient = $apiClient;
+	}
+
+	public function run( Collection $data ) {
+		global $sitepress_settings;
+		$translationServiceData = current( $sitepress_settings['icl_translation_projects'] );
+
+		$result = $this->apiClient->import_icl_translators( $translationServiceData['ts_id'], $translationServiceData['ts_access_key'] );
+
+		if ( Fns::isLeft( $result ) ) {
+			return Either::left( __( 'Error happened! Please try again.', 'sitepress-multilingual-cms' ) );
+		}
+
+		return GetFromICLResponseMapper::map( $result->get()->records );
+	}
+}

wp-content/plugins/sitepress-multilingual-cms/classes/icl-to-ate-migration/endpoints/Translators/GetFromICLResponseMapper.php

@@ -0,0 +1,78 @@
+<?php
+// This is sample of data that mapper returns formatted data same as it.
+//		return Either::of( [
+//			'translators'   => [
+//               [
+//					'user' => [
+//						'id'            => 1,
+//						'first'   		=> 'Translator',
+//						'last'   		=> 'First',
+//						'email'         => 'translator3@ytest.com',
+//						'userName'      => 'translator',
+//						'wpRole'        => 'author',
+//					],
+//					'languagePairs' => [
+//						'en' => [ 'ar', 'bs'],
+//					],
+//				],
+//            ]
+//		] );
+
+namespace WPML\ICLToATEMigration\Endpoints\Translators;
+
+use WPML\FP\Either;
+use WPML\FP\Fns;
+
+class GetFromICLResponseMapper {
+
+	/**
+	 * Returns formatted data of translators.
+	 *
+	 * @param array $records
+	 *
+	 * @return callable|\WPML\FP\Right
+	 */
+	public static function map( $records ) {
+		return Either::of( [ 'translators' => Fns::map( [ self::class, 'constructUserData' ], $records ) ] );
+	}
+
+	/**
+	 * Formats translator data
+	 *
+	 * @param object $record
+	 *
+	 * @return array
+	 */
+	public static function constructUserData( $record ) {
+		$user = get_user_by( 'email', $record->email );
+
+		return [
+			'user'          => [
+				'id'       => $record->icl_id,
+				'first'    => $record->first_name,
+				'last'     => $record->last_name,
+				'email'    => $record->email,
+				'userName' => $user ? $user->data->user_login : strtolower( $record->first_name . '_' . $record->last_name ),
+				'wpRole'   => $user ? current( $user->roles ) : 'subscriber'
+			],
+			'languagePairs' => self::constructUserLanguagePairs( $record->lang_pairs )
+		];
+	}
+
+	/**
+	 * Formats translator language pairs data
+	 *
+	 * @param array $langPairs
+	 *
+	 * @return array
+	 */
+	public static function constructUserLanguagePairs( $langPairs ) {
+		$constructedLangPairs = [];
+
+		foreach ( $langPairs as $langPair ) {
+			$constructedLangPairs[ $langPair->from ][] = $langPair->to;
+		}
+
+		return $constructedLangPairs;
+	}
+}

wp-content/plugins/sitepress-multilingual-cms/classes/icl-to-ate-migration/endpoints/Translators/Save.php

@@ -0,0 +1,97 @@
+<?php
+
+namespace WPML\ICLToATEMigration\Endpoints\Translators;
+
+use WPML\Ajax\IHandler;
+use WPML\Collect\Support\Collection;
+use WPML\FP\Either;
+
+use WPML\FP\Fns;
+use WPML\FP\Left;
+use WPML\FP\Lst;
+use WPML\FP\Obj;
+use WPML\FP\Logic;
+use WPML\Element\API\Languages;
+
+use WPML\FP\Right;
+use function WPML\Container\make;
+use function WPML\FP\invoke;
+use function WPML\FP\partialRight;
+use WPML\TranslationRoles\SaveUser;
+use function WPML\FP\pipe;
+
+class Save extends SaveUser {
+
+	const ERROR_MESSAGE_TRANSLATORS 	= 'There was an error when saving the following translators:';
+	const SUCCESS_MESSAGE_TRANSLATORS 	= 'The translators were saved successfully.';
+
+	/**
+	 * @inheritDoc
+	 */
+	public function run( Collection $data ) {
+		return Either::of( $data )
+			->map( Fns::map( $this->createTranslator() ) )
+			->chain( $this->syncAteIfRequired() )
+			->chain( $this->handleErrors() );
+	}
+
+	/**
+	 * Creates a translator and returns either an error or the translator email.
+	 *
+	 * @return callable(Collection):Either
+	 */
+	private function createTranslator() {
+		return function( $translator ) {
+			$handleError = function($error) use ($translator) {
+				return [ 'translator' => $translator, 'error' => $error ];
+			};
+			$translator = \wpml_collect( $translator );
+
+			return self::getUser( \wpml_collect( $translator ) )
+				->map( Fns::tap( invoke( 'add_cap' )->with( \WPML_Translator_Role::CAPABILITY ) ) )
+				->map( Obj::prop( 'ID' ) )
+				->map( Fns::tap( partialRight( [ make( \WPML_Language_Pair_Records::class ), 'store_active' ], $translator->get( 'languagePairs' ) ) ) )
+				->bimap( $handleError, Fns::always( $translator->get( 'user' )['email'] ) );
+		};
+	}
+
+	/**
+	 * Synchronize ATE translators if one of the results was added.
+	 *
+	 * @return callable(Either[]):Either[]
+	 */
+	private function syncAteIfRequired() {
+		return function( $translatorResults ) {
+			foreach( $translatorResults as $translatorResult ) {
+				if ( $translatorResult instanceof Right ) {
+					do_action( 'wpml_update_translator' );
+					break;
+				}
+			}
+			return $translatorResults;
+		};
+	}
+
+	/**
+	 * If any error happened, it prepares a message and the translators that failed.
+	 *
+	 * @return callable(Either[]):Either
+	 */
+	private function handleErrors() {
+		return function( $translatorResults ) {
+			$getErrorMsg     = pipe( invoke( 'coalesce' )->with( Fns::identity(), Fns::identity() ), invoke( 'get' ) );
+			$getErrorDetails = pipe( Fns::filter( Fns::isLeft() ), Fns::map( $getErrorMsg ) );
+			$errorDetails    = $getErrorDetails( $translatorResults );
+
+			if ( count( $errorDetails ) ) {
+				return Either::left( [
+					'message' => __( self::ERROR_MESSAGE_TRANSLATORS, 'sitepress' ),
+					'details' => $errorDetails
+				] );
+			}
+			return Either::right( __( self::SUCCESS_MESSAGE_TRANSLATORS, 'sitepress' ) );
+		};
+	}
+
+}
+