package com.ryo.metadata.web.extend; import freemarker.core.Environment; import freemarker.template.Configuration; import freemarker.template.TemplateException; import freemarker.template.TemplateModelException; import org.apache.commons.lang3.StringUtils; import java.util.Map; /** * @author houbinbin * @since 16/4/29 */ public class DirectiveUtils { public static final String BLOCK = "__ftl_override__"; public static final String OVERRIDE_CURRENT_NODE = "__ftl_override_current_node"; public static String getOverrideVariableName(String name) { return BLOCK + name; } public static void exposeRapidMacros(Configuration conf) { conf.setSharedVariable(BlockDirective.DIRECTIVE_NAME, new BlockDirective()); conf.setSharedVariable(ExtendsDirective.DIRECTIVE_NAME, new ExtendsDirective()); conf.setSharedVariable(OverrideDirective.DIRECTIVE_NAME, new OverrideDirective()); conf.setSharedVariable(SuperDirective.DIRECTIVE_NAME, new SuperDirective()); } static String getRequiredParam(Map params, String key) throws TemplateException { Object value = params.get(key); if (value == null || StringUtils.isEmpty(value.toString())) { throw new TemplateModelException("not found required parameter:" + key + " for directive"); } return value.toString(); } static String getParam(Map params, String key, String defaultValue) throws TemplateException { Object value = params.get(key); return value == null ? defaultValue : value.toString(); } static TemplateDirectiveBodyOverrideWraper getOverrideBody(Environment env, String name) throws TemplateModelException { TemplateDirectiveBodyOverrideWraper value = (TemplateDirectiveBodyOverrideWraper) env.getVariable(DirectiveUtils.getOverrideVariableName(name)); return value; } static void setTopBodyForParentBody(Environment env, TemplateDirectiveBodyOverrideWraper topBody, TemplateDirectiveBodyOverrideWraper overrideBody) { TemplateDirectiveBodyOverrideWraper parent = overrideBody; while (parent.parentBody != null) { parent = parent.parentBody; } parent.parentBody = topBody; } }