com.liferay.portlet.portletconfiguration.action.ExportImportAction.java Source code

Java tutorial

Introduction

Here is the source code for com.liferay.portlet.portletconfiguration.action.ExportImportAction.java

Source

/**
 * Copyright (c) 2000-2012 Liferay, Inc. 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 com.liferay.portlet.portletconfiguration.action;

import com.liferay.portal.LARFileException;
import com.liferay.portal.LARTypeException;
import com.liferay.portal.LayoutImportException;
import com.liferay.portal.NoSuchLayoutException;
import com.liferay.portal.PortletIdException;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.servlet.ServletResponseUtil;
import com.liferay.portal.kernel.servlet.SessionErrors;
import com.liferay.portal.kernel.staging.StagingUtil;
import com.liferay.portal.kernel.upload.UploadPortletRequest;
import com.liferay.portal.kernel.util.Constants;
import com.liferay.portal.kernel.util.ContentTypes;
import com.liferay.portal.kernel.util.FileUtil;
import com.liferay.portal.kernel.util.GetterUtil;
import com.liferay.portal.kernel.util.ParamUtil;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.model.Layout;
import com.liferay.portal.model.Portlet;
import com.liferay.portal.security.auth.PrincipalException;
import com.liferay.portal.service.LayoutLocalServiceUtil;
import com.liferay.portal.service.LayoutServiceUtil;
import com.liferay.portal.struts.ActionConstants;
import com.liferay.portal.theme.ThemeDisplay;
import com.liferay.portal.util.PortalUtil;
import com.liferay.portal.util.WebKeys;
import com.liferay.portlet.PortletPreferencesFactoryUtil;

import java.io.File;
import java.io.FileInputStream;

import java.util.Calendar;
import java.util.Date;

import javax.portlet.ActionRequest;
import javax.portlet.ActionResponse;
import javax.portlet.PortletConfig;
import javax.portlet.PortletPreferences;
import javax.portlet.RenderRequest;
import javax.portlet.RenderResponse;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.struts.action.ActionForm;
import org.apache.struts.action.ActionForward;
import org.apache.struts.action.ActionMapping;

/**
 * @author Jorge Ferrer
 * @author Raymond Aug
 */
public class ExportImportAction extends EditConfigurationAction {

    @Override
    public void processAction(ActionMapping mapping, ActionForm form, PortletConfig portletConfig,
            ActionRequest actionRequest, ActionResponse actionResponse) throws Exception {

        Portlet portlet = null;

        try {
            portlet = getPortlet(actionRequest);
        } catch (PrincipalException pe) {
            SessionErrors.add(actionRequest, PrincipalException.class.getName());

            setForward(actionRequest, "portlet.portlet_configuration.error");
        }

        String cmd = ParamUtil.getString(actionRequest, Constants.CMD);

        try {
            if (cmd.equals("copy_from_live")) {
                StagingUtil.copyFromLive(actionRequest, portlet);

                sendRedirect(actionRequest, actionResponse);
            } else if (cmd.equals(Constants.EXPORT)) {
                exportData(actionRequest, actionResponse, portlet);

                sendRedirect(actionRequest, actionResponse);
            } else if (cmd.equals(Constants.IMPORT)) {
                importData(actionRequest, actionResponse, portlet);

                sendRedirect(actionRequest, actionResponse);
            } else if (cmd.equals("publish_to_live")) {
                StagingUtil.publishToLive(actionRequest, portlet);

                sendRedirect(actionRequest, actionResponse);
            }
        } catch (Exception e) {
            if (e instanceof NoSuchLayoutException || e instanceof PrincipalException) {

                SessionErrors.add(actionRequest, e.getClass().getName());

                setForward(actionRequest, "portlet.portlet_configuration.error");
            } else {
                throw e;
            }
        }
    }

    @Override
    public ActionForward render(ActionMapping mapping, ActionForm form, PortletConfig portletConfig,
            RenderRequest renderRequest, RenderResponse renderResponse) throws Exception {

        Portlet portlet = null;

        try {
            portlet = getPortlet(renderRequest);
        } catch (PrincipalException pe) {
            SessionErrors.add(renderRequest, PrincipalException.class.getName());

            return mapping.findForward("portlet.portlet_configuration.error");
        }

        renderResponse.setTitle(getTitle(portlet, renderRequest));

        return mapping.findForward(getForward(renderRequest, "portlet.portlet_configuration.export_import"));
    }

    protected void exportData(ActionRequest actionRequest, ActionResponse actionResponse, Portlet portlet)
            throws Exception {

        try {
            ThemeDisplay themeDisplay = (ThemeDisplay) actionRequest.getAttribute(WebKeys.THEME_DISPLAY);

            long plid = ParamUtil.getLong(actionRequest, "plid");
            long groupId = ParamUtil.getLong(actionRequest, "groupId");
            String fileName = ParamUtil.getString(actionRequest, "exportFileName");
            String range = ParamUtil.getString(actionRequest, "range");

            Date startDate = null;
            Date endDate = null;

            if (range.equals("dateRange")) {
                int startDateMonth = ParamUtil.getInteger(actionRequest, "startDateMonth");
                int startDateDay = ParamUtil.getInteger(actionRequest, "startDateDay");
                int startDateYear = ParamUtil.getInteger(actionRequest, "startDateYear");
                int startDateHour = ParamUtil.getInteger(actionRequest, "startDateHour");
                int startDateMinute = ParamUtil.getInteger(actionRequest, "startDateMinute");
                int startDateAmPm = ParamUtil.getInteger(actionRequest, "startDateAmPm");

                if (startDateAmPm == Calendar.PM) {
                    startDateHour += 12;
                }

                startDate = PortalUtil.getDate(startDateMonth, startDateDay, startDateYear, startDateHour,
                        startDateMinute, themeDisplay.getTimeZone(), new PortalException());

                int endDateMonth = ParamUtil.getInteger(actionRequest, "endDateMonth");
                int endDateDay = ParamUtil.getInteger(actionRequest, "endDateDay");
                int endDateYear = ParamUtil.getInteger(actionRequest, "endDateYear");
                int endDateHour = ParamUtil.getInteger(actionRequest, "endDateHour");
                int endDateMinute = ParamUtil.getInteger(actionRequest, "endDateMinute");
                int endDateAmPm = ParamUtil.getInteger(actionRequest, "endDateAmPm");

                if (endDateAmPm == Calendar.PM) {
                    endDateHour += 12;
                }

                endDate = PortalUtil.getDate(endDateMonth, endDateDay, endDateYear, endDateHour, endDateMinute,
                        themeDisplay.getTimeZone(), new PortalException());
            } else if (range.equals("fromLastPublishDate")) {
                Layout layout = LayoutLocalServiceUtil.getLayout(plid);

                PortletPreferences preferences = PortletPreferencesFactoryUtil.getPortletSetup(layout,
                        portlet.getPortletId(), StringPool.BLANK);

                long lastPublishDate = GetterUtil
                        .getLong(preferences.getValue("last-publish-date", StringPool.BLANK));

                if (lastPublishDate > 0) {
                    Calendar cal = Calendar.getInstance(themeDisplay.getTimeZone(), themeDisplay.getLocale());

                    endDate = cal.getTime();

                    cal.setTimeInMillis(lastPublishDate);

                    startDate = cal.getTime();
                }
            }

            File file = LayoutServiceUtil.exportPortletInfoAsFile(plid, groupId, portlet.getPortletId(),
                    actionRequest.getParameterMap(), startDate, endDate);

            HttpServletRequest request = PortalUtil.getHttpServletRequest(actionRequest);
            HttpServletResponse response = PortalUtil.getHttpServletResponse(actionResponse);

            ServletResponseUtil.sendFile(request, response, fileName, new FileInputStream(file),
                    ContentTypes.APPLICATION_ZIP);

            FileUtil.delete(file);

            setForward(actionRequest, ActionConstants.COMMON_NULL);
        } catch (Exception e) {
            if (_log.isDebugEnabled()) {
                _log.debug(e, e);
            }

            SessionErrors.add(actionRequest, e.getClass().getName());
        }
    }

    protected void importData(ActionRequest actionRequest, ActionResponse actionResponse, Portlet portlet)
            throws Exception {

        try {
            UploadPortletRequest uploadPortletRequest = PortalUtil.getUploadPortletRequest(actionRequest);

            long plid = ParamUtil.getLong(uploadPortletRequest, "plid");
            long groupId = ParamUtil.getLong(uploadPortletRequest, "groupId");
            File file = uploadPortletRequest.getFile("importFileName");

            if (!file.exists()) {
                throw new LARFileException("Import file does not exist");
            }

            LayoutServiceUtil.importPortletInfo(plid, groupId, portlet.getPortletId(),
                    actionRequest.getParameterMap(), file);

            addSuccessMessage(actionRequest, actionResponse);
        } catch (Exception e) {
            if ((e instanceof LARFileException) || (e instanceof LARTypeException)
                    || (e instanceof PortletIdException)) {

                SessionErrors.add(actionRequest, e.getClass().getName());
            } else {
                _log.error(e, e);

                SessionErrors.add(actionRequest, LayoutImportException.class.getName());
            }
        }
    }

    private static Log _log = LogFactoryUtil.getLog(ExportImportAction.class);

}