Flow.Launcher/Plugins/Flow.Launcher.Plugin.Program/Views/Commands/ProgramSettingDisplay.cs

using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Flow.Launcher.Plugin.Program.Views.Models;

namespace Flow.Launcher.Plugin.Program.Views.Commands
{
    internal static class ProgramSettingDisplay
    {
        internal static List<ProgramSource> LoadProgramSources()
        {
            // Even though these are disabled, we still want to display them so users can enable later on
            return Main._settings
                       .DisabledProgramSources
                       .Union(Main._settings.ProgramSources)
                       .ToList();
        }

        internal static async Task DisplayAllProgramsAsync()
        {
            await Main._win32sLock.WaitAsync();
            try
            {
                var win32 = Main._win32s
                            .Where(t1 => !ProgramSetting.ProgramSettingDisplayList.Any(x => x.UniqueIdentifier == t1.UniqueIdentifier))
                            .Select(x => new ProgramSource(x));
                ProgramSetting.ProgramSettingDisplayList.AddRange(win32);
            }
            finally
            {
                Main._win32sLock.Release();
            }

            await Main._uwpsLock.WaitAsync();
            try
            {
                var uwp = Main._uwps
                            .Where(t1 => !ProgramSetting.ProgramSettingDisplayList.Any(x => x.UniqueIdentifier == t1.UniqueIdentifier))
                            .Select(x => new ProgramSource(x));
                ProgramSetting.ProgramSettingDisplayList.AddRange(uwp);
            }
            finally
            {
                Main._uwpsLock.Release();
            }
        }

        internal static async Task SetProgramSourcesStatusAsync(List<ProgramSource> selectedProgramSourcesToDisable, bool status)
        {
            foreach(var program in ProgramSetting.ProgramSettingDisplayList)
            {
                if (selectedProgramSourcesToDisable.Any(x => x.UniqueIdentifier == program.UniqueIdentifier && program.Enabled != status))
                {
                    program.Enabled = status;
                }
            }

            await Main._win32sLock.WaitAsync();
            try
            {
                foreach (var program in Main._win32s)
                {
                    if (selectedProgramSourcesToDisable.Any(x => x.UniqueIdentifier == program.UniqueIdentifier && program.Enabled != status))
                    {
                        program.Enabled = status;
                    }
                }
            }
            finally
            {
                Main._win32sLock.Release();
            }

            await Main._uwpsLock.WaitAsync();
            try
            {
                foreach (var program in Main._uwps)
                {
                    if (selectedProgramSourcesToDisable.Any(x => x.UniqueIdentifier == program.UniqueIdentifier && program.Enabled != status))
                    {
                        program.Enabled = status;
                    }
                }
            }
            finally
            {
                Main._uwpsLock.Release();
            }
        }

        internal static void StoreDisabledInSettings()
        {
            // Disabled, not in DisabledProgramSources or ProgramSources
            var tmp = ProgramSetting.ProgramSettingDisplayList
                .Where(t1 => !t1.Enabled
                                && !Main._settings.DisabledProgramSources.Any(x => x.UniqueIdentifier == t1.UniqueIdentifier)
                                && !Main._settings.ProgramSources.Any(x => x.UniqueIdentifier == t1.UniqueIdentifier));

            Main._settings.DisabledProgramSources.AddRange(tmp);
        }

        internal static void RemoveDisabledFromSettings()
        {
            Main._settings.DisabledProgramSources.RemoveAll(t1 => t1.Enabled);
        }

        internal static async Task<bool> IsReindexRequiredAsync(this List<ProgramSource> selectedItems)
        {
            // Not in cache
            await Main._win32sLock.WaitAsync();
            await Main._uwpsLock.WaitAsync();
            try
            {
                if (selectedItems.Any(t1 => t1.Enabled && !Main._uwps.Any(x => t1.UniqueIdentifier == x.UniqueIdentifier))
                && selectedItems.Any(t1 => t1.Enabled && !Main._win32s.Any(x => t1.UniqueIdentifier == x.UniqueIdentifier)))
                    return true;
            }
            finally
            {
                Main._win32sLock.Release();
                Main._uwpsLock.Release();
            }

            // ProgramSources holds list of user added directories, 
            // so when we enable/disable we need to reindex to show/not show the programs
            // that are found in those directories.
            if (selectedItems.Any(t1 => Main._settings.ProgramSources.Any(x => t1.UniqueIdentifier == x.UniqueIdentifier)))
                return true;

            return false;
        }
    }
}