Implement MassPrunable without chunks limit

Author: GromNaN

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);
+    }
 }