com.whaleal.icefrog.extra.template.engine.enjoy.EnjoyEngine 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.enjoy;

import com.jfinal.template.source.FileSourceFactory;
import com.whaleal.icefrog.core.io.FileUtil;
import com.whaleal.icefrog.core.util.IdUtil;
import com.whaleal.icefrog.core.util.ObjectUtil;
import com.whaleal.icefrog.extra.template.Template;
import com.whaleal.icefrog.extra.template.TemplateConfig;
import com.whaleal.icefrog.extra.template.TemplateConfig.ResourceMode;
import com.whaleal.icefrog.extra.template.TemplateEngine;

import java.io.File;

/**
 * Enjoy库的引擎包装
 *
 * @author looly
 */
public class EnjoyEngine implements TemplateEngine {

    private com.jfinal.template.Engine engine;
    private ResourceMode resourceMode;

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

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

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

    /**
     * 构造
     *
     * @param engine {@link com.jfinal.template.Engine}
     */
    public EnjoyEngine( com.jfinal.template.Engine engine ) {
        init(engine);
    }
    // --------------------------------------------------------------------------------- Constructor end

    /**
     * 创建引擎
     *
     * @param config 模板配置
     * @return {@link com.jfinal.template.Engine}
     */
    private static com.jfinal.template.Engine createEngine( TemplateConfig config ) {
        final com.jfinal.template.Engine engine = com.jfinal.template.Engine.create("Hutool-Enjoy-Engine-" + IdUtil.fastSimpleUUID());
        engine.setEncoding(config.getCharsetStr());

        switch (config.getResourceMode()) {
            case STRING:
                // 默认字符串类型资源:
                break;
            case CLASSPATH:
                engine.setToClassPathSourceFactory();
                engine.setBaseTemplatePath(config.getPath());
                break;
            case FILE:
                engine.setSourceFactory(new FileSourceFactory());
                engine.setBaseTemplatePath(config.getPath());
                break;
            case WEB_ROOT:
                engine.setSourceFactory(new FileSourceFactory());
                final File root = FileUtil.file(FileUtil.getWebRoot(), config.getPath());
                engine.setBaseTemplatePath(FileUtil.getAbsolutePath(root));
                break;
            default:
                break;
        }

        return engine;
    }

    @Override
    public TemplateEngine init( TemplateConfig config ) {
        if (null == config) {
            config = TemplateConfig.DEFAULT;
        }
        this.resourceMode = config.getResourceMode();
        init(createEngine(config));
        return this;
    }

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

    @Override
    public Template getTemplate( String resource ) {
        if (null == this.engine) {
            init(TemplateConfig.DEFAULT);
        }
        if (ObjectUtil.equal(ResourceMode.STRING, this.resourceMode)) {
            return EnjoyTemplate.wrap(this.engine.getTemplateByString(resource));
        }
        return EnjoyTemplate.wrap(this.engine.getTemplate(resource));
    }
}