Add route filtering

Author: overint

src/GenerateSwaggerDoc.php

@@ -12,7 +12,8 @@ class GenerateSwaggerDoc extends Command
      * @var string
      */
     protected $signature = 'laravel-swagger:generate
-                            {--format=json : The format of the output, current options are json and yaml}';
+                            {--format=json : The format of the output, current options are json and yaml}
+                            {--filter= : Filter to a specific route prefix, such as /api or /v2/api}';
 
     /**
      * The console command description.
@@ -30,7 +31,7 @@ public function handle()
     {
         $config = config('laravel-swagger');
 
-        $docs = (new Generator($config))->generate();
+        $docs = (new Generator($config, $this->option('filter') ?: null))->generate();
 
         $formattedDocs = (new FormatterManager($docs))
             ->setFormat($this->option('format'))

src/Generator.php

@@ -11,6 +11,8 @@ class Generator
 {
     protected $config;
 
+    protected $routeFilter;
+
     protected $docs;
 
     protected $uri;
@@ -21,9 +23,10 @@ class Generator
 
     protected $action;
 
-    public function __construct($config)
+    public function __construct($config, $routeFilter = null)
     {
         $this->config = $config;
+        $this->routeFilter = $routeFilter;
     }
 
     public function generate()
@@ -33,6 +36,11 @@ public function generate()
         foreach ($this->getAppRoutes() as $route) {
             $this->originalUri = $uri = $this->getRouteUri($route);
             $this->uri = strip_optional_char($uri);
+
+            if ($this->routeFilter && !preg_match('/^' . preg_quote($this->routeFilter, '/') . '/', $this->uri)) {
+                continue;
+            }
+
             $this->action = $route->getAction('uses');
             $methods = $route->methods();