/**
* 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;
}