com.liferay.akismet.hook.service.impl.AkismetWikiPageLocalServiceImpl.java Source code

Introduction

Here is the source code for com.liferay.akismet.hook.service.impl.AkismetWikiPageLocalServiceImpl.java

Source

/**
 * Copyright (c) 2000-present 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.akismet.hook.service.impl;

import com.liferay.akismet.model.AkismetData;
import com.liferay.akismet.service.AkismetDataLocalServiceUtil;
import com.liferay.akismet.util.AkismetConstants;
import com.liferay.akismet.util.AkismetUtil;
import com.liferay.akismet.util.PortletPropsKeys;
import com.liferay.akismet.util.PrefsPortletPropsUtil;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.servlet.HttpHeaders;
import com.liferay.portal.kernel.util.NotificationThreadLocal;
import com.liferay.portal.kernel.util.StringBundler;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.util.StringUtil;
import com.liferay.portal.kernel.workflow.WorkflowConstants;
import com.liferay.portal.model.User;
import com.liferay.portal.service.ServiceContext;
import com.liferay.portal.service.UserLocalServiceUtil;
import com.liferay.portlet.wiki.model.WikiPage;
import com.liferay.portlet.wiki.service.WikiPageLocalService;
import com.liferay.portlet.wiki.service.WikiPageLocalServiceWrapper;

import java.util.Map;

/**
 * @author Peter Shin
 */
public class AkismetWikiPageLocalServiceImpl extends WikiPageLocalServiceWrapper {

    public AkismetWikiPageLocalServiceImpl(WikiPageLocalService wikiPageLocalService) {

        super(wikiPageLocalService);
    }

    @Override
    public WikiPage addPage(long userId, long nodeId, String title, double version, String content, String summary,
            boolean minorEdit, String format, boolean head, String parentTitle, String redirectTitle,
            ServiceContext serviceContext) throws PortalException {

        int workflowAction = serviceContext.getWorkflowAction();

        if (workflowAction == WorkflowConstants.ACTION_SAVE_DRAFT) {
            return super.addPage(userId, nodeId, title, content, summary, minorEdit, serviceContext);
        }

        WikiPage page = super.addPage(userId, nodeId, title, version, content, summary, minorEdit, format, head,
                parentTitle, redirectTitle, serviceContext);

        AkismetData akismetData = updateAkismetData(page, serviceContext);

        if (!isWikiEnabled(userId, nodeId, serviceContext)) {
            return page;
        }

        String akismetContent = page.getTitle() + "\n\n" + page.getContent();

        if (!AkismetUtil.isSpam(userId, akismetContent, akismetData)) {
            return super.updateStatus(userId, page.getResourcePrimKey(), WorkflowConstants.STATUS_APPROVED,
                    serviceContext);
        }

        boolean notificationEnabled = false;

        try {
            notificationEnabled = NotificationThreadLocal.isEnabled();

            NotificationThreadLocal.setEnabled(false);

            page.setSummary(AkismetConstants.WIKI_PAGE_PENDING_APPROVAL);
            page.setStatus(WorkflowConstants.STATUS_APPROVED);

            return super.updateWikiPage(page);
        } finally {
            NotificationThreadLocal.setEnabled(notificationEnabled);
        }
    }

    @Override
    public WikiPage updatePage(long userId, long nodeId, String title, double version, String content,
            String summary, boolean minorEdit, String format, String parentTitle, String redirectTitle,
            ServiceContext serviceContext) throws PortalException {

        int workflowAction = serviceContext.getWorkflowAction();

        if (workflowAction == WorkflowConstants.ACTION_SAVE_DRAFT) {
            return super.updatePage(userId, nodeId, title, version, content, summary, minorEdit, format,
                    parentTitle, redirectTitle, serviceContext);
        }

        WikiPage page = super.updatePage(userId, nodeId, title, version, content, summary, minorEdit, format,
                parentTitle, redirectTitle, serviceContext);

        AkismetData akismetData = updateAkismetData(page, serviceContext);

        if (!isWikiEnabled(userId, nodeId, serviceContext)) {
            return page;
        }

        String akismetContent = page.getTitle() + "\n\n" + page.getContent();

        if (!AkismetUtil.isSpam(userId, akismetContent, akismetData)) {
            return super.updateStatus(userId, page.getResourcePrimKey(), WorkflowConstants.STATUS_APPROVED,
                    serviceContext);
        }

        boolean notificationEnabled = false;

        try {
            notificationEnabled = NotificationThreadLocal.isEnabled();

            NotificationThreadLocal.setEnabled(false);

            page.setSummary(AkismetConstants.WIKI_PAGE_PENDING_APPROVAL);
            page.setStatus(WorkflowConstants.STATUS_APPROVED);

            page = super.updateWikiPage(page);

            WikiPage previousPage = AkismetUtil.getWikiPage(page.getNodeId(), page.getTitle(), page.getVersion(),
                    true);

            if (previousPage == null) {
                return page;
            }

            ServiceContext newServiceContext = new ServiceContext();

            newServiceContext.setFormDate(page.getModifiedDate());

            return super.revertPage(userId, nodeId, title, previousPage.getVersion(), newServiceContext);
        } finally {
            NotificationThreadLocal.setEnabled(notificationEnabled);
        }
    }

    protected String getPermalink(WikiPage page, ServiceContext serviceContext) {

        StringBundler sb = new StringBundler(4);

        sb.append(serviceContext.getPortalURL());
        sb.append(serviceContext.getPathMain());
        sb.append("/wiki/find_page?pageResourcePrimKey=");
        sb.append(page.getResourcePrimKey());

        return sb.toString();
    }

    protected boolean isWikiEnabled(long userId, long nodeId, ServiceContext serviceContext)
            throws PortalException {

        if (!AkismetUtil.hasRequiredInfo(serviceContext)) {
            return false;
        }

        User user = UserLocalServiceUtil.getUser(userId);

        if (!AkismetUtil.isWikiEnabled(user.getCompanyId())) {
            return false;
        }

        int checkThreshold = PrefsPortletPropsUtil.getInteger(user.getCompanyId(),
                PortletPropsKeys.AKISMET_CHECK_THRESHOLD);

        if (checkThreshold > 0) {
            int count = super.getPagesCount(userId, nodeId, WorkflowConstants.STATUS_APPROVED);

            if (count > checkThreshold) {
                return false;
            }
        }

        return true;
    }

    protected AkismetData updateAkismetData(WikiPage page, ServiceContext serviceContext) {

        if (!AkismetUtil.hasRequiredInfo(serviceContext)) {
            return null;
        }

        String permalink = getPermalink(page, serviceContext);

        Map<String, String> headers = serviceContext.getHeaders();

        String referrer = headers.get("referer");
        String userAgent = headers.get(StringUtil.toLowerCase(HttpHeaders.USER_AGENT));

        String userIP = serviceContext.getRemoteAddr();

        return AkismetDataLocalServiceUtil.updateAkismetData(WikiPage.class.getName(), page.getPageId(),
                AkismetConstants.TYPE_WIKI, permalink, referrer, userAgent, userIP, StringPool.BLANK);
    }

}