1 files changed, 59 insertions, 0 deletions
diff --git a/www/wiki/extensions/Scribunto/includes/engines/LuaCommon/MessageLibrary.php b/www/wiki/extensions/Scribunto/includes/engines/LuaCommon/MessageLibrary.php
new file mode 100644
index 00000000..4d9edbd3
--- /dev/null
+++ b/www/wiki/extensions/Scribunto/includes/engines/LuaCommon/MessageLibrary.php
@@ -0,0 +1,59 @@
+<?php
+
+// @codingStandardsIgnoreLine Squiz.Classes.ValidClassName.NotCamelCaps
+class Scribunto_LuaMessageLibrary extends Scribunto_LuaLibraryBase {
+	function register() {
+		$lib = [
+			'plain' => [ $this, 'messagePlain' ],
+			'check' => [ $this, 'messageCheck' ],
+		];
+
+		// Get the correct default language from the parser
+		if ( $this->getParser() ) {
+			$lang = $this->getParser()->getTargetLanguage();
+		} else {
+			global $wgContLang;
+			$lang = $wgContLang;
+		}
+
+		return $this->getEngine()->registerInterface( 'mw.message.lua', $lib, [
+			'lang' => $lang->getCode(),
+		] );
+	}
+
+	private function makeMessage( $data, $setParams ) {
+		if ( isset( $data['rawMessage'] ) ) {
+			$msg = new RawMessage( $data['rawMessage'] );
+		} else {
+			$msg = Message::newFallbackSequence( $data['keys'] );
+		}
+		$msg->inLanguage( $data['lang'] )
+			->useDatabase( $data['useDB'] );
+		if ( $setParams ) {
+			$msg->params( array_values( $data['params'] ) );
+		}
+		return $msg;
+	}
+
+	function messagePlain( $data ) {
+		try {
+			$msg = $this->makeMessage( $data, true );
+			return [ $msg->plain() ];
+		} catch ( MWException $ex ) {
+			throw new Scribunto_LuaError( "msg:plain() failed (" . $ex->getMessage() . ")" );
+		}
+	}
+
+	function messageCheck( $what, $data ) {
+		if ( !in_array( $what, [ 'exists', 'isBlank', 'isDisabled' ] ) ) {
+			throw new Scribunto_LuaError( "invalid what for 'messageCheck'" );
+		}
+
+		try {
+			$msg = $this->makeMessage( $data, false );
+			return [ call_user_func( [ $msg, $what ] ) ];
+		} catch ( MWException $ex ) {
+			throw new Scribunto_LuaError( "msg:$what() failed (" . $ex->getMessage() . ")" );
+		}
+	}
+}