add: Tag behaviour overriding callback.

dedoc · Mar 3, 2024 · 51332b2 · 51332b2
1 parent 407ab08
commit 51332b2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/Scramble.php b/src/Scramble.php
@@ -11,6 +11,8 @@ class Scramble
 
     public static $routeResolver;
 
+    public static $tagResolver;
+
     /**
      * Update open api document before finally rendering it.
      */
@@ -24,6 +26,11 @@ public static function routes(callable $routeResolver)
         static::$routeResolver = $routeResolver;
     }
 
+    public static function tags(callable $tagResolver)
+    {
+        static::$tagResolver = $tagResolver;
+    }
+
     /**
      * @param  array<string, ServerVariable>  $variables
      */

diff --git a/src/Support/OperationExtensions/RequestEssentialsExtension.php b/src/Support/OperationExtensions/RequestEssentialsExtension.php
@@ -4,6 +4,7 @@
 
 use Dedoc\Scramble\Extensions\OperationExtension;
 use Dedoc\Scramble\Infer;
+use Dedoc\Scramble\Scramble;
 use Dedoc\Scramble\Support\Generator\OpenApi;
 use Dedoc\Scramble\Support\Generator\Operation;
 use Dedoc\Scramble\Support\Generator\Parameter;
@@ -51,17 +52,21 @@ public function handle(Operation $operation, RouteInfo $routeInfo)
     {
         [$pathParams, $pathAliases] = $this->getRoutePathParameters($routeInfo->route, $routeInfo->phpDoc());
 
+        $tagResolver = Scramble::$tagResolver ?? function (RouteInfo $routeInfo) {
+            return array_unique([
+                ...$this->extractTagsForMethod($routeInfo),
+                Str::of(class_basename($routeInfo->className()))->replace('Controller', ''),
+            ]);
+        };
+
         $operation
             ->setMethod(strtolower($routeInfo->route->methods()[0]))
             ->setPath(Str::replace(
                 collect($pathAliases)->keys()->map(fn ($k) => '{'.$k.'}')->all(),
                 collect($pathAliases)->values()->map(fn ($v) => '{'.$v.'}')->all(),
                 $routeInfo->route->uri,
             ))
-            ->setTags(array_unique([
-                ...$this->extractTagsForMethod($routeInfo),
-                Str::of(class_basename($routeInfo->className()))->replace('Controller', ''),
-            ]))
+            ->setTags($tagResolver($routeInfo))
             ->servers($this->getAlternativeServers($routeInfo->route))
             ->addParameters($pathParams);