using System; using System.Threading.Tasks; using Bit.Core.Exceptions; using Bit.Core.Models.Table; using Bit.Core.Repositories; using System.Collections.Generic; using Bit.Core.Models.Data; namespace Bit.Core.Services { public class GroupService : IGroupService { private readonly IEventService _eventService; private readonly IOrganizationRepository _organizationRepository; private readonly IOrganizationUserRepository _organizationUserRepository; private readonly IGroupRepository _groupRepository; public GroupService( IEventService eventService, IOrganizationRepository organizationRepository, IOrganizationUserRepository organizationUserRepository, IGroupRepository groupRepository) { _eventService = eventService; _organizationRepository = organizationRepository; _organizationUserRepository = organizationUserRepository; _groupRepository = groupRepository; } public async Task SaveAsync(Group group, IEnumerable collections = null) { var org = await _organizationRepository.GetByIdAsync(group.OrganizationId); if(org == null) { throw new BadRequestException("Organization not found"); } if(!org.UseGroups) { throw new BadRequestException("This organization cannot use groups."); } if(group.Id == default(Guid)) { group.CreationDate = group.RevisionDate = DateTime.UtcNow; if(collections == null) { await _groupRepository.CreateAsync(group); } else { await _groupRepository.CreateAsync(group, collections); } await _eventService.LogGroupEventAsync(group, Enums.EventType.Group_Created); } else { group.RevisionDate = DateTime.UtcNow; await _groupRepository.ReplaceAsync(group, collections ?? new List()); await _eventService.LogGroupEventAsync(group, Enums.EventType.Group_Updated); } } public async Task DeleteAsync(Group group) { await _groupRepository.DeleteAsync(group); await _eventService.LogGroupEventAsync(group, Enums.EventType.Group_Deleted); } public async Task DeleteUserAsync(Group group, Guid organizationUserId) { var orgUser = await _organizationUserRepository.GetByIdAsync(organizationUserId); if(orgUser == null || orgUser.OrganizationId != group.OrganizationId) { throw new NotFoundException(); } await _groupRepository.DeleteUserAsync(group.Id, organizationUserId); await _eventService.LogOrganizationUserEventAsync(orgUser, Enums.EventType.OrganizationUser_UpdatedGroups); } } }