com.liferay.portlet.documentlibrary.action.CompareVersionsAction.java Source code

Introduction

Here is the source code for com.liferay.portlet.documentlibrary.action.CompareVersionsAction.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.documentlibrary.action;

import com.liferay.portal.kernel.io.unsync.UnsyncByteArrayInputStream;
import com.liferay.portal.kernel.repository.model.FileEntry;
import com.liferay.portal.kernel.repository.model.FileVersion;
import com.liferay.portal.kernel.servlet.SessionErrors;
import com.liferay.portal.kernel.util.DiffResult;
import com.liferay.portal.kernel.util.DiffUtil;
import com.liferay.portal.kernel.util.HtmlUtil;
import com.liferay.portal.kernel.util.ParamUtil;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.util.StringUtil;
import com.liferay.portal.security.auth.PrincipalException;
import com.liferay.portal.struts.PortletAction;
import com.liferay.portal.util.WebKeys;
import com.liferay.portlet.documentlibrary.NoSuchFileEntryException;
import com.liferay.portlet.documentlibrary.service.DLAppServiceUtil;
import com.liferay.portlet.documentlibrary.util.DLUtil;
import com.liferay.portlet.documentlibrary.util.DocumentConversionUtil;

import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;

import java.util.List;

import javax.portlet.PortletConfig;
import javax.portlet.RenderRequest;
import javax.portlet.RenderResponse;

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

/**
 * @author Bruno Farache
 */
public class CompareVersionsAction extends PortletAction {

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

        try {
            compareVersions(renderRequest);
        } catch (Exception e) {
            if (e instanceof NoSuchFileEntryException || e instanceof PrincipalException) {

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

                setForward(renderRequest, "portlet.document_library.error");
            } else {
                throw e;
            }
        }

        return mapping.findForward("portlet.document_library.compare_versions");
    }

    protected void compareVersions(RenderRequest renderRequest) throws Exception {

        long fileEntryId = ParamUtil.getLong(renderRequest, "fileEntryId");

        String sourceVersion = ParamUtil.getString(renderRequest, "sourceVersion");
        String targetVersion = ParamUtil.getString(renderRequest, "targetVersion");

        FileEntry fileEntry = DLAppServiceUtil.getFileEntry(fileEntryId);

        String extension = fileEntry.getExtension();

        FileVersion sourceFileVersion = fileEntry.getFileVersion(sourceVersion);

        String sourceTitle = sourceFileVersion.getTitle();

        FileVersion targetFileVersion = fileEntry.getFileVersion(targetVersion);

        String targetTitle = targetFileVersion.getTitle();

        InputStream sourceIs = fileEntry.getContentStream(sourceVersion);
        InputStream targetIs = fileEntry.getContentStream(targetVersion);

        if (extension.equals("htm") || extension.equals("html") || extension.equals("xml")) {

            String escapedSource = HtmlUtil.escape(StringUtil.read(sourceIs));
            String escapedTarget = HtmlUtil.escape(StringUtil.read(targetIs));

            sourceIs = new UnsyncByteArrayInputStream(escapedSource.getBytes(StringPool.UTF8));
            targetIs = new UnsyncByteArrayInputStream(escapedTarget.getBytes(StringPool.UTF8));
        }

        if (DocumentConversionUtil.isEnabled() && DocumentConversionUtil.isConvertBeforeCompare(extension)) {

            String sourceTempFileId = DLUtil.getTempFileId(fileEntryId, sourceVersion);
            String targetTempFileId = DLUtil.getTempFileId(fileEntryId, targetVersion);

            sourceIs = new FileInputStream(
                    DocumentConversionUtil.convert(sourceTempFileId, sourceIs, extension, "txt"));
            targetIs = new FileInputStream(
                    DocumentConversionUtil.convert(targetTempFileId, targetIs, extension, "txt"));
        }

        List<DiffResult>[] diffResults = DiffUtil.diff(new InputStreamReader(sourceIs),
                new InputStreamReader(targetIs));

        renderRequest.setAttribute(WebKeys.SOURCE_NAME, sourceTitle + StringPool.SPACE + sourceVersion);
        renderRequest.setAttribute(WebKeys.TARGET_NAME, targetTitle + StringPool.SPACE + targetVersion);
        renderRequest.setAttribute(WebKeys.DIFF_RESULTS, diffResults);
    }

}