com.foc.web.modules.business.PrnLayout_BrowserWindowOpenerStreamResource.java Source code

Introduction

Here is the source code for com.foc.web.modules.business.PrnLayout_BrowserWindowOpenerStreamResource.java

Source

/*******************************************************************************
 * Copyright 2016 Antoine Nicolas SAMAHA
 * 
 * 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.foc.web.modules.business;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import com.foc.ConfigInfo;
import com.foc.business.printing.PrnLayout;
import com.foc.business.printing.gui.PrintingAction;
import com.foc.util.Utils;
import com.vaadin.server.ConnectorResource;
import com.vaadin.server.DownloadStream;
import com.vaadin.server.StreamResource;

public class PrnLayout_BrowserWindowOpenerStreamResource extends StreamResource {

    private PrintingAction printingAction = null;
    private PrnLayout prnLayout = null;
    private String outputFileNameWithoutExtension = null;
    private String errorMessage = null;
    private int outputFormat = PDF;

    public static final int PDF = 0;
    public static final int DOC = 1;
    public static final int RTF = 2;

    public PrnLayout_BrowserWindowOpenerStreamResource(PrnLayout prnLayout, PrintingAction printingaction) {
        this(prnLayout, printingaction, PDF);
    }

    public PrnLayout_BrowserWindowOpenerStreamResource(PrnLayout prnLayout, PrintingAction printingAction,
            int format) {
        //      super(null, "printnig_"+System.currentTimeMillis()+ (wordDoc ? ".docx" : ".pdf"));
        super(null, "printnig_" + System.currentTimeMillis());
        outputFileNameWithoutExtension = "printnig_" + System.currentTimeMillis();
        this.prnLayout = prnLayout;
        this.printingAction = printingAction;
        this.outputFormat = format;
        setCacheTime(1);
    }

    public void dispose() {
        printingAction = null;
        prnLayout = null;
    }

    protected boolean isWithLogo() {
        return true;
    }

    public PrintingAction getPrintingAction() {
        return printingAction;
    }

    protected void beforeGetStream() {
    }

    @Override
    public DownloadStream getStream() {
        beforeGetStream();
        DownloadStream downloadStream = null;
        if (getPrintingAction() != null && getPrintingAction().getLauncher() != null) {
            if (prnLayout != null) {

                //            if(this.prnLayout_Table.getPrintLogoCheckBox() != null){
                //              boolean withLogo = this.prnLayout_Table.getPrintLogoCheckBox().getValue();
                //              this.prnLayout_Table.getPrintingAction().getLauncher().setWithLogo(withLogo);
                //            }

                getPrintingAction().getLauncher().setWithLogo(isWithLogo());
                getPrintingAction().setLaunchedAutomatically(false);

                byte[] bytes = null;
                if (outputFormat == DOC) {
                    if (outputFileNameWithoutExtension != null)
                        setFilename(outputFileNameWithoutExtension + ".docx");
                    bytes = getPrintingAction().getLauncher().printWordDocument(prnLayout);
                    if (bytes != null)
                        setMIMEType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
                } else if (outputFormat == RTF) {
                    if (outputFileNameWithoutExtension != null)
                        setFilename(outputFileNameWithoutExtension + ".rtf");
                    bytes = getPrintingAction().getLauncher().printRTFDocument(prnLayout);
                    if (bytes != null)
                        setMIMEType("application/rtf");
                } else {
                    if (outputFileNameWithoutExtension != null)
                        setFilename(outputFileNameWithoutExtension + ".pdf");
                    bytes = getPrintingAction().getLauncher().web_FillReport(prnLayout, prnLayout.getFileName());
                    if (bytes != null)
                        setMIMEType("application/pdf");
                }
                if (bytes != null) {
                    setStreamSource(new FStreamSource(bytes));
                }
                downloadStream = super.getStream();
            }
        } else {
            StreamSource source = new StreamSource() {
                @Override
                public InputStream getStream() {
                    try {
                        return new ByteArrayInputStream(getErrorMessageAsHTML().getBytes("UTF-8"));
                    } catch (UnsupportedEncodingException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    }
                    return null;
                }
            };
            // second, create a StreamResource and pass the previous StreamResource:
            StreamResource resource = new StreamResource(source, "file.html");
            resource.setMIMEType("text/html; charset=utf-8");
            downloadStream = ((ConnectorResource) resource).getStream();
        }
        return downloadStream;
    }

    public PrnLayout getPrnLayout() {
        return prnLayout;
    }

    public void setPrnLayout(PrnLayout prnLayout) {
        this.prnLayout = prnLayout;
    }

    //   public boolean isWordDoc() {
    //      return wordDoc;
    //   }
    //
    //   public void setWordDoc(boolean wordDoc) {
    //      this.wordDoc = wordDoc;
    //   }

    public String getErrorMessageAsHTML() {
        String message = getErrorMessage();
        if (Utils.isStringEmpty(message)) {
            if (ConfigInfo.isArabic()) {
                message = "  ";
            } else {
                message = "Could not print report";
            }
        }
        String htmlPrefix = "<div style=\"color:blue;font-family:Arial;font-size:36;position:fixed;float:left;top:50%;left: 50%;transform: translate(-50%, -50%);\" >";
        String htmlSuffix = "</div>";

        return htmlPrefix + message + htmlSuffix;
    }

    public String getErrorMessage() {
        return errorMessage;
    }

    public void setErrorMessage(String errorMessage) {
        this.errorMessage = errorMessage;
    }

}