FileUtilAjax.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * 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 org.turro.elephant.impl.util;
 
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IAction;
import org.turro.elephant.impl.abstracts.AbstractAction;
 
public class FileUtilAjax extends AbstractAction implements IAction {
  
  public static final String
          VALID_ATTACHMENTS = "(?i).*\\.(jpg|gif|png|pdf|doc|txt|zip)$";
  
  public FileUtilAjax() {
  }
 
  @Override
  public int execute() throws Exception {
    String path = getParameter("path"),
           action = getActionParameter();
    boolean selectable = "true".equals(getParameter("sel"));
    if(path == null) {
      return IAction.NO_DATA;
    }
    File realPath = new File(ElephantContext.getRealPath(path));
    if("list-folders".equals(action)) {
      if(realPath.exists()) {
        return listFolders(realPath);
      }
      else {
        return IAction.ERROR;
      }
    }
    else if("list-files".equals(action)) {
      if(realPath.exists()) {
        return listFiles(realPath, selectable);
      }
      else {
        return IAction.ERROR;
      }
    }
    else if("list-attachments".equals(action)) {
      if(realPath.exists()) {
        return listAttachments(realPath, selectable);
      }
      else {
        return IAction.ERROR;
      }
    }
    else if("remove-attachments".equals(action)) {
      return removeAttachments(selectable, realPath);
    }
    else if("file-content".equals(action)) {
      if(realPath.exists()) {
        return fileContent(realPath);
      }
      else {
        return IAction.ERROR;
      }
    }
    else if("save-content".equals(action)) {
      return saveContent(realPath);
    }
    else if("remove-file".equals(action)) {
      if(realPath.exists()) {
        realPath.delete();
      }
      else {
        return IAction.ERROR;
      }
    }
    return IAction.NO_DATA;
  }
 
  private int removeAttachments(final boolean selectable, final File realPath) throws IOException {
    String[] files = constructor.getParameter("files").split(",");
    File f;
    for(int i = 0; i < files.length; i++) {
      f = new File(ElephantContext.getRealPath(files[i]));
      if(f.exists())
        f.delete();
    }
    return listAttachments(realPath, selectable);
  }
 
  private int saveContent(final File realPath) throws IOException {
    PrintWriter out = constructor.getOut();
    String s = constructor.getParameter("content");
    if(s != null) {
      s = s.replaceAll("\\*\\-am\\-\\*","&");
      s = s.replaceAll("\\*\\-eq\\-\\*","=");
      s = s.replaceAll("\\*\\-pl\\-\\*","+");
      OutputStreamWriter osw = FileUtil.getFileWriter(realPath);
      osw.write(s);
      osw.close();      
      out.println("Done");
    }
    else {
      out.println("No data!");
    } 
    return IAction.DONE;
  }
 
  private int fileContent(final File realPath) throws IOException {
    BufferedReader br = FileUtil.getBufferedFile(realPath);
    String s;
    PrintWriter out = constructor.getOut();
    while((s = br.readLine()) != null) {
      out.println(s);
    }
    br.close();      
    return IAction.DONE;
  }
 
  private int listAttachments(final File realPath, final boolean selectable) throws IOException {
    File[] children = realPath.listFiles();
    PrintWriter out = constructor.getOut();
    out.println("<ul>");
    String s;
    for(int i = 0; i < children.length; i++) {
      s = children[i].getAbsolutePath();
      if(s.matches(VALID_ATTACHMENTS)) {
        out.print("<li path='" + ElephantContext.getRelativePath(s) + "'>");
        if(selectable) {
          out.print("<input type='checkbox'/>");
        }
        if(children[i].isFile()) {
          out.print("<a class='file'>");
        }
        out.print(children[i].getName());
        out.print("</a>");        
        out.println("</li>");
      }
    }
    out.println("</ul>");
    return IAction.DONE;
  }
 
  private int listFiles(final File realPath, final boolean selectable) throws IOException {
    File[] children = realPath.listFiles();
    PrintWriter out = constructor.getOut();
    out.println("<ul>");
    for(int i = 0; i < children.length; i++) {
      out.print("<li path='" + ElephantContext.getRelativePath(children[i].getAbsolutePath()) + "'>");
      if(selectable) {
        out.print("<input type='checkbox'/>");
      }
      if(children[i].isDirectory()) {
        out.print("<a class='folder'>");
      }
      else if(children[i].isFile()) {
        out.print("<a class='file'>");
      }
      out.print(children[i].getName());
      out.print("</a>");        
      out.println("</li>");
    }
    out.println("</ul>");
    return IAction.DONE;
  }
 
  private int listFolders(final File realPath) throws IOException {
    File[] children = realPath.listFiles();
    PrintWriter out = constructor.getOut();
    out.println("<ul>");
    for(int i = 0; i < children.length; i++) {
      if(children[i].isDirectory()) {
        out.print("<li class='collapsed' path='" + ElephantContext.getRelativePath(children[i].getAbsolutePath()) + "'>");
        out.print("<a class='folder'>");
        out.print(children[i].getName());
        out.print("</a>");        
        out.println("</li>");
      }
    }
    out.println("</ul>");
    return IAction.DONE;
  }
 
}