/*
* ModeShape (http://www.modeshape.org)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.modeshape.web;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import javax.jcr.Binary;
import javax.jcr.Node;
import javax.jcr.Session;
import javax.jcr.ValueFactory;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.modeshape.common.logging.Logger;
import org.modeshape.web.server.Connector;
/**
*
* @author kulikov
*/
public class BinaryContentUploadServlet extends HttpServlet {
private final static String CONTENT_PARAMETER = "Upload content";
private final static String REPOSITORY_NAME_PARAMETER = "repository";
private final static String WORKSPACE_NAME_PARAMETER = "workspace";
private final static String NODE_PATH_PARAMETER = "path";
private final static String PROPERTY_NAME_PARAMETER = "pname";
private final static String REPOSITORY_CONNECTOR = "connector";
private final static String MIME_TYPE = "jcr:mimeType";
private final static String DESTINATION_URL = "/tree/%s/ws-%s%s";
private ServletFileUpload upload;
private final Logger logger = Logger.getLogger(BinaryContentServlet.class);
/**
* Processes requests for both HTTP
* <code>GET</code> and
* <code>POST</code> methods.
*
* @param request servlet request
* @param response servlet response
* @throws ServletException if a servlet-specific error occurs
* @throws IOException if an I/O error occurs
*/
protected void processRequest(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
List<FileItem> items;
// Parse the request
try {
items = upload.parseRequest(request);
} catch (FileUploadException e) {
throw new ServletException(e);
}
String repository = getParameter(items, REPOSITORY_NAME_PARAMETER);
String workspace = getParameter(items, WORKSPACE_NAME_PARAMETER);
String path = getParameter(items, NODE_PATH_PARAMETER);
String pname = getParameter(items, PROPERTY_NAME_PARAMETER);
if (logger.isDebugEnabled()) {
logger.debug(String.format("Upload %s at %s/%s%s",
pname, repository, workspace, path));
}
String mimeType = getContentType(items);
Connector connector = (Connector)
request.getSession().getAttribute(REPOSITORY_CONNECTOR);
try {
Session session = connector.find(repository).session(workspace);
ValueFactory valueFactory = session.getValueFactory();
Binary bin = valueFactory.createBinary(getStream(items));
Node node = session.getNode(path);
node.setProperty(pname, bin);
node.setProperty(MIME_TYPE, mimeType);
bin.dispose();
} catch (Exception e) {
throw new ServletException(e);
}
String uri = request.getContextPath() +
String.format(DESTINATION_URL, repository, workspace, path);
response.sendRedirect(uri);
}
/**
* Extracts value of the parameter with given name.
*
* @param items
* @param name
* @return parameter value or null.
*/
private String getParameter(List<FileItem> items, String name) {
for (FileItem i : items) {
if (i.isFormField() && i.getFieldName().equals(name)) {
return i.getString();
}
}
return null;
}
/**
* Gets uploaded file as stream.
*
* @param items
* @return
* @throws IOException
*/
private InputStream getStream(List<FileItem> items) throws IOException {
for (FileItem i : items) {
if (!i.isFormField() && i.getFieldName().equals(CONTENT_PARAMETER)) {
return i.getInputStream();
}
}
return null;
}
/**
* Determines content-type of the uploaded file.
*
* @param items
* @return
*/
private String getContentType(List<FileItem> items) {
for (FileItem i : items) {
if (!i.isFormField() && i.getFieldName().equals(CONTENT_PARAMETER)) {
return i.getContentType();
}
}
return null;
}
@Override
public void init() {
DiskFileItemFactory diskFactory = new DiskFileItemFactory();
// Configure a repository (to ensure a secure temp location is used)
ServletContext servletContext = this.getServletConfig().getServletContext();
File tempDir = (File) servletContext.getAttribute("javax.servlet.context.tempdir");
diskFactory.setRepository(tempDir);
// Create a new file upload handler
upload = new ServletFileUpload(diskFactory);
}
// <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
/**
* Handles the HTTP
* <code>GET</code> method.
*
* @param request servlet request
* @param response servlet response
* @throws ServletException if a servlet-specific error occurs
* @throws IOException if an I/O error occurs
*/
@Override
protected void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
processRequest(request, response);
}
/**
* Handles the HTTP
* <code>POST</code> method.
*
* @param request servlet request
* @param response servlet response
* @throws ServletException if a servlet-specific error occurs
* @throws IOException if an I/O error occurs
*/
@Override
protected void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
processRequest(request, response);
}
/**
* Returns a short description of the servlet.
*
* @return a String containing servlet description
*/
@Override
public String getServletInfo() {
return "Short description";
}// </editor-fold>
}