1 files changed, 224 insertions, 42 deletions

diff --git a/MLEB/Translate/utils/MessageUpdateJob.php b/MLEB/Translate/utils/MessageUpdateJob.php
index 1cc0ff6a..8d4ff35a 100644
--- a/MLEB/Translate/utils/MessageUpdateJob.php
+++ b/MLEB/Translate/utils/MessageUpdateJob.php
@@ -8,17 +8,56 @@
  * @license GPL-2.0-or-later
  */
 
+use MediaWiki\Extensions\Translate\Jobs\GenericTranslateJob;
+use MediaWiki\Extensions\Translate\SystemUsers\FuzzyBot;
+use MediaWiki\Extensions\Translate\Utilities\TranslateReplaceTitle;
+
 /**
  * Job for updating translation pages when translation or message definition changes.
  *
  * @ingroup JobQueue
  */
-class MessageUpdateJob extends Job {
+class MessageUpdateJob extends GenericTranslateJob {
+	/**
+	 * Create a normal message update job without a rename process
+	 * @param Title $target
+	 * @param string $content
+	 * @param bool $fuzzy
+	 * @return MessageUpdateJob
+	 */
 	public static function newJob( Title $target, $content, $fuzzy = false ) {
 		$params = [
 			'content' => $content,
 			'fuzzy' => $fuzzy,
 		];
+
+		$job = new self( $target, $params );
+
+		return $job;
+	}
+
+	/**
+	 * Create a message update job containing a rename process
+	 * @param Title $target Target message being modified
+	 * @param string $targetStr Target string
+	 * @param string $replacement Replacement string
+	 * @param bool $fuzzy Whether to fuzzy the message
+	 * @param string $content Content of the source language
+	 * @param array $otherLangContents Content to be updated for other languages
+	 * @return MessageUpdateJob
+	 */
+	public static function newRenameJob(
+		Title $target, $targetStr, $replacement, $fuzzy, $content, $otherLangContents = []
+	) {
+		$params = [
+			'target' => $targetStr,
+			'replacement' => $replacement,
+			'fuzzy' => $fuzzy,
+			'rename' => 'rename',
+			'content' => $content,
+			'otherLangs' => $otherLangContents
+		];
+
 		$job = new self( $target, $params );
 
 		return $job;
@@ -30,70 +69,213 @@ class MessageUpdateJob extends Job {
 	 */
 	public function __construct( $title, $params = [] ) {
 		parent::__construct( __CLASS__, $title, $params );
-		$this->params = $params;
 	}
 
 	public function run() {
-		global $wgTranslateDocumentationLanguageCode;
-
-		$title = $this->title;
 		$params = $this->params;
 		$user = FuzzyBot::getUser();
 		$flags = EDIT_FORCE_BOT;
+		$isRename = $params['rename'] ?? false;
+		$isFuzzy = $params['fuzzy'] ?? false;
+		$otherLangs = $params['otherLangs'] ?? [];
+
+		if ( $isRename ) {
+			$this->title = $this->handleRename( $params['target'], $params['replacement'], $user );
+			if ( $this->title === null ) {
+				// There was a failure, return true, but don't proceed further.
+				$this->logWarning(
+					'Rename process could not find the source title.',
+					[
+						'replacement' => $params['replacement'],
+						'target' => $params['target']
+					]
+				);
+				return true;
+			}
+		}
 
+		$title = $this->title;
 		$wikiPage = WikiPage::factory( $title );
 		$summary = wfMessage( 'translate-manage-import-summary' )
 			->inContentLanguage()->plain();
 		$content = ContentHandler::makeContent( $params['content'], $title );
-		$wikiPage->doEditContent( $content, $summary, $flags, false, $user );
+		$editStatus = $wikiPage->doEditContent( $content, $summary, $flags, false, $user );
+		if ( !$editStatus->isOK() ) {
+			$this->logError(
+				'Failed to update content for source message',
+				[
+					'content' => $content,
+					'errors' => $editStatus->getErrors()
+				]
+			);
+		}
 
-		// NOTE: message documentation is excluded from fuzzying!
-		if ( $params['fuzzy'] ) {
-			$handle = new MessageHandle( $title );
-			$key = $handle->getKey();
+		if ( $isRename ) {
+			// Update other language content if present.
+			$this->processTranslationChanges(
+				$otherLangs, $params['replacement'], $params['namespace'], $summary, $flags, $user
+			);
+		}
 
-			$languages = TranslateUtils::getLanguageNames( 'en' );
-			unset( $languages[$wgTranslateDocumentationLanguageCode] );
-			$languages = array_keys( $languages );
+		if ( $isFuzzy ) {
+			$this->handleFuzzy( $title );
+		}
 
-			$dbw = wfGetDB( DB_MASTER );
-			$fields = [ 'page_id', 'page_latest' ];
-			$conds = [ 'page_namespace' => $title->getNamespace() ];
+		return true;
+	}
 
-			$pages = [];
-			foreach ( $languages as $code ) {
-				$otherTitle = Title::makeTitleSafe( $title->getNamespace(), "$key/$code" );
-				$pages[$otherTitle->getDBkey()] = true;
-			}
-			unset( $pages[$title->getDBkey()] );
-			if ( $pages === [] ) {
-				return true;
-			}
+	/**
+	 * Handles renames
+	 * @param string $target
+	 * @param string $replacement
+	 * @param User $user
+	 * @return Title|null
+	 */
+	private function handleRename( $target, $replacement, User $user ) {
+		$newSourceTitle = null;
 
-			$conds['page_title'] = array_keys( $pages );
+		$sourceMessageHandle = new MessageHandle( $this->title );
+		$movableTitles = TranslateReplaceTitle::getTitlesForMove( $sourceMessageHandle, $replacement );
 
-			$res = $dbw->select( 'page', $fields, $conds, __METHOD__ );
-			$inserts = [];
-			foreach ( $res as $row ) {
-				$inserts[] = [
-					'rt_type' => RevTag::getType( 'fuzzy' ),
-					'rt_page' => $row->page_id,
-					'rt_revision' => $row->page_latest,
-				];
+		if ( $movableTitles === [] ) {
+			$this->logError(
+				'No moveable titles found with target text.',
+				[
+					'title' => $this->title->getPrefixedText(),
+					'replacement' => $replacement,
+					'target' => $target
+				]
+			);
+			return null;
+		}
+
+		$renameSummary = wfMessage( 'translate-manage-import-rename-summary' )
+			->inContentLanguage()->plain();
+
+		/**
+		 * @var Title[] $movableTitles
+		 */
+		foreach ( $movableTitles as $mTitle ) {
+			/**
+			 * @var Title $sourceTitle
+			 * @var Title $replacementTitle
+			 */
+			[ $sourceTitle, $replacementTitle ] = $mTitle;
+			$mv = new MovePage( $sourceTitle, $replacementTitle );
+
+			$status = $mv->move( $user, $renameSummary, false );
+			if ( !$status->isOK() ) {
+				$this->logError(
+					'Error moving message',
+					[
+						'target' => $sourceTitle->getPrefixedText(),
+						'replacement' => $replacementTitle->getPrefixedText(),
+						'errors' => $status->getErrors()
+					]
+				);
 			}
 
-			if ( $inserts === [] ) {
-				return true;
+			[ , $targetCode ] = TranslateUtils::figureMessage( $replacementTitle->getText() );
+			if ( !$newSourceTitle && $sourceMessageHandle->getCode() === $targetCode ) {
+				$newSourceTitle = $replacementTitle;
 			}
+		}
 
-			$dbw->replace(
-				'revtag',
-				[ [ 'rt_type', 'rt_page', 'rt_revision' ] ],
-				$inserts,
-				__METHOD__
+		if ( $newSourceTitle ) {
+			return $newSourceTitle;
+		} else {
+			// This means that the old source Title was never moved
+			// which is not possible but handle it.
+			$this->logError(
+				'Source title was not in the list of moveable titles.',
+				[ 'title' => $this->title->getPrefixedText() ]
 			);
 		}
+	}
 
-		return true;
+	/**
+	 * Handles fuzzying. Message documentation and the source language are excluded from
+	 * fuzzying. The source language is the identified via the $title parameter
+	 * @param Title $title
+	 */
+	private function handleFuzzy( Title $title ) {
+		global $wgTranslateDocumentationLanguageCode;
+		$handle = new MessageHandle( $title );
+
+		$languages = TranslateUtils::getLanguageNames( 'en' );
+
+		// Don't fuzzy the message documentation
+		unset( $languages[$wgTranslateDocumentationLanguageCode] );
+		$languages = array_keys( $languages );
+
+		$dbw = wfGetDB( DB_MASTER );
+		$fields = [ 'page_id', 'page_latest' ];
+		$conds = [ 'page_namespace' => $title->getNamespace() ];
+
+		$pages = [];
+		foreach ( $languages as $code ) {
+			$otherTitle = $handle->getTitleForLanguage( $code );
+			$pages[$otherTitle->getDBkey()] = true;
+		}
+
+		// Unset to ensure that the source language is not fuzzied
+		unset( $pages[$title->getDBkey()] );
+
+		if ( $pages === [] ) {
+			return;
+		}
+
+		$conds['page_title'] = array_keys( $pages );
+
+		$res = $dbw->select( 'page', $fields, $conds, __METHOD__ );
+		$inserts = [];
+		foreach ( $res as $row ) {
+			$inserts[] = [
+				'rt_type' => RevTag::getType( 'fuzzy' ),
+				'rt_page' => $row->page_id,
+				'rt_revision' => $row->page_latest,
+			];
+		}
+
+		if ( $inserts === [] ) {
+			return;
+		}
+
+		$dbw->replace(
+			'revtag',
+			[ [ 'rt_type', 'rt_page', 'rt_revision' ] ],
+			$inserts,
+			__METHOD__
+		);
+	}
+
+	/**
+	 * Updates the translation unit pages in non-source languages.
+	 * @param array $langChanges
+	 * @param string $baseTitle
+	 * @param int $groupNamespace
+	 * @param string $summary
+	 * @param int $flags
+	 * @param User $user
+	 */
+	private function processTranslationChanges(
+		array $langChanges, $baseTitle, $groupNamespace, $summary, $flags, User $user
+	) {
+		foreach ( $langChanges as $code => $contentStr ) {
+			$titleStr = TranslateUtils::title( $baseTitle, $code, $groupNamespace );
+			$title = Title::newFromText( $titleStr, $groupNamespace );
+			$wikiPage = WikiPage::factory( $title );
+			$content = ContentHandler::makeContent( $contentStr, $title );
+			$status = $wikiPage->doEditContent( $content, $summary, $flags, false, $user );
+			if ( !$status->isOK() ) {
+				$this->logError(
+					'Failed to update content for non-source message',
+					[
+						'title' => $title->getPrefixedText(),
+						'errors' => $status->getErrors()
+					]
+				);
+			}
+		}
 	}
 }