Merge pull request #36 from ensi-platform/task-112268

#112268 _source include and exclude

Author: MsNatali

src/Search/SearchQuery.php

@@ -28,6 +28,8 @@ class SearchQuery implements SortableQuery, CollapsibleQuery
     protected ?int $size = null;
     protected ?int $from = null;
     protected array $fields = [];
+    protected array $include = [];
+    protected array $exclude = [];
 
     public function __construct(protected SearchIndex $index)
     {
@@ -120,7 +122,7 @@ protected function execute(
             'from' => $from,
             'query' => $this->boolQuery->toDSL(),
             'track_total_hits' => $totals,
-            '_source' => $source && !$this->fields,
+            '_source' => $this->sourceToDSL($source),
             'fields' => $source && $this->fields ? $this->fields : null,
         ];
 
@@ -140,6 +142,16 @@ protected function execute(
         return $this->index->search(array_filter($dsl));
     }
 
+    protected function sourceToDSL(bool $source): array | bool
+    {
+        return  $source && !$this->fields ?
+            [
+                'include' => $this->include,
+                'exclude' => $this->exclude,
+            ] :
+            false;
+    }
+
     protected function parseHits(array $response): Collection
     {
         return collect(data_get($response, 'hits.hits') ?? []);
@@ -179,6 +191,24 @@ public function take(int $count): static
         return $this;
     }
 
+    public function select(array $include): static
+    {
+        array_map(Assert::stringNotEmpty(...), $include);
+
+        $this->include = $include;
+
+        return $this;
+    }
+
+    public function exclude(array $exclude): static
+    {
+        array_map(Assert::stringNotEmpty(...), $exclude);
+
+        $this->exclude = $exclude;
+
+        return $this;
+    }
+
     public function skip(int $count): static
     {
         Assert::greaterThanEq($count, 0);

tests/Functional/Search/SearchQueryTest.php

@@ -49,6 +49,23 @@ public function testSortBy(): void
         $this->assertDocumentIds([1, 150, 319]);
     }
 
+    public function testSelect(): void
+    {
+        $this->testing->select(['product_id'])->take(1);
+        $result = $this->testing->get();
+
+        $this->assertEquals(1, $result[0]['_source']['product_id']);
+        $this->assertArrayNotHasKey('name', $result[0]['_source']);
+    }
+
+    public function testExclude(): void
+    {
+        $this->testing->exclude(['product_id'])->take(1);
+        $result = $this->testing->get();
+
+        $this->assertArrayNotHasKey('product_id', $result[0]['_source']);
+    }
+
     public function testSortByNested(): void
     {
         $filter = function (BoolQuery $query) {