package de.uhilger.filecms.api; import static de.uhilger.filecms.api.FileMgr.HOME_DIR_NAME; import static de.uhilger.filecms.api.FileMgr.HOME_DIR_PATH; import static de.uhilger.filecms.api.FileMgr.PUB_DIR_NAME; import static de.uhilger.filecms.api.FileMgr.PUB_DIR_PATH; import de.uhilger.filecms.web.Initialiser; import de.uhilger.filesystem.FileRef; import java.io.File; import java.io.IOException; import java.io.InputStream; import java.security.Principal; import java.util.Iterator; import java.util.List; import java.util.logging.Level; import java.util.logging.Logger; 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.disk.DiskFileItemFactory; import org.apache.commons.fileupload.servlet.FileCleanerCleanup; import org.apache.commons.fileupload.servlet.ServletFileUpload; import org.apache.commons.io.FileCleaningTracker; /** * Objekte der Klasee UploadServlet nehmen eine Datei per HTTP Multipart Request * entgegen und speichern sie im webapps-Verzeichnis des Servlet-Containers. * * Annahme ist, dass das von javax.servlet.context.tempdir bezeichnete * Verzeichnis auf CATALINA_BASE/work/Catalina/localhost/[context] zeigt. Von * dort wird dann das webapps-Verzeichnis als CATALINA_BASE/webapps ermittelt. */ public class UploadServlet extends HttpServlet { public static final String DESC = "file-cms uploader"; private static final Logger logger = Logger.getLogger(UploadServlet.class.getName()); /** * Processes requests for both HTTP GET and POST * 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 { //logger.log(Level.INFO, "start"); boolean isMultipart = ServletFileUpload.isMultipartContent(request); if (isMultipart) { //logger.log(Level.INFO, "isMultipart"); try { ServletContext servletContext = this.getServletConfig().getServletContext(); // Configure a repository (to ensure a secure temp location is used) File repository = (File) servletContext.getAttribute("javax.servlet.context.tempdir"); // CATALINA_BASE/work/Catalina/localhost/wbx // Create a factory for disk-based file items DiskFileItemFactory factory = newDiskFileItemFactory(servletContext, repository); // Create a new file upload handler ServletFileUpload upload = new ServletFileUpload(factory); // Parse the request List items = upload.parseRequest(request); // Process the uploaded items Iterator iter = items.iterator(); while (iter.hasNext()) { FileItem item = iter.next(); if (item.isFormField()) { processFormField(item); } else { //File base = repository.getParentFile().getParentFile().getParentFile().getParentFile(); //File webapps = new File(base, "webapps"); //logger.info(webapps.getAbsolutePath()); File targetDir = getTargetDir(request, PUB_DIR_NAME); logger.log(Level.FINE, "rufe processUploadedFile"); processUploadedFile(targetDir, item, true); } } } catch (Exception ex) { logger.log(Level.SEVERE, null, ex); } } else { // kein Upload } /* response.setContentType("text/html;charset=UTF-8"); try (PrintWriter out = response.getWriter()) { out.println(""); out.println(""); out.println(""); out.println("Servlet UploadServlet"); out.println(""); out.println(""); out.println("

Servlet UploadServlet at " + request.getContextPath() + "

"); out.println(""); out.println(""); } */ } private File getTargetDir(HttpServletRequest request, String relPath) { logger.finer(relPath); String targetPath = null; if(relPath.startsWith(PUB_DIR_NAME)) { targetPath = PUB_DIR_PATH + getUserName(request) + "/" + relPath.substring(PUB_DIR_NAME.length()); } else if(relPath.startsWith(HOME_DIR_NAME)) { targetPath = HOME_DIR_PATH + getUserName(request) + "/" + relPath.substring(HOME_DIR_NAME.length()); } else { // kann eigentlich nicht sein.. } logger.finer(targetPath); File targetDir = new File(getBase().getAbsolutePath(), targetPath); return targetDir; } private String getUserName(HttpServletRequest request) { String userName = null; Object p = request.getUserPrincipal(); if (p instanceof Principal) { userName = ((Principal) p).getName(); } return userName; } private FileRef getBase() { FileRef base = null; Object o = getServletContext().getAttribute(Initialiser.FILE_BASE); if (o instanceof FileRef) { base = (FileRef) o; } return base; } private void processUploadedFile(File targetDir, FileItem item, boolean writeToFile) throws Exception { //logger.log(Level.INFO, "start"); // Process a file upload if (!item.isFormField()) { String fieldName = item.getFieldName(); String fileName = item.getName(); String contentType = item.getContentType(); boolean isInMemory = item.isInMemory(); long sizeInBytes = item.getSize(); // Process a file upload if (writeToFile) { //logger.log(Level.INFO, "writeToFile " + fileName); File uploadedFile = new File(targetDir, fileName); item.write(uploadedFile); } else { InputStream uploadedStream = item.getInputStream(); //... uploadedStream.close(); } } } private void processFormField(FileItem item) { //logger.log(Level.INFO, "start"); // Process a regular form field if (item.isFormField()) { String name = item.getFieldName(); String value = item.getString(); //logger.log(Level.INFO, "field name: " + name + ", value: " + value); //... } } public static DiskFileItemFactory newDiskFileItemFactory(ServletContext context, File repository) { FileCleaningTracker fileCleaningTracker = FileCleanerCleanup.getFileCleaningTracker(context); DiskFileItemFactory factory = new DiskFileItemFactory(DiskFileItemFactory.DEFAULT_SIZE_THRESHOLD, repository); factory.setFileCleaningTracker(fileCleaningTracker); return factory; } // /** * Handles the HTTP GET 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 POST 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 DESC; }// }