1 files changed, 80 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/webservices/RESTBaseWebService.php b/www/wiki/extensions/Translate/webservices/RESTBaseWebService.php
new file mode 100644
index 00000000..2ff80c43
--- /dev/null
+++ b/www/wiki/extensions/Translate/webservices/RESTBaseWebService.php
@@ -0,0 +1,80 @@
+<?php
+/**
+ * Contains a class for querying external translation service.
+ *
+ * @file
+ * @author Niklas Laxström
+ * @license GPL-2.0-or-later
+ */
+
+/**
+ * Implements support for cxserver proxied through RESTBase
+ * @ingroup TranslationWebService
+ * @since 2017.10
+ */
+class RESTBaseWebService extends TranslationWebService {
+	public function getType() {
+		return 'mt';
+	}
+
+	protected function mapCode( $code ) {
+		return $code;
+	}
+
+	protected function doPairs() {
+		if ( !isset( $this->config['host'] ) ) {
+			throw new TranslationWebServiceConfigurationException( 'RESTBase host not set' );
+		}
+
+		$pairs = [];
+
+		$url = $this->config['host'] . '/rest_v1/transform/list/tool/mt/';
+		$json = Http::get(
+			$url,
+			[ $this->config['timeout'] ],
+			__METHOD__
+		);
+		$response = FormatJson::decode( $json, true );
+
+		if ( !is_array( $response ) ) {
+			$exception = 'Malformed reply from remote server: ' . $url . ' ' . (string)$json;
+			throw new TranslationWebServiceException( $exception );
+		}
+
+		foreach ( $response['Apertium'] as $source => $targets ) {
+			foreach ( $targets as $target ) {
+				$pairs[$source][$target] = true;
+			}
+		}
+
+		return $pairs;
+	}
+
+	protected function getQuery( $text, $from, $to ) {
+		if ( !isset( $this->config['host'] ) ) {
+			throw new TranslationWebServiceConfigurationException( 'RESTBase host not set' );
+		}
+
+		$text = trim( $text );
+		$text = $this->wrapUntranslatable( $text );
+		$url = $this->config['host'] . "/rest_v1/transform/html/from/$from/to/$to/Apertium";
+
+		return TranslationQuery::factory( $url )
+			->timeout( $this->config['timeout'] )
+			->postWithData( wfArrayToCgi( [ 'html' => $text ] ) );
+	}
+
+	protected function parseResponse( TranslationQueryResponse $reply ) {
+		$body = $reply->getBody();
+
+		$response = FormatJson::decode( $body );
+		if ( !is_object( $response ) ) {
+			throw new TranslationWebServiceException( 'Invalid json: ' . serialize( $body ) );
+		}
+
+		$text = $response->contents;
+		$text = $this->unwrapUntranslatable( $text );
+
+		return trim( $text );
+	}
+}