From 5d80331afe2dd595e889e3607cb4b5a72207957a Mon Sep 17 00:00:00 2001
From: Pierre Gauthier <pigau@smile.fr>
Date: Wed, 4 Dec 2024 15:59:54 +0100
Subject: [PATCH] Add attribute in autocomplete

---
 src/Resources/config/services/search.yml      |  8 +++
 .../templates/search-autocomplete.html        | 17 +++++
 .../layouts/default/config/jsmodules.yml      |  3 +
 src/Search/Autocomplete/Attribute.php         | 68 +++++++++++++++++++
 src/Search/SearchEngine.php                   |  7 +-
 src/Service/ContextProvider.php               | 12 ++++
 6 files changed, 113 insertions(+), 2 deletions(-)
 create mode 100644 src/Search/Autocomplete/Attribute.php

diff --git a/src/Resources/config/services/search.yml b/src/Resources/config/services/search.yml
index 510dbe1..aebd378 100644
--- a/src/Resources/config/services/search.yml
+++ b/src/Resources/config/services/search.yml
@@ -94,3 +94,11 @@ services:
             - '@Gally\OroPlugin\Config\ConfigManager'
         tags:
             - { name: kernel.event_listener, event: Oro\Bundle\ProductBundle\Event\ProcessAutocompleteDataEvent, method: onProcessAutocompleteData }
+
+    Gally\OroPlugin\Search\Autocomplete\Attribute:
+        arguments:
+            - '@Gally\OroPlugin\Service\ContextProvider'
+            - '@Gally\OroPlugin\Config\ConfigManager'
+            - '@router'
+        tags:
+            - { name: kernel.event_listener, event: Oro\Bundle\ProductBundle\Event\ProcessAutocompleteDataEvent, method: onProcessAutocompleteData }
diff --git a/src/Resources/public/default/templates/search-autocomplete.html b/src/Resources/public/default/templates/search-autocomplete.html
index 835d76f..bcd5c4f 100644
--- a/src/Resources/public/default/templates/search-autocomplete.html
+++ b/src/Resources/public/default/templates/search-autocomplete.html
@@ -62,6 +62,23 @@
                     </a>
                 </li>
                 <% }) %>
+                <%_.each(attributes, function(attribute) { %>
+                <li class="search-autocomplete__item" role="option">
+                    <a href="<%- attribute.url %>" class="search-autocomplete-category">
+                        <div class="search-autocomplete-category__image">
+                            <span class="fa-tag fa--no-offset"></span>
+                        </div>
+                        <div class="search-autocomplete-category__info">
+                            <div class="search-autocomplete-category__head">
+                                <%= attribute.field %>
+                            </div>
+                            <div class="search-autocomplete-category__body">
+                                <%= attribute.label %>
+                            </div>
+                        </div>
+                    </a>
+                </li>
+                <% }) %>
                 <li class="search-autocomplete__item" role="option">
                     <button class="btn btn--link btn--no-offset search-autocomplete__submit"
                             type="submit"
diff --git a/src/Resources/views/layouts/default/config/jsmodules.yml b/src/Resources/views/layouts/default/config/jsmodules.yml
index 5d3b11a..bd17933 100644
--- a/src/Resources/views/layouts/default/config/jsmodules.yml
+++ b/src/Resources/views/layouts/default/config/jsmodules.yml
@@ -4,3 +4,6 @@ aliases:
 dynamic-imports:
     orodatagrid:
         - oro/filter/gally-filter
+map:
+    '*':
+        oroproduct/templates/search-autocomplete.html: gallyoro/default/templates/search-autocomplete.html
diff --git a/src/Search/Autocomplete/Attribute.php b/src/Search/Autocomplete/Attribute.php
new file mode 100644
index 0000000..39eede2
--- /dev/null
+++ b/src/Search/Autocomplete/Attribute.php
@@ -0,0 +1,68 @@
+<?php
+/**
+ * DISCLAIMER
+ *
+ * Do not edit or add to this file if you wish to upgrade Gally to newer versions in the future.
+ *
+ * @package   Gally
+ * @author    Gally Team <elasticsuite@smile.fr>
+ * @copyright 2024-present Smile
+ * @license   Open Software License v. 3.0 (OSL-3.0)
+ */
+
+declare(strict_types=1);
+
+namespace Gally\OroPlugin\Search\Autocomplete;
+
+use Gally\OroPlugin\Config\ConfigManager as GallyConfigManager;
+use Gally\OroPlugin\Service\ContextProvider;
+use Oro\Bundle\ProductBundle\Event\ProcessAutocompleteDataEvent;
+use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
+
+/**
+ * Adds category aggregation to product autocomplete.
+ */
+class Attribute
+{
+    public function __construct(
+        private ContextProvider $contextProvider,
+        private GallyConfigManager $gallyConfigManager,
+        private UrlGeneratorInterface $urlGenerator,
+    ) {
+    }
+
+    public function onProcessAutocompleteData(ProcessAutocompleteDataEvent $event): void
+    {
+        $websiteId = $this->contextProvider->getCurrentWebsite()->getId();
+        if (!$this->gallyConfigManager->isGallyEnabled($websiteId)) {
+            return;
+        }
+
+        $searchQuery = $this->contextProvider->getRequest()->getSearchQuery();
+        $attributeData = [];
+        foreach ($this->contextProvider->getResponse()->getAggregations() as $aggregationData) {
+            foreach ($aggregationData['options'] as $option) {
+                $params = [
+                    'f' => [$aggregationData['field'] => ['value' => [$option['value']]]],
+                    'g' => ['search' => $searchQuery],
+                ];
+                $url = $this->urlGenerator->generate(
+                    'oro_product_frontend_product_search',
+                    [
+                        'search' => $searchQuery,
+                        'grid' => ['frontend-product-search-grid' => http_build_query($params)],
+                    ]
+                );
+                $attributeData[] = [
+                    'field' => $aggregationData['label'],
+                    'label' => $option['label'],
+                    'url' => $url,
+                ];
+            }
+        }
+
+        $data = $event->getData();
+        $data['attributes'] = $attributeData;
+        $event->setData($data);
+    }
+}
diff --git a/src/Search/SearchEngine.php b/src/Search/SearchEngine.php
index fd1dfcc..e5bec6e 100644
--- a/src/Search/SearchEngine.php
+++ b/src/Search/SearchEngine.php
@@ -57,9 +57,12 @@ protected function doSearch(Query $query, array $context = [])
     {
         $request = $this->requestBuilder->build($query, $context);
         $response = $this->searchManager->search($request);
-        $this->contextProvider->setResponse($response);
-
         $results = [];
+        if ('product' === $request->getMetadata()->getEntity()) {
+            $this->contextProvider->setRequest($request);
+            $this->contextProvider->setResponse($response);
+        }
+
         foreach ($response->getCollection() as $item) {
             $item['id'] = (int) basename($item['id']);
 
diff --git a/src/Service/ContextProvider.php b/src/Service/ContextProvider.php
index 645a569..b4fb6b2 100644
--- a/src/Service/ContextProvider.php
+++ b/src/Service/ContextProvider.php
@@ -16,6 +16,7 @@
 
 use Gally\OroPlugin\Indexer\Provider\CatalogProvider;
 use Gally\Sdk\Entity\LocalizedCatalog;
+use Gally\Sdk\GraphQl\Request;
 use Gally\Sdk\GraphQl\Response;
 use Oro\Bundle\ConfigBundle\Config\ConfigManager;
 use Oro\Bundle\LocaleBundle\Entity\Localization;
@@ -28,6 +29,7 @@
 class ContextProvider
 {
     private bool $isGallyContext = false;
+    private Request $request;
     private Response $response;
     private ?string $priceFilterUnit = null;
     private bool $isAutocompleteContext = false;
@@ -77,6 +79,16 @@ public function getCurrentContentNode(): ?ContentNodeInterface
         return $isCategoryNode ? $this->requestWebContentVariantProvider->getContentVariant()?->getNode() : null;
     }
 
+    public function getRequest(): Request
+    {
+        return $this->request;
+    }
+
+    public function setRequest(Request $request): void
+    {
+        $this->request = $request;
+    }
+
     public function getResponse(): Response
     {
         return $this->response;