Implement file regex search for ColocatedMappingDriver

Author: rela589n

src/Persistence/Mapping/Driver/ColocatedMappingDriver.php

@@ -43,7 +43,14 @@ trait ColocatedMappingDriver
      */
     protected array $excludePaths = [];
 
-    /** The file extension of mapping documents. */
+    /** The regex used to match mapping files. */
+    protected string $fileRegex = '/^.+\.php$/i';
+
+    /**
+     * The file extension of mapping documents.
+     *
+     * @deprecated Use {@see $fileRegex} instead.
+     */
     protected string $fileExtension = '.php';
 
     /**
@@ -94,16 +101,38 @@ public function getExcludePaths(): array
         return $this->excludePaths;
     }
 
-    /** Gets the file extension used to look for mapping files under. */
+    /** Gets the file regex used to look for mapping files under. */
+    public function getFileRegex(): string
+    {
+        return $this->fileRegex;
+    }
+
+    /** Sets the file regex used to look for mapping files under. */
+    public function setFileRegex(string $fileRegex): void
+    {
+        $this->fileRegex     = $fileRegex;
+        $this->fileExtension = '';
+    }
+
+    /**
+     * Gets the file extension used to look for mapping files under.
+     *
+     * @deprecated Use {@see getFileRegex()} instead.
+     */
     public function getFileExtension(): string
     {
         return $this->fileExtension;
     }
 
-    /** Sets the file extension used to look for mapping files under. */
+    /**
+     * Sets the file extension used to look for mapping files under.
+     *
+     * @deprecated Use {@see setFileRegex()} instead.
+     */
     public function setFileExtension(string $fileExtension): void
     {
         $this->fileExtension = $fileExtension;
+        $this->fileRegex     = '/^.+' . preg_quote($fileExtension) . '$/i';
     }
 
     /**
@@ -145,7 +174,7 @@ public function getAllClassNames(): array
                     new RecursiveDirectoryIterator($path, FilesystemIterator::SKIP_DOTS),
                     RecursiveIteratorIterator::LEAVES_ONLY,
                 ),
-                '/^.+' . preg_quote($this->fileExtension) . '$/i',
+                $this->fileRegex,
                 RecursiveRegexIterator::GET_MATCH,
             );
 

tests/Persistence/Mapping/ColocatedMappingDriverTest.php

@@ -44,6 +44,18 @@ public function testAddGetExcludePaths(): void
         ], $driver->getExcludePaths());
     }
 
+    public function testSetFileRegex(): void
+    {
+        $driver = $this->createDriver(__DIR__ . '/_files/colocated');
+        self::assertSame('/^.+\.php$/i', $driver->getFileRegex());
+
+        $driver->setFileRegex('^(?!.*(Test|Fixture)\.php$).*\.php$');
+
+        self::assertSame('^(?!.*(Test|Fixture)\.php$).*\.php$', $driver->getFileRegex());
+        self::assertSame('', $driver->getFileExtension());
+    }
+
+    /** @deprecated */
     public function testGetSetFileExtension(): void
     {
         $driver = $this->createDriver(__DIR__ . '/_files/colocated');
@@ -52,6 +64,7 @@ public function testGetSetFileExtension(): void
         $driver->setFileExtension('.php1');
 
         self::assertSame('.php1', $driver->getFileExtension());
+        self::assertSame('/^.+\.php1$/i', $driver->getFileRegex());
     }
 
     /** @dataProvider pathProvider */