added ability to change after class definition created (#401)

src/Infer/Analyzer/ClassAnalyzer.php

@@ -2,9 +2,11 @@
 
 namespace Dedoc\Scramble\Infer\Analyzer;
 
+use Dedoc\Scramble\Infer\Context;
 use Dedoc\Scramble\Infer\Definition\ClassDefinition;
 use Dedoc\Scramble\Infer\Definition\ClassPropertyDefinition;
 use Dedoc\Scramble\Infer\Definition\FunctionLikeDefinition;
+use Dedoc\Scramble\Infer\Extensions\Event\ClassDefinitionCreatedEvent;
 use Dedoc\Scramble\Infer\Scope\Index;
 use Dedoc\Scramble\Support\Type\FunctionType;
 use Dedoc\Scramble\Support\Type\TemplateType;
@@ -83,6 +85,8 @@ public function analyze(string $name): ClassDefinition
 
         $this->index->registerClassDefinition($classDefinition);
 
+        Context::getInstance()->extensionsBroker->afterClassDefinitionCreated(new ClassDefinitionCreatedEvent($classDefinition->name, $classDefinition));
+
         return $classDefinition;
     }
 }

src/Infer/Extensions/AfterClassDefinitionCreatedExtension.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Dedoc\Scramble\Infer\Extensions;
+
+use Dedoc\Scramble\Infer\Extensions\Event\ClassDefinitionCreatedEvent;
+
+interface AfterClassDefinitionCreatedExtension extends InferExtension
+{
+    public function shouldHandle(string $name): bool;
+
+    public function afterClassDefinitionCreated(ClassDefinitionCreatedEvent $event);
+}

src/Infer/Extensions/Event/ClassDefinitionCreatedEvent.php

@@ -0,0 +1,14 @@
+<?php
+
+namespace Dedoc\Scramble\Infer\Extensions\Event;
+
+use Dedoc\Scramble\Infer\Definition\ClassDefinition;
+
+class ClassDefinitionCreatedEvent
+{
+    public function __construct(
+        public readonly string $name,
+        public readonly ClassDefinition $classDefinition,
+    ) {
+    }
+}

src/Infer/Extensions/ExtensionsBroker.php

@@ -56,4 +56,16 @@ public function getStaticMethodReturnType($event)
 
         return null;
     }
+
+    public function afterClassDefinitionCreated($event)
+    {
+        $extensions = array_filter($this->extensions, function ($e) use ($event) {
+            return $e instanceof AfterClassDefinitionCreatedExtension
+                && $e->shouldHandle($event->name);
+        });
+
+        foreach ($extensions as $extension) {
+            $extension->afterClassDefinitionCreated($event);
+        }
+    }
 }

src/Scramble.php

@@ -86,6 +86,14 @@ public static function registerExtension(string $extensionClassName): void
         static::$extensions[] = $extensionClassName;
     }
 
+    /**
+     * @param  class-string<ExceptionToResponseExtension|OperationExtension|TypeToSchemaExtension|InferExtension>[]  $extensionClassNames
+     */
+    public static function registerExtensions(array $extensionClassNames): void
+    {
+        static::$extensions = array_merge(static::$extensions, $extensionClassNames);
+    }
+
     /**
      * Modify tag generation behaviour
      *