using System; using System.Linq; using System.Threading.Tasks; using Bit.App.Abstractions; using Bit.App.Models.Data; using Plugin.Settings.Abstractions; using Bit.App.Models.Api; using System.Collections.Generic; using Xamarin.Forms; namespace Bit.App.Services { public class SyncService : ISyncService { private const string LastSyncKey = "lastSync"; private readonly ICipherApiRepository _cipherApiRepository; private readonly IFolderApiRepository _folderApiRepository; private readonly ISiteApiRepository _siteApiRepository; private readonly IFolderRepository _folderRepository; private readonly ISiteRepository _siteRepository; private readonly IAuthService _authService; private readonly ISettings _settings; public SyncService( ICipherApiRepository cipherApiRepository, IFolderApiRepository folderApiRepository, ISiteApiRepository siteApiRepository, IFolderRepository folderRepository, ISiteRepository siteRepository, IAuthService authService, ISettings settings) { _cipherApiRepository = cipherApiRepository; _folderApiRepository = folderApiRepository; _siteApiRepository = siteApiRepository; _folderRepository = folderRepository; _siteRepository = siteRepository; _authService = authService; _settings = settings; } public bool SyncInProgress { get; private set; } public async Task SyncAsync(string id) { if(!_authService.IsAuthenticated) { return false; } SyncStarted(); var cipher = await _cipherApiRepository.GetByIdAsync(id); if(!cipher.Succeeded) { SyncCompleted(false); if(cipher.StatusCode == System.Net.HttpStatusCode.Forbidden || cipher.StatusCode == System.Net.HttpStatusCode.Unauthorized) { MessagingCenter.Send(Application.Current, "Logout", (string)null); } return false; } switch(cipher.Result.Type) { case Enums.CipherType.Folder: var folderData = new FolderData(cipher.Result, _authService.UserId); var existingLocalFolder = _folderRepository.GetByIdAsync(id); if(existingLocalFolder == null) { await _folderRepository.InsertAsync(folderData); } else { await _folderRepository.UpdateAsync(folderData); } break; case Enums.CipherType.Site: var siteData = new SiteData(cipher.Result, _authService.UserId); var existingLocalSite = _siteRepository.GetByIdAsync(id); if(existingLocalSite == null) { await _siteRepository.InsertAsync(siteData); } else { await _siteRepository.UpdateAsync(siteData); } break; default: SyncCompleted(false); return false; } SyncCompleted(true); return true; } public async Task SyncDeleteFolderAsync(string id) { if(!_authService.IsAuthenticated) { return false; } SyncStarted(); await _folderRepository.DeleteAsync(id); SyncCompleted(true); return true; } public async Task SyncDeleteSiteAsync(string id) { if(!_authService.IsAuthenticated) { return false; } SyncStarted(); await _siteRepository.DeleteAsync(id); SyncCompleted(true); return true; } public async Task FullSyncAsync() { if(!_authService.IsAuthenticated) { return false; } SyncStarted(); var now = DateTime.UtcNow; var ciphers = await _cipherApiRepository.GetAsync(); if(!ciphers.Succeeded) { SyncCompleted(false); if(ciphers.StatusCode == System.Net.HttpStatusCode.Forbidden || ciphers.StatusCode == System.Net.HttpStatusCode.Unauthorized) { MessagingCenter.Send(Application.Current, "Logout", (string)null); } return false; } var siteTask = SyncSitesAsync(ciphers.Result.Data.Where(c => c.Type == Enums.CipherType.Site), true); var folderTask = SyncFoldersAsync(ciphers.Result.Data.Where(c => c.Type == Enums.CipherType.Folder), true); await Task.WhenAll(siteTask, folderTask); if(folderTask.Exception != null || siteTask.Exception != null) { SyncCompleted(false); return false; } _settings.AddOrUpdateValue(LastSyncKey, now); SyncCompleted(true); return true; } public async Task IncrementalSyncAsync(TimeSpan syncThreshold) { DateTime? lastSync = _settings.GetValueOrDefault(LastSyncKey); if(lastSync != null && DateTime.UtcNow - lastSync.Value < syncThreshold) { return false; } return await IncrementalSyncAsync(); } public async Task IncrementalSyncAsync() { if(!_authService.IsAuthenticated) { return false; } var now = DateTime.UtcNow; DateTime? lastSync = _settings.GetValueOrDefault(LastSyncKey); if(lastSync == null) { return await FullSyncAsync(); } SyncStarted(); var ciphers = await _cipherApiRepository.GetByRevisionDateWithHistoryAsync(lastSync.Value); if(!ciphers.Succeeded) { SyncCompleted(false); if(ciphers.StatusCode == System.Net.HttpStatusCode.Forbidden || ciphers.StatusCode == System.Net.HttpStatusCode.Unauthorized) { MessagingCenter.Send(Application.Current, "Logout", (string)null); } return false; } var siteTask = SyncSitesAsync(ciphers.Result.Revised.Where(c => c.Type == Enums.CipherType.Site), false); var folderTask = SyncFoldersAsync(ciphers.Result.Revised.Where(c => c.Type == Enums.CipherType.Folder), false); var deleteTask = DeleteCiphersAsync(ciphers.Result.Deleted); await Task.WhenAll(siteTask, folderTask, deleteTask); if(folderTask.Exception != null || siteTask.Exception != null || deleteTask.Exception != null) { SyncCompleted(false); return false; } _settings.AddOrUpdateValue(LastSyncKey, now); SyncCompleted(true); return true; } private async Task SyncFoldersAsync(IEnumerable serverFolders, bool deleteMissing) { var localFolders = await _folderRepository.GetAllByUserIdAsync(_authService.UserId); foreach(var serverFolder in serverFolders) { var existingLocalFolder = localFolders.SingleOrDefault(f => f.Id == serverFolder.Id); if(existingLocalFolder == null) { var data = new FolderData(serverFolder, _authService.UserId); await _folderRepository.InsertAsync(data); } else if(existingLocalFolder.RevisionDateTime != serverFolder.RevisionDate) { var data = new FolderData(serverFolder, _authService.UserId); await _folderRepository.UpdateAsync(data); } } if(!deleteMissing) { return; } foreach(var folder in localFolders.Where(localFolder => !serverFolders.Any(serverFolder => serverFolder.Id == localFolder.Id))) { await _folderRepository.DeleteAsync(folder.Id); } } private async Task SyncSitesAsync(IEnumerable serverSites, bool deleteMissing) { var localSites = await _siteRepository.GetAllByUserIdAsync(_authService.UserId); foreach(var serverSite in serverSites) { var existingLocalSite = localSites.SingleOrDefault(s => s.Id == serverSite.Id); if(existingLocalSite == null) { var data = new SiteData(serverSite, _authService.UserId); await _siteRepository.InsertAsync(data); } else if(existingLocalSite.RevisionDateTime != serverSite.RevisionDate) { var data = new SiteData(serverSite, _authService.UserId); await _siteRepository.UpdateAsync(data); } } if(!deleteMissing) { return; } foreach(var site in localSites.Where(localSite => !serverSites.Any(serverSite => serverSite.Id == localSite.Id))) { await _siteRepository.DeleteAsync(site.Id); } } private async Task DeleteCiphersAsync(IEnumerable cipherIds) { var tasks = new List(); foreach(var cipherId in cipherIds) { tasks.Add(_siteRepository.DeleteAsync(cipherId)); tasks.Add(_folderRepository.DeleteAsync(cipherId)); } await Task.WhenAll(tasks); } private void SyncStarted() { SyncInProgress = true; MessagingCenter.Send(Application.Current, "SyncStarted"); } private void SyncCompleted(bool successfully) { SyncInProgress = false; MessagingCenter.Send(Application.Current, "SyncCompleted", successfully); } } }