com.liferay.message.boards.web.internal.security.permission.MBPermissionPropagatorImpl.java Source code

Introduction

Here is the source code for com.liferay.message.boards.web.internal.security.permission.MBPermissionPropagatorImpl.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.message.boards.web.internal.security.permission;

import com.liferay.message.boards.kernel.model.MBCategory;
import com.liferay.message.boards.kernel.model.MBMessage;
import com.liferay.message.boards.kernel.model.MBThread;
import com.liferay.message.boards.kernel.service.MBCategoryLocalService;
import com.liferay.message.boards.kernel.service.MBMessageLocalService;
import com.liferay.message.boards.web.constants.MBPortletKeys;
import com.liferay.portal.kernel.dao.orm.ActionableDynamicQuery;
import com.liferay.portal.kernel.dao.orm.DynamicQuery;
import com.liferay.portal.kernel.dao.orm.Property;
import com.liferay.portal.kernel.dao.orm.PropertyFactoryUtil;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.security.permission.BasePermissionPropagator;
import com.liferay.portal.kernel.security.permission.PermissionPropagator;
import com.liferay.portal.kernel.util.GetterUtil;
import com.liferay.portal.kernel.workflow.WorkflowConstants;

import java.util.ArrayList;
import java.util.List;

import javax.portlet.ActionRequest;

import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Reference;

/**
 * @author Kenneth Chang
 * @author Hugo Huijser
 */
@Component(immediate = true, property = { "javax.portlet.name=" + MBPortletKeys.MESSAGE_BOARDS,
        "javax.portlet.name=" + MBPortletKeys.MESSAGE_BOARDS_ADMIN }, service = PermissionPropagator.class)
public class MBPermissionPropagatorImpl extends BasePermissionPropagator {

    @Override
    public void propagateRolePermissions(ActionRequest actionRequest, String className, String primKey,
            long[] roleIds) throws PortalException {

        if (className.equals(MBCategory.class.getName())) {
            propagateCategoryRolePermissions(actionRequest, className, primKey, roleIds);
        } else if (className.equals(MBMessage.class.getName())) {
            long messageId = GetterUtil.getLong(primKey);

            MBMessage message = _mbMessageLocalService.getMessage(messageId);

            if (message.isRoot()) {
                propagateThreadRolePermissions(actionRequest, className, messageId, message.getThreadId(), roleIds);
            }
        } else if (className.equals("com.liferay.message.boards")) {
            propagateMBRolePermissions(actionRequest, className, primKey, roleIds);
        }
    }

    protected void propagateCategoryRolePermissions(ActionRequest actionRequest, String className, long primaryKey,
            long categoryId, long[] roleIds) throws PortalException {

        for (long roleId : roleIds) {
            propagateRolePermissions(actionRequest, roleId, className, primaryKey, MBCategory.class.getName(),
                    categoryId);
        }
    }

    protected void propagateCategoryRolePermissions(final ActionRequest actionRequest, final String className,
            String primKey, final long[] roleIds) throws PortalException {

        final long categoryId = GetterUtil.getLong(primKey);

        MBCategory category = _mbCategoryLocalService.getCategory(categoryId);

        List<Object> categoriesAndThreads = _mbCategoryLocalService.getCategoriesAndThreads(category.getGroupId(),
                categoryId);

        for (Object categoryOrThread : categoriesAndThreads) {
            if (categoryOrThread instanceof MBThread) {
                MBThread thread = (MBThread) categoryOrThread;

                List<MBMessage> messages = _mbMessageLocalService.getThreadMessages(thread.getThreadId(),
                        WorkflowConstants.STATUS_ANY);

                for (MBMessage message : messages) {
                    propagateMessageRolePermissions(actionRequest, className, categoryId, message.getMessageId(),
                            roleIds);
                }
            } else {
                category = (MBCategory) categoryOrThread;

                List<Long> categoryIds = new ArrayList<>();

                categoryIds.add(category.getCategoryId());

                categoryIds = _mbCategoryLocalService.getSubcategoryIds(categoryIds, category.getGroupId(),
                        category.getCategoryId());

                for (final long addCategoryId : categoryIds) {
                    propagateCategoryRolePermissions(actionRequest, className, categoryId, addCategoryId, roleIds);

                    ActionableDynamicQuery actionableDynamicQuery = _mbMessageLocalService
                            .getActionableDynamicQuery();

                    actionableDynamicQuery.setAddCriteriaMethod(new ActionableDynamicQuery.AddCriteriaMethod() {

                        @Override
                        public void addCriteria(DynamicQuery dynamicQuery) {
                            Property categoryIdProperty = PropertyFactoryUtil.forName("categoryId");

                            dynamicQuery.add(categoryIdProperty.eq(addCategoryId));
                        }

                    });
                    actionableDynamicQuery.setGroupId(category.getGroupId());
                    actionableDynamicQuery
                            .setPerformActionMethod(new ActionableDynamicQuery.PerformActionMethod<MBMessage>() {

                                @Override
                                public void performAction(MBMessage message) throws PortalException {

                                    propagateMessageRolePermissions(actionRequest, className, categoryId,
                                            message.getMessageId(), roleIds);
                                }

                            });

                    actionableDynamicQuery.performActions();
                }
            }
        }
    }

    protected void propagateMBRolePermissions(final ActionRequest actionRequest, final String className,
            String primKey, final long[] roleIds) throws PortalException {

        final long groupId = GetterUtil.getLong(primKey);

        List<MBCategory> categories = _mbCategoryLocalService.getCategories(groupId);

        for (MBCategory category : categories) {
            propagateCategoryRolePermissions(actionRequest, className, groupId, category.getCategoryId(), roleIds);
        }

        ActionableDynamicQuery actionableDynamicQuery = _mbMessageLocalService.getActionableDynamicQuery();

        actionableDynamicQuery.setGroupId(groupId);
        actionableDynamicQuery.setPerformActionMethod(new ActionableDynamicQuery.PerformActionMethod<MBMessage>() {

            @Override
            public void performAction(MBMessage message) throws PortalException {

                propagateMessageRolePermissions(actionRequest, className, groupId, message.getMessageId(), roleIds);
            }

        });

        actionableDynamicQuery.performActions();
    }

    protected void propagateMessageRolePermissions(ActionRequest actionRequest, String className, long primaryKey,
            long messageId, long[] roleIds) throws PortalException {

        for (long roleId : roleIds) {
            propagateRolePermissions(actionRequest, roleId, className, primaryKey, MBMessage.class.getName(),
                    messageId);
        }
    }

    protected void propagateThreadRolePermissions(ActionRequest actionRequest, String className, long messageId,
            long threadId, long[] roleIds) throws PortalException {

        List<MBMessage> messages = _mbMessageLocalService.getThreadMessages(threadId, WorkflowConstants.STATUS_ANY);

        for (MBMessage message : messages) {
            propagateMessageRolePermissions(actionRequest, className, messageId, message.getMessageId(), roleIds);
        }
    }

    @Reference(unbind = "-")
    protected void setMBCategoryLocalService(MBCategoryLocalService mbCategoryLocalService) {

        _mbCategoryLocalService = mbCategoryLocalService;
    }

    @Reference(unbind = "-")
    protected void setMBMessageLocalService(MBMessageLocalService mbMessageLocalService) {

        _mbMessageLocalService = mbMessageLocalService;
    }

    private MBCategoryLocalService _mbCategoryLocalService;
    private MBMessageLocalService _mbMessageLocalService;

}