A little more cleanup

Author: snake14

Specs/SpecGenerator.php

@@ -30,7 +30,7 @@ public function __construct()
         }
     }
 
-    public function generatePluginDoc(string $pluginName, string $format = 'json'): string
+    public function generatePluginDoc(string $pluginName, string $format = 'json', bool $writeToFile = false): string
     {
         BaseValidator::check('plugin', $pluginName, [new NotEmpty()]);
         Manager::getInstance()->checkIsPluginActivated($pluginName);
@@ -40,8 +40,7 @@ public function generatePluginDoc(string $pluginName, string $format = 'json'):
 
         // Check if the API class has been annotated and use the generated annotations file if it hasn't
         $pluginAnnotationsSource = $pluginDir . '/API.php';
-        $tempGenerator = new Generator(StaticContainer::get(NullLogger::class));
-        $openapi = $tempGenerator->generate([
+        $openapi = (new Generator(StaticContainer::get(NullLogger::class)))->generate([
             $pluginAnnotationsSource,
         ]);
         if (trim($openapi->toYaml()) === 'openapi: ' . OpenApi::DEFAULT_VERSION) {
@@ -62,8 +61,15 @@ public function generatePluginDoc(string $pluginName, string $format = 'json'):
             $pluginAnnotationsSource,
         ]);
 
+        // Update title with plugin name
         $openapi->info->title .= ' for ' . $pluginName . ' plugin';
 
+        // Remove the current server so that it isn't used when saving the spec file. It should only leave demo
+        if ($writeToFile && is_array($openapi->servers) && count($openapi->servers) > 1) {
+            unset($openapi->servers[0]);
+            $openapi->servers = array_values($openapi->servers);
+        }
+
         return strtolower($format) === 'yaml' ? $openapi->toYaml() : $openapi->toJson();
     }
 }