public User importLDAPUserByScreenName(long companyId, String screenName) throws Exception {
long ldapServerId = PortalLDAPUtil.getLdapServerId(companyId, screenName, StringPool.BLANK);
SearchResult result =
(SearchResult)
PortalLDAPUtil.getUser(ldapServerId, companyId, screenName, StringPool.BLANK);
if (result == null) {
if (_log.isWarnEnabled()) {
_log.warn("No user was found in LDAP with screenName " + screenName);
}
return null;
}
LdapContext ldapContext = PortalLDAPUtil.getContext(ldapServerId, companyId);
String fullUserDN = PortalLDAPUtil.getNameInNamespace(ldapServerId, companyId, result);
Attributes attributes =
PortalLDAPUtil.getUserAttributes(ldapServerId, companyId, ldapContext, fullUserDN);
User user = importLDAPUser(ldapServerId, companyId, ldapContext, attributes, StringPool.BLANK);
ldapContext.close();
return user;
}
protected void importFromLDAPByUser(
long ldapServerId,
long companyId,
LdapContext ldapContext,
Properties userMappings,
Properties userExpandoMappings,
Properties contactMappings,
Properties contactExpandoMappings,
Properties groupMappings)
throws Exception {
byte[] cookie = new byte[0];
while (cookie != null) {
List<SearchResult> searchResults = new ArrayList<SearchResult>();
String userMappingsScreenName =
GetterUtil.getString(userMappings.getProperty("screenName")).toLowerCase();
cookie =
PortalLDAPUtil.getUsers(
ldapServerId,
companyId,
ldapContext,
cookie,
0,
new String[] {userMappingsScreenName},
searchResults);
for (SearchResult searchResult : searchResults) {
try {
Attributes userAttributes =
PortalLDAPUtil.getUserAttributes(
ldapServerId,
companyId,
ldapContext,
PortalLDAPUtil.getNameInNamespace(ldapServerId, companyId, searchResult));
User user =
importUser(
companyId,
userAttributes,
userMappings,
userExpandoMappings,
contactMappings,
contactExpandoMappings,
StringPool.BLANK);
importGroups(
ldapServerId,
companyId,
ldapContext,
userAttributes,
user,
userMappings,
groupMappings);
} catch (Exception e) {
_log.error("Unable to import user " + searchResult, e);
}
}
}
}
public User importLDAPUser(
long ldapServerId, long companyId, String emailAddress, String screenName) throws Exception {
LdapContext ldapContext = null;
NamingEnumeration<SearchResult> enu = null;
try {
String postfix = LDAPSettingsUtil.getPropertyPostfix(ldapServerId);
String baseDN = PrefsPropsUtil.getString(companyId, PropsKeys.LDAP_BASE_DN + postfix);
ldapContext = PortalLDAPUtil.getContext(ldapServerId, companyId);
if (ldapContext == null) {
throw new SystemException("Failed to bind to the LDAP server");
}
String filter =
PrefsPropsUtil.getString(companyId, PropsKeys.LDAP_AUTH_SEARCH_FILTER + postfix);
if (_log.isDebugEnabled()) {
_log.debug("Search filter before transformation " + filter);
}
filter =
StringUtil.replace(
filter,
new String[] {"@company_id@", "@email_address@", "@screen_name@"},
new String[] {String.valueOf(companyId), emailAddress, screenName});
if (_log.isDebugEnabled()) {
_log.debug("Search filter after transformation " + filter);
}
Properties userMappings = LDAPSettingsUtil.getUserMappings(ldapServerId, companyId);
String userMappingsScreenName =
GetterUtil.getString(userMappings.getProperty("screenName")).toLowerCase();
SearchControls searchControls =
new SearchControls(
SearchControls.SUBTREE_SCOPE,
1,
0,
new String[] {userMappingsScreenName},
false,
false);
enu = ldapContext.search(baseDN, filter, searchControls);
if (enu.hasMoreElements()) {
if (_log.isDebugEnabled()) {
_log.debug("Search filter returned at least one result");
}
Binding binding = enu.nextElement();
Attributes attributes =
PortalLDAPUtil.getUserAttributes(
ldapServerId,
companyId,
ldapContext,
PortalLDAPUtil.getNameInNamespace(ldapServerId, companyId, binding));
return importLDAPUser(ldapServerId, companyId, ldapContext, attributes, StringPool.BLANK);
} else {
return null;
}
} catch (Exception e) {
if (_log.isWarnEnabled()) {
_log.warn("Problem accessing LDAP server " + e.getMessage());
}
if (_log.isDebugEnabled()) {
_log.debug(e, e);
}
throw new SystemException("Problem accessing LDAP server " + e.getMessage());
} finally {
if (enu != null) {
enu.close();
}
if (ldapContext != null) {
ldapContext.close();
}
}
}
protected void importUsers(
long ldapServerId,
long companyId,
LdapContext ldapContext,
Properties userMappings,
Properties userExpandoMappings,
Properties contactMappings,
Properties contactExpandoMappings,
long userGroupId,
Attribute attribute)
throws Exception {
List<Long> newUserIds = new ArrayList<Long>(attribute.size());
for (int i = 0; i < attribute.size(); i++) {
String fullUserDN = (String) attribute.get(i);
Attributes userAttributes = null;
try {
userAttributes =
PortalLDAPUtil.getUserAttributes(ldapServerId, companyId, ldapContext, fullUserDN);
} catch (NameNotFoundException nnfe) {
_log.error("LDAP user not found with fullUserDN " + fullUserDN, nnfe);
continue;
}
try {
User user =
importUser(
companyId,
userAttributes,
userMappings,
userExpandoMappings,
contactMappings,
contactExpandoMappings,
StringPool.BLANK);
if (user != null) {
if (_log.isDebugEnabled()) {
_log.debug("Adding " + user.getUserId() + " to group " + userGroupId);
}
UserLocalServiceUtil.addUserGroupUsers(userGroupId, new long[] {user.getUserId()});
newUserIds.add(user.getUserId());
}
} catch (Exception e) {
_log.error("Unable to load user " + userAttributes, e);
}
}
List<User> userGroupUsers = UserLocalServiceUtil.getUserGroupUsers(userGroupId);
for (User user : userGroupUsers) {
if (!newUserIds.contains(user.getUserId())) {
UserLocalServiceUtil.deleteUserGroupUser(userGroupId, user.getUserId());
}
}
}
