/**
* Copyright (C) 2009 eXo Platform SAS.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.exoplatform.web.handler;
import org.exoplatform.container.ExoContainer;
import org.exoplatform.container.ExoContainerContext;
import org.exoplatform.upload.UploadResource;
import org.exoplatform.upload.UploadService;
import org.exoplatform.upload.UploadService.UploadLimit;
import org.exoplatform.web.ControllerContext;
import org.exoplatform.web.WebAppController;
import org.exoplatform.web.WebRequestHandler;
import org.gatein.common.text.EntityEncoder;
import java.io.Writer;
import java.net.URLEncoder;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
* Created by The eXo Platform SARL
* Author : Nhu Dinh Thuan
* nhudinhthuan@exoplatform.com
* Dec 9, 2006
*/
public class UploadHandler extends WebRequestHandler
{
static enum UploadServiceAction {
PROGRESS, UPLOAD, DELETE, ABORT
}
public String getHandlerName()
{
return "upload";
}
@Override
public boolean execute(ControllerContext context) throws Exception
{
execute(context.getController(), context.getRequest(), context.getResponse());
return true;
}
public void execute(WebAppController controller, HttpServletRequest req, HttpServletResponse res) throws Exception
{
String action = req.getParameter("action");
String[] uploadIds = req.getParameterValues("uploadId");
res.setHeader("Cache-Control", "no-cache");
ExoContainer container = ExoContainerContext.getCurrentContainer();
UploadService service = (UploadService)container.getComponentInstanceOfType(UploadService.class);
if (action == null || action.length() < 1)
return;
UploadServiceAction uploadActionService = UploadServiceAction.valueOf(action.toUpperCase());
if (uploadActionService == UploadServiceAction.PROGRESS)
{
Writer writer = res.getWriter();
if (uploadIds == null)
return;
StringBuilder value = new StringBuilder();
value.append("{\n upload : {");
for (int i = 0; i < uploadIds.length; i++)
{
UploadResource upResource = service.getUploadResource(uploadIds[i]);
if (upResource == null)
continue;
if (upResource.getStatus() == UploadResource.FAILED_STATUS)
{
UploadLimit limit = service.getUploadLimits().get(uploadIds[i]);
value.append("\n \"").append(uploadIds[i]).append("\": {");
value.append("\n \"status\":").append('\"').append("failed").append("\",");
value.append("\n \"size\":").append('\"').append(limit.getLimit()).append("\",");
value.append("\n \"unit\":").append('\"').append(limit.getUnit()).append("\"");
value.append("\n }");
continue;
}
double percent = 100;
if (upResource.getStatus() == UploadResource.UPLOADING_STATUS)
{
percent = (upResource.getUploadedSize() * 100) / upResource.getEstimatedSize();
}
value.append("\n \"").append(uploadIds[i]).append("\": {");
value.append("\n \"percent\":").append('\"').append((int)percent).append("\",");
String fileName = EntityEncoder.FULL.encode(upResource.getFileName());
value.append("\n \"fileName\":").append('\"').append(encodeName(fileName))
.append("\"");
value.append("\n }");
if (i < uploadIds.length - 1)
value.append(',');
}
value.append("\n }\n}");
writer.append(value);
}
else if (uploadActionService == UploadServiceAction.UPLOAD)
{
service.createUploadResource(req);
}
else if (uploadActionService == UploadServiceAction.DELETE)
{
service.removeUploadResource(uploadIds[0]);
}
else if (uploadActionService == UploadServiceAction.ABORT)
{
service.removeUploadResource(uploadIds[0]);
}
}
@Override
protected boolean getRequiresLifeCycle()
{
return false;
}
public String encodeName(String name) throws Exception
{
String[] arr = name.split(" ");
String str = "";
for (int i = 0; i < arr.length; i++)
{
str += " " + URLEncoder.encode(arr[i], "UTF-8");
}
return str;
}
}