cz.zcu.kiv.eegdatabase.wui.components.utils.FileUtils.java Source code

Introduction

Here is the source code for cz.zcu.kiv.eegdatabase.wui.components.utils.FileUtils.java

Source

/*******************************************************************************
 * This file is part of the EEG-database project
 * 
 *   ==========================================
 *  
 *   Copyright (C) 2013 by University of West Bohemia (http://www.zcu.cz/en/)
 *  
 *  ***********************************************************************************************************************
 *  
 *   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.
 *  
 *  ***********************************************************************************************************************
 *  
 *   FileUtils.java, 2013/10/02 00:01 Jakub Rinkes
 ******************************************************************************/
package cz.zcu.kiv.eegdatabase.wui.components.utils;

import cz.zcu.kiv.eegdatabase.wui.core.file.FileDTO;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.wicket.request.IRequestCycle;
import org.apache.wicket.request.handler.resource.ResourceStreamRequestHandler;
import org.apache.wicket.request.resource.ContentDisposition;
import org.apache.wicket.util.file.Files;
import org.apache.wicket.util.io.IOUtils;
import org.apache.wicket.util.resource.FileResourceStream;
import org.apache.wicket.util.resource.IResourceStream;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * Utilities class for files.
 * 
 * @author Jakub Rinkes
 * 
 */
public class FileUtils {

    protected static Log log = LogFactory.getLog(FileUtils.class);

    /**
     * Prepared request handler for file download.
     * 
     * @param file
     * @return
     */
    public static ResourceStreamRequestHandler prepareDownloadFile(final FileDTO file) {

        if (file == null || file.getFile() == null)
            return null;

        IResourceStream stream = new FileResourceStream(file.getFile());

        ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(stream) {

            @Override
            public void detach(IRequestCycle requestCycle) {
                super.detach(requestCycle);

                if (!Files.remove(file.getFile())) {
                    log.error("Temporary file " + file.getFile().getAbsolutePath() + " cannot be deleted.");
                }
            }
        };
        return resourceStreamRequestHandler.setFileName(file.getFileName())
                .setContentDisposition(ContentDisposition.ATTACHMENT);
    }

    /**
     * Prepare byte array from file.
     * 
     * @param file
     * @return
     */
    public static byte[] getFileContent(File file) {

        if (file == null) {
            return new byte[0];
        } else {
            try {

                FileInputStream fileInputStream = new FileInputStream(file);
                byte[] byteArray = IOUtils.toByteArray(fileInputStream);
                fileInputStream.close();
                return byteArray;

            } catch (FileNotFoundException e) {
                log.error(e.getMessage(), e);
                return new byte[0];
            } catch (IOException e) {
                log.error(e.getMessage(), e);
                return new byte[0];
            }
        }

    }

    public static List<String> getFileLines(File file) throws IOException {
        List<String> lines = new ArrayList<String>();
        FileInputStream fstream = new FileInputStream(file);
        BufferedReader br = new BufferedReader(new InputStreamReader(fstream));

        String strLine;

        //Read File Line By Line
        while ((strLine = br.readLine()) != null) {
            // Print the content on the console
            lines.add(strLine);
        }

        //Close the input stream
        br.close();

        return lines;
    }

    public static boolean deleteOnExitQuietly(File file) {

        if (file == null) {
            return false;
        }

        try {
            file.deleteOnExit();
            return true;
        } catch (Exception e) {
            log.warn(e.getMessage(), e);
            return false;
        }
    }

    public static boolean deleteQuietly(File file) {

        if (file == null) {
            return false;
        }

        try {
            return file.delete();
        } catch (Exception e) {
            log.warn(e.getMessage(), e);
            return false;
        }
    }
}