Closes #19

rossity · Oct 29, 2018 · e54bb02 · e54bb02
1 parent 0efd75c
commit e54bb02
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/Console/GenerateModelCommand.php b/src/Console/GenerateModelCommand.php
@@ -310,10 +310,22 @@ protected function createModel()
                 ],
                 $this->getStub('relation')
             );
-            $modelTemplate = str_replace(['{{relation}}'], [$relationTemplate], $modelTemplate);
+            $modelTemplate = str_replace(
+                [
+                    '{{relation}}',
+                    '{{guardedAttributes}}'
+                ],
+                [
+                    $relationTemplate,
+                    ", '" . snake_case($related['name']) . "_id'"
+                ],
+                $modelTemplate
+            );
+
             $modelTemplate = $this->removeTag($modelTemplate, 'belongsTo');
         } else {
             $modelTemplate = $this->removeBlock($modelTemplate, 'belongsTo');
+            $modelTemplate = $this->removeBlock($modelTemplate, 'guardedAttributes');
         }
 
         $modelTemplate = $this->logActivity ? $this->removeTag($modelTemplate, 'logsActivity') : $this->removeBlock($modelTemplate, 'logsActivity');

diff --git a/stubs/model.stub b/stubs/model.stub
@@ -14,7 +14,7 @@ class {{modelName}} extends Model{{hasMedia}} implements HasMedia{{/hasMedia}}
 
     protected static $logAttributes = ['*'];
     protected static $logOnlyDirty = true;{{/logsActivity}}
-    protected $guarded = [];
+    protected $guarded = ['id'{{guardedAttributes}}];
 
 {{belongsTo}}{{relation}}{{/belongsTo}}
 }