com.gtwm.pb.model.manageData.fields.FileValueDefn.java Source code

Introduction

Here is the source code for com.gtwm.pb.model.manageData.fields.FileValueDefn.java

Source

/*
 *  Copyright 2012 GT webMarque Ltd
 * 
 *  This file is part of agileBase.
 *
 *  agileBase is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  agileBase 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with agileBase.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.gtwm.pb.model.manageData.fields;

import java.util.List;
import javax.servlet.http.HttpServletRequest;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUpload;
import org.apache.commons.fileupload.servlet.ServletRequestContext;
import org.apache.commons.fileupload.FileUploadException;
import org.grlea.log.SimpleLogger;
import com.gtwm.pb.model.interfaces.fields.FileValue;
import com.gtwm.pb.model.interfaces.fields.FileField;
import com.gtwm.pb.util.CantDoThatException;
import com.gtwm.pb.util.Helpers;
import com.gtwm.pb.util.ObjectNotFoundException;

public class FileValueDefn implements FileValue {

    private FileValueDefn() {
    }

    /**
     * Construct a file value object from a known filename
     */
    public FileValueDefn(String filename) {
        this.filename = filename;
    }

    /**
     * Construct a file value reading the uploaded file name for a particlular
     * file field from the HTTP request
     */
    public FileValueDefn(HttpServletRequest request, FileField fileField, List<FileItem> multipartItems)
            throws CantDoThatException, FileUploadException, ObjectNotFoundException {
        if (!FileUpload.isMultipartContent(new ServletRequestContext(request))) {
            throw new CantDoThatException("To upload a file, the form must be posted as multi-part form data");
        }
        String internalFieldName = fileField.getInternalFieldName();
        ITEMLOOP: for (FileItem item : multipartItems) {
            // if item is a file
            if (!item.isFormField()) {
                if (item.getFieldName().equals(internalFieldName)) {
                    this.filename = item.getName().replaceAll("^.*\\\\", "");
                    break ITEMLOOP;
                }
            }
        }
        if (this.filename == null) {
            throw new ObjectNotFoundException("The file field " + fileField + "wasn't found in the user input");
        }
    }

    public String getFilename() {
        if (this.isNull()) {
            return "";
        } else {
            return this.filename;
        }
    }

    public String getExtension() {
        if (this.isNull()) {
            return "";
        } else if (this.filename.contains(".")) {
            return this.filename.replaceAll(".*\\.", "").toLowerCase();
        } else {
            return "";
        }
    }

    public boolean isNull() {
        return (this.filename == null);
    }

    public String getIconName() {
        if (this.isNull()) {
            return "";
        }
        String filename = this.getFilename();
        if (filename.contains(".")) {
            String extensionStr = filename.replaceAll(".*\\.", "").toUpperCase();
            try {
                Extension extension = Extension.valueOf(extensionStr);
                return extensionStr.toLowerCase();
            } catch (IllegalArgumentException iaex) {
                logger.warn("No icon for filetype " + extensionStr);
                return "unknown";
            }
        }
        return "";
    }

    public boolean isImage() {
        return Helpers.isImage(this.filename);
    }

    public String toString() {
        if (this.isNull()) {
            return "";
        } else {
            return this.filename;
        }
    }

    private enum Extension {
        CSV, DOC, DOCX, HTM, PDF, PPT, PSD, TIF, TIFF, TXT, XLS, XLSX, ZIP;
    }

    private String filename = null;

    private static final SimpleLogger logger = new SimpleLogger(FileValueDefn.class);
}