| @@ -2,8 +2,8 @@ | ||
| 2 | 2 | |
| 3 | 3 | namespace BeyondCode\ErdGenerator; | 
| 4 | 4 | |
| 5 | -use phpDocumentor\GraphViz\Graph; | |
| 6 | 5 | use Illuminate\Support\Collection; | 
| 6 | +use phpDocumentor\GraphViz\Graph; | |
| 7 | 7 | use phpDocumentor\GraphViz\Node; | 
| 8 | 8 | |
| 9 | 9 | class GraphBuilder | 
| @@ -6,10 +6,10 @@ | ||
| 6 | 6 | use Illuminate\Filesystem\Filesystem; | 
| 7 | 7 | use Illuminate\Support\Collection; | 
| 8 | 8 | use Illuminate\Support\Str; | 
| 9 | -use PhpParser\Node\Stmt\Class_; | |
| 10 | -use PhpParser\Node\Stmt\Namespace_; | |
| 11 | 9 | use PhpParser\NodeTraverser; | 
| 12 | 10 | use PhpParser\NodeVisitor\NameResolver; | 
| 11 | +use PhpParser\Node\Stmt\Class_; | |
| 12 | +use PhpParser\Node\Stmt\Namespace_; | |
| 13 | 13 | use PhpParser\ParserFactory; | 
| 14 | 14 | use ReflectionClass; | 
| 15 | 15 | |
| @@ -10,6 +10,11 @@ | ||
| 10 | 10 | private $foreignKey; | 
| 11 | 11 | private $name; | 
| 12 | 12 | |
| 13 | + /** | |
| 14 | + * @param string $name | |
| 15 | + * @param string $type | |
| 16 | + * @param string $model | |
| 17 | + */ | |
| 13 | 18 | public function __construct($name, $type, $model, $localKey = null, $foreignKey) | 
| 14 | 19 |      { | 
| 15 | 20 | $this->type = $type; | 
| @@ -18,7 +18,7 @@ discard block | ||
| 18 | 18 | } | 
| 19 | 19 | |
| 20 | 20 | /** | 
| 21 | - * @return mixed | |
| 21 | + * @return string | |
| 22 | 22 | */ | 
| 23 | 23 | public function getModel() | 
| 24 | 24 |      { | 
| @@ -34,7 +34,7 @@ discard block | ||
| 34 | 34 | } | 
| 35 | 35 | |
| 36 | 36 | /** | 
| 37 | - * @return mixed | |
| 37 | + * @return string | |
| 38 | 38 | */ | 
| 39 | 39 | public function getLabel() | 
| 40 | 40 |      { |