Generate model relationships (#72)

Author: peterjaap

src/Generators/FactoryGenerator.php

@@ -63,7 +63,7 @@ protected function buildDefinition(Model $model)
 
         /** @var \Blueprint\Models\Column $column */
         foreach ($model->columns() as $column) {
-            if ($column->name() === 'id') {
+            if ($column->name() === 'id' || $column->name() === 'relationships') {
                 continue;
             }
 

src/Generators/MigrationGenerator.php

@@ -61,6 +61,8 @@ protected function buildDefinition(Model $model)
                 $dataType = 'bigIncrements';
             } elseif ($column->dataType() === 'id') {
                 $dataType = 'unsignedBigInteger';
+            } elseif ($column->name() === 'relationships') {
+                continue;
             }
 
             $definition .= self::INDENT . '$table->' . $dataType . "('{$column->name()}'";

src/Generators/ModelGenerator.php

@@ -46,6 +46,7 @@ protected function populateStub(string $stub, Model $model)
 
         $body = $this->buildProperties($model);
         $body .= PHP_EOL . PHP_EOL;
+        $body .= $this->buildBelongsTo($model);
         $body .= $this->buildRelationships($model);
 
         $stub = str_replace('// ...', trim($body), $stub);
@@ -110,7 +111,7 @@ private function buildProperties(Model $model)
         return trim($properties);
     }
 
-    private function buildRelationships(Model $model)
+    private function buildBelongsTo(Model $model)
     {
         $columns = array_filter($model->columns(), function (Column $column) {
             return $column->name() !== 'id' && $column->dataType() === 'id';
@@ -138,6 +139,41 @@ private function buildRelationships(Model $model)
         return $methods;
     }
 
+    private function buildRelationships(Model $model)
+    {
+        $columns = array_filter($model->columns(), function (Column $column) {
+            return $column->name() === 'relationships';
+        });
+
+        if (empty($columns)) {
+            return '';
+        }
+
+        $methods = '';
+        $template = $this->files->stub('model/method.stub');
+
+        /** @var Column $column */
+        foreach ($columns as $column) {
+            foreach ($column->attributes() as $methodName => $modelName) {
+                if ('belongsTo' === $methodName) {
+                    throw new \Exception('The belongsTo relationship for the '.$modelName.' model on the '.$model->name().' model should be defined using the '.$modelName.'_id: id syntax');
+                }
+                $class = Str::studly($column->attributes()[0] ?? $modelName);
+                $relationship = sprintf("\$this->%s(%s::class)",
+                    $methodName,
+                    '\\' . $model->fullyQualifiedNamespace() . '\\' . $class);
+
+                $modelNameForMethod = Str::contains($methodName, 'Many') ? Str::plural($modelName) : $modelName;
+                $method = str_replace('DummyName', Str::camel($modelNameForMethod), $template);
+                $method = str_replace('null', $relationship, $method);
+
+                $methods .= PHP_EOL . $method;
+            }
+        }
+
+        return $methods;
+    }
+
     protected function getPath(Model $model)
     {
         $path = str_replace('\\', '/', Blueprint::relativeNamespace($model->fullyQualifiedClassName()));
@@ -152,7 +188,8 @@ private function fillableColumns(array $columns)
             'password',
             'deleted_at',
             'created_at',
-            'updated_at'
+            'updated_at',
+            'relationships',
         ]);
     }
 

src/Lexers/ModelLexer.php

@@ -135,35 +135,39 @@ private function buildModel(string $name, array $columns)
         return $model;
     }
 
-    private function buildColumn(string $name, string $definition)
+    private function buildColumn(string $name, $definition)
     {
         $data_type = 'string';
         $modifiers = [];
 
-        $tokens = explode(' ', $definition);
-        foreach ($tokens as $token) {
-            $parts = explode(':', $token);
-            $value = $parts[0];
-
-            if ($value === 'id') {
-                $data_type = 'id';
-                if (isset($parts[1])) {
-                    $attributes = [$parts[1]];
-                }
-            } elseif (isset(self::$dataTypes[strtolower($value)])) {
-                $attributes = $parts[1] ?? null;
-                $data_type = self::$dataTypes[strtolower($value)];
-                if (!empty($attributes)) {
-                    $attributes = explode(',', $attributes);
+        if ($name === 'relationships' && is_array($definition)) {
+            $attributes = $definition;
+        } else {
+            $tokens = explode(' ', $definition);
+            foreach ($tokens as $token) {
+                $parts = explode(':', $token);
+                $value = $parts[0];
+
+                if ($value === 'id') {
+                    $data_type = 'id';
+                    if (isset($parts[1])) {
+                        $attributes = [$parts[1]];
+                    }
+                } elseif (isset(self::$dataTypes[strtolower($value)])) {
+                    $attributes = $parts[1] ?? null;
+                    $data_type = self::$dataTypes[strtolower($value)];
+                    if (!empty($attributes)) {
+                        $attributes = explode(',', $attributes);
+                    }
                 }
-            }
 
-            if (isset(self::$modifiers[strtolower($value)])) {
-                $modifierAttributes = $parts[1] ?? null;
-                if (empty($modifierAttributes)) {
-                    $modifiers[] = self::$modifiers[strtolower($value)];
-                } else {
-                    $modifiers[] = [self::$modifiers[strtolower($value)] => $modifierAttributes];
+                if (isset(self::$modifiers[strtolower($value)])) {
+                    $modifierAttributes = $parts[1] ?? null;
+                    if (empty($modifierAttributes)) {
+                        $modifiers[] = self::$modifiers[strtolower($value)];
+                    } else {
+                        $modifiers[] = [self::$modifiers[strtolower($value)] => $modifierAttributes];
+                    }
                 }
             }
         }