Implemented database backups and restores into the AdvertisingProfitControlData DLL and added a form, DatabaseRecovery, to manage the backups and restores. Updated the installer to include the data DLL file into the install bundle.

2017-06-23 03:12:51 -05:00
parent 308b60bd84
commit 12a6b36370
20 changed files with 5378 additions and 80 deletions
@@ -0,0 +1,176 @@
+using System;
+using System.IO;
+using System.Windows.Forms;
+using AdvertisingProfitControlData;
+
+namespace AdvertsingProfitControl
+{
+    public partial class DatabaseRecovery : Form
+    {
+        public bool RestoredDatabase;       
+
+        public DatabaseRecovery()
+        {
+            InitializeComponent();
+            exisitingBackupsListView.ItemSelectionChanged += UpdateRestoreLocationTextBoxOnItemSelectionChange;
+            restoreLocationTextBox.TextChanged += UpdateRestoreButtonOnRestoreLocationTextChange;
+            //Grab the list of backups and fill the list box.
+            var headerColumn = new ColumnHeader
+            {
+                Width = exisitingBackupsListView.Width - 10,
+                TextAlign = HorizontalAlignment.Center
+            };
+            exisitingBackupsListView.Columns.Add(headerColumn);
+            UpdateBackupList();
+        }
+
+        private void UpdateRestoreButtonOnRestoreLocationTextChange(object sender, EventArgs e)
+        {
+            restoreBackupButton.Enabled = restoreLocationTextBox.Text.Length != 0;
+        }
+
+        private void UpdateRestoreLocationTextBoxOnItemSelectionChange(object sender, ListViewItemSelectionChangedEventArgs e)
+        {
+            //Update the restore location text box when an item is selected.
+            restoreLocationTextBox.Text = ((FileInfo) e.Item.Tag).FullName;
+            deleteBackupFileButton.Enabled = true;
+        }
+
+        private void UpdateBackupList()
+        {
+            var backup = new Backup();
+            var fileList = backup.GetSingleFileBackupFilesList();
+            exisitingBackupsListView.Items.Clear();
+
+            if (fileList.Count > 0)
+            {
+                foreach (var file in fileList)
+                {
+                    var item = new ListViewItem
+                    {
+                        Text = file.Name,
+                        Tag = file
+                    };
+                    exisitingBackupsListView.Items.Add(item);
+                }
+                exisitingBackupsListView.Columns[0].Text = @"Current Backups";
+            }
+            else
+            {
+                exisitingBackupsListView.Columns[0].Text = @"No Backups Exist";          
+            }
+        }
+
+        private void CreateBackupOnButtonClick(object sender, EventArgs e)
+        {
+            var backup = new Backup();
+            var customBackupLocation = string.Empty;
+
+            if (!string.IsNullOrWhiteSpace(customBackupLocationTextBox.Text))
+            {
+                //Check to see if the folder selected or typed exists.
+                var folder = new DirectoryInfo(customBackupLocationTextBox.Text);
+
+                if (folder.Exists)
+                {
+                    customBackupLocation = customBackupLocationTextBox.Text;
+                }
+                else
+                {
+                    MessageBox.Show(
+                        @"The selected backup folder doesn't exist. Please choose a different folder to store the backup.",
+                        @"Backup Folder Location Invalid", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                    backupResultLabel.Text = @"Custom backup location invalid.";
+                    return;
+                }
+            }
+
+            if (backup.SingleFileBackup(customBackupLocation))
+            {
+                backupResultLabel.Text = @"Successfully created backup(s).";
+                customBackupLocationTextBox.Text = string.Empty;
+            }
+            else
+            {
+                MessageBox.Show(backup.ErrorMessage, @"Failed to Create Backup", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                backupResultLabel.Text = @"Failed to create backup(s).";
+            }
+
+            UpdateBackupList();
+        }
+
+        private void UpdateCustomBackupLocationOnBackupLocationButtonClick(object sender, EventArgs e)
+        {
+            var dialog = new FolderBrowserDialog {Description = @"Select a folder to place the backup file into."};
+            dialog.ShowDialog();
+
+            if (dialog.SelectedPath != string.Empty)
+            {
+                customBackupLocationTextBox.Text = dialog.SelectedPath;
+            }
+        }
+
+        private void UpdateRestoreLocationOnCustomBackupLocationButtonClick(object sender, EventArgs e)
+        {
+            var dialog = new OpenFileDialog
+            {
+                Multiselect = false,
+                Filter = @"Backup Files | *." + Backup.DefaultBackupExtension,
+                Title = @"Select your backup file to restore from."
+            };
+
+            dialog.ShowDialog();
+
+            restoreLocationTextBox.Text = dialog.FileName;
+        }
+
+        private void RestoreSelectedBackupOnBackupRestoreButtonClick(object sender, EventArgs e)
+        {
+            var restore = new Restore();
+            //Check for the file's existence before proceeding.
+            if (!File.Exists(restoreLocationTextBox.Text))
+            {
+                MessageBox.Show(@"Database backup file could not be found at the location" + Environment.NewLine + @"'" + restoreLocationTextBox.Text + @"'.", @"Database Backup File Not Found", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                restoreInfoLabel.Text = @"Restore file location invalid.";
+                return;
+            }
+
+            if (restore.SingleFileRestore(restoreLocationTextBox.Text))
+            {
+                restoreResultLabel.Text = @"Successfully restored the database.";
+                restoreLocationTextBox.Text = string.Empty;
+                RestoredDatabase = true;
+            }
+            else
+            {
+                MessageBox.Show(restore.ErrorMessage, @"Failed to Restore Backup", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                restoreInfoLabel.Text = @"Failed to restore the database.";
+            }
+        }
+
+        private void deleteBackupFileButton_Click(object sender, EventArgs e)
+        {
+            if (exisitingBackupsListView.SelectedItems.Count == 0)
+            {
+                deleteBackupFileButton.Enabled = false;
+                return;
+            }
+
+            var selectedItem = exisitingBackupsListView.SelectedItems[0];
+            var fileInfo = (FileInfo) selectedItem.Tag;
+
+            try
+            {
+                File.Delete(fileInfo.FullName);
+                restoreLocationTextBox.Text = string.Empty;
+                deleteBackupFileButton.Enabled = false;
+            }
+            catch (IOException exception)
+            {
+                MessageBox.Show(exception.Message, @"Failed to Delete File " + fileInfo.Name, MessageBoxButtons.OK, MessageBoxIcon.Error);
+            }
+
+            UpdateBackupList();
+        }
+    }
+}