IDBPR-2554 Add whereMoreLikeThis and addFunctionScore

Author: egmanoylin

src/Concerns/DecoratesBoolQuery.php

@@ -3,10 +3,14 @@
 namespace Ensi\LaravelElasticQuery\Concerns;
 
 use Closure;
+use Ensi\LaravelElasticQuery\Contracts\FunctionScoreOptions;
 use Ensi\LaravelElasticQuery\Contracts\MatchOptions;
+use Ensi\LaravelElasticQuery\Contracts\MoreLikeOptions;
 use Ensi\LaravelElasticQuery\Contracts\MultiMatchOptions;
 use Ensi\LaravelElasticQuery\Contracts\WildcardOptions;
 use Ensi\LaravelElasticQuery\Filtering\BoolQueryBuilder;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\FunctionScoreItem;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\MoreLikeThis;
 use Illuminate\Contracts\Support\Arrayable;
 use Illuminate\Support\Traits\ForwardsCalls;
 
@@ -141,4 +145,22 @@ public function addMustBool(callable $fn): static
 
         return $this;
     }
+
+    public function whereMoreLikeThis(array $fields, MoreLikeThis $likeThis, ?MoreLikeOptions $options = null): static
+    {
+        $this->forwardCallTo($this->boolQuery(), __FUNCTION__, func_get_args());
+
+        return $this;
+    }
+
+    /**
+     * @param array<FunctionScoreItem> $functions
+     * @param ?FunctionScoreOptions $options
+     */
+    public function addFunctionScore(array $functions, ?FunctionScoreOptions $options = null): static
+    {
+        $this->forwardCallTo($this->boolQuery(), __FUNCTION__, func_get_args());
+
+        return $this;
+    }
 }

src/Contracts/BoostMode.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Contracts;
+
+class BoostMode
+{
+    public const MULTIPLY = 'multiply';
+    public const REPLACE = 'replace';
+    public const SUM = 'sum';
+    public const AVG = 'avg';
+    public const MAX = 'max';
+    public const MIN = 'min';
+
+    public static function cases(): array
+    {
+        return [
+            self::MULTIPLY,
+            self::SUM,
+            self::AVG,
+            self::REPLACE,
+            self::MAX,
+            self::MIN,
+        ];
+    }
+}

src/Contracts/FunctionScoreOptions.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Contracts;
+
+use Illuminate\Contracts\Support\Arrayable;
+use Webmozart\Assert\Assert;
+
+class FunctionScoreOptions implements Arrayable
+{
+    public function __construct(private array $options = [])
+    {
+    }
+
+    public static function make(
+        ?string $scoreMode = null,
+        ?string $boostMode = null,
+    ): static {
+        Assert::oneOf($scoreMode, ScoreMode::cases());
+        Assert::oneOf($boostMode, BoostMode::cases());
+
+        return new static(array_filter([
+            'score_mode' => $scoreMode,
+            'boost_mode' => $boostMode,
+        ]));
+    }
+
+    public function toArray(): array
+    {
+        return $this->options;
+    }
+}

src/Contracts/MoreLikeOptions.php

@@ -0,0 +1,30 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Contracts;
+
+use Illuminate\Contracts\Support\Arrayable;
+
+class MoreLikeOptions implements Arrayable
+{
+    public function __construct(private array $options = [])
+    {
+    }
+
+    public static function make(
+        ?int $minTermFreq = null,
+        ?int $maxQueryTerms = null,
+        ?string $minimumShouldMatch = null,
+    ): static {
+
+        return new static(array_filter([
+            'min_term_freq' => $minTermFreq,
+            'max_query_terms' => $maxQueryTerms,
+            'minimum_should_match' => $minimumShouldMatch,
+        ]));
+    }
+
+    public function toArray(): array
+    {
+        return $this->options;
+    }
+}

src/Contracts/ScoreMode.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Contracts;
+
+class ScoreMode
+{
+    public const MULTIPLY = 'multiply';
+    public const SUM = 'sum';
+    public const AVG = 'avg';
+    public const FIRST = 'first';
+    public const MAX = 'max';
+    public const MIN = 'min';
+
+    public static function cases(): array
+    {
+        return [
+            self::MULTIPLY,
+            self::SUM,
+            self::AVG,
+            self::FIRST,
+            self::MAX,
+            self::MIN,
+        ];
+    }
+}

src/Filtering/BoolQueryBuilder.php

@@ -6,10 +6,16 @@
 use Ensi\LaravelElasticQuery\Concerns\SupportsPath;
 use Ensi\LaravelElasticQuery\Contracts\BoolQuery;
 use Ensi\LaravelElasticQuery\Contracts\Criteria;
+use Ensi\LaravelElasticQuery\Contracts\FunctionScoreOptions;
 use Ensi\LaravelElasticQuery\Contracts\MatchOptions;
+use Ensi\LaravelElasticQuery\Contracts\MoreLikeOptions;
 use Ensi\LaravelElasticQuery\Contracts\MultiMatchOptions;
 use Ensi\LaravelElasticQuery\Contracts\WildcardOptions;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\FunctionScoreItem;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\MoreLikeThis;
 use Ensi\LaravelElasticQuery\Filtering\Criterias\Exists;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\FunctionScore;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\MoreLike;
 use Ensi\LaravelElasticQuery\Filtering\Criterias\MultiMatch;
 use Ensi\LaravelElasticQuery\Filtering\Criterias\Nested;
 use Ensi\LaravelElasticQuery\Filtering\Criterias\OneMatch;
@@ -245,6 +251,24 @@ protected function makeWildcard(string $field, string $query, ?WildcardOptions $
         return new Wildcard($this->absolutePath($field), $query, $options ?: new WildcardOptions());
     }
 
+    public function whereMoreLikeThis(array $fields, MoreLikeThis $likeThis, ?MoreLikeOptions $options = null): static
+    {
+        $this->must->add(new MoreLike($fields, $likeThis, $options));
+
+        return $this;
+    }
+
+    /**
+     * @param array<FunctionScoreItem> $functions
+     * @param ?FunctionScoreOptions $options
+     */
+    public function addFunctionScore(array $functions, ?FunctionScoreOptions $options = null): static
+    {
+        $this->should->add(new FunctionScore($functions, $options));
+
+        return $this;
+    }
+
     public function addMustBool(callable $fn): static
     {
         $this->must->add(static::make(builder: $fn));

src/Filtering/Criterias/FunctionScore.php

@@ -0,0 +1,37 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Filtering\Criterias;
+
+use Ensi\LaravelElasticQuery\Contracts\Criteria;
+use Ensi\LaravelElasticQuery\Contracts\FunctionScoreOptions;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\FunctionScoreItem;
+use stdClass;
+use Webmozart\Assert\Assert;
+
+class FunctionScore implements Criteria
+{
+    /**
+     * @param array<FunctionScoreItem> $functions
+     * @param FunctionScoreOptions|null $options
+     */
+    public function __construct(
+        private array $functions,
+        private ?FunctionScoreOptions $options = null,
+    ) {
+        array_map(fn ($function) => Assert::isInstanceOfAny($function, [FunctionScoreItem::class]), $functions);
+    }
+
+    public function toDSL(): array
+    {
+        $body = [
+            'query' => ['match_all' => new stdClass()],
+            'functions' => array_map(fn (FunctionScoreItem $function) => $function->toDSL(), $this->functions),
+        ];
+
+        if ($this->options) {
+            $body = array_merge($this->options->toArray(), $body);
+        }
+
+        return ['function_score' => $body];
+    }
+}

src/Filtering/Criterias/FunctionScoreItem.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Filtering\Criterias;
+
+use Ensi\LaravelElasticQuery\Contracts\Criteria;
+
+class FunctionScoreItem implements Criteria
+{
+    public function __construct(
+        private int $weight,
+        private Criteria $filter,
+    ) {
+    }
+
+    public function toDSL(): array
+    {
+        return [
+            'weight' => $this->weight,
+            'filter' => $this->filter->toDSL(),
+        ];
+    }
+}

src/Filtering/Criterias/MoreLike.php

@@ -0,0 +1,34 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Filtering\Criterias;
+
+use Ensi\LaravelElasticQuery\Contracts\Criteria;
+use Ensi\LaravelElasticQuery\Contracts\MoreLikeOptions;
+use Ensi\LaravelElasticQuery\Filtering\Criterias\MoreLikeThis;
+use Webmozart\Assert\Assert;
+
+class MoreLike implements Criteria
+{
+    public function __construct(
+        private array $fields,
+        private MoreLikeThis $likeThis,
+        private ?MoreLikeOptions $options = null,
+    ) {
+        Assert::minCount($fields, 1);
+        array_map(Assert::stringNotEmpty(...), $fields);
+    }
+
+    public function toDSL(): array
+    {
+        $body = [
+            'fields' => $this->fields,
+            'like' => $this->likeThis->toDSL(),
+        ];
+
+        if ($this->options) {
+            $body = array_merge($this->options->toArray(), $body);
+        }
+
+        return ['more_like_this' => $body];
+    }
+}

src/Filtering/Criterias/MoreLikeThis.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace Ensi\LaravelElasticQuery\Filtering\Criterias;
+
+use Ensi\LaravelElasticQuery\Contracts\Criteria;
+
+class MoreLikeThis implements Criteria
+{
+    private array $this = [];
+
+    public function addId(string $id, ?string $index = null): static
+    {
+        $this->this[] = array_filter([
+            '_id' => $id,
+            '_index' => $index,
+        ]);
+
+        return $this;
+    }
+
+    public function addString(string $token): static
+    {
+        $this->this[] = $token;
+
+        return $this;
+    }
+
+    public function toDSL(): array
+    {
+        return $this->this;
+    }
+}