com.whaleal.icefrog.extra.template.engine.beetl.BeetlEngine Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of icefrog-extra Show documentation
icefrog 扩展工具类（提供其它类库的封装）
There is a newer version: 1.1.71
package com.whaleal.icefrog.extra.template.engine.beetl;

import com.whaleal.icefrog.core.io.IORuntimeException;
import com.whaleal.icefrog.extra.template.Template;
import com.whaleal.icefrog.extra.template.TemplateConfig;
import com.whaleal.icefrog.extra.template.TemplateEngine;
import org.beetl.core.Configuration;
import org.beetl.core.GroupTemplate;
import org.beetl.core.ResourceLoader;
import org.beetl.core.resource.*;

import java.io.IOException;

/**
 * Beetl模板引擎封装
 *
 * @author looly
 */
public class BeetlEngine implements TemplateEngine {

    private GroupTemplate engine;

    // --------------------------------------------------------------------------------- Constructor start

    /**
     * 默认构造
     */
    public BeetlEngine() {
    }

    /**
     * 构造
     *
     * @param config 模板配置
     */
    public BeetlEngine( TemplateConfig config ) {
        init(config);
    }

    /**
     * 构造
     *
     * @param engine {@link GroupTemplate}
     */
    public BeetlEngine( GroupTemplate engine ) {
        init(engine);
    }
    // --------------------------------------------------------------------------------- Constructor end

    /**
     * 创建引擎
     *
     * @param config 模板配置
     * @return {@link GroupTemplate}
     */
    private static GroupTemplate createEngine( TemplateConfig config ) {
        if (null == config) {
            config = TemplateConfig.DEFAULT;
        }

        switch (config.getResourceMode()) {
            case CLASSPATH:
                return createGroupTemplate(new ClasspathResourceLoader(config.getPath(), config.getCharsetStr()));
            case FILE:
                return createGroupTemplate(new FileResourceLoader(config.getPath(), config.getCharsetStr()));
            case WEB_ROOT:
                return createGroupTemplate(new WebAppResourceLoader(config.getPath(), config.getCharsetStr()));
            case STRING:
                return createGroupTemplate(new StringTemplateResourceLoader());
            case COMPOSITE:
                //TODO 需要定义复合资源加载器
                return createGroupTemplate(new CompositeResourceLoader());
            default:
                return new GroupTemplate();
        }
    }

    /**
     * 创建自定义的模板组 {@link GroupTemplate}，配置文件使用全局默认

     * 此时自定义的配置文件可在ClassPath中放入beetl.properties配置
     *
     * @param loader {@link ResourceLoader}，资源加载器
     * @return {@link GroupTemplate}
     */
    private static GroupTemplate createGroupTemplate( ResourceLoader loader ) {
        try {
            return createGroupTemplate(loader, Configuration.defaultConfiguration());
        } catch (IOException e) {
            throw new IORuntimeException(e);
        }
    }

    /**
     * 创建自定义的 {@link GroupTemplate}
     *
     * @param loader {@link ResourceLoader}，资源加载器
     * @param conf   {@link Configuration} 配置文件
     * @return {@link GroupTemplate}
     */
    private static GroupTemplate createGroupTemplate( ResourceLoader loader, Configuration conf ) {
        return new GroupTemplate(loader, conf);
    }

    @Override
    public TemplateEngine init( TemplateConfig config ) {
        init(createEngine(config));
        return this;
    }

    /**
     * 初始化引擎
     *
     * @param engine 引擎
     */
    private void init( GroupTemplate engine ) {
        this.engine = engine;
    }

    @Override
    public Template getTemplate( String resource ) {
        if (null == this.engine) {
            init(TemplateConfig.DEFAULT);
        }
        return BeetlTemplate.wrap(engine.getTemplate(resource));
    }
}