#107572

Author: MsNatali

src/Suggesting/Request/Concerns/WithMaxEdits.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+use Webmozart\Assert\Assert;
+
+trait WithMaxEdits
+{
+    protected ?int $maxEdits = null;
+
+    public function maxEdits(int $maxEdits): static
+    {
+        Assert::range($maxEdits, 1, 2);
+
+        $this->maxEdits = $maxEdits;
+
+        return $this;
+    }
+
+}

src/Suggesting/Request/Concerns/WithMaxInspections.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithMaxInspections
+{
+    protected ?int $maxInspections = null;
+
+    public function maxInspections(int $maxInspections): static
+    {
+        $this->maxInspections = $maxInspections;
+
+        return $this;
+    }
+}

src/Suggesting/Request/Concerns/WithMaxTermFreq.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithMaxTermFreq
+{
+    protected ?int $maxTermFreq = null;
+
+    public function maxTermFreq(int $maxTermFreq): static
+    {
+        $this->maxTermFreq = $maxTermFreq;
+
+        return $this;
+    }
+}

src/Suggesting/Request/Concerns/WithMinDocFreq.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithMinDocFreq
+{
+    protected ?int $minDocFreq = null;
+
+    public function minDocFreq(int $minDocFreq): static
+    {
+        $this->minDocFreq = $minDocFreq;
+
+        return $this;
+    }
+}

src/Suggesting/Request/Concerns/WithMinWordLength.php

@@ -0,0 +1,16 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithMinWordLength
+{
+    protected ?int $minWordLength = null;
+
+    public function minWordLength(int $minWordLength): static
+    {
+        $this->minWordLength = $minWordLength;
+
+        return $this;
+    }
+
+}

src/Suggesting/Request/Concerns/WithPrefixLength.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithPrefixLength
+{
+    protected ?int $prefixLength = null;
+
+    public function prefixLength(int $prefixLength): static
+    {
+        $this->prefixLength = $prefixLength;
+
+        return $this;
+    }
+}

src/Suggesting/Request/Concerns/WithSize.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+trait WithSize
+{
+    protected ?int $size = null;
+
+    public function size(int $size): static
+    {
+        $this->size = $size;
+
+        return $this;
+    }
+}

src/Suggesting/Request/Concerns/WithSuggestMode.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request\Concerns;
+
+use Ensi\LaravelElasticQuery\Suggesting\Enums\SuggestMode;
+
+trait WithSuggestMode
+{
+    protected ?string $suggestMode = null;
+
+    public function suggestModeMissing(): static
+    {
+        $this->suggestMode = SuggestMode::MISSING;
+
+        return $this;
+    }
+
+    public function suggestModePopular(): static
+    {
+        $this->suggestMode = SuggestMode::POPULAR;
+
+        return $this;
+    }
+
+    public function suggestModeAlways(): static
+    {
+        $this->suggestMode = SuggestMode::ALWAYS;
+
+        return $this;
+    }
+}

src/Suggesting/Request/DirectGenerator.php

@@ -0,0 +1,54 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Suggesting\Request;
+
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithMaxEdits;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithMaxInspections;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithMaxTermFreq;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithMinDocFreq;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithMinWordLength;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithPrefixLength;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithSize;
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithSuggestMode;
+
+class DirectGenerator
+{
+    use WithSuggestMode, WithSize, WithMaxEdits, WithPrefixLength, WithMinWordLength, WithMaxInspections, WithMinDocFreq, WithMaxTermFreq;
+    protected ?string $preFilter = null;
+    protected ?string $postFilter = null;
+
+    public function __construct(protected string $field)
+    {
+    }
+
+    public function toDSL(): array
+    {
+        return array_filter([
+            "field" => $this->field,
+            "size" => $this->size,
+            "suggest_mode" => $this->suggestMode,
+            "max_edits" => $this->maxEdits,
+            "prefix_length" => $this->prefixLength,
+            "min_word_length" => $this->minWordLength,
+            "max_inspections" => $this->maxInspections,
+            "min_doc_freq" => $this->minDocFreq,
+            "max_term_freq" => $this->maxTermFreq,
+            "pre_filter" => $this->preFilter,
+            "post_filter" => $this->postFilter,
+        ]);
+    }
+
+    public function preFilter(string $preFilter): static
+    {
+        $this->preFilter = $preFilter;
+
+        return $this;
+    }
+
+    public function postFilter(string $postFilter): static
+    {
+        $this->postFilter = $postFilter;
+
+        return $this;
+    }
+}

src/Suggesting/Request/PhraseSuggester.php

@@ -2,22 +2,25 @@
 
 namespace Ensi\LaravelElasticQuery\Suggesting\Request;
 
+use Ensi\LaravelElasticQuery\Suggesting\Request\Concerns\WithSize;
 use Webmozart\Assert\Assert;
 
 class PhraseSuggester implements Suggester
 {
+    use WithSize;
+
     // basic phrase suggest api parameters
     protected ?string $text = null;
     protected ?int $gramSize = null;
     protected ?float $realWordErrorLikelihood = null;
     protected ?float $confidence = null;
     protected ?float $maxErrors = null;
     protected ?string $separator = null;
-    protected ?int $size = null;
     protected ?string $analyzer = null;
     protected ?int $shardSize = null;
     protected ?string $highlightPreTag = null;
     protected ?string $highlightPostTag = null;
+    protected array $directGenerators = [];
 
     public function __construct(protected string $name, protected string $field)
     {
@@ -44,9 +47,9 @@ public function toDSL(): array
                     "pre_tag" => $this->highlightPreTag,
                     "post_tag" => $this->highlightPostTag,
                 ]) ?: null,
+                "direct_generator" => array_map(fn(DirectGenerator $g) => $g->toDSL(), $this->directGenerators),
                 // todo collate
                 // todo smoothing models
-                // todo direct_generator
             ]),
         ];
     }
@@ -56,7 +59,7 @@ public function name(): string
         return $this->name;
     }
 
-    public function text(string $text): self
+    public function text(string $text): static
     {
         Assert::stringNotEmpty(trim($text));
 
@@ -65,66 +68,66 @@ public function text(string $text): self
         return $this;
     }
 
-    public function gramSize(int $gramSize): self
+    public function gramSize(int $gramSize): static
     {
         $this->gramSize = $gramSize;
 
         return $this;
     }
 
-    public function realWordErrorLikelihood(float $realWordErrorLikelihood): self
+    public function realWordErrorLikelihood(float $realWordErrorLikelihood): static
     {
         $this->realWordErrorLikelihood = $realWordErrorLikelihood;
 
         return $this;
     }
 
-    public function confidence(float $confidence): self
+    public function confidence(float $confidence): static
     {
         $this->confidence = $confidence;
 
         return $this;
     }
 
-    public function maxErrors(float $maxErrors): self
+    public function maxErrors(float $maxErrors): static
     {
         $this->maxErrors = $maxErrors;
 
         return $this;
     }
 
-    public function separator(string $separator): self
+    public function separator(string $separator): static
     {
         $this->separator = $separator;
 
         return $this;
     }
 
-    public function size(int $size): self
+    public function analyzer(string $analyzer): static
     {
-        $this->size = $size;
+        $this->analyzer = $analyzer;
 
         return $this;
     }
 
-    public function analyzer(string $analyzer): self
+    public function shardSize(int $shardSize): static
     {
-        $this->analyzer = $analyzer;
+        $this->shardSize = $shardSize;
 
         return $this;
     }
 
-    public function shardSize(int $shardSize): self
+    public function highlight(string $highlightPreTag, string $highlightPostTag): static
     {
-        $this->shardSize = $shardSize;
+        $this->highlightPreTag = $highlightPreTag;
+        $this->highlightPostTag = $highlightPostTag;
 
         return $this;
     }
 
-    public function highlight(string $highlightPreTag, string $highlightPostTag): self
+    public function addDirectGenerator(DirectGenerator $directGenerator): static
     {
-        $this->highlightPreTag = $highlightPreTag;
-        $this->highlightPostTag = $highlightPostTag;
+        $this->directGenerators[] = $directGenerator;
 
         return $this;
     }