From 9311bdf972a1f244e42fdefbb7536725703ae99a Mon Sep 17 00:00:00 2001
From: Tony Tam <fehguy@gmail.com>
Date: Wed, 18 Mar 2015 23:24:47 -0700
Subject: [PATCH] added yaml generator

---
 .../codegen/languages/SwaggerGenerator.java   |  3 ++
 .../languages/SwaggerYamlGenerator.java       | 44 +++++++++++++++++++
 .../src/main/resources/swagger/README.md      |  2 +
 .../swagger/generator/online/Generator.java   |  1 +
 4 files changed, 50 insertions(+)
 create mode 100644 modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerYamlGenerator.java
 create mode 100644 modules/swagger-codegen/src/main/resources/swagger/README.md

diff --git a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerGenerator.java b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerGenerator.java
index 8ad75ae3e211..c0560e2a1bc5 100644
--- a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerGenerator.java
+++ b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerGenerator.java
@@ -23,7 +23,10 @@ public class SwaggerGenerator extends DefaultCodegen implements CodegenConfig {
 
   public SwaggerGenerator() {
     super();
+    templateDir = "swagger";
     outputFolder = "generated-code/swagger";
+
+    supportingFiles.add(new SupportingFile("README.md", "", "README.md"));
   }
 
   @Override
diff --git a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerYamlGenerator.java b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerYamlGenerator.java
new file mode 100644
index 000000000000..7c65168546f9
--- /dev/null
+++ b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/languages/SwaggerYamlGenerator.java
@@ -0,0 +1,44 @@
+package com.wordnik.swagger.codegen.languages;
+
+import com.wordnik.swagger.codegen.*;
+import com.wordnik.swagger.util.*;
+import com.wordnik.swagger.models.Swagger;
+
+import org.apache.commons.io.FileUtils;
+
+import java.io.File;
+
+public class SwaggerYamlGenerator extends DefaultCodegen implements CodegenConfig {
+  public CodegenType getTag() {
+    return CodegenType.DOCUMENTATION;
+  }
+
+  public String getName() {
+    return "swagger-yaml";
+  }
+
+  public String getHelp() {
+    return "Creates a static swagger.yaml file.";
+  }
+
+  public SwaggerYamlGenerator() {
+    super();
+    templateDir = "swagger";
+    outputFolder = "generated-code/swagger";
+
+    supportingFiles.add(new SupportingFile("README.md", "", "README.md"));
+  }
+
+  @Override
+  public void processSwagger(Swagger swagger) {
+    try{
+      String swaggerString = Yaml.mapper().writeValueAsString(swagger);
+      String outputFile = outputFolder + File.separator + "swagger.yaml";
+      FileUtils.writeStringToFile(new File(outputFile), swaggerString);
+      System.out.println("wrote file to " + outputFile);
+    }
+    catch(Exception e) {
+      e.printStackTrace();
+    }
+  }
+}
\ No newline at end of file
diff --git a/modules/swagger-codegen/src/main/resources/swagger/README.md b/modules/swagger-codegen/src/main/resources/swagger/README.md
new file mode 100644
index 000000000000..0c2b69a24dbd
--- /dev/null
+++ b/modules/swagger-codegen/src/main/resources/swagger/README.md
@@ -0,0 +1,2 @@
+# Swagger JSON
+This is a swagger JSON built by the [swagger-codegen](https://github.com/swagger-api/swagger-codegen) project.
\ No newline at end of file
diff --git a/modules/swagger-generator/src/main/java/com/wordnik/swagger/generator/online/Generator.java b/modules/swagger-generator/src/main/java/com/wordnik/swagger/generator/online/Generator.java
index 6bbbfec4f7b1..996a724bb6d3 100644
--- a/modules/swagger-generator/src/main/java/com/wordnik/swagger/generator/online/Generator.java
+++ b/modules/swagger-generator/src/main/java/com/wordnik/swagger/generator/online/Generator.java
@@ -66,6 +66,7 @@ public class Generator {
       List<File> files = new Codegen().opts(clientOptInput).generate();
       if(files.size() > 0) {
         List<File> filesToAdd = new ArrayList<File>();
+        System.out.println("adding to " + outputFolder);
         filesToAdd.add(new File(outputFolder));
         ZipUtil zip = new ZipUtil();
         zip.compressFiles(filesToAdd, outputFilename);