MARKUS

using Avalonia;

using Avalonia.Controls;

using Avalonia.Controls.ApplicationLifetimes;

using Avalonia.Media.Imaging;

using NetSparkleUpdater.Interfaces;

using NetSparkleUpdater.Properties;

using NetSparkleUpdater.UI.Avalonia.ViewModels;

using System;

using System.Collections.Generic;

using System.IO;

using System.Threading;

namespace NetSparkleUpdater.UI.Avalonia

{

    /// <summary>

    /// UI factory for default interface

    /// </summary>

    public class UIFactory : IUIFactory

    {

        private WindowIcon _applicationIcon = null;

        private string _releaseNotesSeparatorTemplate;

        private string _releaseNotesHeadAddition;

        private Bitmap _iconBitmap;

        public UIFactory()

        {

            HideReleaseNotes = false;

            HideRemindMeLaterButton = false;

            HideSkipButton = false;

        }

        public UIFactory(WindowIcon applicationIcon, string releaseNotesSeparatorTemplate = "", string releaseNotesHeadAddition = "")

        {

            _applicationIcon = applicationIcon;

            if (applicationIcon != null)

            {

                using (var stream = new MemoryStream())

                {

                    applicationIcon?.Save(stream);

                    stream.Position = 0;

                    _iconBitmap = new Bitmap(stream);

                }

            }

            _releaseNotesSeparatorTemplate = releaseNotesSeparatorTemplate;

            _releaseNotesHeadAddition = releaseNotesHeadAddition;

            HideReleaseNotes = false;

            HideRemindMeLaterButton = false;

            HideSkipButton = false;

        }

        /// <summary>

        /// Hides the release notes view when an update is found.

        /// </summary>

        public bool HideReleaseNotes { get; set; }

        /// <summary>

        /// Hides the skip this update button when an update is found.

        /// </summary>

        public bool HideSkipButton { get; set; }

        /// <summary>

        /// Hides the remind me later button when an update is found.

        /// </summary>

        public bool HideRemindMeLaterButton { get; set; }

        /// <summary>

        /// Create sparkle form implementation

        /// </summary>

        /// <param name="sparkle">The <see cref="SparkleUpdater"/> instance to use</param>

        /// <param name="updates">Sorted array of updates from latest to earliest</param>

        /// <param name="isUpdateAlreadyDownloaded">If true, make sure UI text shows that the user is about to install the file instead of download it.</param>

        public virtual IUpdateAvailable CreateUpdateAvailableWindow(SparkleUpdater sparkle, List<AppCastItem> updates, bool isUpdateAlreadyDownloaded = false)

        {

            var viewModel = new UpdateAvailableWindowViewModel();

            var window = new UpdateAvailableWindow(viewModel, _iconBitmap)

            {

                Icon = _applicationIcon

            };

            if (HideReleaseNotes)

            {

                (window as IUpdateAvailable).HideReleaseNotes();

            }

            if (HideSkipButton)

            {

                (window as IUpdateAvailable).HideSkipButton();

            }

            if (HideRemindMeLaterButton)

            {

                (window as IUpdateAvailable).HideRemindMeLaterButton();

            }

            viewModel.Initialize(sparkle, updates, isUpdateAlreadyDownloaded, _releaseNotesSeparatorTemplate, _releaseNotesHeadAddition);

            return window;

        }

        /// <summary>

        /// Create download progress window

        /// </summary>

        /// <param name="item">Appcast item to download</param>

        public virtual IDownloadProgress CreateProgressWindow(AppCastItem item)

        {

            var viewModel = new DownloadProgressWindowViewModel()

            {

                ItemToDownload = item

            };

            return new DownloadProgressWindow(viewModel, _iconBitmap)

            {

                Icon = _applicationIcon

            };

        }

        /// <summary>

        /// Inform user in some way that NetSparkle is checking for updates

        /// </summary>

        public virtual ICheckingForUpdates ShowCheckingForUpdates()

        {

            return new CheckingForUpdatesWindow(_iconBitmap)

            { 

                Icon = _applicationIcon

            };

        }

        /// <summary>

        /// Initialize UI. Called when Sparkle is constructed and/or when the UIFactory is set.

        /// </summary>

        public virtual void Init()

        {

        }

        /// <summary>

        /// Show user a message saying downloaded update format is unknown

        /// </summary>

        /// <param name="downloadFileName">The filename to be inserted into the message text</param>

        public virtual void ShowUnknownInstallerFormatMessage(string downloadFileName)

        {

            ShowMessage(Resources.DefaultUIFactory_MessageTitle,

                string.Format(Resources.DefaultUIFactory_ShowUnknownInstallerFormatMessageText, downloadFileName));

        }

        /// <summary>

        /// Show user that current installed version is up-to-date

        /// </summary>

        public virtual void ShowVersionIsUpToDate()

        {

            ShowMessage(Resources.DefaultUIFactory_MessageTitle, Resources.DefaultUIFactory_ShowVersionIsUpToDateMessage);

        }

        /// <summary>

        /// Show message that latest update was skipped by user

        /// </summary>

        public virtual void ShowVersionIsSkippedByUserRequest()

        {

            ShowMessage(Resources.DefaultUIFactory_MessageTitle, Resources.DefaultUIFactory_ShowVersionIsSkippedByUserRequestMessage);

        }

        /// <summary>

        /// Show message that appcast is not available

        /// </summary>

        /// <param name="appcastUrl">the URL for the appcast file</param>

        public virtual void ShowCannotDownloadAppcast(string appcastUrl)

        {

            ShowMessage(Resources.DefaultUIFactory_ErrorTitle, Resources.DefaultUIFactory_ShowCannotDownloadAppcastMessage);

        }

        public virtual bool CanShowToastMessages()

        {

            return false;

        }

        /// <summary>

        /// Show 'toast' window to notify new version is available

        /// </summary>

        /// <param name="updates">Appcast updates</param>

        /// <param name="clickHandler">handler for click</param>

        public virtual void ShowToast(List<AppCastItem> updates, Action<List<AppCastItem>> clickHandler)

        {

        }

        /// <summary>

        /// Show message on download error

        /// </summary>

        /// <param name="message">Error message from exception</param>

        /// <param name="appcastUrl">the URL for the appcast file</param>

        public virtual void ShowDownloadErrorMessage(string message, string appcastUrl)

        {

            ShowMessage(Resources.DefaultUIFactory_ErrorTitle, string.Format(Resources.DefaultUIFactory_ShowDownloadErrorMessage, message));

        }

        private void ShowMessage(string title, string message)

        {

            var messageWindow = new MessageNotificationWindow(new MessageNotificationWindowViewModel(message), _iconBitmap)

            {

                Title = title,

                Icon = _applicationIcon

            };

            messageWindow.WindowStartupLocation = WindowStartupLocation.CenterScreen;

            messageWindow.Show(); // TODO: This was ShowDialog; will this break anything?

        }

        /// <summary>

        /// Shut down the UI so we can run an update.

        /// If in WPF, System.Windows.Application.Current.Shutdown().

        /// If in WinForms, Application.Exit().

        /// </summary>

        public void Shutdown()

        {

            (Application.Current?.ApplicationLifetime as IClassicDesktopStyleApplicationLifetime)?.Shutdown();

        }

    }

}