Override the models the prune command looks for

Author: tobytwigger

src/Support/Config.php

@@ -4,8 +4,18 @@
 
 class Config
 {
+    protected static \Closure $callback;
+
+    public static function resolvePrunableModelsUsing(\Closure $callback)
+    {
+        static::$callback = $callback;
+    }
+
     public static function getPrunableModels(): array
     {
+        if(isset(static::$callback)) {
+            return call_user_func(static::$callback);
+        }
         return config('prunable-fields.models', []);
     }
 }

tests/MassPrunableFieldsTest.php

@@ -56,3 +56,17 @@
         fn (ModelsFieldsPruned $e) => $e->count === 1 && $e->model === MassPrunableUser::class
     );
 })->with('user_with_mass_prunable_fields');
+
+test('should allow the prunable models to be overridden at runtime', function(MassPrunableUser $model) {
+    \Maize\PrunableFields\Support\Config::resolvePrunableModelsUsing(fn() => [MassPrunableUser::class]);
+    
+    Event::fake();
+
+    pruneFields([]);
+
+    Event::assertDispatched(
+        ModelsFieldsPruned::class,
+        fn (ModelsFieldsPruned $e) => $e->count === 1 && $e->model === MassPrunableUser::class
+    );
+
+})->with('user_with_mass_prunable_fields');