using System; using System.Linq; using System.Collections.Generic; using System.Threading.Tasks; using Bit.Core.Domains; using Bit.Core.Repositories; namespace Bit.Core.Services { public class CipherService : ICipherService { private readonly ICipherRepository _cipherRepository; private readonly IPushService _pushService; public CipherService( ICipherRepository cipherRepository, IPushService pushService) { _cipherRepository = cipherRepository; _pushService = pushService; } public async Task SaveAsync(Cipher cipher) { if(cipher.Id == default(Guid)) { await _cipherRepository.CreateAsync(cipher); // push await _pushService.PushSyncCipherCreateAsync(cipher); } else { cipher.RevisionDate = DateTime.UtcNow; await _cipherRepository.ReplaceAsync(cipher); // push await _pushService.PushSyncCipherUpdateAsync(cipher); } } public async Task DeleteAsync(Cipher cipher) { await _cipherRepository.DeleteAsync(cipher); // push await _pushService.PushSyncCipherDeleteAsync(cipher); } public async Task ImportCiphersAsync( List folders, List ciphers, IEnumerable> folderRelationships) { // create all the folders var folderTasks = new List(); foreach(var folder in folders) { folderTasks.Add(_cipherRepository.CreateAsync(folder)); } await Task.WhenAll(folderTasks); // associate the newly created folders to the ciphers foreach(var relationship in folderRelationships) { var cipher = ciphers.ElementAtOrDefault(relationship.Key); var folder = folders.ElementAtOrDefault(relationship.Value); if(cipher == null || folder == null) { continue; } cipher.FolderId = folder.Id; } // create all the ciphers await _cipherRepository.CreateAsync(ciphers); // push var userId = folders.FirstOrDefault()?.UserId ?? ciphers.FirstOrDefault()?.UserId; if(userId.HasValue) { await _pushService.PushSyncCiphersAsync(userId.Value); } } } }