UploadDocument.java

Go to the documentation of this file.

/*
 * Project: Server for annotations sharing
 * Author: Ing. Jaroslav Dytrych idytrych@fit.vutbr.cz
 * File: UploadDocument.java
 * Description: Backbean for page for uploading documents
 */

/**
 * @file UploadDocument.java
 *
 * @brief Backbean for page for uploading documents
 */

package cz.vutbr.fit.knot.annotations.web;

import cz.vutbr.fit.knot.annotations.app.AppBean;
import cz.vutbr.fit.knot.annotations.app.Constants;
import cz.vutbr.fit.knot.annotations.entity.StoredDocument;
import cz.vutbr.fit.knot.annotations.entity.User;
import cz.vutbr.fit.knot.annotations.entity.UserGroup;
import java.io.IOException;
import java.io.Serializable;
import java.io.StringWriter;
import java.util.List;
import javax.faces.bean.ManagedBean;
import javax.faces.bean.ViewScoped;
import javax.persistence.EntityManager;
import javax.persistence.EntityTransaction;
import org.apache.commons.io.IOUtils;
import org.apache.myfaces.custom.fileupload.UploadedFile;

/**
 * Backbean for page for uploading documents
 *
 * @brief Backbean for for page for uploading documents
 * @author idytrych
 */
@ManagedBean
@ViewScoped
public class UploadDocument implements Serializable {
  /**
   * Error message displayed in form
   */
  private String errorMessage = "";
  /**
   * Success message displayed in form
   */
  private String successMessage = "";
  /**
   * Uploaded file
   */
  private UploadedFile uploadedFile;
  /**
   * File name
   */
  private String fileName;
  /**
   * User group in which uploaded types will be shared
   */
  private String uGroup;

  /**
   * Action listener for upload button
   *
   * @return Returns page outcome (identificator of next page or null to stay here)
   */
  public String btnUploadAction() {

    errorMessage = "";
    successMessage = "";

    if (uGroup.isEmpty()) {
      errorMessage = MessageProvider.getMessage("groupMustBeSet");
      return null;
    }
    // query database for user group
    Object[] params = new Object[2];  // parameters for database query
    params[0] = "name";
    params[1] = uGroup;
    List gList = AppBean.getPersistenceManager().queryDB("UserGroup.findByName", params);
    UserGroup userGroup = null;
    if (gList != null && !gList.isEmpty()) {  // if group was found
      userGroup = (UserGroup) gList.get(0);
    } else {
      errorMessage = MessageProvider.getMessage("unknownGroup");
      return null;
    }

    // Informations about uploaded file
    String fileType = uploadedFile.getContentType();
    String fileNameStr = uploadedFile.getName();
    // Long fileSize = uploadedFile.getSize();  // bytes

    // Create writer with file data
    StringWriter fWriter = new StringWriter();
    try {
      IOUtils.copy(uploadedFile.getInputStream(), fWriter);
    } catch (IOException ex) {
      errorMessage = MessageProvider.getMessage("uploadingError");
      return null;
    }

    // check document type
    if (!Constants.DOC_UPLOAD_TYPES.contains(fileType)) {
      errorMessage = MessageProvider.getMessage("unsupportedDocType") + ": " + fileType;
      return null;
    }
    
    // gets logged in user
    User user = SessionManager.getSession().getLoggedUser();

    // Create document entity
    StoredDocument newDoc = new StoredDocument(fileNameStr, fileType, user, userGroup, fWriter.toString());

    // persist document to database
    EntityManager em = AppBean.getPersistenceManager().getEM();
    EntityTransaction transaction = em.getTransaction();
    try {
      transaction.begin();
      newDoc.setUri(newDoc.getGeneratedUri());  // bad URI - must be updated later (but must be filled to commit)
      newDoc = em.merge(newDoc);  // persist document
      em.flush();
      transaction.commit();
    } catch (Exception e) {
      transaction.rollback();
      errorMessage = errorMessage + MessageProvider.getMessage("documentPersistingError") + "<br/>";
      return null;
    } finally {
      em.close();
    }

    // Generate and store URI
    newDoc.setUri(newDoc.getGeneratedUri());
    AppBean.getPersistenceManager().saveEntityChanges(newDoc);

    // display message about success
    successMessage = MessageProvider.getMessage("uploadSuccessfull");

    return "documents";
  }  // btnUploadAction()

  /**
   * Action listener for back button
   *
   * @return Returns page outcome (identificator of next page)
   */
  public String btnCancelAction() {
    return "documents";
  }

  /**
   * Gets uploaded file
   *
   * @return Returns uploaded file
   */
  public UploadedFile getUploadedFile() {
    return uploadedFile;
  }

  /**
   * Sets uploaded file
   *
   * @param uploadedFile Uploaded file
   */
  public void setUploadedFile(UploadedFile uploadedFile) {
    this.uploadedFile = uploadedFile;
  }

  /**
   * Gets uploaded file name
   *
   * @return Returns uploaded file name
   */
  public String getFileName() {
    return fileName;
  }

  /**
   * Gets error message displayed in form
   *
   * @return Returns error message displayed in form
   */
  public String getErrorMessage() {
    return errorMessage;
  }

  /**
   * Sets error message to be displayed in form
   *
   * @param errorMessage Error message to be displayed in form
   */
  public void setErrorMessage(String errorMessage) {
    this.errorMessage = errorMessage;
  }

  /**
   * Gets success message displayed in form
   *
   * @return Returns success message displayed in form
   */
  public String getSuccessMessage() {
    return successMessage;
  }

  /**
   * Sets success message to be displayed in form
   *
   * @param successMessage Success message to be displayed in form
   */
  public void setSuccessMessage(String successMessage) {
    this.successMessage = successMessage;
  }

  /**
   * Gets user group in which uploaded types will be shared
   *
   * @return Returns user group in which uploaded types will be shared
   */
  public String getuGroup() {
    return uGroup;
  }

  /**
   * Sets user group in which uploaded types will be shared
   *
   * @param uGroup User group in which uploaded types will be shared
   */
  public void setuGroup(String uGroup) {
    this.uGroup = uGroup;
  }

}  // public class UploadDocument