using System;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Windows.Forms;

namespace BackgroundWorkerExample
{
    public partial class Form1 : Form
    {
        private BackgroundWorker _worker;
        public Form1()
        {
            InitializeComponent();
            ChangeButtonsState(false);
        }

        private void Button1_Click(object sender, EventArgs e)
        {
            //Очищаем резултат
            FolderSizeTextBox.Text = string.Empty;
            //Создаем воркер
            _worker = new BackgroundWorker
            {
                WorkerSupportsCancellation = true
            };
            _worker.DoWork += BackgroundWorker_DoWork;
            _worker.RunWorkerCompleted += BackgroundWorker_RunWorkerCompleted;
            _worker.RunWorkerAsync(SourceFolderTextBox.Text);
            //Меняем состояние кнопок
            ChangeButtonsState(true);
        }

        private void BackgroundWorker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e)
        {
            //Получаем результат
            long folderSize = (long)e.Result;
            //Сохраняем его в TextBox
            FolderSizeTextBox.Text = folderSize.ToString();
            //Меняем состояние кнопок
            ChangeButtonsState(false);
            //Высвобождаем worker
            _worker.Dispose();
            _worker = null;
        }
        /// <summary> Меняет состояние кнопок </summary>
        /// <param name="isWorking">Состояние выполнения</param>
        private void ChangeButtonsState(bool isWorking)
        {
            StartButton.Enabled = !isWorking;
            //Если Worker не поддерживает отмену, кнопка всегда будет выключенной
            CancelButton.Enabled = _worker?.WorkerSupportsCancellation == true && isWorking;
        }

        private void BackgroundWorker_DoWork(object sender, DoWorkEventArgs e)
        {
            //Сохраняем в результаты асинхронного выполнения размер папки
            e.Result = GetFolderSize((string)e.Argument);
        }

        /// <summary> Рекурсивно вычисляет размер папки </summary>
        /// <param name="folderPath">Путь к папке</param>
        /// <returns>Размер папки</returns>
        private long GetFolderSize(string folderPath)
        {
            long size = 0;

            try
            {
                string[] files = Directory.GetFiles(folderPath, "*");
                foreach (string file in files)
                {
                    //Если запрошена отмена кидаем исключение
                    if (_worker.CancellationPending) throw new OperationCanceledException();
                    var fileInfo = new FileInfo(file);
                    size += fileInfo.Length;
                }

                string[] folders = Directory.GetDirectories(folderPath, "*");
                foreach (string folder in folders)
                {
                    //Если запрошена отмена кидаем исключение
                    if (_worker.CancellationPending) throw new OperationCanceledException();
                    size += GetFolderSize(folder);
                }
            }
            catch (OperationCanceledException)
            {
                //так как операция отменена, результаты операции не известны
                return -1;
            }
            catch(Exception ex)
            {
                Debug.Print(ex.Message);
            }
            return size;
        }
        
        private void CancelButton_Click(object sender, EventArgs e)
        {
            //Если воркер поддерживает отмену, вызываем её
            if (_worker.WorkerSupportsCancellation) _worker.CancelAsync();
        }
    }
}