Add global callback handler

Adds a way to call `Model::handleLazyLoadingViolationUsing()` from a service provider to add a way to handle violations for all models instead of overriding `violatedLazyLoading` on the models themselves.

Author: bonroyage

src/Illuminate/Database/Eloquent/Concerns/HasAttributes.php

@@ -468,6 +468,11 @@ public function isRelation($key)
      */
     protected function violatedLazyLoading($key)
     {
+        if (isset(static::$violatedLazyLoadingCallback)) {
+            call_user_func(static::$violatedLazyLoadingCallback, $this, $key);
+            return;
+        }
+
         throw new LazyLoadingViolationException($this, $key);
     }
 

src/Illuminate/Database/Eloquent/Model.php

@@ -158,6 +158,13 @@ abstract class Model implements Arrayable, ArrayAccess, Jsonable, JsonSerializab
      */
     protected static $modelsShouldPreventLazyLoading = false;
 
+    /**
+     * The callback that is responsible for handing lazy loading violations.
+     *
+     * @var callable|null
+     */
+    protected static $violatedLazyLoadingCallback;
+
     /**
      * The name of the "created at" column.
      *
@@ -358,6 +365,16 @@ public static function preventLazyLoading($value = true)
         static::$modelsShouldPreventLazyLoading = $value;
     }
 
+    /**
+     * Register a callback that is responsible for handling lazy loading violations.
+     *
+     * @param callable $callback
+     */
+    public static function handleLazyLoadingViolationUsing(callable $callback)
+    {
+        static::$violatedLazyLoadingCallback = $callback;
+    }
+
     /**
      * Fill the model with an array of attributes.
      *

tests/Integration/Database/EloquentStrictLoadingTest.php

@@ -100,6 +100,35 @@ public function testStrictModeThrowsAnExceptionOnLazyLoadingInRelations()
 
         $models[0]->modelTwos[0]->modelThrees;
     }
+
+    public function testStrictModeWithCustomCallbackOnLazyLoading()
+    {
+        $this->expectsEvents(ViolatedLazyLoadingEvent::class);
+
+        Model::handleLazyLoadingViolationUsing(function ($model, $key) {
+            event(new ViolatedLazyLoadingEvent($model, $key));
+        });
+
+        EloquentStrictLoadingTestModel1::create();
+        EloquentStrictLoadingTestModel1::create();
+
+        $models = EloquentStrictLoadingTestModel1::get();
+
+        $models[0]->modelTwos;
+    }
+
+    public function testStrictModeWithOverriddenHandlerOnLazyLoading()
+    {
+        $this->expectException(\RuntimeException::class);
+        $this->expectExceptionMessage('Violated');
+
+        EloquentStrictLoadingTestModel1WithCustomHandler::create();
+        EloquentStrictLoadingTestModel1WithCustomHandler::create();
+
+        $models = EloquentStrictLoadingTestModel1WithCustomHandler::get();
+
+        $models[0]->modelTwos;
+    }
 }
 
 class EloquentStrictLoadingTestModel1 extends Model
@@ -114,6 +143,23 @@ public function modelTwos()
     }
 }
 
+class EloquentStrictLoadingTestModel1WithCustomHandler extends Model
+{
+    public $table = 'test_model1';
+    public $timestamps = false;
+    protected $guarded = [];
+
+    public function modelTwos()
+    {
+        return $this->hasMany(EloquentStrictLoadingTestModel2::class, 'model_1_id');
+    }
+
+    protected function violatedLazyLoading($key)
+    {
+        throw new \RuntimeException("Violated {$key}");
+    }
+}
+
 class EloquentStrictLoadingTestModel2 extends Model
 {
     public $table = 'test_model2';
@@ -132,3 +178,15 @@ class EloquentStrictLoadingTestModel3 extends Model
     public $timestamps = false;
     protected $guarded = [];
 }
+
+class ViolatedLazyLoadingEvent
+{
+    public $model;
+    public $key;
+
+    public function __construct($model, $key)
+    {
+        $this->model = $model;
+        $this->key = $key;
+    }
+}