org.beangle.ems.system.web.action.FileAction.java Source code

Introduction

Here is the source code for org.beangle.ems.system.web.action.FileAction.java

Source

/*
 * Beangle, Agile Development Scaffold and Toolkit
 *
 * Copyright (c) 2005-2014, Beangle Software.
 *
 * Beangle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Beangle 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Beangle.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.beangle.ems.system.web.action;

import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.LineNumberReader;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.lang3.time.DateFormatUtils;
import org.beangle.commons.collection.CollectUtils;
import org.beangle.commons.http.mime.MimeTypeProvider;
import org.beangle.commons.lang.Strings;
import org.beangle.commons.web.io.StreamDownloader;
import org.beangle.struts2.action.ActionSupport;

public class FileAction extends ActionSupport {

    private MimeTypeProvider mimeTypeProvider;

    private StreamDownloader streamDownloader;

    public String index() {
        put("path", getPath());
        return forward();
    }

    public String list() {
        String path = getPath();
        File curFile = new File(path);
        if (curFile.exists()) {
            File[] files = list(curFile);
            put("files", files);
            put("path", path);
            if (null != curFile.getParent()) {
                put("parent", curFile.getParent());
            }
        }
        put("dateformat", new DateFormatUtils());
        put("mimeType", new FileMimeType(mimeTypeProvider));
        return forward();
    }

    public String download() throws IOException {
        String path = get("path");
        FileMimeType fileMimeType = new FileMimeType(mimeTypeProvider);
        if (Strings.isNotBlank(path)) {
            File file = new File(path);
            if (!file.isFile()) {
                return null;
            }
            boolean download = getBool("download");
            if (!download && fileMimeType.isTextType(file)) {
                List<String> lines = CollectUtils.newArrayList();
                LineNumberReader reader = new LineNumberReader(new FileReader(file));
                String line = reader.readLine();
                while (null != line) {
                    lines.add(line);
                    line = reader.readLine();
                }
                put("lines", lines);
                put("file", file);
                reader.close();
                return forward("content");
            } else {
                streamDownloader.download(getRequest(), getResponse(), file);
            }
        }
        return null;
    }

    private File[] list(File curFile) {
        try {
            String path = curFile.getAbsolutePath();
            String[] sFiles = curFile.list();
            File[] files = new File[sFiles.length];
            for (int n = 0; n < sFiles.length; n++) {
                files[n] = new File(path + "/" + sFiles[n]);
            }
            Arrays.sort(files);
            return files;
        } catch (Exception e) {
            return null;
        }
    }

    private String getPath() {
        String path = get("path");
        if (Strings.isEmpty(path)) {
            path = getRequest().getSession().getServletContext().getRealPath("");
        }
        String sRet = path;// .replace('\\', '/');
        File file = new File(path);
        try {
            sRet = file.getCanonicalPath();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        if (file.isDirectory()) {
            if (!sRet.endsWith(File.separator))
                sRet += File.separator;
        }
        return sRet;
    }

    public String newFolder() throws IOException {
        String path = getPath();
        String newFolder = get("newFolder");
        File file = new File(path + newFolder);
        file.createNewFile();
        return redirect("list", "info.action.success", "&path=" + path);
    }

    public void setMimeTypeProvider(MimeTypeProvider mimeTypeProvider) {
        this.mimeTypeProvider = mimeTypeProvider;
    }

    public void setStreamDownloader(StreamDownloader streamDownloader) {
        this.streamDownloader = streamDownloader;
    }

}