Fixes #32 Attaches relationships to models in factory definitions

Author: axit-joost

src/Generators/FactoryGenerator.php

@@ -4,6 +4,7 @@
 
 use Blueprint\Contracts\Generator;
 use Blueprint\Models\Model;
+use Illuminate\Support\Str;
 
 class FactoryGenerator implements Generator
 {
@@ -61,10 +62,19 @@ protected function buildDefinition(Model $model)
                 continue;
             }
 
-            $definition .= self::INDENT . "'{$column->name()}' => ";
-            $faker = $this->fakerData($column->name()) ?? $this->fakerDataType($column->dataType());
-            $definition .= '$faker->' . $faker;
-            $definition .= ',' . PHP_EOL;
+            if ($column->dataType() === 'id') {
+				$name = Str::substr($column->name(), 0, -3);
+				$class = Str::studly($column->attributes()[0] ?? $name);
+
+				$definition .= self::INDENT . "'{$column->name()}' => ";
+				$definition .= sprintf("factory(\App\%s::class)", $class);
+				$definition .= ',' . PHP_EOL;
+			} else {
+				$definition .= self::INDENT . "'{$column->name()}' => ";
+				$faker = $this->fakerData($column->name()) ?? $this->fakerDataType($column->dataType());
+				$definition .= '$faker->' . $faker;
+				$definition .= ',' . PHP_EOL;
+			}
         }
 
         return trim($definition);