org.beangle.ems.system.web.action.FileAction Maven / Gradle / Ivy

Go to download
/*
 * Beangle, Agile Development Scaffold and Toolkit
 *
 * Copyright (c) 2005-2016, 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 .
 */
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 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;
  }

}