Implement autocomplete

Author: GromNaN

src/Eloquent/Builder.php

@@ -5,6 +5,7 @@
 namespace MongoDB\Laravel\Eloquent;
 
 use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
+use Illuminate\Support\Collection;
 use MongoDB\BSON\Document;
 use MongoDB\Driver\CursorInterface;
 use MongoDB\Driver\Exception\WriteException;
@@ -16,6 +17,7 @@
 use function array_key_exists;
 use function array_merge;
 use function collect;
+use function compact;
 use function is_array;
 use function is_object;
 use function iterator_to_array;
@@ -69,6 +71,18 @@ public function aggregate($function = null, $columns = ['*'])
         return $result ?: $this;
     }
 
+    public function search(...$args)
+    {
+        $results = $this->toBase()->search(...$args);
+
+        return $this->model->hydrate($results->all());
+    }
+
+    public function autocomplete(string $path, string $query, bool|array $fuzzy = false, string $tokenOrder = 'any'): Collection
+    {
+        return $this->toBase()->autocomplete(...compact('path', 'query', 'fuzzy', 'tokenOrder'));
+    }
+
     /** @inheritdoc */
     public function update(array $values, array $options = [])
     {

src/Query/Builder.php

@@ -23,6 +23,7 @@
 use MongoDB\BSON\ObjectID;
 use MongoDB\BSON\Regex;
 use MongoDB\BSON\UTCDateTime;
+use MongoDB\Builder\Search;
 use MongoDB\Builder\Stage\FluentFactoryTrait;
 use MongoDB\Builder\Type\SearchOperatorInterface;
 use MongoDB\Driver\Cursor;
@@ -41,6 +42,7 @@
 use function blank;
 use function call_user_func;
 use function call_user_func_array;
+use function compact;
 use function count;
 use function ctype_xdigit;
 use function date_default_timezone_get;
@@ -1522,10 +1524,23 @@ public function search(
         ?array $sort = null,
         ?bool $returnStoredSource = null,
         ?array $tracking = null,
-    ): Collection|LazyCollection {
-        return $this->aggregate()
-            ->search(...array_filter(func_get_args(), fn ($arg) => $arg !== null))
-            ->get();
+    ): Collection {
+        return $this->aggregate()->search(...array_filter(func_get_args(), fn ($arg) => $arg !== null))->get();
+    }
+
+    /** @return Collection<string> */
+    public function autocomplete(string $path, string $query, bool|array $fuzzy = false, string $tokenOrder = 'any'): Collection
+    {
+        $args = compact('path', 'query', 'fuzzy', 'tokenOrder');
+        if ($args['fuzzy'] === true) {
+            $args['fuzzy'] = ['maxEdits' => 2];
+        } elseif ($args['fuzzy'] === false) {
+            unset($args['fuzzy']);
+        }
+
+        return $this->aggregate()->search(
+            Search::autocomplete(...$args),
+        )->get()->pluck($path);
     }
 
     /**

tests/AtlasSearchTest.php

@@ -2,7 +2,9 @@
 
 namespace MongoDB\Laravel\Tests;
 
+use Illuminate\Database\Eloquent\Collection;
 use Illuminate\Support\Facades\Schema;
+use MongoDB\Builder\Search;
 use MongoDB\Collection as MongoDBCollection;
 use MongoDB\Driver\Exception\ServerException;
 use MongoDB\Laravel\Schema\Builder;
@@ -42,6 +44,7 @@ public function setUp(): void
         ]);
 
         $collection = $this->getConnection('mongodb')->getCollection('books');
+
         assert($collection instanceof MongoDBCollection);
         try {
             $collection->createSearchIndex([
@@ -135,4 +138,60 @@ public function testGetIndexes()
 
         self::assertSame($expected, $indexes);
     }
+
+    public function testEloquentBuilderSearch()
+    {
+        $results = Book::search(Search::text('title', 'systems'));
+
+        self::assertInstanceOf(Collection::class, $results);
+        self::assertCount(3, $results);
+        self::assertInstanceOf(Book::class, $results->first());
+        self::assertSame([
+            'Operating System Concepts',
+            'Database System Concepts',
+            'Modern Operating Systems',
+        ], $results->pluck('title')->all());
+    }
+
+    public function testDatabaseBuilderSearch()
+    {
+        $results = $this->getConnection('mongodb')->table('books')
+            ->search(Search::text('title', 'systems'));
+
+        self::assertInstanceOf(\Illuminate\Support\Collection::class, $results);
+        self::assertCount(3, $results);
+        self::assertIsArray($results->first());
+        self::assertSame([
+            'Operating System Concepts',
+            'Database System Concepts',
+            'Modern Operating Systems',
+        ], $results->pluck('title')->all());
+    }
+
+    public function testEloquentBuilderAutocomplete()
+    {
+        $results = Book::autocomplete('title', 'system');
+
+        self::assertInstanceOf(\Illuminate\Support\Collection::class, $results);
+        self::assertCount(3, $results);
+        self::assertSame([
+            'Operating System Concepts',
+            'Database System Concepts',
+            'Modern Operating Systems',
+        ], $results->all());
+    }
+
+    public function testDatabaseBuilderAutocomplete()
+    {
+        $results = $this->getConnection('mongodb')->table('books')
+            ->autocomplete('title', 'system');
+
+        self::assertInstanceOf(\Illuminate\Support\Collection::class, $results);
+        self::assertCount(3, $results);
+        self::assertSame([
+            'Operating System Concepts',
+            'Database System Concepts',
+            'Modern Operating Systems',
+        ], $results->all());
+    }
 }