edu.uiowa.icts.spring.ldap.LDAPHome Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-utils Show documentation
ICTS Utilities for Spring Hibernate Projects
The newest version!
package edu.uiowa.icts.spring.ldap;

/*
 * #%L
 * spring-utils
 * %%
 * Copyright (C) 2010 - 2015 University of Iowa Institute for Clinical and Translational Science (ICTS)
 * %%
 * 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.
 * #L%
 */

import java.util.List;

import javax.naming.NamingException;
import javax.naming.directory.Attributes;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.ldap.core.AttributesMapper;
import org.springframework.ldap.core.ContextMapper;
import org.springframework.ldap.core.DirContextAdapter;
import org.springframework.ldap.core.DirContextOperations;
import org.springframework.ldap.core.DistinguishedName;
import org.springframework.ldap.core.LdapTemplate;
import org.springframework.stereotype.Repository;

@Repository
@SuppressWarnings( { "unchecked", "rawtypes" } )
public class LDAPHome implements LDAPService {

	@Autowired
	private LdapTemplate ldapTemplate;

	public void setLdapTemplate( LdapTemplate ldapTemplate ) {
		this.ldapTemplate = ldapTemplate;
	}

	public List findByUsername( String username ) {
		return ldapTemplate.search( "", "(cn=" + username + "*)", getContextMapper() );
	}

	public List getByUsername( String username ) {
		return ldapTemplate.search( "", "(cn=" + username + ")", getContextMapper() );
	}

	public List searchFirstName( String firstName ) {
		return ldapTemplate.search( "", "(givenName=" + firstName + "*)", getContextMapper() );
	}

	public List searchLastName( String lastName ) {
		return ldapTemplate.search( "", "(sn=" + lastName + "*)", getContextMapper() );
	}

	public List findByFullname( String firstname, String lastname ) {
		return ldapTemplate.search( "", "(&(givenName=" + firstname + "*)(sn=" + lastname + "*))", getContextMapper() );
	}

	public List search( String value ) {
		List list = ldapTemplate.search( "", "(cn=" + value + "*)",
			new AttributesMapper() {
				public Object mapFromAttributes( Attributes attrs )
					throws NamingException {
					return attrs.get( "DN" ).get();
				}
			} );
		return list;
	}

	public List getAllPersonNames() {
		List list = ldapTemplate.search( "", "(cn=*)", new AttributesMapper() {
			public Object mapFromAttributes( Attributes attrs )
				throws NamingException {
				return attrs.get( "DN" ).get();
			}
		} );
		return list;
	}

	protected ContextMapper getContextMapper() {
		return new PersonContextMapper();
	}

	private static class PersonContextMapper implements ContextMapper {
		public Object mapFromContext( Object ctx ) {

			DirContextAdapter context = (DirContextAdapter) ctx;
			DirContextOperations operations = (DirContextOperations) ctx;

			LDAPPerson ldapPerson = new LDAPPerson();
			ldapPerson.setFullName( context.getStringAttribute( "displayName" ) );
			ldapPerson.setLastName( context.getStringAttribute( "sn" ) );
			ldapPerson.setTitle( context.getStringAttribute( "title" ) );
			ldapPerson.setDepartment( context.getStringAttribute( "department" ) );
			ldapPerson.setCollege( context.getStringAttribute( "company" ) );
			ldapPerson.setCompany( context.getStringAttribute( "company" ) );
			ldapPerson.setTelephoneNumber( context.getStringAttribute( "telephoneNumber" ) );
			ldapPerson.setMail( context.getStringAttribute( "mail" ) );
			ldapPerson.setGivenName( context.getStringAttribute( "givenName" ) );
			ldapPerson.setUsername( context.getStringAttribute( "cn" ) );

			String[] groupArray = operations.getStringAttributes( "memberOf" );
			if ( groupArray != null && groupArray.length > 0 ) {
				String[] roleNames = new String[ groupArray.length ];
				for ( int g = 0; g < groupArray.length; g++ ) {
					roleNames[g] = new DistinguishedName( groupArray[g] ).removeLast().getValue();
				}
				ldapPerson.setRoleNames( roleNames );
			}

			// p.setDn(context.getStringAttribute("distinguishedName"));
			// p.setDescription(context.getStringAttribute("description"));
			return ldapPerson;
		}
	}
}