Migrate to Stylet and refactor view/view-model framework

DiscordChatExporter.Gui/ViewModels/RootViewModel.cs

@@ -0,0 +1,227 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using System.Net;
+using System.Reflection;
+using DiscordChatExporter.Core.Exceptions;
+using DiscordChatExporter.Core.Models;
+using DiscordChatExporter.Core.Services;
+using DiscordChatExporter.Gui.ViewModels.Framework;
+using MaterialDesignThemes.Wpf;
+using Stylet;
+using Tyrrrz.Extensions;
+
+namespace DiscordChatExporter.Gui.ViewModels
+{
+    public class RootViewModel : Screen
+    {
+        private readonly IViewModelFactory _viewModelFactory;
+        private readonly DialogManager _dialogManager;
+        private readonly SettingsService _settingsService;
+        private readonly UpdateService _updateService;
+        private readonly DataService _dataService;
+        private readonly ExportService _exportService;
+
+        private readonly Dictionary<Guild, IReadOnlyList<Channel>> _guildChannelsMap =
+            new Dictionary<Guild, IReadOnlyList<Channel>>();
+
+        public SnackbarMessageQueue Notifications { get; } = new SnackbarMessageQueue(TimeSpan.FromSeconds(5));
+
+        public bool IsEnabled { get; private set; } = true;
+
+        public bool IsProgressIndeterminate => Progress < 0;
+
+        public double Progress { get; private set; }
+
+        public bool IsBotToken { get; set; }
+
+        public string TokenValue { get; set; }
+
+        public IReadOnlyList<Guild> AvailableGuilds { get; private set; }
+
+        public Guild SelectedGuild { get; set; }
+
+        public IReadOnlyList<Channel> AvailableChannels =>
+            SelectedGuild != null ? _guildChannelsMap[SelectedGuild] : Array.Empty<Channel>();
+
+        public RootViewModel(IViewModelFactory viewModelFactory, DialogManager dialogManager,
+            SettingsService settingsService, UpdateService updateService, DataService dataService,
+            ExportService exportService)
+        {
+            _viewModelFactory = viewModelFactory;
+            _dialogManager = dialogManager;
+            _settingsService = settingsService;
+            _updateService = updateService;
+            _dataService = dataService;
+            _exportService = exportService;
+
+            // Set title
+            var version = Assembly.GetExecutingAssembly().GetName().Version;
+            DisplayName = $"DiscordChatExporter v{version}";            
+        }
+
+        protected override async void OnViewLoaded()
+        {
+            // Load settings
+            _settingsService.Load();
+
+            // Get last token
+            if (_settingsService.LastToken != null)
+            {
+                IsBotToken = _settingsService.LastToken.Type == AuthTokenType.Bot;
+                TokenValue = _settingsService.LastToken.Value;
+            }
+
+            // Check and prepare update
+            try
+            {
+                var updateVersion = await _updateService.CheckPrepareUpdateAsync();
+                if (updateVersion != null)
+                {
+                    Notifications.Enqueue(
+                        $"Update to DiscordChatExporter v{updateVersion} will be installed when you exit",
+                        "INSTALL NOW", () =>
+                        {
+                            _updateService.FinalizeUpdate(true);
+                            RequestClose();
+                        });
+                }
+            }
+            catch
+            {
+                Notifications.Enqueue("Failed to perform application auto-update");
+            }
+        }
+
+        protected override void OnClose()
+        {
+            // Save settings
+            _settingsService.Save();
+
+            // Finalize updates if necessary
+            _updateService.FinalizeUpdate(false);
+        }
+
+        public async void ShowSettings()
+        {
+            // Create dialog
+            var dialog = _viewModelFactory.CreateSettingsViewModel();
+
+            // Show dialog
+            await _dialogManager.ShowDialogAsync(dialog);
+        }
+
+        public void ShowAbout()
+        {
+            Process.Start("https://github.com/Tyrrrz/DiscordChatExporter");
+        }
+
+        public bool CanPopulateGuildsAndChannels => IsEnabled && TokenValue.IsNotBlank();
+
+        public async void PopulateGuildsAndChannels()
+        {
+            IsEnabled = false;
+            Progress = -1;
+
+            // Sanitize token
+            TokenValue = TokenValue.Trim('"');
+
+            // Create token
+            var token = new AuthToken(
+                IsBotToken ? AuthTokenType.Bot : AuthTokenType.User,
+                TokenValue);
+
+            // Save token
+            _settingsService.LastToken = token;
+
+            // Clear existing
+            _guildChannelsMap.Clear();
+
+            try
+            {
+                // Get DM channels
+                {
+                    var channels = await _dataService.GetDirectMessageChannelsAsync(token);
+                    var guild = Guild.DirectMessages;
+                    _guildChannelsMap[guild] = channels.OrderBy(c => c.Name).ToArray();
+                }
+
+                // Get guild channels
+                {
+                    var guilds = await _dataService.GetUserGuildsAsync(token);
+                    foreach (var guild in guilds)
+                    {
+                        var channels = await _dataService.GetGuildChannelsAsync(token, guild.Id);
+                        _guildChannelsMap[guild] = channels.Where(c => c.Type == ChannelType.GuildTextChat)
+                            .OrderBy(c => c.Name)
+                            .ToArray();
+                    }
+                }
+            }
+            catch (HttpErrorStatusCodeException ex) when (ex.StatusCode == HttpStatusCode.Unauthorized)
+            {
+                Notifications.Enqueue("Unauthorized – make sure the token is valid");
+            }
+            catch (HttpErrorStatusCodeException ex) when (ex.StatusCode == HttpStatusCode.Forbidden)
+            {
+                Notifications.Enqueue("Forbidden – account may be locked by 2FA");
+            }
+
+            AvailableGuilds = _guildChannelsMap.Keys.ToArray();
+            SelectedGuild = AvailableGuilds.FirstOrDefault();
+
+            Progress = 0;
+            IsEnabled = true;
+        }
+
+        public bool CanExportChannel => IsEnabled;
+
+        public async void ExportChannel(Channel channel)
+        {
+            IsEnabled = false;
+            Progress = -1;
+
+            // Get last used token
+            var token = _settingsService.LastToken;
+
+            // Create dialog
+            var dialog = _viewModelFactory.CreateExportSetupViewModel();
+            dialog.Guild = SelectedGuild;
+            dialog.Channel = channel;
+
+            // Show dialog
+            if (await _dialogManager.ShowDialogAsync(dialog) == true)
+            {
+                // Export
+                try
+                {
+                    // Create progress handler
+                    var progressHandler = new Progress<double>(p => Progress = p);
+
+                    // Get chat log
+                    var chatLog = await _dataService.GetChatLogAsync(token, dialog.Guild, dialog.Channel,
+                        dialog.From, dialog.To, progressHandler);
+
+                    // Export
+                    _exportService.ExportChatLog(chatLog, dialog.FilePath, dialog.SelectedFormat,
+                        dialog.PartitionLimit);
+
+                    // Notify completion
+                    Notifications.Enqueue("Export complete");
+                }
+                catch (HttpErrorStatusCodeException ex) when (ex.StatusCode == HttpStatusCode.Forbidden)
+                {
+                    Notifications.Enqueue("You don't have access to this channel");
+                }
+                catch (HttpErrorStatusCodeException ex) when (ex.StatusCode == HttpStatusCode.NotFound)
+                {
+                    Notifications.Enqueue("This channel doesn't exist");
+                }
+            }
+
+            Progress = 0;
+            IsEnabled = true;
+        }
+    }
+}