Create eloquent datatable from relation

Author: Sébastien Nikolaou

src/EloquentDataTable.php

@@ -9,6 +9,7 @@
 use Illuminate\Database\Eloquent\Relations\HasOneOrMany;
 use Illuminate\Database\Eloquent\Relations\HasOneThrough;
 use Illuminate\Database\Eloquent\Relations\MorphTo;
+use Illuminate\Database\Eloquent\Relations\Relation;
 use Yajra\DataTables\Exceptions\Exception;
 
 /**
@@ -25,16 +26,12 @@ class EloquentDataTable extends QueryDataTable
      */
     public function __construct($model)
     {
-        switch ($model) {
-            case $model instanceof Model:
-                $builder = $model->newQuery();
-                break;
-            case $model instanceof EloquentBuilder:
-                $builder = $model;
-                break;
-            default:
-                throw new Exception('Invalid model type. Must be an instance of Eloquent Model or Eloquent Builder.');
-        }
+        $builder = match (true) {
+            $model instanceof Model => $model->newQuery(),
+            $model instanceof Relation => $model->getQuery(),
+            $model instanceof EloquentBuilder => $model,
+            default => throw new Exception('Invalid model type. Must be an instance of Eloquent Model, Eloquent Relation, or Eloquent Builder.'),
+        };
 
         parent::__construct($builder->getQuery());
 
@@ -100,10 +97,13 @@ protected function compileQuerySearch($query, string $column, string $keyword, s
         }
 
         if ($this->isMorphRelation($relation)) {
-            $query->{$boolean.'WhereHasMorph'}($relation, '*',
+            $query->{$boolean.'WhereHasMorph'}(
+                $relation,
+                '*',
                 function (EloquentBuilder $query) use ($newColumn, $keyword) {
                     parent::compileQuerySearch($query, $newColumn, $keyword, '');
-                });
+                }
+            );
         } else {
             $query->{$boolean.'WhereHas'}($relation, function (EloquentBuilder $query) use ($newColumn, $keyword) {
                 parent::compileQuerySearch($query, $newColumn, $keyword, '');

tests/Integration/EloquentDataTableTest.php

@@ -121,6 +121,17 @@ public function it_can_return_formatted_columns()
         $this->assertEquals(Carbon::parse($user->created_at)->format('Y-m-d'), $data['created_at_formatted']);
     }
 
+    /** @test */
+    public function it_accepts_a_relation()
+    {
+        $user = User::first();
+
+        $dataTable = app('datatables')->eloquent($user->posts());
+        $response = $dataTable->toJson();
+        $this->assertInstanceOf(EloquentDataTable::class, $dataTable);
+        $this->assertInstanceOf(JsonResponse::class, $response);
+    }
+
     protected function setUp(): void
     {
         parent::setUp();