1 files changed, 135 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/translationaids/TranslationAidDataProvider.php b/www/wiki/extensions/Translate/translationaids/TranslationAidDataProvider.php
new file mode 100644
index 00000000..2f6672ef
--- /dev/null
+++ b/www/wiki/extensions/Translate/translationaids/TranslationAidDataProvider.php
@@ -0,0 +1,135 @@
+<?php
+/**
+ * Translation aid code.
+ *
+ * @file
+ * @author Niklas Laxström
+ * @license GPL-2.0-or-later
+ */
+
+use Wikimedia\Rdbms\IDatabase;
+
+/**
+ * @since 2018.01
+ */
+class TranslationAidDataProvider {
+	private $handle;
+	private $group;
+
+	private $definition;
+	private $translations;
+
+	public function __construct( MessageHandle $handle ) {
+		$this->handle = $handle;
+		$this->group = $handle->getGroup();
+	}
+
+	/**
+	 * Get the message definition. Cached for performance.
+	 *
+	 * @return string
+	 */
+	public function getDefinition() {
+		if ( $this->definition !== null ) {
+			return $this->definition;
+		}
+
+		// Optional performance optimization
+		if ( method_exists( $this->group, 'getMessageContent' ) ) {
+			$this->definition = $this->group->getMessageContent( $this->handle );
+		} else {
+			$this->definition = $this->group->getMessage(
+				$this->handle->getKey(),
+				$this->group->getSourceLanguage()
+			);
+		}
+
+		return $this->definition;
+	}
+
+	/**
+	 * @return Content
+	 */
+	public function getDefinitionContent() {
+		return ContentHandler::makeContent( $this->getDefinition(), $this->handle->getTitle() );
+	}
+
+	/**
+	 * Get the translations in all languages. Cached for performance.
+	 * Fuzzy translation are not included.
+	 *
+	 * @return array Language code => Translation
+	 */
+	public function getGoodTranslations() {
+		if ( $this->translations !== null ) {
+			return $this->translations;
+		}
+
+		$data = self::loadTranslationData( wfGetDB( DB_REPLICA ), $this->handle );
+		$translations = [];
+		$prefixLength = strlen( $this->handle->getTitleForBase()->getDBKey() . '/' );
+
+		foreach ( $data as $page => $translation ) {
+			// Could use MessageHandle here, but that queries the message index.
+			// Instead we can get away with simple string manipulation.
+			$code = substr( $page, $prefixLength );
+			if ( !Language::isKnownLanguageTag( $code ) ) {
+				continue;
+			}
+
+			$translations[ $code ] = $translation;
+		}
+
+		$this->translations = $translations;
+
+		return $translations;
+	}
+
+	private static function loadTranslationData( IDatabase $db, MessageHandle $handle ) {
+		if ( method_exists( 'Revision', 'getQueryInfo' ) ) {
+			$queryInfo = Revision::getQueryInfo( [ 'page', 'text' ] );
+			$tables = $queryInfo[ 'tables' ];
+			$fields = $queryInfo[ 'fields' ];
+			$conds = [];
+			$options = [];
+			$joins = $queryInfo[ 'joins' ];
+		} else {
+			// BC for <= MW 1.31
+			$tables = [ 'page', 'text', 'revision' ];
+			$fields = array_merge(
+				Revision::selectFields(),
+				Revision::selectPageFields(),
+				Revision::selectTextFields()
+			);
+			$conds = [];
+			$options = [];
+			$joins = [
+				'page' => Revision::pageJoinCond(),
+				'text' => [ 'INNER JOIN', [ 'rev_text_id=old_id' ] ]
+			];
+		}
+
+		// The list of pages we want to select, and their latest versions
+		$conds['page_namespace'] = $handle->getTitle()->getNamespace();
+		$base = $handle->getKey();
+		$conds[] = 'page_title ' . $db->buildLike( "$base/", $db->anyString() );
+		$conds[] = 'rev_id=page_latest';
+
+		// For fuzzy tags we also need:
+		$tables[] = 'revtag';
+		$conds[ 'rt_type' ] = null;
+		$joins[ 'revtag' ] = [
+			'LEFT JOIN',
+			[ 'page_id=rt_page', 'page_latest=rt_revision', 'rt_type' => 'fuzzy' ]
+		];
+
+		$rows = $db->select( $tables, $fields, $conds, __METHOD__, $options, $joins );
+
+		$pages = [];
+		foreach ( $rows as $row ) {
+			$pages[$row->page_title] = Revision::getRevisionText( $row );
+		}
+
+		return $pages;
+	}
+}