From e08a5a9d83aeb979181ef1469ddc13ae90d6b134 Mon Sep 17 00:00:00 2001
From: hrachya <hrachya@HYEGHSEALNXD1>
Date: Tue, 26 May 2015 17:01:07 -0700
Subject: [PATCH] Add 'modelPackage' and 'apiPackage' as command line options
 Add setters for modelPackage and apiPackage

---
 .../swagger/codegen/DefaultCodegen.java        | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
index 97c5798f2234..6571990efc73 100644
--- a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
+++ b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
@@ -87,6 +87,14 @@ public class DefaultCodegen {
     if(additionalProperties.containsKey("t")) {
       this.setTemplateDir((String)additionalProperties.get("t"));
     }
+    
+    if(additionalProperties.containsKey("modelPackage")) {
+      this.setModelPackage((String)additionalProperties.get("modelPackage"));
+    }
+    
+    if(additionalProperties.containsKey("apiPackage")) {
+      this.setApiPackage((String)additionalProperties.get("apiPackage"));
+    }
   }
 
   // override with any special post-processing
@@ -183,6 +191,14 @@ public class DefaultCodegen {
     this.templateDir = templateDir;
   }
 
+  public void setModelPackage(String modelPackage) {
+    this.modelPackage = modelPackage;
+  }
+
+  public void setApiPackage(String apiPackage) {
+    this.apiPackage = apiPackage;
+  }
+
   public String toApiFilename(String name) {
     return toApiName(name);
   }
@@ -288,6 +304,8 @@ public class DefaultCodegen {
     importMapping.put("LocalTime", "org.joda.time.*");
     
     cliOptions.add(new CliOption("t", "template-dir", true, "folder containing the template files"));
+    cliOptions.add(new CliOption("modelPackage", "model-package", true, "Package for generated models"));
+    cliOptions.add(new CliOption("apiPackage", "api-package", true, "Package for generated api classes"));
   }