TweetDuck/Core/Other/Settings/Dialogs/DialogSettingsManage.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Windows.Forms;
using TweetDuck.Configuration;
using TweetDuck.Core.Management;
using TweetDuck.Core.Utils;
using TweetDuck.Plugins;

namespace TweetDuck.Core.Other.Settings.Dialogs{
    sealed partial class DialogSettingsManage : Form{
        private enum State{
            Deciding, Reset, Import, Export
        }

        private ProfileManager.Items SelectedItems{
            get => _selectedItems;

            set{
                // this will call events and SetFlag, which also updates the UI
                foreach(KeyValuePair<CheckBox, ProfileManager.Items> kvp in checkBoxMap){
                    kvp.Key.Checked = value.HasFlag(kvp.Value);
                }
            }
        }

        private bool SelectedItemsForceRestart{
            get => _selectedItems.HasFlag(ProfileManager.Items.Session);
        }
        
        public bool IsRestarting { get; private set; }
        public bool ShouldReloadBrowser { get; private set; }
        
        private readonly PluginManager plugins;
        private readonly Dictionary<CheckBox, ProfileManager.Items> checkBoxMap = new Dictionary<CheckBox, ProfileManager.Items>(4);
        private readonly bool openImportImmediately;

        private State currentState;
        private ProfileManager importManager;
        private bool requestedRestartFromConfig;

        private ProfileManager.Items _selectedItems = ProfileManager.Items.None;

        public DialogSettingsManage(PluginManager plugins, bool openImportImmediately = false){
            InitializeComponent();

            this.plugins = plugins;
            this.currentState = State.Deciding;

            this.checkBoxMap[cbProgramConfig] = ProfileManager.Items.UserConfig;
            this.checkBoxMap[cbSystemConfig] = ProfileManager.Items.SystemConfig;
            this.checkBoxMap[cbSession] = ProfileManager.Items.Session;
            this.checkBoxMap[cbPluginData] = ProfileManager.Items.PluginData;

            this.openImportImmediately = openImportImmediately;

            if (openImportImmediately){
                radioImport.Checked = true;
                btnContinue_Click(null, EventArgs.Empty);
            }
        }

        private void radioDecision_CheckedChanged(object sender, EventArgs e){
            btnContinue.Enabled = true;
        }

        private void checkBoxSelection_CheckedChanged(object sender, EventArgs e){
            CheckBox cb = (CheckBox)sender;
            SetFlag(checkBoxMap[cb], cb.Checked);
        }

        private void btnContinue_Click(object sender, EventArgs e){
            string file;

            switch(currentState){
                case State.Deciding:
                    // Reset
                    if (radioReset.Checked){
                        currentState = State.Reset;

                        Text = "Restore Defaults";
                        SelectedItems = ProfileManager.Items.UserConfig;
                    }

                    // Import
                    else if (radioImport.Checked){
                        using(OpenFileDialog dialog = new OpenFileDialog{
                            AutoUpgradeEnabled = true,
                            DereferenceLinks = true,
                            Title = "Import TweetDuck Profile",
                            Filter = "TweetDuck Profile (*.tdsettings)|*.tdsettings"
                        }){
                            if (dialog.ShowDialog() != DialogResult.OK){
                                if (openImportImmediately){
                                    Close();
                                }

                                return;
                            }

                            file = dialog.FileName;
                        }

                        importManager = new ProfileManager(file, plugins);
                        currentState = State.Import;

                        Text = "Import Profile";
                        SelectedItems = importManager.FindImportItems();

                        foreach(CheckBox cb in checkBoxMap.Keys){
                            cb.Enabled = cb.Checked;
                        }
                    }

                    // Export
                    else if (radioExport.Checked){
                        currentState = State.Export;

                        Text = "Export Profile";
                        btnContinue.Text = "Export Profile";
                        SelectedItems = ProfileManager.Items.UserConfig | ProfileManager.Items.PluginData;
                    }
                    
                    // Continue...
                    panelDecision.Visible = false;
                    panelSelection.Visible = true;
                    Height += panelSelection.Height-panelDecision.Height;
                    break;

                case State.Reset:
                    if (FormMessage.Warning("Reset TweetDuck Options", "This will reset the selected items. Are you sure you want to proceed?", FormMessage.Yes, FormMessage.No)){
                        Program.Config.ProgramRestartRequested += Config_ProgramRestartRequested;

                        if (SelectedItems.HasFlag(ProfileManager.Items.UserConfig)){
                            Program.Config.User.Reset();
                        }

                        if (SelectedItems.HasFlag(ProfileManager.Items.SystemConfig)){
                            Program.Config.System.Reset();
                        }

                        Program.Config.ProgramRestartRequested -= Config_ProgramRestartRequested;

                        if (SelectedItems.HasFlag(ProfileManager.Items.PluginData)){
                            Program.Config.Plugins.Reset();

                            try{
                                Directory.Delete(Program.PluginDataPath, true);
                            }catch(Exception ex){
                                Program.Reporter.HandleException("Profile Reset", "Could not delete plugin data.", true, ex);
                            }
                        }

                        if (SelectedItemsForceRestart){
                            RestartProgram(SelectedItems.HasFlag(ProfileManager.Items.Session) ? new string[]{ Arguments.ArgDeleteCookies } : StringUtils.EmptyArray);
                        }
                        else if (requestedRestartFromConfig){
                            if (FormMessage.Information("Profile Reset", "The application must restart for some of the restored options to take place. Do you want to restart now?", FormMessage.Yes, FormMessage.No)){
                                RestartProgram();
                            }
                        }

                        ShouldReloadBrowser = true;

                        DialogResult = DialogResult.OK;
                        Close();
                    }

                    break;

                case State.Import:
                    if (importManager.Import(SelectedItems)){
                        Program.Config.ProgramRestartRequested += Config_ProgramRestartRequested;
                        Program.Config.ReloadAll();
                        Program.Config.ProgramRestartRequested -= Config_ProgramRestartRequested;
                        
                        if (SelectedItemsForceRestart){
                            RestartProgram(SelectedItems.HasFlag(ProfileManager.Items.Session) ? new string[]{ Arguments.ArgImportCookies } : StringUtils.EmptyArray);
                        }
                        else if (requestedRestartFromConfig){
                            if (FormMessage.Information("Profile Import", "The application must restart for some of the imported options to take place. Do you want to restart now?", FormMessage.Yes, FormMessage.No)){
                                RestartProgram();
                            }
                        }
                    }
                    
                    ShouldReloadBrowser = true;

                    DialogResult = DialogResult.OK;
                    Close();
                    break;

                case State.Export:
                    using(SaveFileDialog dialog = new SaveFileDialog{
                        AddExtension = true,
                        AutoUpgradeEnabled = true,
                        OverwritePrompt = true,
                        DefaultExt = "tdsettings",
                        FileName = "TweetDuck.tdsettings",
                        Title = "Export TweetDuck Profile",
                        Filter = "TweetDuck Profile (*.tdsettings)|*.tdsettings"
                    }){
                        if (dialog.ShowDialog() != DialogResult.OK){
                            return;
                        }

                        file = dialog.FileName;
                    }
                    
                    new ProfileManager(file, plugins).Export(SelectedItems);

                    DialogResult = DialogResult.OK;
                    Close();
                    break;
            }
        }

        private void btnCancel_Click(object sender, EventArgs e){
            DialogResult = DialogResult.Cancel;
            Close();
        }

        private void Config_ProgramRestartRequested(object sender, EventArgs e){
            requestedRestartFromConfig = true;
        }

        private void SetFlag(ProfileManager.Items flag, bool enable){
            _selectedItems = enable ? _selectedItems | flag : _selectedItems & ~flag;
            btnContinue.Enabled = _selectedItems != ProfileManager.Items.None;
            
            if (currentState == State.Import){
                btnContinue.Text = SelectedItemsForceRestart ? "Import && Restart" : "Import Profile";
            }
            else if (currentState == State.Reset){
                btnContinue.Text = SelectedItemsForceRestart ? "Restore && Restart" : "Restore Defaults";
            }
        }

        private void RestartProgram(params string[] extraArgs){
            IsRestarting = true;
            Program.Restart(extraArgs);
        }
    }
}