1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\CatalogSearch\Model\Autocomplete;
use Magento\Search\Model\ResourceModel\Query\Collection;
use Magento\Search\Model\QueryFactory;
use Magento\Search\Model\Autocomplete\DataProviderInterface;
use Magento\Search\Model\Autocomplete\ItemFactory;
use Magento\Framework\App\Config\ScopeConfigInterface as ScopeConfig;
use Magento\Store\Model\ScopeInterface;
/**
* Catalog search auto-complete data provider.
*/
class DataProvider implements DataProviderInterface
{
/**
* Autocomplete limit
*/
const CONFIG_AUTOCOMPLETE_LIMIT = 'catalog/search/autocomplete_limit';
/**
* Query factory
*
* @var QueryFactory
*/
protected $queryFactory;
/**
* Autocomplete result item factory
*
* @var ItemFactory
*/
protected $itemFactory;
/**
* Limit
*
* @var int
*/
protected $limit;
/**
* @param QueryFactory $queryFactory
* @param ItemFactory $itemFactory
* @param ScopeConfig $scopeConfig
*/
public function __construct(
QueryFactory $queryFactory,
ItemFactory $itemFactory,
ScopeConfig $scopeConfig
) {
$this->queryFactory = $queryFactory;
$this->itemFactory = $itemFactory;
$this->limit = (int) $scopeConfig->getValue(
self::CONFIG_AUTOCOMPLETE_LIMIT,
ScopeInterface::SCOPE_STORE
);
}
/**
* @inheritdoc
*/
public function getItems()
{
$collection = $this->getSuggestCollection();
$query = $this->queryFactory->get()->getQueryText();
$result = [];
foreach ($collection as $item) {
$resultItem = $this->itemFactory->create([
'title' => $item->getQueryText(),
'num_results' => $item->getNumResults(),
]);
if ($resultItem->getTitle() == $query) {
array_unshift($result, $resultItem);
} else {
$result[] = $resultItem;
}
}
return ($this->limit) ? array_splice($result, 0, $this->limit) : $result;
}
/**
* Retrieve suggest collection for query
*
* @return Collection
*/
private function getSuggestCollection()
{
return $this->queryFactory->get()->getSuggestCollection();
}
}