com.cybermkd.plugin.beetl.BeetlRender Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ICEREST-PLUGIN Show documentation
ICEREST官方插件集合
The newest version!
package com.cybermkd.plugin.beetl;

import com.cybermkd.common.Render;
import com.cybermkd.common.http.HttpRequest;
import com.cybermkd.common.http.HttpResponse;
import com.cybermkd.common.util.Stringer;
import com.cybermkd.log.Logger;
import org.beetl.core.GroupTemplate;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Map;
import java.util.Set;


public class BeetlRender extends Render {
    private static final Logger logger = Logger.getLogger(BeetlRender.class);
    private transient static final String encoding = "UTF-8";
    private transient static final String contentType = "text/html; charset=" + encoding;
    protected String view;
    GroupTemplate gt = null;

    public BeetlRender(GroupTemplate gt, String view) {
        this.gt = gt;
        this.view = view;
    }

    @Override
    public void render(HttpRequest request, HttpResponse response, Object out) {
        response.setContentType(contentType);
        if (out == null) {
            return;
        }
        HttpServletRequest httpServletRequest = request.unwrap(HttpServletRequest.class);
        HttpServletResponse httpServletResponse = response.unwrap(HttpServletResponse.class);
        if (out instanceof ModelView) {
            ModelView modelView = (ModelView) out;

            if (Stringer.notBlank(modelView.getRedirect())) {
                try {
                    httpServletResponse.sendRedirect(modelView.getRedirect());
                } catch (IOException e) {
                    e.printStackTrace();
                }
            } else {
                Set entries = modelView.getPara().entrySet();
                for (Map.Entry entry : entries) {
                    httpServletRequest.setAttribute((String) entry.getKey(), entry.getValue());
                }
                WebRender webRender = new WebRender(gt);
                webRender.render(modelView.getView(), httpServletRequest, httpServletResponse);
            }
        } else if (out instanceof String) {
            if ("Unauthorized".equals(out)) {
                try {
                    httpServletResponse.sendRedirect("/");
                } catch (IOException e) {
                    e.printStackTrace();
                }
            } else {
                this.write(request, response, (String) out);
            }
        }

    }
}