QuadrigaUserRoleMapper.java

package edu.asu.spring.quadriga.web.login;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.springframework.ldap.core.DirContextOperations;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.ldap.userdetails.PersonContextMapper;
import org.springframework.transaction.CannotCreateTransactionException;

import edu.asu.spring.quadriga.domain.IQuadrigaRole;
import edu.asu.spring.quadriga.domain.IUser;
import edu.asu.spring.quadriga.exceptions.QuadrigaStorageException;
import edu.asu.spring.quadriga.service.IUserManager;

/**
 * This class is responsible for adding Quadriga specific roles to authenticated
 * users.
 * 
 * @author Julia Damerow
 * 
 */
public class QuadrigaUserRoleMapper extends PersonContextMapper {

	IUserManager userManager;

	public IUserManager getUserManager() {
		return userManager;
	}

	public void setUserManager(IUserManager userManager) {
		this.userManager = userManager;
	}

	/**
	 * This user is called with the username of the user that tries to login to
	 * Quadriga. It asks the {@link IUserManager} for the details about the user
	 * then creates and adds the corresponding {@link QuadrigaGrantedAuthority}
	 * /ies.
	 */
	@Override
	public UserDetails mapUserFromContext(DirContextOperations ctx,
			String username, Collection<? extends GrantedAuthority> authorities) {

		List<GrantedAuthority> authorityList = new ArrayList<GrantedAuthority>();
		authorityList.addAll(authorities);

		try {
			fillAuthorityList(authorityList, username);
		} catch (QuadrigaStorageException e) {
				e.printStackTrace();
		}

		UserDetails details = super.mapUserFromContext(ctx, username,
				authorityList);
		

		return details;
	}

	public void fillAuthorityList(List<GrantedAuthority> authorities,
			String username) throws QuadrigaStorageException {
		// Check the status of the user in the Quadriga DB
		IUser user = null;
		try
		{
		user = userManager.getUser(username);

		// add QuadrigaGrantedAuthorities with roles of user
		if (user.getQuadrigaRoles() != null) {
			for (IQuadrigaRole role : user.getQuadrigaRoles()) {
				authorities.add(new QuadrigaGrantedAuthority(role.getId()));
			}
		}
		}
		catch(CannotCreateTransactionException ex)
		{
			throw new QuadrigaStorageException();
		}
	}

}