/**
    * Copyright 2011-2013 The Kuali Foundation Licensed under the
    * Educational Community 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.osedu.org/licenses/ECL-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.kuali.mobility.security.user.dao;
  
  import org.apache.log4j.Logger;
  import org.kuali.mobility.security.group.api.Group;
  import org.kuali.mobility.security.group.api.GroupDao;
  import org.kuali.mobility.security.user.api.User;
  import org.kuali.mobility.security.user.api.UserDao;
  import org.kuali.mobility.security.user.entity.GroupMembership;
  import org.kuali.mobility.security.user.entity.UserImpl;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.beans.factory.annotation.Qualifier;
  import org.springframework.stereotype.Repository;
  import org.springframework.transaction.annotation.Transactional;
  
  import javax.persistence.EntityManager;
  import javax.persistence.PersistenceContext;
  import javax.persistence.Query;
  import java.util.ArrayList;
  import java.util.List;
  import javax.persistence.EntityNotFoundException;
  import javax.persistence.NonUniqueResultException;
  
  /**
   * @author Kuali Mobility Team (mobility.collab@kuali.org)
   */
  @Repository
  public class UserDaoImpl implements UserDao {
  
  	private static final Logger LOG = Logger.getLogger(UserDaoImpl.class);
  
  	@PersistenceContext
  	private EntityManager entityManager;
  
  	@Autowired
  	@Qualifier("kmeGroupDao")
  	private GroupDao groupDao;
  
  	@Override
  	public User loadUserById(Long id) {
  		Query query = getEntityManager().createNamedQuery("User.lookupById");
  		query.setParameter("id", id);
  		UserImpl user = null;
  
  		try {
  			user = (UserImpl) query.getSingleResult();
  		} catch (EntityNotFoundException e) {
  			LOG.error(e.getLocalizedMessage());
  		} catch (NonUniqueResultException e) {
  			LOG.error(e.getLocalizedMessage());
  		} catch (Exception e) {
  			LOG.error(e.getLocalizedMessage());
  		}
  
  		if (null != user) {
  			loadGroupMembership(user);
  		}
  		return user;
  	}
  
  	@Override
  	public User loadUserByLoginName(String loginName) {
  		Query query = getEntityManager().createNamedQuery("User.lookupByLoginName");
  		query.setParameter("loginName", loginName);
  		UserImpl user = null;
  
  		try {
  			user = (UserImpl) query.getSingleResult();
  		} catch (EntityNotFoundException e) {
  			LOG.error(e.getLocalizedMessage());
  		} catch (NonUniqueResultException e) {
  			LOG.error(e.getLocalizedMessage());
  		} catch (Exception e) {
  			LOG.error(e.getLocalizedMessage());
  		}
  
  		if (null != user) {
  			loadGroupMembership(user);
  		}
  		return user;
  	}
  
  	@Override
  	@Transactional
  	public Long saveUser(User user) {
  		Long id;
 		if (null == user) {
 			id = null;
 		} else {
 			if (null == user.getId()) {
 				getEntityManager().persist(user);
 			} else {
 				getEntityManager().merge(user);
 			}
 			id = user.getId();
 			Query query = getEntityManager().createNamedQuery("GroupMembership.deleteMembershipByUserId");
 			query.setParameter("userId", id);
 			query.executeUpdate();
 			for (Group g : user.getGroups()) {
 				GroupMembership groupMembership = new GroupMembership();
 				groupMembership.setGroupId(g.getId());
 				groupMembership.setUserId(id);
 				getEntityManager().persist(groupMembership);
 			}
 		}
 		return id;
 	}
 
 	private void loadGroupMembership(UserImpl user) {
 		if (user != null && user.getId() != null) {
 			List<Group> groups = new ArrayList<Group>();
 			Query query = getEntityManager().createNamedQuery("GroupMembership.loadMembershipByUserId");
 			query.setParameter("userId", user.getId());
 			List<GroupMembership> groupMemberships = query.getResultList();
 			for (GroupMembership gm : groupMemberships) {
 				Group g = getGroupDao().getGroup(gm.getGroupId());
 				groups.add(g);
 			}
 			user.setGroups(groups);
 		}
 	}
 
 	private void loadGroupMembership(List<UserImpl> users) {
 		for (UserImpl u : users) {
 			loadGroupMembership(u);
 		}
 	}
 
 	public EntityManager getEntityManager() {
 		return entityManager;
 	}
 
 	public void setEntityManager(EntityManager entityManager) {
 		this.entityManager = entityManager;
 	}
 
 	public GroupDao getGroupDao() {
 		return groupDao;
 	}
 
 	public void setGroupDao(GroupDao groupDao) {
 		this.groupDao = groupDao;
 	}
 }