cc.kune.core.server.manager.file.FileDownloadManagerUtils.java Source code

Introduction

Here is the source code for cc.kune.core.server.manager.file.FileDownloadManagerUtils.java

Source

/*
 *
 * Copyright (C) 2007-2015 Licensed to the Comunes Association (CA) under
 * one or more contributor license agreements (see COPYRIGHT for details).
 * The CA licenses this file to you under the GNU Affero General Public
 * License version 3, (the "License"); you may not use this file except in
 * compliance with the License. This file is part of kune.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 \*/
package cc.kune.core.server.manager.file;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringWriter;
import java.util.List;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

// TODO: Auto-generated Javadoc
/**
 * The Class FileDownloadManagerUtils.
 * 
 * @author vjrj@ourproject.org (Vicente J. Ruiz Jurado)
 */
public class FileDownloadManagerUtils {

    /** The Constant LOG. */
    public static final Log LOG = LogFactory.getLog(FileDownloadManagerUtils.class);

    /**
     * Gets the inpu stream as string.
     * 
     * @param iStream
     *          the i stream
     * @return the inpu stream as string
     * @throws IOException
     *           Signals that an I/O exception has occurred.
     */
    public static String getInpuStreamAsString(final InputStream iStream) throws IOException {
        final StringWriter writer = new StringWriter();
        IOUtils.copy(iStream, writer, "UTF-8");
        return writer.toString();
    }

    /**
     * Gets the input stream in resource bases.
     * 
     * @param resourceBases
     *          the resource bases
     * @param filename
     *          the filename
     * @return the input stream in resource bases
     */
    public static InputStream getInputStreamInResourceBases(final List<String> resourceBases,
            final String filename) {
        InputStream in = null;
        final File icon = searchFileInResourceBases(resourceBases, filename);
        try {
            if (icon != null) {
                in = new BufferedInputStream(new FileInputStream(icon));
            }
        } catch (final FileNotFoundException e) {
            LOG.error(String.format("Cannot read filename: %s in %s", filename, resourceBases.toString()));
        }
        return in;
    }

    /**
     * Return file.
     * 
     * @param in
     *          the in
     * @param out
     *          the out
     * @throws FileNotFoundException
     *           the file not found exception
     * @throws IOException
     *           Signals that an I/O exception has occurred.
     */
    public static void returnFile(final InputStream in, final OutputStream out)
            throws FileNotFoundException, IOException {
        try {
            final byte[] buf = new byte[4 * 1024]; // 4K buffer
            int bytesRead;
            while ((bytesRead = in.read(buf)) != -1) {
                out.write(buf, 0, bytesRead);
            }
        } finally {
            if (in != null) {
                in.close();
            }
        }
    }

    /**
     * Return file.
     * 
     * @param filename
     *          the filename
     * @param out
     *          the out
     * @throws FileNotFoundException
     *           the file not found exception
     * @throws IOException
     *           Signals that an I/O exception has occurred.
     */
    public static void returnFile(final String filename, final OutputStream out)
            throws FileNotFoundException, IOException {
        InputStream in = null;
        try {
            in = new BufferedInputStream(new FileInputStream(filename));
            final byte[] buf = new byte[4 * 1024]; // 4K buffer
            int bytesRead;
            while ((bytesRead = in.read(buf)) != -1) {
                out.write(buf, 0, bytesRead);
            }
        } finally {
            if (in != null) {
                in.close();
            }
        }
    }

    /**
     * Return not found404.
     * 
     * @param resp
     *          the resp
     * @throws IOException
     *           Signals that an I/O exception has occurred.
     */
    public static void returnNotFound404(final HttpServletResponse resp) throws IOException {
        resp.getWriter().println("Content not found");
        resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
    }

    /**
     * Search file in resource bases.
     * 
     * @param resourceBases
     *          the resource bases
     * @param filename
     *          the filename
     * @return the file
     */
    public static File searchFileInResourceBases(final List<String> resourceBases, final String filename) {
        File icon = null;
        for (final String path : resourceBases) {
            final String pathAndfilename = path + (path.endsWith(File.separator) ? "" : File.separator) + filename;
            final File file = new File(pathAndfilename);
            if (file.exists()) {
                icon = file;
                break;
            }
        }
        return icon;
    }
}