added formatters

Author: mtrajano

composer.json

@@ -16,6 +16,9 @@
             "Mtrajano\\LaravelSwagger\\": "src/"
         }
     },
+    "suggest": {
+        "ext-yaml": "Required to use the YAML output option"
+    },
     "extra": {
         "laravel": {
             "providers": [

src/FormatterManager.php

@@ -0,0 +1,43 @@
+<?php
+
+namespace Mtrajano\LaravelSwagger;
+
+class FormatterManager
+{
+    private $docs;
+
+    private $formatter;
+
+    public function __construct($docs)
+    {
+        $this->docs = $docs;
+
+        $this->formatter = new Formatters\JsonFormatter($docs);
+    }
+
+    public function setFormat($format)
+    {
+        $format = strtolower($format);
+
+        $this->formatter = $this->getFormatter($format);
+
+        return $this;
+    }
+
+    protected function getFormatter($format)
+    {
+        switch($format) {
+            case 'json':
+                return new Formatters\JsonFormatter($this->docs);
+            case 'yaml':
+                return new Formatters\YamlFormatter($this->docs);
+            default:
+                throw new LaravelSwaggerException('Invalid format passed');
+        }
+    }
+
+    public function format()
+    {
+        return $this->formatter->format();
+    }
+}

src/Formatters/Formatter.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Mtrajano\LaravelSwagger\Formatters;
+
+abstract class Formatter
+{
+    protected $docs;
+
+    public function __construct($docs)
+    {
+        $this->docs = $docs;
+    }
+
+    abstract public function format();
+}

src/Formatters/JsonFormatter.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Mtrajano\LaravelSwagger\Formatters;
+
+class JsonFormatter extends Formatter
+{
+    public function format()
+    {
+        if (!extension_loaded('json')) {
+            throw new \Exception('JSON extension must be loaded to use the json output format');
+        }
+
+        return json_encode($this->docs, JSON_PRETTY_PRINT);
+    }
+}

src/Formatters/YamlFormatter.php

@@ -0,0 +1,15 @@
+<?php
+
+namespace Mtrajano\LaravelSwagger\Formatters;
+
+class YamlFormatter extends Formatter
+{
+    public function format()
+    {
+        if (!extension_loaded('yaml')) {
+            throw new \Exception('YAML extension must be loaded to use the yaml output format');
+        }
+
+        return yaml_emit($this->docs);
+    }
+}

src/GenerateSwaggerDoc.php

@@ -11,7 +11,8 @@ class GenerateSwaggerDoc extends Command
      *
      * @var string
      */
-    protected $signature = 'laravel-swagger:generate';
+    protected $signature = 'laravel-swagger:generate
+                            {--format=json : The format of the output, current options are json and yaml}';
 
     /**
      * The console command description.
@@ -31,6 +32,10 @@ public function handle()
 
         $docs = (new Generator($config))->generate();
 
-        echo json_encode($docs, JSON_PRETTY_PRINT) . "\n";
+        $formattedDocs = (new FormatterManager($docs))
+            ->setFormat($this->option('format'))
+            ->format();
+
+        $this->line($formattedDocs);
     }
 }

src/LaravelSwaggerException.php

@@ -0,0 +1,7 @@
+<?php
+
+namespace Mtrajano\LaravelSwagger;
+
+class LaravelSwaggerException extends \Exception
+{
+}