RepositoryBean.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.repository;
 
import java.io.File;
import java.util.Iterator;
import java.util.List;
import java.util.TreeSet;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IConstructor;
 
public class RepositoryBean {
  public static final String 
          REPOSITORY_JSP = "/WEB-INF/elephant/admin/repository.jsp",
          VALID_PATTERN = ".*\\.(#)$",
          VALID_IMAGES = "gif|jpg|png|mpg",
          VALID_HTML = "html|htm",
          VALID_DOCUMENTS = "txt|cvs|pdf|odt|ott|odm|oth|ods|ots|odg|otg|odp|otp|odi|odf|odc|odb|doc|xls";
 
  private IConstructor constructor;
  private String context, subFolder;
  private boolean selectable = true;
  private String roleUpload = null, 
          roleList = null,
          roleDelete = null,
          roleCreateDir = null,
          validUploads = null, 
          rootPath = null;
  
  private String[] htmlFiles;
  
  public RepositoryBean() {
  }
 
  public void setRoleUpload(String value) {
    roleUpload = value;
  }
  
  public void setRoleList(String value) {
    roleList = value;
  }
  
  public void setRoleDelete(String value) {
    roleDelete = value;
  }
  
  public void setRoleCreateDir(String value) {
    roleCreateDir = value;
  }
  
  public void setRootPath(String value) {
    rootPath = value;
  }
  
  public void setValidUploads(String value) {
    String sep ="", values[] = value.split(" *, *");
    validUploads = "";
    for(int i = 0; i < values.length; i++) {
      if(values[i].equals("images")) {
        validUploads += sep + VALID_IMAGES;
        sep ="|";
      }
      else if(values[i].equals("html")) {
        validUploads += sep + VALID_HTML;
        sep ="|";
      }
      if(values[i].equals("documents")) {
        validUploads += sep + VALID_DOCUMENTS;
        sep ="|";
      }
    }
    validUploads = VALID_PATTERN.replaceAll("#", validUploads);
  }
  
  public void deleteFile(String file) {
    File f = new File(getRealPath(file));
    if(f.exists()) {
      org.amic.util.file.FileUtil.deleteFile(f);
    }
  }
  
  public void createFolder(String folder) {
    File f = new File(getRealPath(folder));
    f.mkdir();
  }
  
  public String[] getFiles() {
    if(!constructor.isInRole(roleList)) {
      htmlFiles = new String[0];
    }
    if(htmlFiles == null) {    
      File f = new File(getRealPath(null));
      File[] children = f.listFiles();
      TreeSet set = new TreeSet();
      String folderLink;
      if(subFolder != null && subFolder.length() > 0) {
        folderLink = ElephantContext.getRootWebPath() + "/xprepository?context=" + context;
        set.add(
                "<li>" +
                "<a class='folder' href='" + folderLink + "'>..</a>" +
                "</li>");
      }
      for(int i = 0; children != null && i < children.length; i++) {
        folderLink = ElephantContext.getRootWebPath() + "/xprepository?context=" + context + "&subfolder=" + children[i].getName();
        set.add(
                "<li>" +
                (isSelectable() ? "<input type='checkbox' name='check' value ='" + children[i].getName() + "'/>" : "") +
                (children[i].isDirectory() ?
                  "<a class='folder' href='" + folderLink + "'>" :
                  "<a repository=\"" + getImageUrl(children[i].getName()) + "\" class='file'>") +
                children[i].getName() +
                "</a>" +
                "</li>");
      }
      htmlFiles = (String[])set.toArray(new String[0]);
    }
    return htmlFiles;
  }
  
  public String getRealFolder() {
    return getRealPath(null);
  }
  
  public String convertToValidName(String fileName) {
    return fileName.replaceAll("[^a-zA-Z0-9\\-\\_\\.]", "_");
  }
  
  public String getContext() {
    return context;
  }
 
  public void setContext(String context) {
    this.context = context;
  }
 
  public IConstructor getConstructor() {
    return constructor;
  }
 
  public void setConstructor(IConstructor constructor) throws Exception {
    this.constructor = constructor;
    if(ServletFileUpload.isMultipartContent(constructor.getRequest()) && constructor.isInRole(roleUpload)) {
      FileItemFactory factory = new DiskFileItemFactory();
      ServletFileUpload upload = new ServletFileUpload(factory);
      List items = upload.parseRequest(constructor.getRequest());
      Iterator iter = items.iterator();
      while (iter.hasNext()) {
        FileItem item = (FileItem) iter.next();
        if(!item.isFormField()) {
          String onlyName = item.getName();
          if(onlyName != null && onlyName.length() > 0) {
            int p = onlyName.lastIndexOf('/');
            if(p == -1) p = onlyName.lastIndexOf('\\');
            String realName = getRealPath(convertToValidName(onlyName.substring(p + 1)));
            if(realName.matches(validUploads)) {
              File uploadedFile = new File(realName);
              item.write(uploadedFile);
            }
          }
        }
      }
    }
    else if(constructor.getParameter("removesel") != null && constructor.isInRole(roleDelete)) {
      String[] check = constructor.getRequest().getParameterValues("check");
      if(check != null) {
        for(int i = 0; i < check.length; i++) {
          deleteFile(check[i]);
        }
      }
    }
    else if(constructor.getParameter("createdir") != null && constructor.isInRole(roleCreateDir)) {
      createFolder(convertToValidName(constructor.getParameter("newdir")));
    }
  }
  
  public String getImageUrl(String file) {
    String folderPath = "repository(" +
            (subFolder == null || subFolder.length() == 0 ? "" : "/" + subFolder);
    return folderPath + "/" + file + ")";
  }
 
  private String getRealPath(String file) {
    String folderPath = rootPath + 
            (subFolder == null || subFolder.length() == 0 ? "" : "/" + subFolder);
    if(file == null) {
      return ElephantContext.getRealPath(folderPath);
    }
    else {
      return ElephantContext.getRealPath(folderPath + "/" + file);
    }
  }
 
  public boolean isSelectable() {
    return selectable;
  }
 
  public void setSelectable(boolean selectable) {
    this.selectable = selectable;
  }
 
  public String getSubFolder() {
    return subFolder;
  }
 
  public void setSubFolder(String subFolder) {
    this.subFolder = subFolder;
  }
  
}