Horizon – Diff between revs 3 and 5

Subversion Repositories:

?pathlinks?

Rev 3 – Rev 6 – Go to most recent revision – Details – Blame – Last modification – View Log

 using System.Threading.Tasks;
 using System.Windows.Forms;
 using Horizon.Database;
 using Horizon.Utilities;
 using Microsoft.WindowsAPICodePack.Dialogs;
+using Org.BouncyCastle.Crypto;
 using Serilog;
 namespace Horizon.Snapshots
+{
         #endregion
         #region Constructors, Destructors and Finalizers
-        public SnapshotManagerForm()
+        private SnapshotManagerForm()
+        {
+        {
             toolStripProgressBar1.Minimum = 0;
             toolStripProgressBar1.Maximum = (int)await _snapshotDatabase.CountSnapshots(_cancellationToken);
-            var snapshotQueue = new ConcurrentQueue<Snapshot>();
-            var snapshotsQueuedTaskCompletionSource = new TaskCompletionSource<object>();
+            var snapshotQueue = new ConcurrentQueue<Snapshot>();
-            async void IdleHandler(object idleHandlerSender, EventArgs idleHandlerArgs)
-            {
-                await snapshotsQueuedTaskCompletionSource.Task;
+            void IdleHandler(object idleHandlerSender, EventArgs idleHandlerArgs)
+            {
                 try
+                {
-                    if (!snapshotQueue.TryDequeue(out var snapshot))
+                    if (snapshotQueue.IsEmpty)
+                    {
                         Application.Idle -= IdleHandler;
+                        dataGridView1.Sort(dataGridView1.Columns["TimeColumn"], ListSortDirection.Descending);
+                        toolStripStatusLabel1.Text = "Done.";
+                    }
                         dataGridView1.Sort(dataGridView1.Columns["TimeColumn"], ListSortDirection.Descending);
-                        toolStripStatusLabel1.Text = "Done.";
+                    if (!snapshotQueue.TryDequeue(out var snapshot))
                 catch (Exception exception)
+                {
                     Log.Error(exception, "Could not update data grid view.");
+                }
+            }
-            Application.Idle += IdleHandler;
             try
+            {
-                foreach (var snapshot in await _snapshotDatabase.LoadSnapshots(_cancellationToken))
+                await foreach (var snapshot in _snapshotDatabase.LoadSnapshots(_cancellationToken).WithCancellation(_cancellationToken))
+                {
                     snapshotQueue.Enqueue(snapshot);
+                }
-                snapshotsQueuedTaskCompletionSource.TrySetResult(new { });
+                Application.Idle += IdleHandler;
+            }
             catch (Exception exception)
+            {
+            }
             e.Effect = DragDropEffects.None;
+        }
+        private async Task CreateSnapshots(IReadOnlyList<string> files, Bitmap screenCapture, TrackedFolders.TrackedFolders trackedFolders, IProgress<CreateSnapshotProgress> progress, CancellationToken cancellationToken)
+        {
+            Parallel.ForEach(files, new ParallelOptions { MaxDegreeOfParallelism = 512 }, async file =>
+            {
+                var color = Color.Empty;
+                if (_mainForm.TrackedFolders.TryGet(file, out var folder))
+                {
+                    color = folder.Color;
+                }
+                var fileInfo = File.GetAttributes(file);
+                if (fileInfo.HasFlag(FileAttributes.Directory))
+                {
+                    foreach (var directoryFile in Directory.GetFiles(file, "*.*", SearchOption.AllDirectories))
+                    {
+                        var name = Path.GetFileName(directoryFile);
+                        var path = Path.Combine(Path.GetDirectoryName(directoryFile), name);
+                        try
+                        {
+                            await _snapshotDatabase.CreateSnapshot(name, path, screenCapture, color,
+                                _cancellationToken);
+                            progress.Report(new CreateSnapshotProgressSuccess(file));
+                        }
+                        catch (Exception exception)
+                        {
+                            progress.Report(new CreateSnapshotProgressFailure(file, exception));
+                        }
+                    }
+                    return;
+                }
+                var fileName = Path.GetFileName(file);
+                var pathName = Path.Combine(Path.GetDirectoryName(file), fileName);
+                try
+                {
+                    await _snapshotDatabase.CreateSnapshot(fileName, pathName, screenCapture, color,
+                        _cancellationToken);
+                    progress.Report(new CreateSnapshotProgressSuccess(file));
+                }
+                catch (Exception exception)
+                {
+                    progress.Report(new CreateSnapshotProgressFailure(file, exception));
+                }
             });
+        }
         private async void DataGridView1_DragDrop(object sender, DragEventArgs e)
+        {
             if (!e.Data.GetDataPresent(DataFormats.FileDrop))
             toolStripProgressBar1.Minimum = 0;
             toolStripProgressBar1.Maximum = files.Length;
             toolStripStatusLabel1.Text = "Snapshotting files...";
+            var screenCapture = ScreenCapture.Capture((CaptureMode)_mainForm.Configuration.CaptureMode);
+            var progress = new Progress<CreateSnapshotProgress>(createSnapshotProgress =>
+            {
+                switch (createSnapshotProgress)
+                {
+                    case CreateSnapshotProgressSuccess createSnapshotProgressSuccess:
+                        toolStripStatusLabel1.Text = $"Snapshot taken of {createSnapshotProgressSuccess.File}.";
+                        break;
+                    case CreateSnapshotProgressFailure createSnapshotProgressFailure:
+                        if (createSnapshotProgressFailure.Exception is SQLiteException { ResultCode: SQLiteErrorCode.Constraint })
+                        {
+                            toolStripStatusLabel1.Text = $"Snapshot of file {createSnapshotProgressFailure.File} already exists.";
+                            break;
+                        }
+                        toolStripStatusLabel1.Text = $"Could not snapshot file {createSnapshotProgressFailure.File}";
+                        Log.Warning(createSnapshotProgressFailure.Exception, $"Could not snapshot file {createSnapshotProgressFailure.File}");
+                        break;
+                }
+                toolStripProgressBar1.Increment(1);
+                statusStrip1.Update();
+            });
+            await Task.Factory.StartNew(async () =>
+            {
+                await CreateSnapshots(files, screenCapture, _mainForm.TrackedFolders, progress, _cancellationToken);
+            }, _cancellationToken);
+            /*
-            var screenCapture = ScreenCapture.Capture((CaptureMode)_mainForm.Configuration.CaptureMode);
-            for (var i = 0; i < files.Length; ++i)
+            foreach (var file in files)
+            {
                 var color = Color.Empty;
-                if (_mainForm.TrackedFolders.TryGet(files[i], out var folder))
+                if (_mainForm.TrackedFolders.TryGet(file, out var folder))
+                {
                     color = folder.Color;
                 }
                 var fileInfo = File.GetAttributes(file);
-                if (Directory.Exists(files[i]))
+                if (fileInfo.HasFlag(FileAttributes.Directory))
+                {
-                    foreach (var directoryFile in Directory.GetFiles(files[i], "*.*", SearchOption.AllDirectories))
+                    foreach (var directoryFile in Directory.GetFiles(file, "*.*", SearchOption.AllDirectories))
+                    {
                         var name = Path.GetFileName(directoryFile);
                         var path = Path.Combine(Path.GetDirectoryName(directoryFile), name);
                         try
+                        {
-                        {
+                            await _snapshotDatabase.CreateSnapshot(name, path, screenCapture, color,
-                            await _snapshotDatabase.CreateSnapshot(name, path, screenCapture, color,
+                                _cancellationToken);
                                 _cancellationToken);
-                            toolStripProgressBar1.Value = i + 1;
+                            toolStripStatusLabel1.Text = $"Snapshot taken of {file}.";
-                            toolStripStatusLabel1.Text = $"Snapshot taken of {files[i]}.";
+                            toolStripProgressBar1.Increment(1);
                             statusStrip1.Update();
+                        }
                         catch (SQLiteException exception)
                         {
-                            if (exception.ResultCode == SQLiteErrorCode.Constraint)
                             if (exception.ResultCode == SQLiteErrorCode.Constraint)
+                            {
                                 Log.Information(exception, "Snapshot already exists.");
-                                toolStripProgressBar1.Value = i + 1;
+                                toolStripStatusLabel1.Text = "Snapshot already exists.";
-                                toolStripStatusLabel1.Text = "Snapshot already exists.";
+                                toolStripProgressBar1.Increment(1);
                                 statusStrip1.Update();
+                    }
                     continue;
                 }
-                var fileName = Path.GetFileName(files[i]);
+                var fileName = Path.GetFileName(file);
-                var pathName = Path.Combine(Path.GetDirectoryName(files[i]), fileName);
+                var pathName = Path.Combine(Path.GetDirectoryName(file), fileName);
                 try
                 {
                     await _snapshotDatabase.CreateSnapshot(fileName, pathName, screenCapture, color,
                         _cancellationToken);
-                    toolStripProgressBar1.Value = i + 1;
+                    toolStripStatusLabel1.Text = $"Snapshot taken of {file}.";
-                    toolStripStatusLabel1.Text = $"Snapshot taken of {files[i]}.";
+                    toolStripProgressBar1.Increment(1);
                     statusStrip1.Update();
+                }
                 catch (SQLiteException exception)
+                {
                     if (exception.ResultCode == SQLiteErrorCode.Constraint)
-                    {
                         Log.Information(exception, "Snapshot already exists.");
+                        Log.Information(exception, "Snapshot already exists.");
-                        toolStripProgressBar1.Value = i + 1;
+                        toolStripStatusLabel1.Text = "Snapshot already exists.";
-                        toolStripStatusLabel1.Text = "Snapshot already exists.";
+                        toolStripProgressBar1.Increment(1);
                         statusStrip1.Update();
+                    }
+                }
                 catch (Exception exception)
+                {
+                    Log.Warning(exception, "Could not create snapshot");
                     Log.Warning(exception, "Could not create snapshot");
+            }
             */
+        }