plugins/org.eclipse.osee.ats.core.client/src/org/eclipse/osee/ats/core/client/internal/user/AtsUserServiceClientImpl.java

/*******************************************************************************
 * Copyright (c) 2013 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.ats.core.client.internal.user;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import org.eclipse.osee.ats.api.IAtsWorkItem;
import org.eclipse.osee.ats.api.config.IAtsConfigurationProvider;
import org.eclipse.osee.ats.api.data.AtsRelationTypes;
import org.eclipse.osee.ats.api.user.IAtsUser;
import org.eclipse.osee.ats.api.user.JaxAtsUser;
import org.eclipse.osee.ats.core.client.IAtsUserServiceClient;
import org.eclipse.osee.ats.core.client.util.AtsGroup;
import org.eclipse.osee.ats.core.users.AbstractAtsUserService;
import org.eclipse.osee.ats.core.util.AtsUtilCore;
import org.eclipse.osee.framework.core.enums.Active;
import org.eclipse.osee.framework.core.enums.CoreArtifactTypes;
import org.eclipse.osee.framework.core.enums.CoreAttributeTypes;
import org.eclipse.osee.framework.jdk.core.type.OseeCoreException;
import org.eclipse.osee.framework.skynet.core.User;
import org.eclipse.osee.framework.skynet.core.UserManager;
import org.eclipse.osee.framework.skynet.core.artifact.Artifact;
import org.eclipse.osee.framework.skynet.core.artifact.search.ArtifactQuery;

/**
 * Non-artifact base user service
 *
 * @author Donald G Dunne
 */
public class AtsUserServiceClientImpl extends AbstractAtsUserService implements IAtsUserServiceClient {

   private IAtsConfigurationProvider configurationProvider;

   public AtsUserServiceClientImpl() {
      // For OSGI Instantiation
   }

   public AtsUserServiceClientImpl(IAtsConfigurationProvider configurationProvider) {
      this.configurationProvider = configurationProvider;
   }

   public void setConfigurationsService(IAtsConfigurationProvider configurationProvider) {
      this.configurationProvider = configurationProvider;
   }

   @Override
   public IAtsUser getUserFromOseeUser(User user) throws OseeCoreException {
      IAtsUser atsUser = userIdToAtsUser.get(user.getUserId());
      if (atsUser == null) {
         atsUser = createFromArtifact(user);
         userIdToAtsUser.put(user.getUserId(), atsUser);
      }
      return atsUser;
   }

   @Override
   public User getOseeUser(IAtsUser atsUser) throws OseeCoreException {
      User oseeUser = null;
      if (atsUser.getStoreObject() != null) {
         oseeUser = (User) atsUser.getStoreObject();
      } else {
         oseeUser = getOseeUserById(atsUser.getUserId());
      }
      return oseeUser;
   }

   @Override
   public User getCurrentOseeUser() throws OseeCoreException {
      IAtsUser user = getCurrentUser();
      return getOseeUser(user);
   }

   @Override
   public Collection<? extends User> toOseeUsers(Collection<? extends IAtsUser> users) throws OseeCoreException {
      List<User> results = new LinkedList<>();
      for (IAtsUser user : users) {
         results.add(getOseeUser(user));
      }
      return results;
   }

   @Override
   public Collection<IAtsUser> getAtsUsers(Collection<? extends Artifact> artifacts) throws OseeCoreException {
      List<IAtsUser> users = new LinkedList<>();
      for (Artifact artifact : artifacts) {
         if (artifact instanceof User) {
            User user = (User) artifact;
            IAtsUser atsUser = getUserFromOseeUser(user);
            users.add(atsUser);
         }
      }
      return users;
   }

   @Override
   public Collection<User> getOseeUsers(Collection<? extends IAtsUser> users) throws OseeCoreException {
      List<User> results = new LinkedList<>();
      for (IAtsUser user : users) {
         results.add(getOseeUser(user));
      }
      return results;
   }

   @Override
   public User getOseeUserById(String userId) throws OseeCoreException {
      return getOseeUser(getUserById(userId));
   }

   @Override
   public List<User> getOseeUsersSorted(Active active) {
      List<IAtsUser> activeUsers = getUsers(active);
      List<User> oseeUsers = new ArrayList<>();
      oseeUsers.addAll(getOseeUsers(activeUsers));
      Collections.sort(oseeUsers);
      return oseeUsers;
   }

   @Override
   public List<IAtsUser> getSubscribed(IAtsWorkItem workItem) throws OseeCoreException {
      ArrayList<IAtsUser> arts = new ArrayList<>();
      for (Artifact art : ((Artifact) workItem.getStoreObject()).getRelatedArtifacts(
         AtsRelationTypes.SubscribedUser_User)) {
         arts.add(getUserById((String) art.getSoleAttributeValue(CoreAttributeTypes.UserId)));
      }
      return arts;
   }

   @Override
   public IAtsUser getUserById(long accountId) {
      return getUserFromOseeUser(UserManager.getUserByArtId((int) accountId));
   }

   @Override
   public String getCurrentUserId() {
      return UserManager.getUser().getUserId();
   }

   @Override
   public boolean isAtsAdmin(IAtsUser user) {
      Boolean admin = userIdToAdmin.get(user.getUserId());
      if (admin == null) {
         admin = AtsGroup.AtsAdmin.isMember(user);
         userIdToAdmin.put(user.getUserId(), admin);
      }
      return admin;
   }

   @Override
   public List<? extends IAtsUser> getUsers() {
      return configurationProvider.getConfigurations().getUsers();
   }

   @Override
   protected IAtsUser loadUserFromDbByUserId(String userId) {
      Artifact userArt = ArtifactQuery.getArtifactFromTypeAndAttribute(CoreArtifactTypes.User,
         CoreAttributeTypes.UserId, userId, AtsUtilCore.getAtsBranch());
      return createFromArtifact(userArt);
   }

   private JaxAtsUser createFromArtifact(Artifact userArt) {
      JaxAtsUser atsUser = new JaxAtsUser();
      atsUser.setName(userArt.getName());
      atsUser.setStoreObject(userArt);
      atsUser.setUserId(userArt.getSoleAttributeValue(CoreAttributeTypes.UserId, ""));
      atsUser.setEmail(userArt.getSoleAttributeValue(CoreAttributeTypes.Email, ""));
      atsUser.setActive(userArt.getSoleAttributeValue(CoreAttributeTypes.Active, true));
      atsUser.setUuid(userArt.getUuid());
      return atsUser;
   }

   @Override
   protected IAtsUser loadUserFromDbByUserName(String name) {
      Artifact userArt = ArtifactQuery.getArtifactFromTypeAndAttribute(CoreArtifactTypes.User, CoreAttributeTypes.Name,
         name, AtsUtilCore.getAtsBranch());
      return createFromArtifact(userArt);
   }

}