Parse only php files

Author: Jordan Hoff

src/ModelFinder.php

@@ -2,11 +2,12 @@
 
 namespace BeyondCode\ErdGenerator;
 
+use Illuminate\Support\Str;
 use PhpParser\NodeTraverser;
 use PhpParser\ParserFactory;
 use PhpParser\Node\Stmt\Class_;
-use PhpParser\Node\Stmt\Namespace_;
 use Illuminate\Support\Collection;
+use PhpParser\Node\Stmt\Namespace_;
 use Illuminate\Filesystem\Filesystem;
 use PhpParser\NodeVisitor\NameResolver;
 use Illuminate\Database\Eloquent\Model as EloquentModel;
@@ -28,7 +29,9 @@ public function getModelsInDirectory(string $directory): Collection
             $this->filesystem->allFiles($directory) :
             $this->filesystem->files($directory);
 
-        return Collection::make($files)->map(function ($path) {
+        return Collection::make($files)->filter(function ($path) {
+            return Str::endsWith($path, '.php');
+        })->map(function ($path) {
             return $this->getFullyQualifiedClassNameFromFile($path);
         })->filter(function (string $className) {
             return !empty($className);