WowUp/WowUp.WPF/ViewModels/AddonListItemViewModel.cs

using Serilog;
using System;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Media;
using WowUp.Common.Enums;
using WowUp.Common.Extensions;
using WowUp.WPF.Entities;
using WowUp.WPF.Extensions;
using WowUp.WPF.Services.Contracts;
using WowUp.WPF.Utilities;

namespace WowUp.WPF.ViewModels
{
    public class AddonListItemViewModel : BaseViewModel
    {
        private readonly IAddonService _addonService;
        private readonly IAnalyticsService _analyticsService;
        private readonly SolidColorBrush _rareBrush;
        private readonly SolidColorBrush _epicBrush;

        private Addon _addon;
        public Addon Addon
        {
            get => _addon;
            set
            {
                _addon = value;
                SetupDisplayState();
            }
        }

        public Command ActionCommand { get; set; }
        public Command InstallCommand { get; set; }
        public Command UpdateCommand { get; set; }
        public Command OpenLinkCommand { get; set; }
        public Command ReInstallCommand { get; set; }
        public Command UninstallCommand { get; set; }
        public Command IgnoreCheckedCommand { get; set; }
        public Command StableCheckedCommand { get; set; }
        public Command BetaCheckedCommand { get; set; }
        public Command AlphaCheckedCommand { get; set; }
        public Command AutoUpdateCheckedCommand { get; set; }

        private bool _showInstallButton;
        public bool ShowInstallButton
        {
            get => _showInstallButton;
            set { SetProperty(ref _showInstallButton, value); }
        }

        private bool _showUpdateButton;
        public bool ShowUpdateButton
        {
            get => _showUpdateButton;
            set { SetProperty(ref _showUpdateButton, value); }
        }

        private bool _showProgressBar;
        public bool ShowProgressBar
        {
            get => _showProgressBar;
            set { SetProperty(ref _showProgressBar, value); }
        }

        public bool _showStatusText;
        public bool ShowStatusText
        {
            get => _showStatusText;
            set { SetProperty(ref _showStatusText, value); }
        }

        public bool _showUninstall;
        public bool ShowUninstall
        {
            get => _showUninstall;
            set { SetProperty(ref _showUninstall, value); }
        }

        public bool _showReInstall;
        public bool ShowReInstall
        {
            get => _showReInstall;
            set { SetProperty(ref _showReInstall, value); }
        }

        private string _progressText;
        public string ProgressText
        {
            get => _progressText;
            set { SetProperty(ref _progressText, value); }
        }

        private decimal _progressPercent;
        public decimal ProgressPercent
        {
            get => _progressPercent;
            set { SetProperty(ref _progressPercent, value); }
        }

        private string _name;
        public string Name
        {
            get => _name;
            set { SetProperty(ref _name, value); }
        }

        private string _currentVersion;
        public string CurrentVersion
        {
            get => _currentVersion;
            set { SetProperty(ref _currentVersion, value); }
        }

        private string _latestVersion;
        public string LatestVersion
        {
            get => _latestVersion;
            set { SetProperty(ref _latestVersion, value); }
        }

        private string _author;
        public string Author
        {
            get => _author;
            set { SetProperty(ref _author, value); }
        }

        private string _gameVersion;
        public string GameVersion
        {
            get => _gameVersion;
            set { SetProperty(ref _gameVersion, value); }
        }

        private string _providerName;
        public string ProviderName
        {
            get => _providerName;
            set { SetProperty(ref _providerName, value); }
        }

        private string _thumbnailUrl;
        public string ThumbnailUrl
        {
            get => _thumbnailUrl;
            set { SetProperty(ref _thumbnailUrl, value); }
        }

        private string _externalUrl;
        public string ExternalUrl
        {
            get => _externalUrl;
            set { SetProperty(ref _externalUrl, value); }
        }

        private AddonDisplayState _displayState;
        public AddonDisplayState DisplayState
        {
            get => _displayState;
            set { SetProperty(ref _displayState, value); }
        }

        private bool _isIgnored;
        public bool IsIgnored
        {
            get => _isIgnored;
            set { SetProperty(ref _isIgnored, value); }
        }

        private bool _isAutoUpdated;
        public bool IsAutoUpdated
        {
            get => _isAutoUpdated;
            set { SetProperty(ref _isAutoUpdated, value); }
        }

        private AddonChannelType _channelType;
        public AddonChannelType ChannelType
        {
            get => _channelType;
            set { SetProperty(ref _channelType, value); }
        }

        private bool _showChannelName;
        public bool ShowChannelName
        {
            get => _showChannelName;
            set { SetProperty(ref _showChannelName, value); }
        }

        private string _channelName;
        public string ChannelName
        {
            get => _channelName;
            set { SetProperty(ref _channelName, value); }
        }

        private bool _canCheckStable;
        public bool CanCheckStable
        {
            get => _canCheckStable;
            set { SetProperty(ref _canCheckStable, value); }
        }

        private bool _canCheckBeta;
        public bool CanCheckBeta
        {
            get => _canCheckBeta;
            set { SetProperty(ref _canCheckBeta, value); }
        }

        private bool _canCheckAlpha;
        public bool CanCheckAlpha
        {
            get => _canCheckAlpha;
            set { SetProperty(ref _canCheckAlpha, value); }
        }

        private bool _isStableChannel;
        public bool IsStableChannel
        {
            get => _isStableChannel;
            set { SetProperty(ref _isStableChannel, value); }
        }

        private bool _isBetaChannel;
        public bool IsBetaChannel
        {
            get => _isBetaChannel;
            set { SetProperty(ref _isBetaChannel, value); }
        }

        private bool _isAlphaChannel;
        public bool IsAlphaChannel
        {
            get => _isAlphaChannel;
            set { SetProperty(ref _isAlphaChannel, value); }
        }

        private string _statusText;
        public string StatusText
        {
            get => _statusText;
            set { SetProperty(ref _statusText, value); }
        }

        private Brush _channelNameBrush;
        public Brush ChannelNameBrush
        {
            get => _channelNameBrush;
            set { SetProperty(ref _channelNameBrush, value); }
        }

        public bool CanInstall => _addon.CanInstall();
        public bool CanUpdate => _addon.CanUpdate();

        public AddonListItemViewModel(
            IAddonService addonService,
            IAnalyticsService analyticsService)
            : base()
        {
            _addonService = addonService;
            _analyticsService = analyticsService;

            InstallCommand = new Command(async () => await InstallAddon());
            UpdateCommand = new Command(async () => await UpdateAddon());
            OpenLinkCommand = new Command(() => ExternalUrl.OpenUrlInBrowser());
            ReInstallCommand = new Command(() => OnReInstall());
            UninstallCommand = new Command(() => OnUninstall());
            IgnoreCheckedCommand = new Command(() => OnIgnoreChanged());
            StableCheckedCommand = new Command(() => OnChannelChanged(AddonChannelType.Stable));
            BetaCheckedCommand = new Command(() => OnChannelChanged(AddonChannelType.Beta));
            AlphaCheckedCommand = new Command(() => OnChannelChanged(AddonChannelType.Alpha));
            AutoUpdateCheckedCommand = new Command(() => OnAutoUpdateChanged());

            _rareBrush = Application.Current.Resources["RareBrush"] as SolidColorBrush;
            _epicBrush = Application.Current.Resources["EpicBrush"] as SolidColorBrush;
        }

        private void SetupDisplayState()
        {
            Name = _addon.Name;
            CurrentVersion = string.IsNullOrEmpty(_addon.InstalledVersion)
                ? "None"
                : _addon.InstalledVersion;
            LatestVersion = _addon.LatestVersion;
            Author = _addon.Author;
            ProviderName = _addon.ProviderName;
            GameVersion = _addon.GameVersion;
            ThumbnailUrl = string.IsNullOrEmpty(_addon.ThumbnailUrl)
                ? "/Assets/wowup_logo_1.png"
                : _addon.ThumbnailUrl;

            IsIgnored = _addon.IsIgnored;

            ChannelType = _addon.ChannelType;
            ChannelName = _addon.ChannelType.GetDisplayName();
            ShowChannelName = _addon.ChannelType != AddonChannelType.Stable;
            CanCheckStable = _addon.ChannelType != AddonChannelType.Stable;
            CanCheckBeta = _addon.ChannelType != AddonChannelType.Beta;
            CanCheckAlpha = _addon.ChannelType != AddonChannelType.Alpha;

            IsStableChannel = _addon.ChannelType == AddonChannelType.Stable;
            IsBetaChannel = _addon.ChannelType == AddonChannelType.Beta;
            IsAlphaChannel = _addon.ChannelType == AddonChannelType.Alpha;

            IsAutoUpdated = _addon.AutoUpdateEnabled;

            if (ChannelType == AddonChannelType.Beta)
            {
                ChannelNameBrush = _rareBrush;
            }
            else if (ChannelType == AddonChannelType.Alpha)
            {
                ChannelNameBrush = _epicBrush;
            }

            ExternalUrl = _addon.ExternalUrl;
            DisplayState = _addon.GetDisplayState();
            ShowInstallButton = DisplayState == AddonDisplayState.Install;
            ShowUpdateButton = DisplayState == AddonDisplayState.Update;
            ShowStatusText = DisplayState == AddonDisplayState.UpToDate || DisplayState == AddonDisplayState.Ignored;
            ShowReInstall = DisplayState == AddonDisplayState.UpToDate;
            ShowUninstall = DisplayState != AddonDisplayState.Unknown;
            ShowProgressBar = false;
            StatusText = GetStatusText(DisplayState);
        }

        public async Task UpdateAddon()
        {
            ShowStatusText = false;
            ShowUpdateButton = false;

            try
            {
                await _addonService.InstallAddon(_addon.Id, OnInstallUpdate);
            }
            catch (Exception ex)
            {
                Log.Error("Failed to update addon", ex);
                ShowUpdateButton = true;
            }
        }

        public async Task InstallAddon()
        {
            ShowStatusText = false;
            ShowInstallButton = false;
            ShowUpdateButton = false;

            try
            {
                await _addonService.InstallAddon(_addon.Id, OnInstallUpdate);

                await _analyticsService.TrackUserAction("Addons", "InstallScanned", _addon.Name);
            }
            catch (Exception ex)
            {
                Console.WriteLine(ex);
                ShowInstallButton = true;
            }
        }

        private async void OnReInstall()
        {
            await InstallAddon();
        }

        private async void OnUninstall()
        {
            var messageBoxResult = MessageBox.Show(
                $"Are you sure you want to remove {Addon.Name}? This will remove all related folders from your World of Warcraft folder.",
                "Uninstall Addon?",
                MessageBoxButton.YesNo);

            if (messageBoxResult != MessageBoxResult.Yes)
            {
                return;
            }

            try
            {
                await _addonService.UninstallAddon(Addon);
            }
            catch (Exception ex)
            {
                Log.Error(ex, $"Failed to uninstall addon {Addon.Name}");
                MessageBox.Show("An error occurred during uninstall.");
            }
        }

        private void OnIgnoreChanged()
        {
            _addon.IsIgnored = !_addon.IsIgnored;
            _addonService.UpdateAddon(_addon);
            SetupDisplayState();

            if (_addon.IsIgnored)
            {
                _analyticsService.TrackUserAction("Addons", "Ignore", _addon.Name);
            }
        }

        private void OnAutoUpdateChanged()
        {
            _addon.AutoUpdateEnabled = !_addon.AutoUpdateEnabled;
            _addonService.UpdateAddon(_addon);
            SetupDisplayState();

            if (_addon.AutoUpdateEnabled)
            {
                _analyticsService.TrackUserAction("Addons", "AutoUpdate", _addon.Name);
            }
        }

        private void OnChannelChanged(AddonChannelType channelType)
        {
            _addon.ChannelType = channelType;

            _addonService.UpdateAddon(_addon);

            SetupDisplayState();

            _analyticsService.TrackUserAction("Addons", "Channel", channelType.ToString());
        }

        private void OnInstallUpdate(AddonInstallState installState, decimal percent)
        {
            try
            {
                ProgressText = GetInstallStateText(installState);
                ProgressPercent = percent;
                ShowProgressBar = true;

                if (installState == AddonInstallState.Complete)
                {
                    _addon = _addonService.GetAddon(_addon.Id);
                    SetupDisplayState();
                }
            }
            catch(Exception ex)
            {
                Log.Error("Failed to handle addon install event", ex);
            }
        }

        private string GetStatusText(AddonDisplayState displayState)
        {
            switch (displayState)
            {
                case AddonDisplayState.UpToDate:
                    return "Up to Date";
                case AddonDisplayState.Ignored:
                    return "Ignored";
                case AddonDisplayState.Update:
                case AddonDisplayState.Install:
                case AddonDisplayState.Unknown:
                default:
                    return string.Empty;
            }
        }

        private string GetInstallStateText(AddonInstallState installState)
        {
            switch (installState)
            {
                case AddonInstallState.Pending:
                    return "Pending";
                case AddonInstallState.Downloading:
                    return "Downloading";
                case AddonInstallState.BackingUp:
                    return "BackingUp";
                case AddonInstallState.Installing:
                    return "Installing";
                case AddonInstallState.Complete:
                    return "Complete";
                default:
                    return "Unknown";
            }
        }
    }
}