src/modules/cmdpal/ext/Microsoft.CmdPal.Ext.Apps/Storage/PackageRepository.cs - main - githubmirror/PowerToys

src/modules/cmdpal/ext/Microsoft.CmdPal.Ext.Apps/Storage/PackageRepository.cs
124 строки4 KB
Michael Jolley
CmdPal: Linq clean-up (All Apps) (#41551)
26 сен 2025, 19:15
26 сен 2025, 19:153bdb530
// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System;
using ManagedCommon;
using Microsoft.CmdPal.Ext.Apps.Programs;
using Microsoft.CmdPal.Ext.Apps.Utils;
using Windows.ApplicationModel;

namespace Microsoft.CmdPal.Ext.Apps.Storage;

/// <summary>
/// A repository for storing packaged applications such as UWP apps or appx packaged desktop apps.
/// This repository will also monitor for changes to the PackageCatalog and update the repository accordingly
/// </summary>
internal sealed partial class PackageRepository : ListRepository<IUWPApplication>, IProgramRepository
{
    private readonly IPackageCatalog _packageCatalog;

    private bool _isDirty;

    public bool ShouldReload()
    {
        return _isDirty;
    }

    public void ResetReloadFlag()
    {
        _isDirty = false;
    }

    // private readonly PluginInitContext _context;
    public PackageRepository(IPackageCatalog packageCatalog)
    {
        _packageCatalog = packageCatalog ?? throw new ArgumentNullException(nameof(packageCatalog), "PackageRepository expects an interface to be able to subscribe to package events");

        _packageCatalog.PackageInstalling += OnPackageInstalling;
        _packageCatalog.PackageUninstalling += OnPackageUninstalling;
        _packageCatalog.PackageUpdating += OnPackageUpdating;
    }

    public void OnPackageInstalling(PackageCatalog p, PackageInstallingEventArgs args)
    {
        if (args.IsComplete)
        {
            AddPackage(args.Package);
        }
    }

    public void OnPackageUninstalling(PackageCatalog p, PackageUninstallingEventArgs args)
    {
        if (args.Progress == 0)
        {
            RemovePackage(args.Package);
        }
    }

    public void OnPackageUpdating(PackageCatalog p, PackageUpdatingEventArgs args)
    {
        if (args.Progress == 0)
        {
            RemovePackage(args.SourcePackage);
        }

        if (args.IsComplete)
        {
            AddPackage(args.TargetPackage);
        }
    }

    private void AddPackage(Package package)
    {
        var packageWrapper = PackageWrapper.GetWrapperFromPackage(package);
        if (string.IsNullOrEmpty(packageWrapper.InstalledLocation))
        {
            return;
        }

        try
        {
            var uwp = new UWP(packageWrapper);
            uwp.InitializeAppInfo(packageWrapper.InstalledLocation);
            foreach (var app in uwp.Apps)
            {
                app.UpdateLogoPath(ThemeHelper.GetCurrentTheme());
                Add(app);
                _isDirty = true;
            }
        }

        // InitializeAppInfo will throw if there is no AppxManifest.xml for the package.
        // Note there are sometimes multiple packages per product and this doesn't necessarily mean that we haven't found the app.
        // e.g. "Could not find file 'C:\\Program Files\\WindowsApps\\Microsoft.WindowsTerminalPreview_2020.616.45.0_neutral_~_8wekyb3d8bbwe\\AppxManifest.xml'."
        catch (System.IO.FileNotFoundException ex)
        {
            Logger.LogError(ex.Message);
        }
    }

    private void RemovePackage(Package package)
    {
        // find apps associated with this package.
        var packageWrapper = PackageWrapper.GetWrapperFromPackage(package);
        var uwp = new UWP(packageWrapper);

        foreach (var app in Items)
        {
            if (!app.Package.Equals(uwp))
            {
                continue;
            }

            Remove(app);
            _isDirty = true;
        }
    }

    public void IndexPrograms()
    {
        var applications = UWP.All();
        SetList(applications);
    }
}