Merge branch 'master' of github.com:beyondcode/laravel-er-diagram-generator

Author: mpociot

README.md

@@ -19,6 +19,12 @@ You can install Graphviz on MacOS via homebrew:
 brew install graphviz
 ```
 
+Or, if you are using Homestead:
+
+```bash
+sudo apt-get install graphviz
+```
+
 ## Installation
 
 You can install the package via composer:

src/ModelFinder.php

@@ -5,6 +5,7 @@
 use PhpParser\NodeTraverser;
 use PhpParser\ParserFactory;
 use PhpParser\Node\Stmt\Class_;
+use PhpParser\Node\Stmt\Namespace_;
 use Illuminate\Support\Collection;
 use Illuminate\Filesystem\Filesystem;
 use PhpParser\NodeVisitor\NameResolver;
@@ -42,10 +43,14 @@ protected function getFullyQualifiedClassNameFromFile(string $path): string
         $code = file_get_contents($path);
 
         $statements = $parser->parse($code);
-
         $statements = $traverser->traverse($statements);
+        
+        // get the first namespace declaration in the file
+        $root_statement = collect($statements)->filter(function ($statement) {
+            return $statement instanceof Namespace_;
+        })->first();
 
-        return collect($statements[0]->stmts)
+        return collect($root_statement->stmts)
                 ->filter(function ($statement) {
                     return $statement instanceof Class_;
                 })
@@ -55,4 +60,4 @@ protected function getFullyQualifiedClassNameFromFile(string $path): string
                 ->first() ?? '';
     }
 
-}
+}