org.lsug.quota.web.internal.upload.QuotaDefaultUploadHandler.java Source code

Introduction

Here is the source code for org.lsug.quota.web.internal.upload.QuotaDefaultUploadHandler.java

Source

/**
 * Copyright (c) 2013-present Liferay Spain User Group All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package org.lsug.quota.web.internal.upload;

import com.liferay.document.library.kernel.exception.FileSizeException;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.exception.SystemException;
import com.liferay.portal.kernel.json.JSONObject;
import com.liferay.portal.kernel.portlet.JSONPortletResponseUtil;
import com.liferay.portal.kernel.repository.model.FileEntry;
import com.liferay.portal.kernel.upload.LiferayFileItemException;
import com.liferay.portal.kernel.upload.UploadException;
import com.liferay.portal.kernel.upload.UploadPortletRequest;
import com.liferay.portal.kernel.upload.UploadRequestSizeException;
import com.liferay.portal.kernel.util.Portal;
import com.liferay.portal.kernel.util.WebKeys;
import com.liferay.upload.UploadFileEntryHandler;
import com.liferay.upload.UploadHandler;
import com.liferay.upload.UploadResponseHandler;

import java.io.IOException;

import javax.portlet.PortletRequest;
import javax.portlet.PortletResponse;

import org.lsug.quota.exception.QuotaExceededException;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Reference;

/**
 * 
 * @author Juan Gonzalez
 *
 */
@Component(immediate = true, service = QuotaDefaultUploadHandler.class)
public class QuotaDefaultUploadHandler implements UploadHandler {

    private QuotaDefaultUploadResponseHandler _quotaDefaultUploadResponseHandler;

    @Override
    public void upload(UploadFileEntryHandler uploadFileEntryHandler, UploadResponseHandler uploadResponseHandler,
            PortletRequest portletRequest, PortletResponse portletResponse) throws PortalException {

        try {
            JSONObject responseJSONObject = _getResponseJSONObject(uploadFileEntryHandler, uploadResponseHandler,
                    portletRequest);

            JSONPortletResponseUtil.writeJSON(portletRequest, portletResponse, responseJSONObject);
        } catch (IOException ioe) {
            throw new SystemException(ioe);
        }
    }

    private JSONObject _getResponseJSONObject(UploadFileEntryHandler uploadFileEntryHandler,
            UploadResponseHandler uploadResponseHandler, PortletRequest portletRequest)
            throws IOException, PortalException {

        try {
            UploadPortletRequest uploadPortletRequest = _getUploadPortletRequest(portletRequest);

            FileEntry fileEntry = uploadFileEntryHandler.upload(uploadPortletRequest);

            return uploadResponseHandler.onSuccess(uploadPortletRequest, fileEntry);
        } catch (PortalException pe) {
            if (pe instanceof QuotaExceededException) {
                return _quotaDefaultUploadResponseHandler.onFailure(portletRequest, pe);
            } else {
                return uploadResponseHandler.onFailure(portletRequest, pe);
            }
        }
    }

    private UploadPortletRequest _getUploadPortletRequest(PortletRequest portletRequest) throws PortalException {

        UploadPortletRequest uploadPortletRequest = _portal.getUploadPortletRequest(portletRequest);

        UploadException uploadException = (UploadException) portletRequest.getAttribute(WebKeys.UPLOAD_EXCEPTION);

        if (uploadException != null) {
            Throwable cause = uploadException.getCause();

            if (uploadException.isExceededFileSizeLimit()) {
                throw new FileSizeException(cause);
            }

            if (uploadException.isExceededLiferayFileItemSizeLimit()) {
                throw new LiferayFileItemException(cause);
            }

            if (uploadException.isExceededUploadRequestSizeLimit()) {
                throw new UploadRequestSizeException(cause);
            }

            throw new PortalException(cause);
        }

        return uploadPortletRequest;
    }

    @Reference(unbind = "-")
    protected void setQuotaDefaultUploadResponseHandler(
            QuotaDefaultUploadResponseHandler quotaDefaultUploadResponseHandler) {
        _quotaDefaultUploadResponseHandler = quotaDefaultUploadResponseHandler;
    }

    @Reference
    private Portal _portal;
}