Add --output-format to filter command (json or yaml)

Signed-off-by: Si Beaumont <[email protected]>

Author: simonjbeaumont

Sources/swift-openapi-generator/FilterCommand.swift

@@ -36,6 +36,9 @@ struct _FilterCommand: AsyncParsableCommand {
     @Option(help: "Path to a YAML configuration file.")
     var config: URL
 
+    @Option(help: "Output format, either \(OutputFormat.yaml.rawValue) or \(OutputFormat.json.rawValue).")
+    var outputFormat: OutputFormat = .yaml
+
     @Argument(help: "Path to the OpenAPI document, either in YAML or JSON.")
     var docPath: URL
 
@@ -54,7 +57,14 @@ struct _FilterCommand: AsyncParsableCommand {
             return
         }
         let filteredDocument = try timing("Filtering document", documentFilter.filter(document))
-        FileHandle.standardOutput.write(try YAMLEncoder().encode(filteredDocument))
+        switch outputFormat {
+        case .yaml:
+            FileHandle.standardOutput.write(try YAMLEncoder().encode(filteredDocument))
+        case .json:
+            let encoder = JSONEncoder()
+            encoder.outputFormatting = [.prettyPrinted, .withoutEscapingSlashes]
+            FileHandle.standardOutput.write(try encoder.encode(filteredDocument))
+        }
     }
 }
 
@@ -80,3 +90,8 @@ private let sampleConfig = _UserConfig(
         schemas: ["Greeting"]
     )
 )
+
+enum OutputFormat: String, ExpressibleByArgument {
+    case json
+    case yaml
+}