com.blade.mvc.view.template.JetbrickTemplateEngine Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of blade-template-jetbrick Show documentation
There is a newer version: 0.1.3
Show newest version
/**
 * Copyright (c) 2016, biezhi 王爵 ([email protected])
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.blade.mvc.view.template;

import com.blade.BladeException;
import com.blade.mvc.WebContext;
import com.blade.mvc.http.Request;
import com.blade.mvc.http.Response;
import com.blade.mvc.http.Session;
import com.blade.mvc.ui.ModelAndView;
import com.blade.mvc.ui.template.TemplateEngine;
import jetbrick.template.*;
import jetbrick.template.resolver.GlobalResolver;

import java.io.Writer;
import java.util.Map;
import java.util.Properties;

/**
 * JetbrickTemplateEngine
 *
 * @author biezhi
 * @since 1.0
 */
public class JetbrickTemplateEngine implements TemplateEngine {

    private JetEngine jetEngine;
    private Properties config = new Properties();
    private String suffix = ".html";

    public JetbrickTemplateEngine() {
        config.put(JetConfig.TEMPLATE_SUFFIX, suffix);

        Class bootClass = WebContext.blade().bootClass();
        if (null != bootClass) {
            config.put(JetConfig.AUTOSCAN_PACKAGES, bootClass.getPackage().getName());
        }

        String $classpathLoader = "jetbrick.template.loader.ClasspathResourceLoader";
        config.put(JetConfig.TEMPLATE_LOADERS, "$classpathLoader");
        config.put("$classpathLoader", $classpathLoader);
        config.put("$classpathLoader.root", "/templates/");
        config.put("$classpathLoader.reloadable", "true");
    }

    public JetbrickTemplateEngine(Properties config) {
        this.config = config;
        jetEngine = JetEngine.create(config);
    }

    public JetbrickTemplateEngine(String conf) {
        jetEngine = JetEngine.create(conf);
    }

    public JetbrickTemplateEngine(JetEngine jetEngine) {
        if (null == jetEngine) {
            throw new IllegalArgumentException("jetEngine must not be null");
        }
        this.jetEngine = jetEngine;
    }

    @Override
    public void render(ModelAndView modelAndView, Writer writer) throws TemplateException {
        if (null == jetEngine) {
            this.jetEngine = JetEngine.create(config);
        }
        Map modelMap = modelAndView.getModel();

        Request request = WebContext.request();
        Response response = WebContext.response();
        Session session = request.session();

        modelMap.putAll(request.attributes());

        if (null != session) {
            modelMap.putAll(session.attributes());
        }

        JetContext context = new JetContext(modelMap.size());
        context.putAll(modelMap);

        String templateName = modelAndView.getView().endsWith(suffix) ? modelAndView.getView() : modelAndView.getView() + suffix;
        try {
            JetTemplate template = jetEngine.getTemplate(templateName);
            template.render(context, writer);
        } catch (Exception e) {
            throw new BladeException(e);
        }
    }

    public JetEngine getJetEngine() {
        return jetEngine;
    }

    public void setJetEngine(JetEngine jetEngine) {
        this.jetEngine = jetEngine;
    }

    public JetGlobalContext getGlobalContext() {
        if (null == jetEngine) {
            this.jetEngine = JetEngine.create(config);
        }
        return jetEngine.getGlobalContext();
    }

    public GlobalResolver getGlobalResolver() {
        if (null == jetEngine) {
            this.jetEngine = JetEngine.create(config);
        }
        return jetEngine.getGlobalResolver();
    }

    public Properties getConfig() {
        return config;
    }

    public TemplateEngine addConfig(String key, String value) {
        config.put(key, value);
        return this;
    }

    public String getSuffix() {
        return suffix;
    }

    public void setSuffix(String suffix) {
        this.suffix = suffix;
    }
}