specify Stmt ClassName.

Author: smeghead

src/Php/PhpAccessModifier.php

@@ -3,7 +3,9 @@
 
 use PhpParser\Node\Stmt;
 use PhpParser\Node\Stmt\ {
+    ClassConst,
     ClassMethod,
+    Property,
 };
 
 class PhpAccessModifier {
@@ -14,7 +16,7 @@ class PhpAccessModifier {
     protected bool $final = false;
     protected bool $static = false;
 
-    public function __construct(Stmt $stmt) {
+    public function __construct(ClassConst|Property|ClassMethod $stmt) {
         $this->public = $stmt->isPublic();
         $this->protected = $stmt->isProtected();
         $this->private = $stmt->isPrivate();

src/Php/PhpClass.php

@@ -1,7 +1,6 @@
 <?php declare(strict_types=1);
 namespace Smeghead\PhpClassDiagram\Php;
 
-use PhpParser\NodeAbstract;
 use PhpParser\Node\ {
     NullableType,
     Identifier,
@@ -11,18 +10,20 @@
 use PhpParser\Node\Name\FullyQualified;
 use PhpParser\Node\Stmt;
 use PhpParser\Node\Stmt\ {
+    Class_,
     Namespace_,
     ClassLike,
     ClassMethod,
     Property,
     GroupUse,
+    Interface_,
     Use_,
 };
 
 class PhpClass {
     /** @var string[] directory parts */
     protected array $dirs;
-    protected Stmt $syntax;
+    protected Class_|Interface_ $syntax;
     protected array $full;
 
     public function __construct(string $filename, Stmt $syntax, array $full) {

src/Php/PhpMethod.php

@@ -4,6 +4,7 @@
 use PhpParser\Node\ {
     NullableType,
     Name,
+    Param,
 };
 use PhpParser\Node\Stmt\ {
     ClassMethod,
@@ -22,7 +23,7 @@ public function __construct(ClassMethod $method, PhpClass $class) {
         if ( ! empty($doc)) {
             $docString = $doc->getText();
         }
-        $params = array_map(function($x) use ($class, $docString){
+        $params = array_map(function(Param $x) use ($class, $docString){
             $type = PhpTypeExpression::buildByMethodParam($x, $class->getNamespace(), $docString, $x->var->name, $class->getUses());
             return new PhpMethodParameter($x->var->name, $type);
         }, $method->getParams());