auto generating columns from model fillable

rappasoft · Jan 9, 2022 · 4c681b1 · 4c681b1
1 parent 5ec4a23
commit 4c681b1
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 5 deletions.
diff --git a/src/Commands/MakeCommand.php b/src/Commands/MakeCommand.php
@@ -6,6 +6,7 @@
 use Illuminate\Support\Facades\File;
 use Illuminate\Support\Str;
 use Livewire\Commands\ComponentParser;
+use Illuminate\Database\Eloquent\Model;
 use Livewire\Commands\MakeCommand as LivewireMakeCommand;
 
 /**
@@ -141,8 +142,8 @@ public function classContents(): string
             $template = file_get_contents(__DIR__.DIRECTORY_SEPARATOR.'table-with-model.stub');
 
             $contents = str_replace(
-                ['[namespace]', '[class]', '[model]', '[model_import]'],
-                [$this->parser->classNamespace(), $this->parser->className(), $this->model, $this->getModelImport()],
+                ['[namespace]', '[class]', '[model]', '[model_import]', '[columns]'],
+                [$this->parser->classNamespace(), $this->parser->className(), $this->model, $this->getModelImport(), $this->generateColumns($this->getModelImport())],
                 $template
             );
         } else {
@@ -204,4 +205,40 @@ public function getModelImport(): string
 
         return 'App\Models\\' . $this->model;
     }
+
+    /**
+     * @param string $modelName
+     * @return string
+     * @throws Exception
+     */
+    private function generateColumns(string $modelName): string
+    {
+        $model = new $modelName();
+
+        if ($model instanceof Model === false) {
+            throw new \Exception('Invalid model given.');
+        }
+
+        $getFillable    = array_merge(
+            [$model->getKeyName()],
+            $model->getFillable(),
+            ['created_at', 'updated_at']
+        );
+
+        $columns        = "[\n";
+
+        foreach ($getFillable as $field) {
+            if (in_array($field, $model->getHidden())) {
+                continue;
+            }
+
+            $title = Str::of($field)->replace('_', ' ')->ucfirst();
+
+            $columns    .= '            Column::make("' . $title . '", "' . $field . '")' . "\n" . '                ->sortable(),' . "\n\n";
+        }
+
+        $columns .= "        ]\n";
+
+        return $columns;
+    }
 }
diff --git a/src/Commands/table-with-model.stub b/src/Commands/table-with-model.stub
@@ -12,9 +12,7 @@ class [class] extends DataTableComponent
 
     public function columns(): array
     {
-        return [
-            Column::make('Column Name'),
-        ];
+        return [columns];
     }
 
     public function query(): Builder