advertising-profit-control/AdvertsingProfitControl/DatabaseRecovery.cs

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();
        }
    }
}