Implement MassPrunable without chunks limit

CHANGELOG.md

@@ -22,6 +22,7 @@ All notable changes to this project will be documented in this file.
 - Fix Query on `whereDate`, `whereDay`, `whereMonth`, `whereYear`, `whereTime` to use MongoDB operators [#2570](https://github.com/mongodb/laravel-mongodb/pull/2376) by [@Davpyu](https://github.com/Davpyu) and [@GromNaN](https://github.com/GromNaN).
 - `Model::unset()` does not persist the change. Call `Model::save()` to persist the change [#2578](https://github.com/mongodb/laravel-mongodb/pull/2578) by [@GromNaN](https://github.com/GromNaN).
 - Support delete one document with `Query\Builder::limit(1)->delete()` [#2591](https://github.com/mongodb/laravel-mongodb/pull/2591) by [@GromNaN](https://github.com/GromNaN)
+- Add trait `MongoDB\Laravel\Eloquent\MassPrunable` to replace Eloquent trait on MongoDB models [#2598](https://github.com/mongodb/laravel-mongodb/pull/2598) by [@GromNaN](https://github.com/GromNaN)
 
 ## [3.9.2] - 2022-09-01
 

src/Eloquent/MassPrunable.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace MongoDB\Laravel\Eloquent;
+
+use Illuminate\Database\Eloquent\MassPrunable as EloquentMassPrunable;
+use Illuminate\Database\Events\ModelsPruned;
+
+/**
+ * @see \Illuminate\Database\Eloquent\MassPrunable
+ */
+trait MassPrunable
+{
+    use EloquentMassPrunable;
+
+    /**
+     * Prune all prunable models in the database.
+     */
+    public function pruneAll(): int
+    {
+        $query = $this->prunable();
+        $total = in_array(SoftDeletes::class, class_uses_recursive(get_class($this)))
+                    ? $query->forceDelete()
+                    : $query->delete();
+
+        event(new ModelsPruned(static::class, $total));
+
+        return $total;
+    }
+}

tests/Eloquent/MassPrunableTest.php

@@ -0,0 +1,63 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Eloquent;
+
+use Illuminate\Database\Console\PruneCommand;
+use Illuminate\Database\Eloquent\MassPrunable;
+use Illuminate\Database\Eloquent\Prunable;
+use MongoDB\Laravel\Tests\Models\Soft;
+use MongoDB\Laravel\Tests\Models\User;
+use MongoDB\Laravel\Tests\TestCase;
+
+class MassPrunableTest extends TestCase
+{
+    public function tearDown(): void
+    {
+        User::truncate();
+        Soft::truncate();
+    }
+
+    public function testPruneWithQuery(): void
+    {
+        $this->assertTrue($this->isPrunable(User::class));
+
+        User::insert([
+            ['name' => 'John Doe', 'age' => 35],
+            ['name' => 'Jane Doe', 'age' => 32],
+            ['name' => 'Tomy Doe', 'age' => 11],
+        ]);
+
+        $model = new User();
+        $total = $model->pruneAll();
+        $this->assertEquals(2, $total);
+        $this->assertEquals(1, User::count());
+    }
+
+    public function testPruneSoftDelete(): void
+    {
+        $this->assertTrue($this->isPrunable(Soft::class));
+
+        Soft::insert([
+            ['name' => 'John Doe'],
+            ['name' => 'Jane Doe'],
+        ]);
+
+        $model = new Soft();
+        $total = $model->pruneAll();
+        $this->assertEquals(2, $total);
+        $this->assertEquals(0, Soft::count());
+        $this->assertEquals(0, Soft::withTrashed()->count());
+    }
+
+    /**
+     * @see PruneCommand::isPrunable()
+     */
+    protected function isPrunable($model)
+    {
+        $uses = class_uses_recursive($model);
+
+        return in_array(Prunable::class, $uses) || in_array(MassPrunable::class, $uses);
+    }
+}

tests/Models/Soft.php

@@ -4,6 +4,8 @@
 
 namespace MongoDB\Laravel\Tests\Models;
 
+use MongoDB\Laravel\Eloquent\Builder;
+use MongoDB\Laravel\Eloquent\MassPrunable;
 use MongoDB\Laravel\Eloquent\Model as Eloquent;
 use MongoDB\Laravel\Eloquent\SoftDeletes;
 
@@ -15,9 +17,15 @@
 class Soft extends Eloquent
 {
     use SoftDeletes;
+    use MassPrunable;
 
     protected $connection = 'mongodb';
     protected $collection = 'soft';
     protected static $unguarded = true;
     protected $casts = ['deleted_at' => 'datetime'];
+
+    public function prunable(): Builder
+    {
+        return $this->newQuery();
+    }
 }

tests/Models/User.php

@@ -12,7 +12,9 @@
 use Illuminate\Database\Eloquent\Casts\Attribute;
 use Illuminate\Notifications\Notifiable;
 use Illuminate\Support\Str;
+use MongoDB\Laravel\Eloquent\Builder;
 use MongoDB\Laravel\Eloquent\HybridRelations;
+use MongoDB\Laravel\Eloquent\MassPrunable;
 use MongoDB\Laravel\Eloquent\Model as Eloquent;
 
 /**
@@ -35,6 +37,7 @@ class User extends Eloquent implements AuthenticatableContract, CanResetPassword
     use CanResetPassword;
     use HybridRelations;
     use Notifiable;
+    use MassPrunable;
 
     protected $connection = 'mongodb';
     protected $casts = [
@@ -106,4 +109,9 @@ protected function username(): Attribute
             set: fn ($value) => Str::slug($value)
         );
     }
+
+    public function prunable(): Builder
+    {
+        return $this->where('age', '>', 18);
+    }
 }