support yaml to json

Author: man-zhang

bb-exp-util/src/main/kotlin/org/evomaster/json2yaml/Main.kt

@@ -45,32 +45,42 @@ class Main {
                     jsonToYaml(args[1])
                 }
                 "updateURLAndPort" -> {
-                    if (args.size != 4 && args.size != 3)
-                        throw IllegalArgumentException("to update url and port, the args should be 3 or 4, ie., `updateURLAndPort <openapi path> <port> <openapi V2ToV3>`")
+                    if (args.size > 5 || args.size < 3)
+                        throw IllegalArgumentException("to update url and port, the args should be 3--5, ie., `updateURLAndPort <openapi path> <port> <openapi V2ToV3> <output format, json or yaml>`")
+                    val format = if (args.size != 5) null else args[4]
                     try {
                         args[2].toInt()
-                        if (args.size == 4)
+                        if (args.size >= 4)
                             args[3].toBoolean()
+                        if (format != null && listOf("json","yaml").none { format.equals(it, ignoreCase = true) })
+                            throw IllegalArgumentException("output format could be either json or yaml")
                     }catch (ex: NumberFormatException){
                         throw IllegalArgumentException("port must be a int number and <openapi V2ToV3> must be boolean")
                     }
-                    updateURLAndPort(args[1], args[2], args.size == 4 && args[3].toBoolean())
+                    updateURLAndPort(args[1], args[2], args.size == 4 && args[3].toBoolean(), format)
                 }
                 else ->{
                     throw IllegalArgumentException("only able to configure `authForResTest`, `jsonToYaml` and `updateURLAndPort`")
                 }
             }
         }
 
-        fun updateURLAndPort(openapiFile: String, port: String, convertToOpenAPIV3: Boolean,saveFile: String = openapiFile){
+        fun updateURLAndPort(openapiFile: String, port: String, convertToOpenAPIV3: Boolean, format: String?, saveFile: String = openapiFile){
             val path = Paths.get(openapiFile)
             if (!Files.exists(path)) throw IllegalArgumentException("File cannot found")
 
             val isJson = openapiFile.endsWith(".json")
 
             val mapper = if (isJson) JSON_MAPPER else YAML_MAPPER
             val openApi = mapper.readTree(path.toFile()) as ObjectNode
-            val writer = if (isJson) mapper.writer() else YAML_WRITER
+
+            val jsonoutput = (format == null && isJson) || format.equals("json", ignoreCase = true)
+
+            val writer = if (jsonoutput == isJson) mapper.writer() else if (jsonoutput) OPENAPI_V3_WRITER else YAML_WRITER
+
+            val output = if (isJson && !jsonoutput) path.fileName.name.replace(".json", ".yaml")
+                        else if (!isJson && jsonoutput) path.fileName.name.replace(".yaml", ".json")
+                        else path.fileName.name
 
             if (openApi.has("swagger")){
                 openApi.put("host", "localhost:$port")
@@ -95,12 +105,20 @@ class Main {
                 }
             }
 
+            val dir = path.parent.toAbsolutePath().toString().run {
+                if (!endsWith(File.separator))
+                    "${this}${File.separator}"
+                else
+                    this
+            }
+
+            val savePath = "$dir$output"
 
-            writer.writeValue(Paths.get(saveFile).toFile(), openApi)
+            writer.writeValue(Paths.get(savePath).toFile(), openApi)
 
             if (openApi.has("swagger") && convertToOpenAPIV3){
-                val openapiV3 = OpenAPIV3Parser().read(saveFile)
-                (if (isJson) OPENAPI_V3_WRITER else YAML_WRITER).writeValue(Paths.get(saveFile).toFile(), openapiV3)
+                val openapiV3 = OpenAPIV3Parser().read(savePath)
+                (if (jsonoutput) OPENAPI_V3_WRITER else YAML_WRITER).writeValue(Paths.get(savePath).toFile(), openapiV3)
             }
         }