/* Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.activiti.rest.api.identity;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

import org.activiti.engine.ActivitiException;
import org.activiti.engine.IdentityService;
import org.activiti.engine.impl.GroupQueryProperty;
import org.activiti.engine.query.QueryProperty;
import org.activiti.rest.api.ActivitiUtil;
import org.activiti.rest.api.DataResponse;
import org.activiti.rest.api.SecuredResource;
import org.restlet.resource.Get;
import org.restlet.resource.Post;

/**
 * @author Tijs Rademakers
 */
public class UserGroupsResource extends SecuredResource {

  Map<String, QueryProperty> properties = new HashMap<String, QueryProperty>();

  public UserGroupsResource() {
    properties.put("id", GroupQueryProperty.GROUP_ID);
    properties.put("name", GroupQueryProperty.NAME);
    properties.put("type", GroupQueryProperty.TYPE);
  }

  @Get
  public DataResponse getGroups() {
    if (authenticate() == false)
      return null;

    String userId = (String) getRequest().getAttributes().get("userId");
    if (userId == null) {
      throw new ActivitiException("No userId provided");
    }

    DataResponse dataResponse = new UserGroupsPaginateList().paginateList(
        getQuery(), ActivitiUtil.getIdentityService().createGroupQuery()
            .groupMember(userId), "id", properties);
    return dataResponse;
  }

  @Post
  public StateResponse setGroups(ArrayList<String> groupIds) {
    if (authenticate() == false)
      return null;
    String userId = (String) getRequest().getAttributes().get("userId");
    if (userId == null) {
      throw new ActivitiException("No userId provided");
    }
    if (groupIds == null) {
      throw new ActivitiException("No groupIds provided");
    }

    IdentityService identityService = ActivitiUtil.getIdentityService();
    // Check if user exists
    if (identityService.createUserQuery().userId(userId).singleResult() == null)
      throw new ActivitiException("The user '" + userId + " does not exist.");

    // Check first if all groups exist
    for (String groupId : groupIds) {
      if (identityService.createGroupQuery().groupId(groupId).singleResult() == null)
        throw new ActivitiException("Group '" + groupId + "' does not exist.");
    }
    for (String groupId : groupIds) {
      // Add only if not already member
      if (identityService.createUserQuery().userId(userId)
          .memberOfGroup(groupId).singleResult() == null)
        identityService.createMembership(userId, groupId);
    }
    return new StateResponse().setSuccess(true);
  }

}