loafle/openapi-generator
3
0
Fork 0
forked from loafle/openapi-generator-original
Code Pull Requests Activity

Expose Handlebars infiniteLoops setting (#7926)

Browse Source
This commit is contained in:
Slavek Kabrda 2020-12-19 14:54:46 +01:00 committed by GitHub
parent a2e5d274d6
commit 129881d04a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/openapi-generator/src/main/java/org/openapitools/codegen/templating/HandlebarsEngineAdapter.java
View File

@ -45,6 +45,7 @@ public class HandlebarsEngineAdapter extends AbstractTemplatingEngineAdapter {
// We use this as a simple lookup for valid file name extensions. This adapter will inspect .mustache (built-in) and infer the relevant handlebars filename // We use this as a simple lookup for valid file name extensions. This adapter will inspect .mustache (built-in) and infer the relevant handlebars filename
private final String[] canCompileFromExtensions = new String[]{".handlebars",".hbs",".mustache"}; private final String[] canCompileFromExtensions = new String[]{".handlebars",".hbs",".mustache"};
private boolean infiniteLoops = false;
/** /**
* Provides an identifier used to load the adapter. This could be a name, uuid, or any other string. * Provides an identifier used to load the adapter. This could be a name, uuid, or any other string.
@ -82,6 +83,7 @@ public class HandlebarsEngineAdapter extends AbstractTemplatingEngineAdapter {
StringHelpers.register(handlebars); StringHelpers.register(handlebars);
handlebars.registerHelpers(ConditionalHelpers.class); handlebars.registerHelpers(ConditionalHelpers.class);
handlebars.registerHelpers(org.openapitools.codegen.templating.handlebars.StringHelpers.class); handlebars.registerHelpers(org.openapitools.codegen.templating.handlebars.StringHelpers.class);
handlebars.setInfiniteLoops(infiniteLoops);
Template tmpl = handlebars.compile(templateFile); Template tmpl = handlebars.compile(templateFile);
return tmpl.apply(context); return tmpl.apply(context);
} }
@ -121,5 +123,16 @@ public class HandlebarsEngineAdapter extends AbstractTemplatingEngineAdapter {
// disallow any extension-only files like ".hbs" or ".mustache", and only consider a file compilable if it's handlebars or mustache (from which we later infer the handlebars filename) // disallow any extension-only files like ".hbs" or ".mustache", and only consider a file compilable if it's handlebars or mustache (from which we later infer the handlebars filename)
return Arrays.stream(canCompileFromExtensions).anyMatch(suffix -> !suffix.equalsIgnoreCase(filename) && filename.endsWith(suffix)); return Arrays.stream(canCompileFromExtensions).anyMatch(suffix -> !suffix.equalsIgnoreCase(filename) && filename.endsWith(suffix));
} }
/**
* Enable/disable infiniteLoops setting for the Handlebars engine. Enabling this allows for recursive partial inclusion.
*
* @param infiniteLoops Whether to enable (true) or disable (false)
* @return this object
*/
public HandlebarsEngineAdapter infiniteLoops(boolean infiniteLoops) {
this.infiniteLoops = infiniteLoops;
return this;
}
} }