From 27697afa9f3c6c4ed9db1de9fb840907a87bfd64 Mon Sep 17 00:00:00 2001
From: Deepak Michael <mikee76@gmail.com>
Date: Thu, 18 Aug 2011 10:33:29 +0530
Subject: [PATCH] Swagger: AS3 sdk - Using another template for output wrappers

---
 conf/as3/templates/WrapperObject.st           | 29 +++++++++++++++++++
 .../swagger/codegen/LibraryCodeGenerator.java |  9 ++++--
 2 files changed, 36 insertions(+), 2 deletions(-)
 create mode 100644 conf/as3/templates/WrapperObject.st

diff --git a/conf/as3/templates/WrapperObject.st b/conf/as3/templates/WrapperObject.st
new file mode 100644
index 00000000000..73f853c9f89
--- /dev/null
+++ b/conf/as3/templates/WrapperObject.st
@@ -0,0 +1,29 @@
+package $packageName$ {
+
+$imports:{ import |
+import $import$;
+}$
+
+    /**
+     * $model.description$
+     * NOTE: This class is auto generated by the drive code generator program so please do not edit the class manually.
+     * @author deepak
+     *
+     */
+    public class $className$ extends $extends$ {
+
+    $fields:{ field |
+
+    /**
+     * $field.description$
+     * $if(field.required)$@Required$endif$
+     * $if(field.allowableValues)$[AllowableValues(value="$field.allowedValuesString$"]$endif$
+     */
+$if(!field.fieldDefinition.collectionItemType)$
+     [XmlElement(name="$field.fieldDefinition.name$")]$endif$
+$if(field.fieldDefinition.collectionItemType)$
+     [XmlElements(name="$field.fieldDefinition.collectionItemName$", type="$field.fieldDefinition.collectionItemType$")]$endif$
+     public var $field.fieldDefinition.name$: $field.fieldDefinition.returnType$ $field.fieldDefinition.initialization$;$\r$}$
+
+    }
+}
\ No newline at end of file
diff --git a/src/main/java/com/wordnik/swagger/codegen/LibraryCodeGenerator.java b/src/main/java/com/wordnik/swagger/codegen/LibraryCodeGenerator.java
index 540b081d001..101f97d5c87 100644
--- a/src/main/java/com/wordnik/swagger/codegen/LibraryCodeGenerator.java
+++ b/src/main/java/com/wordnik/swagger/codegen/LibraryCodeGenerator.java
@@ -46,6 +46,7 @@ public class LibraryCodeGenerator {
 	private static String MODEL_OBJECT_TEMPLATE = "ModelObject";
     private static String API_OBJECT_TEMPLATE = "ResourceObject";
     private static final String ENUM_OBJECT_TEMPLATE = "EnumObject";
+    private static final String WRAPPER_OBJECT_TEMPLATE = "WrapperObject";
 
     private static final String PACKAGE_NAME = "packageName";
     private ApiConfiguration config = null;
@@ -273,7 +274,11 @@ public class LibraryCodeGenerator {
 
     private void generateOutputWrappers(List<Resource> resources, StringTemplateGroup templateGroup) {
         List<String> generatedClasses = new ArrayList<String>();
-        StringTemplate template;
+        StringTemplate template = templateGroup.getInstanceOf(WRAPPER_OBJECT_TEMPLATE);
+        if(template == null){
+            System.out.println("WrapperObject template not found to generate output wrappers");
+            return;
+        }
 
         for(Resource resource: resources) {
             if(resource.getEndPoints() != null) {
@@ -298,7 +303,7 @@ public class LibraryCodeGenerator {
                                                 }
                                             }
                                         }
-                                        template = templateGroup.getInstanceOf(MODEL_OBJECT_TEMPLATE);
+                                        template = templateGroup.getInstanceOf(WRAPPER_OBJECT_TEMPLATE);
 
                                         template.setAttribute("fields", model.getFields());
                                         template.setAttribute("imports", imports);