Add support for enums

researchgate · Apr 4, 2024 · 1f60489 · 1f60489
1 parent 2c403ad
commit 1f60489
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 14 deletions.
diff --git a/src/ClassScanner.php b/src/ClassScanner.php
@@ -42,8 +42,7 @@ public function __construct(FilesystemAccess $filesystem, Output $output)
     }
 
     /**
-     * parses all given files for classes and interfaces that are defined or used in this
-     * files.
+     * parses all given files for classes and interfaces that are defined or used in these files.
      *
      * @param array $files
      * @param string $root
@@ -80,7 +79,7 @@ private function cleanContent($fileContent)
     {
         $fileContent = str_replace('\\\'', '  ', $fileContent);
         $fileContent = str_replace('\\"', '  ', $fileContent);
-        $fileContent = preg_replace("/([a-zA-Z])\'([a-zA-Z])/", '$1$2', $fileContent);
+        $fileContent = preg_replace("/([a-zA-Z])'([a-zA-Z])/", '$1$2', $fileContent);
         $getWhitespaces = function ($count) {
             $s = '';
             for ($i = 0; $i < $count; ++$i) {
@@ -115,15 +114,6 @@ private function cleanContent($fileContent)
         $fileContent = $cleanWithWhitespaces('/(<<<(?P<tag>_[A-Za-z]+).*(?P=tag);)/sU', $fileContent);
         $fileContent = $cleanWithWhitespaces('/(<<<(?P<tag>[A-Za-z_]+).*(?P=tag);)/sU', $fileContent);
 
-        if (false) {
-            $fileContent = preg_replace("/(\/\*.*\*\/)/sU", '', $fileContent);
-            $fileContent = preg_replace("/(\?>.*<\?)/sU", '', $fileContent);
-            $fileContent = preg_replace("/(\?>.*$)/sU", '', $fileContent);
-            $fileContent = preg_replace("/(\'.*\')/sU", '', $fileContent);
-            $fileContent = preg_replace('/(".*")/sU', '', $fileContent);
-            $fileContent = preg_replace("/(\/\/.*)/", '', $fileContent);
-        }
-
         return $fileContent;
     }
 
@@ -177,9 +167,9 @@ public function parseUsedEntities($file, $namespace, $fileContent, $originalFile
         // Extends
         $this->parseFileWithRegexForUsedEntities($file, $namespace, $fileContent, $originalFileContent, '/\sextends\s+([a-zA-Z0-9_\\\]+)\W/i', $this->usedEntities, $reservedClassKeywords);
         // static call
-        $this->parseFileWithRegexForUsedEntities($file, $namespace, $fileContent, $originalFileContent, '/\W([\$a-zA-Z0-9_\\\]+)::/i', $this->usedEntities, $reservedClassKeywords);
+        $this->parseFileWithRegexForUsedEntities($file, $namespace, $fileContent, $originalFileContent, '/\W([$a-zA-Z0-9_\\\]+)::/i', $this->usedEntities, $reservedClassKeywords);
         // Typehints
-        $this->parseFileWithRegexForUsedEntities($file, $namespace, $fileContent, $originalFileContent, '/[\,\(]\s*([a-zA-Z0-9_\\\]+)\s+\$[a-zA-Z0-9_]+/i', $this->usedEntities, $reservedClassKeywords);
+        $this->parseFileWithRegexForUsedEntities($file, $namespace, $fileContent, $originalFileContent, '/[,(]\s*([a-zA-Z0-9_\\\]+)\s+\$[a-zA-Z0-9_]+/i', $this->usedEntities, $reservedClassKeywords);
 
         // Return Typehints
         $this->parseFileWithRegexForUsedEntities(
@@ -251,6 +241,7 @@ public function parseDefinedEntities($file, $namespace, $fileContent, $originalF
     {
         $this->parseFileWithRegexForDefinedEntities($file, $namespace, $fileContent, $originalFileContent, '/^\s*(abstract\s+|final\s+)?class\s+([a-zA-Z0-9_]+)\W/mi', $this->definedEntities, 2);
         $this->parseFileWithRegexForDefinedEntities($file, $namespace, $fileContent, $originalFileContent, '/^\s*interface\s+([a-zA-Z0-9_]+)\W/mi', $this->definedEntities);
+        $this->parseFileWithRegexForDefinedEntities($file, $namespace, $fileContent, $originalFileContent, '/^\s*enum\s+([a-zA-Z0-9_]+)\W/mi', $this->definedEntities);
     }
 
     public function parseUseStatements($file, $namespace, $fileContent, $originalFileContent)

diff --git a/test/rg/tools/phpnsc/ClassScannerTest.php b/test/rg/tools/phpnsc/ClassScannerTest.php
@@ -53,6 +53,11 @@ interface   InterfaceTwo
 '/root/folder/namespace/InterfaceOne.php' => '
 <?php
 interface InterfaceOne
+',
+'/root/folder/namespace/EnumOne.php' => '
+<?php
+enum EnumOne
+{}
 ',
         );
         $files = array_keys($this->filesystem->filesystem);
@@ -75,6 +80,9 @@ interface InterfaceOne
             'InterfaceTwo' => array(
                 'namespaces' => array('vendor\namespace'),
             ),
+            'EnumOne' => array(
+                'namespaces' => array('vendor\namespace'),
+            ),
         );
 
         $this->assertEquals($expectedEntities, $this->classScanner->getDefinedEntities());