Merge pull request #21 from macisamuele/maci-add-title-fallback

Update SharedCodegen to fallback to title usage if x-model is not present
Yelp · Mar 17, 2019 · 08a09e3 · 08a09e3
2 parents 74d2a2b + e2772a3
commit 08a09e3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/plugin/src/main/java/com/yelp/codegen/SharedCodegen.kt b/plugin/src/main/java/com/yelp/codegen/SharedCodegen.kt
@@ -122,10 +122,14 @@ abstract class SharedCodegen : DefaultCodegen(), CodegenConfig {
     }
 
     /**
-     * Returns the x-model alternative name if it was defined
+     * Returns the x-model alternative name if it was defined.
+     * If the x-model alternative name is not found then the call will
+     * use the defined title, if any, or returns the input name
      */
     fun matchXModel(name: String): String {
-        return xModelMatches[name] ?: name
+        return xModelMatches[name] ?: (
+            this.swagger?.definitions?.get(name)?.title ?: name
+        )
     }
 
     /**