/*
* Licensed under the GPL License. You may not use this file except in
* compliance with the License. You may obtain a copy of the License at
*
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*
* THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED
* WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
* MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
*/
package com.googlecode.psiprobe.controllers.deploy;
import com.googlecode.psiprobe.controllers.TomcatContainerController;
import com.googlecode.psiprobe.controllers.jsp.DisplayJspController;
import com.googlecode.psiprobe.model.jsp.Summary;
import java.io.File;
import java.net.URL;
import java.util.Iterator;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.catalina.Context;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileUpload;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.servlet.ServletRequestContext;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.view.InternalResourceView;
/**
* Uploads and installs web application from a .WAR.
*
* @author Vlad Ilyushchenko
*/
public class UploadWarController extends TomcatContainerController {
protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response) throws Exception {
if (FileUpload.isMultipartContent(new ServletRequestContext(request))) {
File tmpWar = null;
String contextName = null;
boolean update = false;
boolean compile = false;
boolean discard = false;
//
// parse multipart request and extract the file
//
FileItemFactory factory = new DiskFileItemFactory(1048000, new File(System.getProperty("java.io.tmpdir")));
ServletFileUpload upload = new ServletFileUpload(factory);
upload.setSizeMax(-1);
upload.setHeaderEncoding("UTF8");
try {
for (Iterator it = upload.parseRequest(request).iterator(); it.hasNext();) {
FileItem fi = (FileItem) it.next();
if (!fi.isFormField()) {
if (fi.getName() != null && fi.getName().length() > 0) {
tmpWar = new File(System.getProperty("java.io.tmpdir"), FilenameUtils.getName(fi.getName()));
fi.write(tmpWar);
}
} else if ("context".equals(fi.getFieldName())) {
contextName = fi.getString();
} else if ("update".equals(fi.getFieldName()) && "yes".equals(fi.getString())) {
update = true;
} else if ("compile".equals(fi.getFieldName()) && "yes".equals(fi.getString())) {
compile = true;
} else if ("discard".equals(fi.getFieldName()) && "yes".equals(fi.getString())) {
discard = true;
}
}
} catch (Exception e) {
logger.fatal("Could not process file upload", e);
request.setAttribute("errorMessage", getMessageSourceAccessor().getMessage("probe.src.deploy.war.uploadfailure",
new Object[]{e.getMessage()}));
if (tmpWar != null && tmpWar.exists()) {
tmpWar.delete();
}
tmpWar = null;
}
String errMsg = null;
if (tmpWar != null) {
try {
if (tmpWar.getName().endsWith(".war")) {
if (contextName == null || contextName.length() == 0) {
String warFileName = tmpWar.getName().replaceAll("\\.war$", "");
contextName = "/" + warFileName;
}
contextName = getContainerWrapper().getTomcatContainer().formatContextName(contextName);
//
// pass the name of the newly deployed context to the presentation layer
// using this name the presentation layer can render a url to view compilation details
//
String visibleContextName = "".equals(contextName) ? "/" : contextName;
request.setAttribute("contextName", visibleContextName);
if (update && getContainerWrapper().getTomcatContainer().findContext(contextName) != null) {
logger.debug("updating "+contextName + ": removing the old copy");
getContainerWrapper().getTomcatContainer().remove(contextName);
}
if (getContainerWrapper().getTomcatContainer().findContext(contextName) == null) {
//
// move the .war to tomcat application base dir
//
String destWarFilename = getContainerWrapper().getTomcatContainer().formatContextFilename(contextName);
File destWar = new File(getContainerWrapper().getTomcatContainer().getAppBase(), destWarFilename + ".war");
FileUtils.moveFile(tmpWar, destWar);
//
// let Tomcat know that the file is there
//
getContainerWrapper().getTomcatContainer().installWar(contextName, new URL("jar:file:" + destWar.getAbsolutePath() + "!/"));
Context ctx = getContainerWrapper().getTomcatContainer().findContext(contextName);
if (ctx == null) {
errMsg = getMessageSourceAccessor().getMessage("probe.src.deploy.war.notinstalled", new Object[]{visibleContextName});
} else {
request.setAttribute("success", Boolean.TRUE);
if (discard) {
getContainerWrapper().getTomcatContainer().discardWorkDir(ctx);
}
if (compile) {
Summary summary = new Summary();
summary.setName(ctx.getName());
getContainerWrapper().getTomcatContainer().listContextJsps(ctx, summary, true);
request.getSession(true).setAttribute(DisplayJspController.SUMMARY_ATTRIBUTE, summary);
request.setAttribute("compileSuccess", Boolean.TRUE);
}
}
} else {
errMsg = getMessageSourceAccessor().getMessage("probe.src.deploy.war.alreadyExists", new Object[]{visibleContextName});
}
} else {
errMsg = getMessageSourceAccessor().getMessage("probe.src.deploy.war.notWar.failure");
}
} catch (Exception e) {
errMsg = getMessageSourceAccessor().getMessage("probe.src.deploy.war.failure", new Object[]{e.getMessage()});
logger.error("Tomcat throw an exception when trying to deploy", e);
} finally {
if (errMsg != null) {
request.setAttribute("errorMessage", errMsg);
}
tmpWar.delete();
}
}
}
return new ModelAndView(new InternalResourceView(getViewName()));
}
}