first commit

wp-content/plugins/sitepress-multilingual-cms/addons/wpml-page-builders/classes/Shared/AutoUpdate/Hooks.php

@@ -0,0 +1,153 @@
+<?php
+
+namespace WPML\PB\AutoUpdate;
+
+use WPML\FP\Fns;
+use WPML\FP\Logic;
+use WPML\FP\Lst;
+use WPML\FP\Maybe;
+use WPML\FP\Relation;
+use WPML\PB\Shutdown\Hooks as ShutdownHooks;
+use function WPML\FP\invoke;
+use function WPML\FP\partialRight;
+use function WPML\FP\pipe;
+
+class Hooks implements \IWPML_Backend_Action, \IWPML_Frontend_Action, \IWPML_DIC_Action {
+
+	const HASH_SEP = '-';
+
+	/** @var \WPML_PB_Integration $pbIntegration */
+	private $pbIntegration;
+
+	/** @var \WPML_Translation_Element_Factory $elementFactory */
+	private $elementFactory;
+
+	/** @var \WPML_Page_Builders_Page_Built $pageBuilt */
+	private $pageBuilt;
+
+	/** @var array $translationStatusesUpdaters */
+	private $translationStatusesUpdaters = [];
+
+	public function __construct(
+		\WPML_PB_Integration $pbIntegration,
+		\WPML_Translation_Element_Factory $elementFactory,
+		\WPML_Page_Builders_Page_Built $pageBuilt
+	) {
+		$this->pbIntegration  = $pbIntegration;
+		$this->elementFactory = $elementFactory;
+		$this->pageBuilt      = $pageBuilt;
+	}
+
+	public function add_hooks() {
+		if ( Settings::isEnabled() && $this->isTmLoaded() ) {
+			add_filter( 'wpml_tm_delegate_translation_statuses_update', [ $this, 'enqueueTranslationStatusUpdate'], 10, 3 );
+			add_filter( 'wpml_tm_post_md5_content', [ $this, 'getMd5ContentFromPackageStrings' ], 10, 2 );
+			add_action( 'shutdown', [ $this, 'afterRegisterAllStringsInShutdown' ], ShutdownHooks::PRIORITY_REGISTER_STRINGS + 1 );
+		}
+	}
+
+	public function isTmLoaded() {
+		return defined( 'WPML_TM_VERSION' );
+	}
+
+	/**
+	 * @param bool     $isDelegated
+	 * @param int      $originalPostId
+	 * @param callable $statusesUpdater
+	 *
+	 * @return bool
+	 */
+	public function enqueueTranslationStatusUpdate( $isDelegated, $originalPostId, $statusesUpdater ) {
+		$this->translationStatusesUpdaters[ $originalPostId ] = $statusesUpdater;
+		return true;
+	}
+
+	/**
+	 * @param string   $content
+	 * @param \WP_Post $post
+	 *
+	 * @return string
+	 */
+	public function getMd5ContentFromPackageStrings( $content, $post ) {
+		// $joinPackageStringHashes :: \WPML_Package → string
+		$joinPackageStringHashes = pipe(
+			invoke( 'get_package_strings' )->with( true ),
+			Lst::pluck( 'value' ),
+			Lst::sort( Relation::gt() ),
+			Lst::join( self::HASH_SEP )
+		);
+
+		return Maybe::of( $post->ID )
+			->map( [ self::class, 'getPackages' ] )
+			->map( Fns::map( $joinPackageStringHashes ) )
+			->filter()
+			->map( Lst::join( self::HASH_SEP ) )
+			->getOrElse( $content );
+	}
+
+	/**
+	 * @param int $postId
+	 *
+	 * @return \WPML_Package[]
+	 */
+	public static function getPackages( $postId ) {
+		return apply_filters( 'wpml_st_get_post_string_packages', [], $postId );
+	}
+
+	/**
+	 * We need to update translation statuses after string registration
+	 * to make sure we build the content hash with the new strings.
+	 */
+	public function afterRegisterAllStringsInShutdown() {
+		if ( $this->translationStatusesUpdaters ) {
+			do_action( 'wpml_cache_clear' );
+
+			foreach ( $this->translationStatusesUpdaters as $originalPostId => $translationStatusesUpdater ) {
+				if ( \get_post( $originalPostId ) ) {
+					call_user_func( $translationStatusesUpdater );
+					$this->resaveTranslations( $originalPostId );
+				}
+			}
+		}
+	}
+
+	/**
+	 * @param int $postId
+	 */
+	private function resaveTranslations( $postId ) {
+		if ( ! $this->isPageBuilder( $postId ) ) {
+			return;
+		}
+
+		// $ifOriginal :: \WPML_Post_Element → bool
+		$ifOriginal = pipe( invoke( 'get_source_language_code' ), Logic::not() );
+
+		// $ifCompleted :: \WPML_Post_Element → bool
+		$ifCompleted = pipe( [ TranslationStatus::class, 'get' ], Relation::equals( ICL_TM_COMPLETE ) );
+
+		// $resaveElement :: \WPML_Post_Element → null
+		$resaveElement = \WPML\FP\Fns::unary( partialRight( [ $this->pbIntegration, 'resave_post_translation_in_shutdown' ], false ) );
+
+		wpml_collect( $this->elementFactory->create_post( $postId )->get_translations() )
+			->reject( $ifOriginal )
+			->filter( $ifCompleted )
+			->each( $resaveElement );
+	}
+
+	/**
+	 * @param int $postId
+	 *
+	 * @return bool
+	 */
+	private function isPageBuilder( $postId ) {
+		$isPbPostWithoutStrings = function( $postId ) {
+			$post = get_post( $postId );
+
+			return $post instanceof \WP_Post
+			       && $this->pageBuilt->is_page_builder_page( $post );
+		};
+
+		return self::getPackages( $postId )
+			|| $isPbPostWithoutStrings( $postId );
+	}
+}

wp-content/plugins/sitepress-multilingual-cms/addons/wpml-page-builders/classes/Shared/AutoUpdate/Settings.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace WPML\PB\AutoUpdate;
+
+class Settings {
+
+	/**
+	 * This is part of the "Translation Auto-Update" feature
+	 * The "Translation Auto-Update" feature will be released in the next major version.
+	 * We need a way for users to allow disabling it quickly, if necessary.
+	 *
+	 * @return bool
+	 */
+	public static function isEnabled() {
+		if ( defined( 'WPML_TRANSLATION_AUTO_UPDATE_ENABLED' ) ) {
+			return (bool) constant( 'WPML_TRANSLATION_AUTO_UPDATE_ENABLED' );
+		}
+
+		return true;
+	}
+}

wp-content/plugins/sitepress-multilingual-cms/addons/wpml-page-builders/classes/Shared/AutoUpdate/TranslationStatus.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace WPML\PB\AutoUpdate;
+
+use WPML\FP\Maybe;
+use WPML_Post_Element;
+use function WPML\Container\make;
+use function WPML\FP\invoke;
+
+class TranslationStatus {
+
+	/**
+	 * @param WPML_Post_Element $element
+	 *
+	 * @return int|null
+	 */
+	public static function get( WPML_Post_Element $element ) {
+		return Maybe::fromNullable( make( '\WPML_TM_Translation_Status' ) )
+			->map( invoke( 'filter_translation_status' )->with( null, $element->get_trid(), $element->get_language_code() ) )
+			->getOrElse( null );
+	}
+}