com.liferay.portlet.sites.action.EditTeamAssignmentsAction.java Source code

Introduction

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

import com.liferay.portal.NoSuchTeamException;
import com.liferay.portal.kernel.servlet.SessionErrors;
import com.liferay.portal.kernel.util.Constants;
import com.liferay.portal.kernel.util.ParamUtil;
import com.liferay.portal.kernel.util.StringUtil;
import com.liferay.portal.kernel.util.Validator;
import com.liferay.portal.security.auth.PrincipalException;
import com.liferay.portal.service.UserGroupServiceUtil;
import com.liferay.portal.service.UserServiceUtil;
import com.liferay.portal.struts.PortletAction;

import javax.portlet.ActionRequest;
import javax.portlet.ActionResponse;
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 Brian Wing Shun Chan
 */
public class EditTeamAssignmentsAction extends PortletAction {

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

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

        try {
            if (cmd.equals("team_user_groups")) {
                updateTeamUserGroups(actionRequest);
            } else if (cmd.equals("team_users")) {
                updateTeamUsers(actionRequest);
            }

            if (Validator.isNotNull(cmd)) {
                String redirect = ParamUtil.getString(actionRequest, "assignmentsRedirect");

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

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

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

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

        try {
            ActionUtil.getTeam(renderRequest);
        } catch (Exception e) {
            if (e instanceof NoSuchTeamException || e instanceof PrincipalException) {

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

                return mapping.findForward("portlet.sites_admin.error");
            } else {
                throw e;
            }
        }

        return mapping.findForward(getForward(renderRequest, "portlet.sites_admin.edit_team_assignments"));
    }

    protected void updateTeamUserGroups(ActionRequest actionRequest) throws Exception {

        long teamId = ParamUtil.getLong(actionRequest, "teamId");

        long[] addUserGroupIds = StringUtil.split(ParamUtil.getString(actionRequest, "addUserGroupIds"), 0L);
        long[] removeUserGroupIds = StringUtil.split(ParamUtil.getString(actionRequest, "removeUserGroupIds"), 0L);

        UserGroupServiceUtil.addTeamUserGroups(teamId, addUserGroupIds);
        UserGroupServiceUtil.unsetTeamUserGroups(teamId, removeUserGroupIds);
    }

    protected void updateTeamUsers(ActionRequest actionRequest) throws Exception {

        long teamId = ParamUtil.getLong(actionRequest, "teamId");

        long[] addUserIds = StringUtil.split(ParamUtil.getString(actionRequest, "addUserIds"), 0L);
        long[] removeUserIds = StringUtil.split(ParamUtil.getString(actionRequest, "removeUserIds"), 0L);

        UserServiceUtil.addTeamUsers(teamId, addUserIds);
        UserServiceUtil.unsetTeamUsers(teamId, removeUserIds);
    }

}