1 files changed, 93 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/webservices/CaighdeanWebService.php b/www/wiki/extensions/Translate/webservices/CaighdeanWebService.php
new file mode 100644
index 00000000..cb472d93
--- /dev/null
+++ b/www/wiki/extensions/Translate/webservices/CaighdeanWebService.php
@@ -0,0 +1,93 @@
+<?php
+/**
+ * Contains a class for querying external translation service.
+ *
+ * @file
+ * @author Niklas Laxström
+ * @license GPL-2.0-or-later
+ */
+
+/**
+ * Implements support Caighdean translator api.
+ * @see https://github.com/kscanne/caighdean/blob/master/API.md
+ * @ingroup TranslationWebService
+ * @since 2017.04
+ */
+class CaighdeanWebService extends TranslationWebService {
+	public function getType() {
+		return 'mt';
+	}
+
+	public function mapCode( $code ) {
+		return $code;
+	}
+
+	protected function doPairs() {
+		$pairs = [
+			'gd' => [ 'ga' => true ],
+			'gv' => [ 'ga' => true ],
+		];
+
+		return $pairs;
+	}
+
+	protected function getQuery( $text, $from, $to ) {
+		if ( !isset( $this->config['url'] ) ) {
+			throw new TranslationWebServiceConfigurationException( '`url` not set in configuration' );
+		}
+
+		$text = trim( $text );
+		if ( $text === '' ) {
+			throw new TranslationWebServiceInvalidInputException( 'Input is empty' );
+		}
+
+		$data = wfArrayToCgi( [
+			'foinse' => $from,
+			'teacs' => $text,
+		] );
+
+		// Maximum payload is 16 KiB. Based ont testing 16000 bytes is safe by leaving 224
+		// bytes for other things.
+		if ( strlen( $data ) > 16000 ) {
+			throw new TranslationWebServiceInvalidInputException( 'Input is over 16000 bytes long' );
+		}
+
+		return TranslationQuery::factory( $this->config['url'] )
+			->timeout( $this->config['timeout'] )
+			->postWithData( $data )
+			->attachProcessingInstructions( $text );
+	}
+
+	protected function parseResponse( TranslationQueryResponse $reply ) {
+		$body = $reply->getBody();
+		$response = FormatJson::decode( $body );
+		if ( !is_array( $response ) ) {
+			throw new TranslationWebServiceException( 'Invalid json: ' . serialize( $body ) );
+		}
+
+		$text = '';
+		$originalText = $reply->getQuery()->getProcessingInstructions();
+		foreach ( $response as list( $sourceToken, $targetToken ) ) {
+			$separator = ' ';
+			$pos = strpos( $originalText, $sourceToken );
+			// Try to keep the effects local. If we fail to match at token, we could accidentally
+			// scan very far ahead in the text, find a false match and not find matches for all
+			// of the tokens in the between.
+			if ( $pos !== false && $pos < 50 ) {
+				// Remove the portion of text we have processed. $pos should be zero, unless
+				// we failed to match something earlier.
+				$originalText = substr( $originalText, $pos + strlen( $sourceToken ) );
+				if ( preg_match( '/^\s+/', $originalText, $match ) ) {
+					$separator = $match[ 0 ];
+					$originalText = substr( $originalText, strlen( $separator ) );
+				} else {
+					$separator = '';
+				}
+			}
+
+			$text .= $targetToken . $separator;
+		}
+
+		return $text;
+	}
+}