PowerToys/Wox.Plugin.SystemPlugins/Program/ProgramSourceSetting.xaml.cs

using System;
using System.ComponentModel;
using System.Linq;
using System.Windows;
using System.Windows.Controls;
using Wox.Infrastructure.Storage.UserSettings;
using MessageBox = System.Windows.MessageBox;

namespace Wox.Plugin.SystemPlugins.Program
{
    public partial class ProgramSourceSetting : Window
    {
        private ProgramSetting settingWindow;
        private bool update;
        private ProgramSource updateProgramSource;

        public ProgramSourceSetting(ProgramSetting settingWidow)
        {
            this.settingWindow = settingWidow;
            InitializeComponent();

            this.cbType.ItemsSource = Programs.SourceTypes.Select(o => o.Key).ToList();
        }

        public void UpdateItem(ProgramSource programSource)
        {
            updateProgramSource = UserSettingStorage.Instance.ProgramSources.FirstOrDefault(o => o == programSource);
            if (updateProgramSource == null)
            {
                MessageBox.Show("Invalid program source");
                Close();
                return;
            }

            update = true;
            lblAdd.Text = "Update";
            cbEnable.IsChecked = programSource.Enabled;
            cbType.SelectedItem = programSource.Type;
            cbType.IsEnabled = false;
            tbLocation.Text = programSource.Location;
            tbBonusPoints.Text = programSource.BonusPoints.ToString();
        }

        private void BtnCancel_OnClick(object sender, RoutedEventArgs e)
        {
            Close();
        }

        private void btnAdd_OnClick(object sender, RoutedEventArgs e)
        {
            string location = tbLocation.Text;
            if (this.tbLocation.IsEnabled == true && string.IsNullOrEmpty(location))
            {
                MessageBox.Show("Please input Type field");
                return;
            }

            string type = cbType.SelectedItem as string;
            if (string.IsNullOrEmpty(type))
            {
                MessageBox.Show("Please input Type field");
                return;
            }

            int bonusPoint = 0;
            int.TryParse(this.tbBonusPoints.Text, out bonusPoint);

            if (!update)
            {
                ProgramSource p = new ProgramSource()
                {
                    Location = this.tbLocation.IsEnabled ? location : null,
                    Enabled = cbEnable.IsChecked ?? false,
                    Type = type,
                    BonusPoints = bonusPoint
                };
                if (UserSettingStorage.Instance.ProgramSources.Exists(o => o.ToString() == p.ToString() && o != p))
                {
                    MessageBox.Show("Program source already exists!");
                    return;
                }
                UserSettingStorage.Instance.ProgramSources.Add(p);
                MessageBox.Show(string.Format("Add {0} program source successfully!", p.ToString()));
            }
            else
            {
                if (UserSettingStorage.Instance.ProgramSources.Exists(o => o.ToString() == updateProgramSource.ToString() && o != updateProgramSource))
                {
                    MessageBox.Show("Program source already exists!");
                    return;
                }
                updateProgramSource.Location = this.tbLocation.IsEnabled ? location : null;
                updateProgramSource.Type = type;
                updateProgramSource.Enabled = cbEnable.IsChecked ?? false;
                updateProgramSource.BonusPoints = bonusPoint;
                MessageBox.Show(string.Format("Update {0} program source successfully!", updateProgramSource.ToString()));
            }
            UserSettingStorage.Instance.Save();
            settingWindow.ReloadProgramSourceView();
            Close();
        }

        private void cbType_SelectionChanged(object sender, SelectionChangedEventArgs e)
        {
            string item = cbType.SelectedItem as String;
            Type type;
            if (item != null && Programs.SourceTypes.TryGetValue(item, out type))
            {
                var attrs = type.GetCustomAttributes(typeof(BrowsableAttribute), false);
                if (attrs.Length > 0 && (attrs[0] as BrowsableAttribute).Browsable == false)
                {
                    this.tbLocation.IsEnabled = false;
                    return;
                }
            }
            this.tbLocation.IsEnabled = true;
        }
    }
}