Z-Archive » Maybank Projects » M2U UPASS Project

						if(profile instanceof Im2uUser){

							//cater creation of additional objectClass

							MinimalUPassControllerV2.getMaybankLdapDAO().updateUser(username, attrMap, new String[]{MaybankLdapConstant.OBJECT_CLASS_IM2U});

						}else if(profile instanceof StockUser){

							//cater creation of additional objectClass

							MinimalUPassControllerV2.getMaybankLdapDAO().updateUser(username, attrMap, new String[]{MaybankLdapConstant.OBJECT_CLASS_ONLINE_STOCK1, MaybankLdapConstant.OBJECT_CLASS_ONLINE_STOCK2});

						}else if(profile instanceof M2uUser){

							//cater creation of additional objectClass

							MinimalUPassControllerV2.getMaybankLdapDAO().updateUser(username, attrMap, new String[]{MaybankLdapConstant.OBJECT_CLASS_M2U});

						}else{

							MinimalUPassControllerV2.getMaybankLdapDAO().updateUser(username, attrMap);

						}

package my.com.upass.spring.ldap;

import java.util.HashMap;

import java.util.Map;

public class MaybankLdap {

	private String[] objectClasses;

	private Map attrMap;

	public MaybankLdap() {

		attrMap = new HashMap();

	}

	public MaybankLdap(Map attrMap) {

		this.attrMap = attrMap;

	}

	public String[] getObjectClasses() {

		return objectClasses;

	}

	public void setObjectClasses(String[] objectClasses) {

		this.objectClasses = objectClasses;

	}

	public Map getAttrMap() {

		return attrMap;

	}

	public void setAttrMap(Map attrMap) {

		this.attrMap = attrMap;

	}

}

package my.com.upass.spring.ldap;

import java.util.HashMap;

import java.util.Map;

import javax.naming.NamingEnumeration;

import javax.naming.NamingException;

import javax.naming.directory.Attribute;

import javax.naming.directory.Attributes;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.ldap.core.DirContextOperations;

import org.springframework.ldap.core.support.AbstractContextMapper;

public class MaybankLdapAttributeMapper  extends AbstractContextMapper {

	private Log logger = LogFactory.getLog(MaybankLdapAttributeMapper.class);

	public Object doMapFromContext(DirContextOperations context) {

		MaybankLdap maybankLdap = new MaybankLdap();

		maybankLdap.setObjectClasses(context.getStringAttributes("objectClass"));

		try {

			Map attrMap = new HashMap();

			Attributes attributes = context.getAttributes();

			NamingEnumeration e = attributes.getAll();

			while(e.hasMore()){

				Attribute attr = (Attribute)e.next();

				if(!("objectClass".equals(attr.getID()) || "userPassword".equals(attr.getID()))){

					attrMap.put(attr.getID(), (String)attr.get(0));

				}

			}

			maybankLdap.setAttrMap(attrMap);

		} catch (NamingException e) {

			logger.warn("Error parsing LDAP attributes.");

		}

		return maybankLdap;

	}

}

	public void updateUser(String mbbuserid, Map/*<String, String>*/ attributesMap, String[] objectClasses) throws UPassException;

	public void rebindUser(String mbbuserid, String password, String[] objectClasses, Map attributesMap) throws UPassException;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import javax.naming.directory.DirContext;

import javax.naming.directory.ModificationItem;

	public void rebindUser(String mbbuserid, String password, String[] objectClasses, Map/*<String, String>*/ attributesMap) throws UPassException {

		Name dn = buildDn(mbbuserid);

		//retrieve profile from LDAP

		MaybankLdap maybankLdap = (MaybankLdap) ldapTemplate.lookup(dn, new MaybankLdapAttributeMapper());

		//existing objectClass 

		Set objectClassSet = new HashSet(Arrays.asList(maybankLdap.getObjectClasses()));

		//add new or replace the objectClass

		if(objectClasses != null && objectClasses.length > 0){

			for(int i=0; i<objectClasses.length; i++){

				objectClassSet.add(objectClasses[i]);

			}

		}

		//existing attribute map

		Map existingAttr = maybankLdap.getAttrMap();

		//add new or replace the attributes

		existingAttr.putAll(attributesMap);

		//construct LDAP entry

		Attributes attrs = new BasicAttributes();

		BasicAttribute basicAttribute = new BasicAttribute("objectclass");

		//add objectClass in LDAP Attribute

		Iterator itr = objectClassSet.iterator();

		while(itr.hasNext()){

			basicAttribute.add((String)itr.next());

		}

		attrs.put(basicAttribute);

		//add attributes in LDAP Attribute

		itr = existingAttr.entrySet().iterator();

	    while (itr.hasNext()) {

	        Map.Entry pairs = (Map.Entry)itr.next();

	        attrs.put((String)pairs.getKey(), (String)pairs.getValue());

	    }

		try {

			ldapTemplate.rebind(dn, null, attrs);

		} catch (NamingException e) {

			logger.warn("LDAP Error code:"+LdapStatusCodeParser.getCode(e.getExplanation()) + " - " + e.getExplanation());

			throw new UPassException(MinimalConstants.ERR_LDAP);

		}

	}

	public void updateUser(String mbbuserid, Map attributesMap, String[] objectClasses) throws UPassException {

		Name dn = buildDn(mbbuserid);

		//retrieve profile from LDAP

		MaybankLdap maybankLdap = (MaybankLdap) ldapTemplate.lookup(dn, new MaybankLdapAttributeMapper());

		//existing objectClass 

		Set objectClassSet = new HashSet(Arrays.asList(maybankLdap.getObjectClasses()));

		List itemList = new ArrayList();

		if(!objectClassSet.contains(objectClasses[0])){

			for(int i=0; i<objectClasses.length; i++){

				itemList.add(new ModificationItem(DirContext.ADD_ATTRIBUTE, new BasicAttribute("objectClass", objectClasses[i])));

			}

		}

		//DirContext.REPLACE_ATTRIBUTE will create attribute if it does not exist

		if(objectClasses[0].equals(MaybankLdapConstant.OBJECT_CLASS_IM2U)){

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("wsFlag", (String)attributesMap.get("wsFlag"))));

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("wsIdentCode", (String)attributesMap.get("wsIdentCode"))));

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("wsRegTimeStamp", (String)attributesMap.get("wsRegTimeStamp"))));

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("wsMySgID", (String)attributesMap.get("wsMySgID"))));

		}else if(objectClasses[0].equals(MaybankLdapConstant.OBJECT_CLASS_ONLINE_STOCK1) || objectClasses[0].equals(MaybankLdapConstant.OBJECT_CLASS_ONLINE_STOCK2)){

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("IDNo", (String)attributesMap.get("IDNo"))));

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("CustType", (String)attributesMap.get("CustType"))));

		}else if(objectClasses[0].equals(MaybankLdapConstant.OBJECT_CLASS_M2U)){

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("ibpan", (String)attributesMap.get("ibpan"))));

			itemList.add(new ModificationItem(DirContext.REPLACE_ATTRIBUTE, new BasicAttribute("ibpan2", (String)attributesMap.get("ibpan2"))));

		}

		try {

			if(itemList != null && !itemList.isEmpty()){

				if(logger.isDebugEnabled()){

					try {

						Iterator itr = itemList.iterator();

						while(itr.hasNext()){

							ModificationItem debugItem = (ModificationItem)itr.next();

							logger.debug("LDAP Operation:"+debugItem.getModificationOp()+" "+debugItem.getAttribute().getID() + " = "+(String)debugItem.getAttribute().get());

						}

					} catch (Exception e) {}

				}

				ldapTemplate.modifyAttributes(dn, (ModificationItem[])itemList.toArray(new ModificationItem[itemList.size()]));

			}

		} catch (NamingException e) {

			logger.warn("LDAP Error code:"+LdapStatusCodeParser.getCode(e.getExplanation()) + " - " + e.getExplanation());

			throw new UPassException(MinimalConstants.ERR_LDAP);

		}

	}