org.unitime.timetable.gwt.server.UploadServlet.java Source code

Introduction

Here is the source code for org.unitime.timetable.gwt.server.UploadServlet.java

Source

/*
 * Licensed to The Apereo Foundation under one or more contributor license
 * agreements. See the NOTICE file distributed with this work for
 * additional information regarding copyright ownership.
 *
 * The Apereo Foundation licenses this file to you 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.unitime.timetable.gwt.server;

import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.util.List;
import java.util.TreeSet;

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.unitime.timetable.ApplicationProperties;
import org.unitime.timetable.defaults.ApplicationProperty;
import org.unitime.timetable.gwt.server.CalendarServlet.HttpParams;
import org.unitime.timetable.gwt.server.CalendarServlet.Params;
import org.unitime.timetable.gwt.server.CalendarServlet.QParams;
import org.unitime.timetable.model.Event;
import org.unitime.timetable.model.EventNote;
import org.unitime.timetable.model.dao.EventDAO;
import org.unitime.timetable.security.SessionContext;
import org.unitime.timetable.security.context.HttpSessionContext;
import org.unitime.timetable.security.rights.Right;

/**
 * @author Tomas Muller
 */
public class UploadServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;
    private static final int DEFAULT_MAX_SIZE = 4096 * 1024;

    public static final String SESSION_LAST_FILE = "LAST_FILE";

    protected SessionContext getSessionContext() {
        return HttpSessionContext.getSessionContext(getServletContext());
    }

    public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        Params params = null;
        String q = request.getParameter("q");
        if (q != null) {
            params = new QParams(q);
        } else {
            params = new HttpParams(request);
        }
        if (params.getParameter("event") != null) {
            Long eventId = Long.parseLong(params.getParameter("event"));
            String fileName = params.getParameter("name");
            Long noteId = (params.getParameter("note") == null ? null : Long.valueOf(params.getParameter("note")));
            if (q == null)
                getSessionContext().checkPermissionAnyAuthority(Long.valueOf(eventId), "Event", Right.EventDetail);
            Event event = EventDAO.getInstance().get(eventId);
            TreeSet<EventNote> notes = new TreeSet<EventNote>();
            if (event != null)
                for (EventNote note : event.getNotes()) {
                    if (note.getAttachedName() == null || note.getAttachedName().isEmpty())
                        continue;
                    if (fileName != null) {
                        if (fileName.equals(note.getAttachedName())
                                && (noteId == null || noteId.equals(note.getUniqueId())))
                            notes.add(note);
                    } else if (noteId != null) {
                        if (noteId.equals(note.getUniqueId()))
                            notes.add(note);
                    } else {
                        notes.add(note);
                    }
                }
            if (!notes.isEmpty()) {
                EventNote note = notes.last();

                response.setContentType(note.getAttachedContentType());
                response.setHeader("Content-Disposition",
                        "attachment; filename=\"" + note.getAttachedName() + "\"");
                OutputStream out = response.getOutputStream();
                out.write(note.getAttachedFile());
                out.flush();
                out.close();

                return;
            }
        }

        throw new ServletException("Nothing to download.");
    }

    public void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        try {
            String maxSizeProperty = ApplicationProperty.MaxUploadSize.value();
            int maxSize = (maxSizeProperty == null ? DEFAULT_MAX_SIZE : Integer.parseInt(maxSizeProperty));

            ServletFileUpload upload = new ServletFileUpload(
                    new DiskFileItemFactory(maxSize, ApplicationProperties.getTempFolder()));
            upload.setSizeMax(maxSize);

            List<FileItem> files = (List<FileItem>) upload.parseRequest(request);

            String message = null;
            if (files.size() == 1) {
                FileItem file = files.get(0);
                if (file.getSize() <= 0) {
                    request.getSession().removeAttribute(SESSION_LAST_FILE);
                    message = "No file is selected.";
                } else {
                    request.getSession().setAttribute(SESSION_LAST_FILE, file);
                    message = "File " + file.getName() + " (" + file.getSize() + " bytes) selected.";
                }
            } else {
                request.getSession().removeAttribute(SESSION_LAST_FILE);
                message = "No file is selected.";
            }

            response.setContentType("text/html; charset=UTF-8");
            response.setCharacterEncoding("UTF-8");
            PrintWriter out = response.getWriter();
            out.print(message);
            out.flush();
            out.close();
        } catch (FileUploadException e) {
            response.setContentType("text/html; charset=UTF-8");
            response.setCharacterEncoding("UTF-8");
            PrintWriter out = response.getWriter();
            out.print("ERROR:Upload failed: " + e.getMessage());
            out.flush();
            out.close();
        }
    }
}