com.eufar.asmm.server.UploadFunction.java Source code

Introduction

Here is the source code for com.eufar.asmm.server.UploadFunction.java

Source

/***LICENSE START
 * Copyright 2011 EUROPEAN UNION
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved by
 * the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * 
 * http://ec.europa.eu/idabc/eupl
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 * 
 * Date: 03 January 2011
 * Authors: Marzia Grasso, Angelo Quaglia, Massimo Craglia
LICENSE END***/

package com.eufar.asmm.server;

import java.io.IOException;
import java.util.List;
import java.util.Iterator;

import javax.servlet.Servlet;
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.FileItemFactory;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.io.FilenameUtils;

@SuppressWarnings({ "hiding", "rawtypes" })
public class UploadFunction<FileItem> extends HttpServlet implements Servlet {
    private static final long serialVersionUID = 1L;

    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        System.out.println("UploadFunction - the function started");
        response.setContentType("text/html;charset=UTF-8");
        response.addHeader("Cache-Control", "no-cache,no-store");
        FileItemFactory factory = new DiskFileItemFactory();
        ServletFileUpload upload = new ServletFileUpload(factory);
        try {
            List items = upload.parseRequest(request);
            Iterator iter = items.iterator();
            while (iter.hasNext()) {
                Object obj = iter.next();
                org.apache.commons.fileupload.FileItem item = (org.apache.commons.fileupload.FileItem) obj;
                if (FilenameUtils.getExtension(item.getName()).matches("(xml|XML)")) {
                    if (item.isFormField()) {
                        String name = item.getName();
                        String value = "";
                        if (name.compareTo("textBoxFormElement") == 0) {
                            value = item.getString();
                        } else {
                            value = item.getString();
                        }
                        response.getWriter().write(name + "=" + value + "\n");
                    } else {
                        byte[] fileContents = item.get();
                        String message = new String(fileContents);
                        response.setCharacterEncoding("UTF-8");
                        response.setContentType("text/html");
                        response.getWriter().write(message);
                        System.out.println("UploadFunction - file uploaded");
                    }
                } else {
                    System.out.println("UploadFunction - file rejected: wrong format");
                    response.setCharacterEncoding("UTF-8");
                    response.setContentType("text/html");
                    response.getWriter().write("format");
                }
            }
        } catch (Exception ex) {
            System.out.println("UploadFunction - a problem occured: " + ex);
            response.getWriter().write("ERROR:" + ex.getMessage());
        }
    }
}