Add support for code style in FileGenerator - Close #67

Author: sandrokeil

src/FileCodeGenerator.php

@@ -43,11 +43,13 @@ public function __construct(
      *
      * @param FileCollection $fileCollection
      * @param callable|null $currentFileAst Callable to return current file AST, if null, file will be overwritten
+     * @param callable|null $applyCodeStyle Apply additional code style
      * @return array<string, string> List of filename => code
      */
     public function generateFiles(
         FileCollection $fileCollection,
-        callable $currentFileAst = null
+        callable $currentFileAst = null,
+        callable $applyCodeStyle = null
     ): array {
         $files = [];
 
@@ -62,6 +64,10 @@ public function generateFiles(
             };
         }
 
+        if ($applyCodeStyle === null) {
+            $applyCodeStyle = static fn (string $code) => $code;
+        }
+
         $previousNamespace = '__invalid//namespace__';
 
         foreach ($fileCollection as $classBuilder) {
@@ -75,9 +81,9 @@ public function generateFiles(
             $nodeTraverser = new NodeTraverser();
             $classBuilder->injectVisitors($nodeTraverser, $this->parser);
 
-            $files[$filename] = $this->printer->prettyPrintFile(
+            $files[$filename] = ($applyCodeStyle)($this->printer->prettyPrintFile(
                 $nodeTraverser->traverse($currentFileAst($classBuilder, $classInfo))
-            );
+            ));
         }
 
         return $files;