mark context non internal, fix non-closure document transformers

src/Generator.php

@@ -4,6 +4,7 @@
 
 use Dedoc\Scramble\Attributes\ExcludeAllRoutesFromDocs;
 use Dedoc\Scramble\Attributes\ExcludeRouteFromDocs;
+use Dedoc\Scramble\Contracts\DocumentTransformer;
 use Dedoc\Scramble\Exceptions\RouteAware;
 use Dedoc\Scramble\OpenApiVisitor\SchemaEnforceVisitor;
 use Dedoc\Scramble\Support\ContainerUtils;
@@ -23,6 +24,7 @@
 use Illuminate\Support\Collection;
 use Illuminate\Support\Facades\Route as RouteFacade;
 use Illuminate\Support\Str;
+use InvalidArgumentException;
 use Throwable;
 
 class Generator
@@ -96,7 +98,19 @@ public function __invoke(?GeneratorConfig $config = null)
                     TypeTransformer::class => $typeTransformer,
                 ]);
 
-            $openApiTransformer($openApi, $context);
+            if (is_callable($openApiTransformer)) {
+                $openApiTransformer($openApi, $context);
+
+                continue;
+            }
+
+            if ($openApiTransformer instanceof DocumentTransformer) {
+                $openApiTransformer->handle($openApi, $context);
+
+                continue;
+            }
+
+            throw new InvalidArgumentException('(callable(OpenApi, OpenApiContext): void)|DocumentTransformer type for document transformer expected, received '.$openApiTransformer::class);
         }
 
         return $openApi->toArray();

src/OpenApiContext.php

@@ -4,7 +4,6 @@
 
 use Dedoc\Scramble\Support\Generator\OpenApi;
 
-/** @internal */
 class OpenApiContext
 {
     public function __construct(

src/ScrambleServiceProvider.php

@@ -21,6 +21,8 @@
 use Dedoc\Scramble\Support\ExceptionToResponseExtensions\HttpExceptionToResponseExtension;
 use Dedoc\Scramble\Support\ExceptionToResponseExtensions\NotFoundExceptionToResponseExtension;
 use Dedoc\Scramble\Support\ExceptionToResponseExtensions\ValidationExceptionToResponseExtension;
+use Dedoc\Scramble\Support\Generator\Operation;
+use Dedoc\Scramble\Support\Generator\Parameter;
 use Dedoc\Scramble\Support\Generator\TypeTransformer;
 use Dedoc\Scramble\Support\IndexBuilders\IndexBuilder;
 use Dedoc\Scramble\Support\InferExtensions\AbortHelpersExceptionInfer;