naps2/NAPS2.Core/Update/AutoUpdaterUI.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Windows.Forms;
using NAPS2.Config;
using NAPS2.Lang.Resources;
using NAPS2.WinForms;

namespace NAPS2.Update
{
    public class AutoUpdaterUI
    {
        private static readonly TimeSpan UpdateCheckInterval = TimeSpan.FromDays(0);

        private readonly IUserConfigManager userConfigManager;
        private readonly AppConfigManager appConfigManager;
        private readonly IAutoUpdater autoUpdater;
        private readonly IFormFactory formFactory;

        public AutoUpdaterUI(IUserConfigManager userConfigManager, AppConfigManager appConfigManager, IAutoUpdater autoUpdater, IFormFactory formFactory)
        {
            this.userConfigManager = userConfigManager;
            this.appConfigManager = appConfigManager;
            this.autoUpdater = autoUpdater;
            this.formFactory = formFactory;
        }

        public void OnApplicationStart(IAutoUpdaterClient client)
        {
            if (userConfigManager.Config.LastUpdateCheckDate == null)
            {
                userConfigManager.Config.LastUpdateCheckDate = DateTime.Now;
                userConfigManager.Save();
            }
            if (DateTime.Now - userConfigManager.Config.LastUpdateCheckDate > UpdateCheckInterval)
            {
                PromptToEnableAutomaticUpdates();
                CheckForUpdate(client);
            }
        }

        private void PromptToEnableAutomaticUpdates()
        {
            if (GetAutoUpdateStatus() == AutoUpdateStatus.Unspecified)
            {
                switch (MessageBox.Show(MiscResources.EnableAutoUpdates, MiscResources.AutoUpdates, MessageBoxButtons.YesNo, MessageBoxIcon.Question))
                {
                    case DialogResult.Yes:
                        userConfigManager.Config.AutoUpdate = AutoUpdateStatus.Enabled;
                        break;
                    case DialogResult.No:
                        userConfigManager.Config.AutoUpdate = AutoUpdateStatus.Disabled;
                        break;
                }
                userConfigManager.Save();
            }
        }

        private void CheckForUpdate(IAutoUpdaterClient client)
        {
            if (GetAutoUpdateStatus() == AutoUpdateStatus.Enabled)
            {
                autoUpdater.CheckForUpdate().ContinueWith(updateInfo =>
                {
                    if (updateInfo.Result.HasUpdate)
                    {
                        client.UpdateAvailable(updateInfo.Result.VersionInfo);
                    }
                });
            }
        }

        public void PerformUpdate(IAutoUpdaterClient client, VersionInfo versionInfo)
        {
            switch (formFactory.Create<FUpdate>().ShowDialog())
            {
                case DialogResult.Yes: // Install
                    // TODO: The app process might need to be killed/restarted before/after installing
                    autoUpdater.DownloadAndInstallUpdate(versionInfo).ContinueWith(result =>
                    {
                        if (result.Result)
                        {
                            client.InstallComplete();
                        }
                        else
                        {
                            MessageBox.Show(MiscResources.InstallFailed, MiscResources.InstallFailedTitle,
                                MessageBoxButtons.OK, MessageBoxIcon.Error);
                        }
                    });
                    break;
                case DialogResult.OK: // Download
                    var saveDialog = new SaveFileDialog
                    {
                        FileName = versionInfo.FileName
                    };
                    if (saveDialog.ShowDialog() == DialogResult.OK)
                    {
                        // TODO: Display progress while downloading
                        autoUpdater.DownloadUpdate(versionInfo, saveDialog.FileName);
                    }
                    break;
            }
        }

        private AutoUpdateStatus GetAutoUpdateStatus()
        {
            return userConfigManager.Config.AutoUpdate == AutoUpdateStatus.Unspecified
                ? appConfigManager.Config.AutoUpdate
                : userConfigManager.Config.AutoUpdate;
        }
    }
}