-
Notifications
You must be signed in to change notification settings - Fork 88
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implements #61 Simple Automatic Backup (Remote and/or Local)
- Loading branch information
Showing
25 changed files
with
712 additions
and
99 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,160 @@ | ||
| using System; | ||
| using System.IO; | ||
| using System.Linq; | ||
| using System.Text.RegularExpressions; | ||
| using System.Threading.Tasks; | ||
| using KeeAnywhere.Configuration; | ||
| using KeeAnywhere.StorageProviders; | ||
|
|
||
| namespace KeeAnywhere.Backup | ||
| { | ||
| public class BackupProvider : FileOperationsProxyProvider | ||
| { | ||
| private readonly StorageUri _requestedUri; | ||
| private readonly ConfigurationService _configService; | ||
|
|
||
| public BackupProvider(IStorageProvider baseProvider, StorageUri requestedUri, ConfigurationService configService) | ||
| :base(baseProvider) | ||
| { | ||
| if (requestedUri == null) throw new ArgumentNullException("requestedUri"); | ||
| if (configService == null) throw new ArgumentNullException("configService"); | ||
| _requestedUri = requestedUri; | ||
| _configService = configService; | ||
| } | ||
|
|
||
| public static IStorageProviderFileOperations WrapInBackupProvider(IStorageProvider provider, StorageUri requestedUri, ConfigurationService configService) | ||
| { | ||
| var backupProvider = new BackupProvider(provider, requestedUri, configService); | ||
|
|
||
| return backupProvider; | ||
| } | ||
|
|
||
| public override async Task Save(Stream stream, string path) | ||
| { | ||
| if (_configService.PluginConfiguration.IsBackupToRemoteEnabled) | ||
| { | ||
| await BackupRemote(path); | ||
| await RotateRemote(path); | ||
| } | ||
|
|
||
| var tempStream = new MemoryStream(); | ||
| await stream.CopyToAsync(tempStream); | ||
| tempStream.Position = 0; | ||
|
|
||
| await this.BaseProvider.Save(tempStream, path); | ||
|
|
||
| if (_configService.PluginConfiguration.IsBackupToLocalEnabled) | ||
| { | ||
| BackupLocal(stream, path); | ||
| RotateLocal(path); | ||
| } | ||
| } | ||
|
|
||
| private async Task RotateRemote(string path) | ||
| { | ||
| var folder = CloudPath.GetDirectoryName(path); | ||
| var items = await this.BaseProvider.GetChildrenByParentPath(folder); | ||
|
|
||
| var pattern = "^" + Regex.Escape(GetBackupFilenamePattern(path)) | ||
| .Replace("\\*", ".*") | ||
| .Replace("\\?", ".") + "$"; | ||
|
|
||
| var regex = new Regex(pattern); | ||
| var files = items.Where(item => regex.IsMatch(item.Name)).Select(item => item.Name).ToArray(); | ||
| Array.Sort(files); | ||
|
|
||
| for (var i = 0; i < files.Length - _configService.PluginConfiguration.BackupCopies; i++) | ||
| { | ||
| await this.BaseProvider.Delete(CloudPath.Combine(folder, files[i])); | ||
| } | ||
| } | ||
|
|
||
| private async Task BackupRemote(string path) | ||
| { | ||
| var backupFilename = GetBackupFilename(CloudPath.GetFileName(path)); | ||
| var backupFolder = CloudPath.GetDirectoryName(path); | ||
| var backupPath = CloudPath.Combine(backupFolder, backupFilename); | ||
|
|
||
| try | ||
| { | ||
| await this.BaseProvider.Copy(path, backupPath); | ||
| } | ||
| catch (Exception) | ||
| { | ||
| // Try to copy file to backup: it fails if file is saved for the first time | ||
| } | ||
| } | ||
|
|
||
| private void RotateLocal(string path) | ||
| { | ||
| var folder = GetBackupFolder(); | ||
| var filenameFilter = GetBackupFilenamePattern(path); | ||
|
|
||
| var files = Directory.GetFiles(folder, filenameFilter, SearchOption.TopDirectoryOnly); | ||
| Array.Sort(files); | ||
|
|
||
| for (var i = 0; i < files.Length - _configService.PluginConfiguration.BackupCopies; i++) | ||
| { | ||
| File.Delete(files[i]); | ||
| } | ||
| } | ||
|
|
||
| private void BackupLocal(Stream stream, string path) | ||
| { | ||
| stream.Position = 0; | ||
|
|
||
| var folder = GetBackupFolder(); | ||
|
|
||
| var localFilename = GetLocalFilename(path); | ||
| var localPath = Path.Combine(folder, localFilename); | ||
|
|
||
| if (File.Exists(localPath)) | ||
| { | ||
| var backupFilename = GetBackupFilename(GetLocalFilename(path)); | ||
| var backupPath = Path.Combine(folder, backupFilename); | ||
| File.Move(localPath, backupPath); | ||
| } | ||
|
|
||
| using (var backupStream = File.OpenWrite(localPath)) | ||
| stream.CopyTo(backupStream); | ||
| } | ||
|
|
||
| public string GetBackupFolder() | ||
| { | ||
| var accountFolder = string.Format("{0} - {1}", _requestedUri.Scheme, _requestedUri.GetAccountName()); | ||
| var path = Path.Combine(_configService.PluginConfiguration.BackupToLocalFolder, accountFolder); | ||
|
|
||
| if (!Directory.Exists(path)) | ||
| Directory.CreateDirectory(path); | ||
|
|
||
| return path; | ||
| } | ||
|
|
||
| public string GetLocalFilename(string path) | ||
| { | ||
| var filename = CloudPath.MaskInvalidFileNameChars(CloudPath.GetFileName(path)); | ||
|
|
||
| return filename; | ||
| } | ||
|
|
||
| public string GetBackupFilename(string filename) | ||
| { | ||
| var baseFilename = Path.GetFileNameWithoutExtension(filename); | ||
|
|
||
| filename = string.Format("{0}_Backup_{1:yyyy-MM-dd-HH-mm-ss}.kdbx", baseFilename, DateTime.Now); | ||
|
|
||
| return filename; | ||
| } | ||
|
|
||
| public string GetBackupFilenamePattern(string path) | ||
| { | ||
| var filename = GetLocalFilename(path); | ||
| var baseFilename = Path.GetFileNameWithoutExtension(filename); | ||
|
|
||
| filename = string.Format("{0}_Backup_????-??-??-??-??-??.kdbx", baseFilename); | ||
|
|
||
| return filename; | ||
| } | ||
|
|
||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,16 @@ | ||
| using System.IO; | ||
|
|
||
| namespace KeeAnywhere.Offline | ||
| { | ||
| public struct CacheFileInfo | ||
| { | ||
| public CacheFileInfo(Stream stream, string hash) | ||
| { | ||
| Stream = stream; | ||
| Hash = hash; | ||
| } | ||
|
|
||
| public string Hash { get; set; } | ||
| public Stream Stream { get; set; } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.