Skip to content

Commit 7f5566a

Browse files
committed
use file template loader class for custom templates
1 parent e31243b commit 7f5566a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1013,7 +1013,12 @@ private com.github.jknack.handlebars.Template getHandlebars(String templateFile)
10131013
if (templateFile.startsWith(config.templateDir())) {
10141014
templateFile = templateFile.replaceFirst(config.templateDir(), StringUtils.EMPTY);
10151015
}
1016-
final TemplateLoader templateLoader = new ClassPathTemplateLoader("/" + config.templateDir(), ".mustache");
1016+
TemplateLoader templateLoader = null;
1017+
if (config.additionalProperties().get(CodegenConstants.TEMPLATE_DIR) != null) {
1018+
templateLoader = new FileTemplateLoader(config.templateDir(), ".mustache");
1019+
} else {
1020+
templateLoader = new ClassPathTemplateLoader("/" + config.templateDir(), ".mustache");
1021+
}
10171022
final Handlebars handlebars = new Handlebars(templateLoader);
10181023
config.addHandlebarHelpers(handlebars);
10191024

0 commit comments

Comments
 (0)