wip

src/Feature.php

@@ -25,6 +25,9 @@ class Feature extends \ElasticPress\Feature {
 	/** @var FeatureSupport\RelatedPosts */
 	private $frontendRelatedPosts;
 
+	/** @var FeatureSupport\Autosuggest */
+	private $frontendAutosuggest;
+
 	/** @var Stats\Health */
 	private $statsHealth;
 
@@ -59,6 +62,7 @@ public function __construct(
 		Sync\CLI                    $syncCli,
 		FeatureSupport\Search       $frontendSearch,
 		FeatureSupport\RelatedPosts $frontendRelatedPosts,
+		FeatureSupport\Autosuggest  $frontendAutosuggest,
 		Stats\Health                $statsHealth,
 		Stats\Report                $statsReport
 	) {
@@ -69,6 +73,7 @@ public function __construct(
 		$this->syncCli              = $syncCli;
 		$this->frontendSearch       = $frontendSearch;
 		$this->frontendRelatedPosts = $frontendRelatedPosts;
+		$this->frontendAutosuggest  = $frontendAutosuggest;
 		$this->statsHealth          = $statsHealth;
 		$this->statsReport          = $statsReport;
 
@@ -87,6 +92,7 @@ public function setup() {
 		$this->syncCli->addHooks();
 		$this->frontendSearch->addHooks();
 		$this->frontendRelatedPosts->addHooks();
+		$this->frontendAutosuggest->addHooks();
 		$this->statsHealth->addHooks();
 		$this->statsReport->addHooks();
 	}

src/FeatureSupport/Autosuggest.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace WPML\ElasticPress\FeatureSupport;
+
+use ElasticPress\Features;
+use ElasticPress\Indexables;
+
+use WPML\ElasticPress\Constants;
+use WPML\ElasticPress\FeatureSupport;
+
+class Autosuggest extends FeatureSupport {
+
+	const FEATURE_SLUG = 'autosuggest';
+
+	public function addHooks() {
+		if ( ! $this->isFeatureActive() ) {
+			return;
+		}
+
+		add_filter( 'ep_autosuggest_options', [ $this, 'useIndexByLanguage' ], Constants::LATE_HOOK_PRIORITY );
+	}
+
+	/**
+	 * @param array  $options
+	 *
+	 * @return array
+	 */
+	public function useIndexByLanguage( $options ) {
+		$endpointUrl = $options['endpointUrl'];
+		$defaultIndexName = Indexables::factory()->get( 'post' )->get_index_name();
+		$this->indicesManager->setCurrentIndexLanguage( $this->currentLanguage );
+		$currentLanguageIndexName = Indexables::factory()->get( 'post' )->get_index_name();
+		$this->indicesManager->clearCurrentIndexLanguage();
+		$options['endpointUrl'] = str_replace( $defaultIndexName, $currentLanguageIndexName, $endpointUrl );
+		return $options;
+	}
+
+}

src/Plugin.php

@@ -100,6 +100,11 @@ public static function init() {
 					$indicesManager,
 					$currentLanguage
 				),
+				new FeatureSupport\Autosuggest(
+					$features,
+					$indicesManager,
+					$currentLanguage
+				),
 				new Stats\Health(
 					$indexables,
 					$networkActivated,