Storage.java

/*****************************************************************************************
Infosistema - OpenBaas
Copyright(C) 2002-2014 Infosistema, S.A.
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/>.
www.infosistema.com
info@openbaas.com
Av. José Gomes Ferreira, 11 3rd floor, s.34
Miraflores
1495-139 Algés Portugal
****************************************************************************************/
package infosistema.openbaas.data.models;

import infosistema.openbaas.utils.Log;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import java.util.Date;
import java.util.List;

import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.StreamingOutput;

import org.apache.commons.fileupload.FileItem;
import org.codehaus.jettison.json.JSONObject;

/**
 * Media Model object
 * 
 * Abstract class that implements several actions that audio/image and video require.
 * 
 * 
 * @author Miguel Aniceto
 * @version 0.0 
 */
public class Storage extends Media {

	private String _id;
	private String creationDate;
	private long size;
	private String dir;
	private String fileName;
	private String location;
	/**
	 * Constructor
	 * 
	 */
	public Storage(String _id, String dir, long size, String location){
		this._id = _id;
		this.dir = dir;
		this.size = size;
		creationDate = new Date().toString();
		this.location = location;
	}

	public Storage(String _id) {
		this._id = _id;
	}
	public Storage(){
		
	}
	/**
	 * Retrieves Audio/Video File in MB.
	 * @return audio File Size
	 */
	public long getSize() {
		return size;
	}
	
	/**
	 * Upload the File to the server.
	 * @return FileID
	 */
	public String uploadFile(){
		return null;
		//TO DO
	}
	/**
	 * Returns the imager identifier.
	 * @return String Image ID.
	 */
	public String getID() {
		return this._id;
	}
	public String getDir(){
		return this.dir;
	}
	/**
	 * Returns the time of creation for the image.
	 * @return Date Time
	 */
	public String getTimeOfCreation() {
		return creationDate;
	}
	
	public Response download(String appId, String id, String dir){
		String filePath = dir;
		if (filePath != null && !"".equals(filePath)) {
			File file = new File(filePath);
			StreamingOutput stream = null;
			try {
				final InputStream in = new FileInputStream(file);
				stream = new StreamingOutput() {
					public void write(OutputStream out) throws IOException,
							WebApplicationException {
						try {
							int read = 0;
							byte[] bytes = new byte[1024];

							while ((read = in.read(bytes)) != -1) {
								out.write(bytes, 0, read);
							}
						} catch (Exception e) {
							Log.error("", this, "download", "An error ocorred.", e); 
							throw new WebApplicationException(e);
						}
					}
				};
				in.close();
			} catch (Exception e) {
				Log.error("", this, "download", "An error ocorred.", e); 
				return Response.status(404).entity(appId).build();
			}
			return Response
					.ok(stream)
					.header("content-disposition",
							"attachment; filename = " + file.getName()).build();
		}
		return Response.ok("file path null").build();
	}
	public String validateType(JSONObject json){
		return null;
	}
	public String upload(List<FileItem> items){
		String resultStatus = "";
        for (FileItem item : items) {
            if (!item.isFormField()) {
                try {
                    item.write(new File(dir + ".mp3"));
                    resultStatus = "fileupload success";
                } catch (Exception e) {
    				Log.error("", this, "upload", "An error ocorred.", e); 
                }
            }
        }
        return resultStatus;
	}
	public void setDir(String dir){
		this.dir = dir;
	}
	public void setSize(long size){
		this.size = size;
	}
	public void setCreationDate(String creationDate){
		this.creationDate = creationDate;
	}
	public void setFileName(String fileName){
		this.fileName = fileName;
	}
	public String getFileName(){
		return this.fileName;
	}
	public String getLocation(){
		return this.location;
	}
	public void setLocation(String location){
		this.location = location;
	}
	public void setId(String id){
		this._id = id;
	}
}