using SecurityApi.Core.Models;

using SecurityApi.Core.Repositories;

using System;

using System.Collections.Concurrent;

using System.Collections.Generic;

using System.ComponentModel.Composition;

using System.DirectoryServices.AccountManagement;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

using AutoMapper;

using SecurityApi.EntityFramework;

using System.DirectoryServices;

​

namespace SecurityApi.Repositories

{

    [Export(typeof(IADUserRepository))]

    public class ADUserRepository : IADUserRepository

    {

        private Object etlLock = new Object();

​

        public async Task<List<Core.Models.ADUser>> GetADUsersFromAD(string domainName)

        {

            var domainId = await new DomainRepository().GetDomainId(domainName);

            var adUsers = new ConcurrentBag<Core.Models.ADUser>();

            var adGroups = new ConcurrentBag<Core.Models.ADGroup>();

​

            return await Task.Run(() =>

            {

                using (var context = new PrincipalContext(ContextType.Domain, domainName))

                {

                    // init vars

                    var up = new UserPrincipal(context);

                    var ps = new PrincipalSearcher(up);

                    Core.Models.ADUser adUser;

​

                    Parallel.ForEach(

                        ps.FindAll().ToList(), 

                        new ParallelOptions() { MaxDegreeOfParallelism = Environment.ProcessorCount }, 

                        result =>

                    {

                        if (!result.DistinguishedName.Contains("OU=Users")) return;

​

                        //set ad user properties

                        var userPrincipal = result as UserPrincipal;

                        adUser.Name = userPrincipal.Name;

                        adUser.FirstName = userPrincipal.GivenName;

                        adUser.LastName = userPrincipal.Surname;

​

                        lock (etlLock)

                        {

                            //get a directory entry representation for accessing special properties

                            var directoryEntry = result.GetUnderlyingObject() as DirectoryEntry;

​

                            //user principal name

                            adUser.UserPrincipalName = directoryEntry.Properties["UserPrincipalName"].Value?.ToString();

​

                            //employee id

                            adUser.EmployeeId = directoryEntry.Properties["EmployeeId"].Value?.ToString();

                        }

​

                        #endregion

​

                        adUsers.Add(adUser);

                    });

                }

​

                return adUsersList;

            });

        }

​

​

    }

}

​