io.jenkins.backend.jiraldapsyncer.LdapClient.java Source code

Introduction

Here is the source code for io.jenkins.backend.jiraldapsyncer.LdapClient.java

Source

/*
 * The MIT License
 *
 * Copyright (c) 2016, CloudBees, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 */
package io.jenkins.backend.jiraldapsyncer;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

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.AttributesMapper;
import org.springframework.ldap.core.LdapTemplate;

public class LdapClient {
    public static final String ROLE = "ldapClient";

    private static final Log LOG = LogFactory.getLog(LdapClient.class);

    private LdapTemplate ldapTemplate;

    private class EmployeeAttributesMapper implements AttributesMapper {
        public Object mapFromAttributes(Attributes attrs) throws NamingException {
            RemoteUser user = new RemoteUser();
            user.setName((String) getOne(attrs, "cn"));

            String givenName = (String) getOne(attrs, "givenName");
            String sn = (String) getOne(attrs, "sn");

            if (sn.equals("-"))
                user.setFullname(givenName);
            else
                user.setFullname(givenName + ' ' + sn);

            user.setEmail((String) getOne(attrs, "mail"));

            return user;
        }

        private Object getOne(Attributes attrs, String attributeName) throws NamingException {
            Attribute a = attrs.get(attributeName);
            return a != null ? a.get() : null;
        }
    }

    public RemoteUser getUser(String cn) {
        List results = ldapTemplate.search("", "(cn=" + cn + ")", new EmployeeAttributesMapper());
        return results.isEmpty() ? null : (RemoteUser) results.get(0);
    }

    public Map<String, RemoteUser> getAllUsers() {
        Map<String, RemoteUser> ldapUsers = new HashMap<String, RemoteUser>();

        // retrieve users from LDAP
        List results = ldapTemplate.search("", "(objectclass=person)", new EmployeeAttributesMapper());

        // put them into hashmap
        for (Object result : results) {
            RemoteUser user = (RemoteUser) result;
            ldapUsers.put(user.getName(), user);
        }

        return ldapUsers;
    }

    public static void main(String[] args) {
        ServiceLocator locator = new ServiceLocator();
        LdapClient ldapClient = (LdapClient) locator.lookupService(LdapClient.ROLE);
        Map<String, RemoteUser> ldapUsers = ldapClient.getAllUsers();
        LOG.info("LDAP directory contains " + ldapUsers.size() + " users");
    }

    public LdapTemplate getLdapTemplate() {
        return ldapTemplate;
    }

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

}