/* 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.UserQueryProperty;
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 Ernesto Revilla
 */
public class GroupUsersResource extends SecuredResource {

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

  public GroupUsersResource() {
    properties.put("id", UserQueryProperty.USER_ID);
    properties.put("firstName", UserQueryProperty.FIRST_NAME);
    properties.put("lastName", UserQueryProperty.LAST_NAME);
    properties.put("email", UserQueryProperty.EMAIL);
  }

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

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

    DataResponse dataResponse = new GroupUsersPaginateList().paginateList(
        getQuery(), ActivitiUtil.getIdentityService().createUserQuery()
            .memberOfGroup(groupId), "id", properties);
    return dataResponse;
  }

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

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

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

}