graph = new Graph(); foreach (config('erd-generator.graph') as $key => $value) { $this->graph->{"set${key}"}($value); } $this->addModelsToGraph($models); return $this->graph; } protected function getTableColumnsFromModel(EloquentModel $model) { try { $table = $model->getConnection()->getTablePrefix() . $model->getTable(); $schema = $model->getConnection()->getDoctrineSchemaManager($table); $databasePlatform = $schema->getDatabasePlatform(); $databasePlatform->registerDoctrineTypeMapping('enum', 'string'); $database = null; if (strpos($table, '.')) { list($database, $table) = explode('.', $table); } return $schema->listTableColumns($table, $database); } catch (\Throwable $e) { } return []; } protected function getModelLabel(EloquentModel $model, string $label) { $table = '<' . PHP_EOL; $table .= '' . PHP_EOL; if (config('erd-generator.use_db_schema')) { $columns = $this->getTableColumnsFromModel($model); foreach ($columns as $column) { $label = $column->getName(); if (config('erd-generator.use_column_types')) { $label .= ' ('.$column->getType()->getName().')'; } $table .= '' . PHP_EOL; } } $table .= '

' . $label . '

' . $label . '

>'; return $table; } protected function addModelsToGraph(Collection $models) { // Add models to graph $models->map(function (Model $model) { $eloquentModel = app($model->getModel()); $this->addNodeToGraph($eloquentModel, $model->getNodeName(), $model->getLabel()); }); // Create relations $models->map(function ($model) { $this->addRelationToGraph($model); }); } protected function addNodeToGraph(EloquentModel $eloquentModel, string $nodeName, string $label) { $node = Node::create($nodeName); $node->setLabel($this->getModelLabel($eloquentModel, $label)); foreach (config('erd-generator.node') as $key => $value) { $node->{"set${key}"}($value); } $this->graph->setNode($node); } protected function addRelationToGraph(Model $model) { $modelNode = $this->graph->findNode($model->getNodeName()); /** @var ModelRelation $relation */ foreach ($model->getRelations() as $relation) { $relatedModelNode = $this->graph->findNode($relation->getModelNodeName()); if ($relatedModelNode !== null) { $this->connectByRelation($model, $relation, $modelNode, $relatedModelNode); } } } /** * @param Node $modelNode * @param Node $relatedModelNode * @param ModelRelation $relation */ protected function connectNodes(Node $modelNode, Node $relatedModelNode, ModelRelation $relation): void { $edge = Edge::create($modelNode, $relatedModelNode); $edge->setFromPort($relation->getLocalKey()); $edge->setToPort($relation->getForeignKey()); $edge->setLabel(' '); $edge->setXLabel($relation->getType() . PHP_EOL . $relation->getName()); foreach (config('erd-generator.edge') as $key => $value) { $edge->{"set${key}"}($value); } foreach (config('erd-generator.relations.' . $relation->getType(), []) as $key => $value) { $edge->{"set${key}"}($value); } $this->graph->link($edge); } /** * @param Model $model * @param ModelRelation $relation * @param Node $modelNode * @param Node $relatedModelNode * @return void */ protected function connectBelongsToMany( Model $model, ModelRelation $relation, Node $modelNode, Node $relatedModelNode ): void { $relationName = $relation->getName(); $eloquentModel = app($model->getModel()); /** @var BelongsToMany $eloquentRelation */ $eloquentRelation = $eloquentModel->$relationName(); if (!$eloquentRelation instanceof BelongsToMany) { return; } $pivotClass = $eloquentRelation->getPivotClass(); try { /** @var EloquentModel $relationModel */ $pivotModel = app($pivotClass); $pivotModel->setTable($eloquentRelation->getTable()); $label = (new \ReflectionClass($pivotClass))->getShortName(); $pivotTable = $eloquentRelation->getTable(); $this->addNodeToGraph($pivotModel, $pivotTable, $label); $pivotModelNode = $this->graph->findNode($pivotTable); $relation = new ModelRelation( $relationName, 'BelongsToMany', $model->getModel(), $eloquentRelation->getParent()->getKeyName(), $eloquentRelation->getForeignPivotKeyName() ); $this->connectNodes($modelNode, $pivotModelNode, $relation); $relation = new ModelRelation( $relationName, 'BelongsToMany', $model->getModel(), $eloquentRelation->getRelatedPivotKeyName(), $eloquentRelation->getRelated()->getKeyName() ); $this->connectNodes($pivotModelNode, $relatedModelNode, $relation); } catch (\ReflectionException $e){} } /** * @param Model $model * @param ModelRelation $relation * @param Node $modelNode * @param Node $relatedModelNode */ protected function connectByRelation( Model $model, ModelRelation $relation, Node $modelNode, Node $relatedModelNode ): void { if ($relation->getType() === 'BelongsToMany') { $this->connectBelongsToMany($model, $relation, $modelNode, $relatedModelNode); return; } $this->connectNodes($modelNode, $relatedModelNode, $relation); } }