/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* 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.gatein.management.gadget.mop.exportimport.server;
import static org.gatein.management.gadget.mop.exportimport.server.ContainerRequestHandler.doInRequest;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.fileupload.FileItem;
import org.exoplatform.container.ExoContainer;
import org.gatein.common.logging.Logger;
import org.gatein.common.logging.LoggerFactory;
import org.gatein.management.api.ContentType;
import org.gatein.management.api.PathAddress;
import org.gatein.management.api.controller.ManagedRequest;
import org.gatein.management.api.controller.ManagedResponse;
import org.gatein.management.api.controller.ManagementController;
import org.gatein.management.api.operation.OperationNames;
import gwtupload.server.UploadAction;
import gwtupload.server.exceptions.UploadActionException;
/**
* {@code FileUploadServlet}
* <p>
* The file upload servlet based on GWT upload, used for import sites.
* </p>
* Created on Jan 3, 2011, 3:43:36 PM
*
* @author <a href="mailto:nbenothm@redhat.com">Nabil Benothman</a>
* @version 1.0
*/
public class FileUploadServlet extends UploadAction {
private static final Logger log = LoggerFactory.getLogger(FileUploadServlet.class);
private static final long serialVersionUID = 1L;
private Hashtable<String, String> receivedContentTypes = new Hashtable<String, String>();
/**
* Maintain a list with received files and their content types.
*/
private Hashtable<String, File> receivedFiles = new Hashtable<String, File>();
/**
* Override executeAction to save the received files in a custom place and delete this items from session.
*/
@Override
public String executeAction(HttpServletRequest request, List<FileItem> sessionFiles) throws UploadActionException {
StringBuilder response = new StringBuilder("<response>\n");
int count = 0;
for (FileItem item : sessionFiles) {
// if (false == item.isFormField()) {
if (!item.isFormField()) {
count++;
try {
// Create a new file based on the remote file name in the client
String saveName = item.getName().replaceAll("[\\\\/><\\|\\s\"'{}()\\[\\]]+", "_");
// Create a temporary file placed in the default system temp folder
File file = File.createTempFile(saveName, ".zip");
item.write(file);
// Save a list with the received files
receivedFiles.put(item.getFieldName(), file);
receivedContentTypes.put(item.getFieldName(), item.getContentType());
String importMode = request.getParameter("importMode");
// process the uploaded file
processImport(request.getParameter("pc"), new FileInputStream(file), importMode);
// Compose a xml message with the full file information which can be parsed in client side
response.append("<file-").append(count).append("-field>").append(item.getFieldName()).append("</file-")
.append(count).append("-field>\n");
response.append("<file-").append(count).append("-name>").append(item.getName()).append("</file-")
.append(count).append("-name>\n");
response.append("<file-").append(count).append("-size>").append(item.getSize()).append("</file-")
.append(count).append("-size>\n");
response.append("<file-").append(count).append("-type>").append(item.getContentType()).append("</file-")
.append(count).append("type>\n");
} catch (Exception e) {
throw new UploadActionException(e);
}
}
}
// Remove files from session because we have a copy of them
removeSessionFileItems(request);
// Send information of the received files to the client.
return response.append("</response>\n").toString();
}
/**
* Get the content of an uploaded file.
*/
@Override
public void getUploadedFile(HttpServletRequest request, HttpServletResponse response) throws IOException {
String fieldName = request.getParameter(PARAM_SHOW);
File f = receivedFiles.get(fieldName);
if (f != null) {
response.setContentType(receivedContentTypes.get(fieldName));
FileInputStream is = new FileInputStream(f);
copyFromInputStreamToOutputStream(is, response.getOutputStream());
} else {
renderXmlResponse(request, response, ERROR_ITEM_NOT_FOUND);
}
}
/**
* Remove a file when the user sends a delete request.
*/
@Override
public void removeItem(HttpServletRequest request, String fieldName) throws UploadActionException {
File file = receivedFiles.get(fieldName);
receivedFiles.remove(fieldName);
receivedContentTypes.remove(fieldName);
if (file != null) {
boolean deleted = file.delete();
if (!deleted)
log.warn("Could not delete file " + file);
}
}
private void processImport(final String containerName, final InputStream in, final String importMode) throws Exception {
doInRequest(containerName, new ContainerCallback<Void>() {
@Override
public Void doInContainer(ExoContainer container) throws Exception {
ManagementController controller = getComponent(container, ManagementController.class);
Map<String, List<String>> attributes = new HashMap<String, List<String>>(1);
attributes.put("importMode", Collections.singletonList(importMode));
ManagedRequest request = ManagedRequest.Factory.create(OperationNames.IMPORT_RESOURCE,
PathAddress.pathAddress("mop"), attributes, in, ContentType.ZIP);
ManagedResponse response = controller.execute(request);
if (!response.getOutcome().isSuccess()) {
throw new Exception(response.getOutcome().getFailureDescription());
}
return null;
}
});
}
}