org.primefaces.extensions.component.documentviewer.DocumentViewerRenderer Maven / Gradle / Ivy

Go to download
package org.primefaces.extensions.component.documentviewer;

import org.apache.commons.lang3.StringUtils;
import org.primefaces.renderkit.CoreRenderer;
import org.primefaces.util.ComponentUtils;
import org.primefaces.util.DynamicResourceBuilder;

import javax.faces.application.Resource;
import javax.faces.application.ResourceHandler;
import javax.faces.component.UIComponent;
import javax.faces.context.FacesContext;
import javax.faces.context.ResponseWriter;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import org.primefaces.application.resource.DynamicContentType;

public class DocumentViewerRenderer extends CoreRenderer {

    @Override
    public void encodeEnd(FacesContext context, UIComponent component) throws IOException {
        DocumentViewer documentViewer = (DocumentViewer) component;
        encodeMarkup(context, documentViewer);
    }

    private void encodeMarkup(FacesContext context,DocumentViewer documentViewer) throws IOException {

        ResponseWriter writer = context.getResponseWriter();

        writer.startElement("iframe", documentViewer);
        writer.writeAttribute("id", documentViewer.getClientId(), null);
        writer.writeAttribute("style", documentViewer.getStyle(), null);
        writer.writeAttribute("width",documentViewer.getWidth() != null ? documentViewer.getWidth() : "100%", null);
        writer.writeAttribute("height", documentViewer.getHeight(), null);
        writer.writeAttribute("allowfullscreen", "", null);
        writer.writeAttribute("webkitallowfullscreen", "", null);
        writer.writeAttribute("src",generateSrc(context, documentViewer), null);
        writer.endElement("iframe");

    }

    private String generateSrc(FacesContext context,DocumentViewer documentViewer)	throws IOException {
        String imageSrc;
        try {
            imageSrc = URLEncoder.encode(getDocumentSource(context, documentViewer), "UTF-8");
        } catch (Exception ex) {
            throw new IOException(ex);
        }

        StringBuilder srcBuilder = new StringBuilder();
        srcBuilder.append(getResourceURL(documentViewer,context));
        srcBuilder.append("&file=");
        srcBuilder.append(imageSrc);
        srcBuilder.append(generateHashString(documentViewer,context));

        return srcBuilder.toString();
    }

    private String generateHashString(DocumentViewer documentViewer,FacesContext context) {

        List params = new ArrayList(1);
        params.add("locale=" + getCalculatedLocale(documentViewer, context));
        if(documentViewer.getPage() != null){
            params.add("page="+documentViewer.getPage());
        }
		
		if(documentViewer.getZoom() != null){
            params.add("zoom=" + documentViewer.getZoom());
        }

        if(!params.isEmpty()){
            return "#" + StringUtils.join(params, "&");
        }else{
            return "";
        }
    }

    private String getResourceURL(DocumentViewer documentViewer, FacesContext context) {
        ResourceHandler handler = context.getApplication().getResourceHandler();
        return context.getExternalContext().encodeResourceURL(handler.createResource("documentviewer/viewer.html", "primefaces-extensions").getRequestPath());
    }

    private Locale getCalculatedLocale(DocumentViewer documentViewer,FacesContext context){
        Object locale = documentViewer.getLocale();
        if(locale == null){
            return context.getViewRoot().getLocale();
        }else{
            if(locale instanceof Locale){
                return (Locale) locale;
            }else if(locale instanceof String){
                return ComponentUtils.toLocale((String) locale);
            }else{
                throw new IllegalArgumentException("Type:" + locale.getClass() + " is not a valid locale type for calendar:" + documentViewer.getClientId(context));
            }
        }
    }

    protected String getDocumentSource(FacesContext context,DocumentViewer documentViewer) throws UnsupportedEncodingException {

        String name = documentViewer.getName();

        if (name != null) {
            String libName = documentViewer.getLibrary();
            ResourceHandler handler = context.getApplication().getResourceHandler();
            Resource res = handler.createResource(name, libName);

            if (res == null) {
                return "RES_NOT_FOUND";
            } else {
                String requestPath = res.getRequestPath();
                return context.getExternalContext().encodeResourceURL(requestPath);
            }
        } else {
            return DynamicResourceBuilder.build(context,documentViewer.getValue(), documentViewer, documentViewer.isCache(), DynamicContentType.STREAMED_CONTENT, true);
        }
    }
}