Merge pull request #39 from smeghead/support-enum

support enum.

Author: smeghead

CHANGELOG.md

@@ -1,8 +1,12 @@
 # CHANGELOG
 
+### Features
+
+ * support enum.
+
 ### Bug fix
 
- * Fixed an error when loading traits.
+ * Fixed an error when loading traits. #37
 
 
 ## v0.3.0 (2023-02-23)

src/DiagramElement/Entry.php

@@ -34,7 +34,8 @@ public function getClass(): PhpClass {
     public function dump($level = 0): array {
         $indent = str_repeat('  ', $level);
         $lines = [];
-        $meta = $this->class->getClassType()->getMeta() === 'Stmt_Interface' ? 'interface' : 'class';
+        // $meta = $this->class->getClassType()->getMeta() === 'Stmt_Interface' ? 'interface' : 'class';
+        $meta = $this->class->getClassType()->getMetaName();
         if ($this->options->classProperties() || $this->options->classMethods()) {
             $lines[] = sprintf('%s%s %s {', $indent, $meta, $this->class->getLogicalName());
             if ($this->options->classProperties()) {

src/Php/PhpType.php

@@ -24,6 +24,19 @@ public function getMeta(): string {
         return $this->meta;
     }
 
+    public function getMetaName(): string {
+        switch ($this->meta) {
+            case 'Stmt_Interface':
+                return 'interface';
+            case 'Stmt_Enum':
+                return 'enum';
+            case 'Stmt_Trait':
+                return 'class';
+            default:
+                return 'class';
+        }
+    }
+
     /**
      * @return string[] namespace
      */

test/PhpReflectionTest.php

@@ -298,4 +298,18 @@ public function testTrait(): void {
         $this->assertSame('string', $data->getMethods()[0]->getParams()[0]->getType()->getName(), 'parameter type.');
         $this->assertSame([], $data->getMethods()[0]->getParams()[0]->getType()->getTypes()[0]->getNamespace(), 'parameter type namespace.');
     }
+    /**
+     * @requires PHP >= 8.1
+     * PHP8.0 dose not have `enum`. 
+     */
+    public function testEnum(): void {
+        $options = new Options([]);
+        $directory = sprintf('%s/enum', $this->fixtureDir);
+        $filename = sprintf('%s/enum/TestEnum.php', $this->fixtureDir);
+        $parsed = PhpReader::parseFile($directory, $filename, $options);
+
+        $data = $parsed[0]->getInfo();
+        $this->assertSame('Suit', $data->getClassType()->getName(), '1st class type name.');
+        $this->assertSame(['Hoge', 'TestEnum'], $data->getClassType()->getNamespace(), '1st namespace name.');
+    }
 }

test/fixtures/enum/TestEnum.php

@@ -0,0 +1,17 @@
+<?php
+namespace Hoge\TestEnum;
+
+enum Suit
+{
+    case Hearts;
+    case Diamonds;
+    case Clubs;
+    case Spades;
+}
+
+function do_stuff(Suit $s)
+{
+    // ...
+}
+
+do_stuff(Suit::Spades);