diff --git a/src/Generator/Commands/ModelGenerator.php b/src/Generator/Commands/ModelGenerator.php index 4583fd23b..7857fc31a 100644 --- a/src/Generator/Commands/ModelGenerator.php +++ b/src/Generator/Commands/ModelGenerator.php @@ -7,6 +7,7 @@ use Apiato\Core\Generator\Printer; use Apiato\Core\Generator\Traits\HasTestTrait; use Illuminate\Support\Str; +use Nette\PhpGenerator\Literal; use Nette\PhpGenerator\PhpFile; use Symfony\Component\Console\Input\InputOption; @@ -105,18 +106,26 @@ protected function getFileContent(): string ->setExtends($parentModelFullPath); // properties + $class->addProperty('table') + ->setVisibility('protected') + ->setValue($this->table); + $class->addProperty('fillable') ->setVisibility('protected') - ->setValue([]); + ->setValue(new Literal("[\n]")); - $class->addProperty('resourceKey') - ->setType('string') + $class->addProperty('hidden') ->setVisibility('protected') - ->setValue($this->fileName); + ->setValue(new Literal("[\n]")); - $class->addProperty('table') + $class->addProperty('casts') ->setVisibility('protected') - ->setValue($this->table); + ->setValue(new Literal("[\n]")); + + $class->addMethod('getResourceKey') + ->setReturnType('string') + ->setBody("return '$this->fileName';") + ->setPublic(); return $printer->printFile($file); }