Extractor: Fixed extracting enum method body [Closes #115] (#116)

Author: sisklu

src/PhpGenerator/Extractor.php

@@ -66,8 +66,7 @@ public function extractMethodBodies(string $className): array
 		$nodeFinder = new NodeFinder;
 		$classNode = $nodeFinder->findFirst(
 			$this->statements,
-			fn(Node $node) => ($node instanceof Node\Stmt\Class_ || $node instanceof Node\Stmt\Trait_)
-				&& $node->namespacedName->toString() === $className,
+			fn(Node $node) => $node instanceof Node\Stmt\ClassLike && $node->namespacedName->toString() === $className,
 		);
 
 		$res = [];

tests/PhpGenerator/Extractor.getMethodBodies.phpt

@@ -35,6 +35,17 @@ abstract class Another
 		echo 123;
 	}
 }
+
+enum Color
+{
+	case Red;
+	case Blue;
+
+	public function getName(): string
+	{
+		return $this->name;
+	}
+}
 ');
 
 $bodies = $extractor->extractMethodBodies('NS\Undefined');
@@ -45,3 +56,8 @@ Assert::same([
 	'bar1' => "\$a = 10;\necho 123;",
 	'bar2' => 'echo "hello";',
 ], $bodies);
+
+$bodies = $extractor->extractMethodBodies('NS\Color');
+Assert::same([
+	'getName' => 'return $this->name;',
+], $bodies);