#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: UserHelper.cs // Created By: Jamie Rees // // 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. // ************************************************************************/ #endregion using System.Collections.Generic; using System.Linq; using Ombi.Core.Models; using Ombi.Helpers; using Ombi.Helpers.Permissions; using Ombi.Store.Repository; namespace Ombi.Core.Users { public class UserHelper : IUserHelper { public UserHelper(IUserRepository userRepository, IPlexUserRepository plexUsers, ISecurityExtensions security) { LocalUserRepository = userRepository; PlexUserRepository = plexUsers; Security = security; } private IUserRepository LocalUserRepository { get; } private IPlexUserRepository PlexUserRepository { get; } private ISecurityExtensions Security { get; } public IEnumerable GetUsers() { var model = new List(); var localUsers = LocalUserRepository.GetAll(); var plexUsers = PlexUserRepository.GetAll(); foreach (var user in localUsers) { var props = ByteConverterHelper.ReturnObject(user.UserProperties); model.Add(new UserHelperModel { Type = UserType.LocalUser, Username = user.UserName, UserAlias = props.UserAlias, EmailAddress = props.EmailAddress, Permissions = (Permissions)user.Permissions }); } model.AddRange(plexUsers.Select(user => new UserHelperModel { Type = UserType.LocalUser, Username = user.Username, UserAlias = user.UserAlias, EmailAddress = user.EmailAddress, Permissions = (Permissions)user.Permissions })); return model; } public IEnumerable GetUsersWithPermission(Permissions permission) { var model = new List(); var localUsers = LocalUserRepository.GetAll().ToList(); var plexUsers = PlexUserRepository.GetAll().ToList(); var filteredLocal = localUsers.Where(x => ((Permissions)x.Permissions).HasFlag(permission)); var filteredPlex = plexUsers.Where(x => ((Permissions)x.Permissions).HasFlag(permission)); foreach (var user in filteredLocal) { var props = ByteConverterHelper.ReturnObject(user.UserProperties); model.Add(new UserHelperModel { Type = UserType.LocalUser, Username = user.UserName, UserAlias = props.UserAlias, EmailAddress = props.EmailAddress, Permissions = (Permissions)user.Permissions, Features = (Features)user.Features }); } model.AddRange(filteredPlex.Select(user => new UserHelperModel { Type = UserType.LocalUser, Username = user.Username, UserAlias = user.UserAlias, EmailAddress = user.EmailAddress, Permissions = (Permissions)user.Permissions, Features = (Features)user.Features })); return model; } public IEnumerable GetUsersWithFeature(Features features) { var model = new List(); var localUsers = LocalUserRepository.GetAll().ToList(); var plexUsers = PlexUserRepository.GetAll().ToList(); var filteredLocal = localUsers.Where(x => ((Features)x.Features).HasFlag(features)); var filteredPlex = plexUsers.Where(x => ((Features)x.Features).HasFlag(features)); foreach (var user in filteredLocal) { var props = ByteConverterHelper.ReturnObject(user.UserProperties); model.Add(new UserHelperModel { Type = UserType.LocalUser, Username = user.UserName, UserAlias = props.UserAlias, EmailAddress = props.EmailAddress, Permissions = (Permissions)user.Permissions, Features = (Features)user.Features }); } model.AddRange(filteredPlex.Select(user => new UserHelperModel { Type = UserType.LocalUser, Username = user.Username, UserAlias = user.UserAlias, EmailAddress = user.EmailAddress, Permissions = (Permissions)user.Permissions, Features = (Features)user.Features })); return model; } } }