FileUpload.java

/**
 * Copyright 2014 David L. Whitehurst
 * 
 * Licensed under the Apache License, Version 2.0 
 * (the "License"); You may not use this file except 
 * in compliance with the License. You may obtain a 
 * copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, 
 * software distributed under the License is distributed on an 
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
 * either express or implied. See the License for the specific 
 * language governing permissions and limitations under the 
 * License.
 * 
 */

package org.musicrecital.webapp.pages;

import org.apache.commons.fileupload.FileUploadException;
import org.apache.tapestry5.ComponentResources;
import org.apache.tapestry5.alerts.AlertManager;
import org.apache.tapestry5.alerts.Duration;
import org.apache.tapestry5.alerts.Severity;
import org.apache.tapestry5.annotations.DiscardAfter;
import org.apache.tapestry5.annotations.InjectPage;
import org.apache.tapestry5.annotations.Log;
import org.apache.tapestry5.annotations.Property;
import org.apache.tapestry5.ioc.Messages;
import org.apache.tapestry5.ioc.annotations.Inject;
import org.apache.tapestry5.services.Context;
import org.musicrecital.Constants;
import org.musicrecital.webapp.data.FileData;
import org.musicrecital.webapp.services.SecurityContext;
import org.slf4j.Logger;

import javax.servlet.http.HttpServletRequest;
import java.io.File;

/**
 * This class handles the uploading of a file and writing it to
 * the filesystem.  Eventually, it will also addChild support for persisting the
 * files information into the database.
 *
 * @author <a href="mailto:matt@raibledesigns.com">Matt Raible</a>
 * @author Serge Eby
 * @version $Id: f9040dcba72c2416bc54c4b9114bc505029d8d2f $
 */
public class FileUpload {

    @Inject
    private Logger logger;

    @Inject
    private Messages messages;

    @Inject
    private HttpServletRequest request;

    @Inject
    private AlertManager alertManager;

    @Inject
    private Context context;

    @Inject
    private SecurityContext securityContext;

    @Inject
    private ComponentResources resources;

    @InjectPage
    private FileDisplay fileDisplay;

    @Property
    private FileData fileData;


    @Log
    void onPrepare() {
        fileData = new FileData();
    }

    @Log
    void onValidate() {
        if (fileData.getFile() == null) {
            return;
        }
    }

    @Log
    Object onCancel() {
        return Home.class;
    }

    @DiscardAfter
    Object onSuccess() {

        File uploadPath = buildUploadPath();

        File copied = new File(uploadPath, fileData.getFileName());
        fileData.write(copied);

        // Populate fileData object
        String path = uploadPath.getAbsolutePath() + Constants.FILE_SEP + fileData.getFileName();
        String url =
                request.getContextPath() + "/resources/" +
                        securityContext.getUsername() + "/" + fileData.getFileName();


        fileData.setPath(path);
        fileData.setUrl(url);

        fileDisplay.setFileData(fileData);

        return fileDisplay;
    }


    Object onUploadException(FileUploadException exception) {
        logger.error(String.format("Upload exception: %s ", exception.getMessage()));
        alertManager.alert(Duration.TRANSIENT,
                           Severity.ERROR,
                           messages.get("maxLengthExceeded"));
        return this;
    }


    private File buildUploadPath() {

        // the directory to upload to
        String uploadDir = request.getServletContext().getRealPath("/resources");

        // The following seems to happen when running jetty:run
        if (uploadDir == null) {
            uploadDir = new File("src/main/webapp/resources").getAbsolutePath();
        }
        uploadDir += Constants.FILE_SEP + securityContext.getUsername() + Constants.FILE_SEP;

        // Create the directory if it doesn't exist
        File dirPath = new File(uploadDir);

        if (!dirPath.exists()) {
            dirPath.mkdirs();
        }

        return dirPath;
    }

}