PowerToys/Wox/Storage/TopMostRecordStorage.cs

using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using Wox.Infrastructure.Storage;

namespace Wox.Storage
{
    public class TopMostRecordStorage : JsonStrorage<TopMostRecordStorage>
    {
        public Dictionary<string, TopMostRecord> records = new Dictionary<string, TopMostRecord>();

        protected override string ConfigFolder
        {
            get { return Path.Combine(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location), "Config"); }
        }

        protected override string ConfigName
        {
            get { return "TopMostRecords"; }
        }

        internal bool IsTopMost(Plugin.Result result)
        {
            return records.Any(o => o.Value.Title == result.Title
                && o.Value.SubTitle == result.SubTitle
                && o.Value.PluginID == result.PluginID
                && o.Key == result.OriginQuery.RawQuery);
        }

        internal void Remove(Plugin.Result result)
        {
            if (records.ContainsKey(result.OriginQuery.RawQuery))
            {
                records.Remove(result.OriginQuery.RawQuery);
                Save();
            }
        }

        internal void AddOrUpdate(Plugin.Result result)
        {
            if (records.ContainsKey(result.OriginQuery.RawQuery))
            {
                records[result.OriginQuery.RawQuery].Title = result.Title;
                records[result.OriginQuery.RawQuery].SubTitle = result.SubTitle;
                records[result.OriginQuery.RawQuery].PluginID = result.PluginID;
            }
            else
            {
                records.Add(result.OriginQuery.RawQuery, new TopMostRecord()
                    {
                        PluginID = result.PluginID,
                        Title = result.Title,
                        SubTitle = result.SubTitle,
                    });
            }

            Save();
        }
    }


    public class TopMostRecord
    {
        public string Title { get; set; }
        public string SubTitle { get; set; }
        public string PluginID { get; set; }
    }
}