add params, fields and highlight to QueryBuilder

Author: igoooor

src/QueryBuilder.php

@@ -25,8 +25,14 @@ class QueryBuilder
 
     private ?QueryInterface $query = null;
 
+    private ?array $fields = null;
+
+    private ?array $highlight = null;
+
     private ?QueryInterface $postFilter = null;
 
+    private array $params = [];
+
     public function setSource(array|bool|string|null $source): self
     {
         $this->source = $source;
@@ -69,11 +75,33 @@ public function setPostFilter(QueryInterface $query): self
         return $this;
     }
 
+    public function setFields(?array $fields): self
+    {
+        $this->fields = $fields;
+
+        return $this;
+    }
+
+    public function setHighlight(?array $highlight): self
+    {
+        $this->highlight = $highlight;
+
+        return $this;
+    }
+
+    public function setParams(array $params): self
+    {
+        $this->params = $params;
+
+        return $this;
+    }
+
     public function build(): array
     {
-        $query = [
-            'body' => [],
-        ];
+        $query = $this->params;
+        if (!isset($query['body'])) {
+            $query['body'] = [];
+        }
 
         if (null !== $this->index) {
             $query['index'] = $this->index;
@@ -99,6 +127,14 @@ public function build(): array
             $query['body']['post_filter'] = $this->postFilter->build();
         }
 
+        if (null !== $this->fields) {
+            $query['body']['fields'] = $this->fields;
+        }
+
+        if (null !== $this->highlight) {
+            $query['body']['highlight'] = $this->highlight;
+        }
+
         $this->buildSortTo($query['body'])
             ->buildAggregationsTo($query['body'])
             ->buildCollapseTo($query['body']);