loafle/openapi-generator-original
3
0
Fork 1
mirror of https://github.com/OpenAPITools/openapi-generator.git synced 2025-07-04 14:40:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support overriding API/model templates with library template

Browse Source
This commit is contained in:
xhh 2015-08-08 11:14:06 +08:00
parent a41361c959
commit b686b53259
2 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
modules/swagger-codegen/src/main/java/io/swagger/codegen/AbstractGenerator.java
View File

@ -60,6 +60,23 @@ public abstract class AbstractGenerator {
throw new RuntimeException("can't load template " + name); throw new RuntimeException("can't load template " + name);
} }
/**
* Get the template file path with template dir prepended, and use the
* library template if exists.
*/
public String getFullTemplateFile(CodegenConfig config, String templateFile) {
String library = config.getLibrary();
if (library != null && !"".equals(library)) {
String libTemplateFile = config.templateDir() + File.separator +
"libraries" + File.separator + library + File.separator +
templateFile;
if (templateExists(libTemplateFile)) {
return libTemplateFile;
}
}
return config.templateDir() + File.separator + templateFile;
}
public boolean templateExists(String name) { public boolean templateExists(String name) {
return this.getClass().getClassLoader().getResource(getCPResourcePath(name)) != null; return this.getClass().getClassLoader().getResource(getCPResourcePath(name)) != null;
} }

20
modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java
View File

@ -139,7 +139,8 @@ public class DefaultGenerator extends AbstractGenerator implements Generator {
if (!config.shouldOverwrite(filename)) { if (!config.shouldOverwrite(filename)) {
continue; continue;
} }
String template = readTemplate(config.templateDir() + File.separator + templateName); String templateFile = getFullTemplateFile(config, templateName);
String template = readTemplate(templateFile);
Template tmpl = Mustache.compiler() Template tmpl = Mustache.compiler()
.withLoader(new Mustache.TemplateLoader() { .withLoader(new Mustache.TemplateLoader() {
public Reader getTemplate(String name) { public Reader getTemplate(String name) {
@ -191,7 +192,8 @@ public class DefaultGenerator extends AbstractGenerator implements Generator {
continue; continue;
} }
String template = readTemplate(config.templateDir() + File.separator + templateName); String templateFile = getFullTemplateFile(config, templateName);
String template = readTemplate(templateFile);
Template tmpl = Mustache.compiler() Template tmpl = Mustache.compiler()
.withLoader(new Mustache.TemplateLoader() { .withLoader(new Mustache.TemplateLoader() {
public Reader getTemplate(String name) { public Reader getTemplate(String name) {
@ -262,19 +264,7 @@ public class DefaultGenerator extends AbstractGenerator implements Generator {
continue; continue;
} }
String templateFile = null; String templateFile = getFullTemplateFile(config, support.templateFile);
String library = config.getLibrary();
if (library != null && !"".equals(library)) {
String libTemplateFile = config.templateDir() + File.separator +
"libraries" + File.separator + library + File.separator +
support.templateFile;
if (templateExists(libTemplateFile)) {
templateFile = libTemplateFile;
}
}
if (templateFile == null) {
templateFile = config.templateDir() + File.separator + support.templateFile;
}
if (templateFile.endsWith("mustache")) { if (templateFile.endsWith("mustache")) {
String template = readTemplate(templateFile); String template = readTemplate(templateFile);