microsoft · krschau · Apr 22, 2024 · Apr 16, 2024 · Apr 16, 2024 · Apr 17, 2024
diff --git a/src/App.xaml.cs b/src/App.xaml.cs
@@ -104,6 +104,7 @@ public App()
             // Other Activation Handlers
             services.AddTransient<IActivationHandler, ProtocolActivationHandler>();
             services.AddTransient<IActivationHandler, DSCFileActivationHandler>();
+            services.AddTransient<IActivationHandler, AppInstallActivationHandler>();
 
             // Services
             services.AddSingleton<ILocalSettingsService, LocalSettingsService>();

diff --git a/src/Services/AppInstallActivationHandler.cs b/src/Services/AppInstallActivationHandler.cs
@@ -0,0 +1,113 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+using System.Diagnostics;
+using System.Web;
+using DevHome.Activation;
+using DevHome.Common.Extensions;
+using DevHome.Common.Services;
+using DevHome.Settings.ViewModels;
+using DevHome.SetupFlow.Services;
+using DevHome.SetupFlow.ViewModels;
+using Microsoft.UI.Xaml;
+using Serilog;
+using Windows.ApplicationModel.Activation;
+using Windows.Storage;
+
+namespace DevHome.Services;
+
+/// <summary>
+/// Class that handles the activation of the application when an add-apps-to-cart URI protcol is used.
+/// </summary>
+public class AppInstallActivationHandler : ActivationHandler<ProtocolActivatedEventArgs>
+{
+    private readonly ILogger _log = Log.ForContext("SourceContext", nameof(AppInstallActivationHandler));
+    private const string AppSearchUri = "add-apps-to-cart";
+    private readonly INavigationService _navigationService;
+    private readonly SetupFlowViewModel _setupFlowViewModel;
+    private readonly IWindowsPackageManager _windowsPackageManager;
+    private readonly PackageProvider _packageProvider;
+    private readonly SetupFlowOrchestrator _setupFlowOrchestrator;
+    private static readonly char[] Separator = [','];
+
+    public AppInstallActivationHandler(
+        INavigationService navigationService,
+        SetupFlowViewModel setupFlowViewModel,
+        PackageProvider packageProvider,
+        IWindowsPackageManager wpm,
+        SetupFlowOrchestrator setupFlowOrchestrator)
+    {
+        _navigationService = navigationService;
+        _setupFlowViewModel = setupFlowViewModel;
+        _packageProvider = packageProvider;
+        _windowsPackageManager = wpm;
+        _setupFlowOrchestrator = setupFlowOrchestrator;
+    }
+
+    protected override bool CanHandleInternal(ProtocolActivatedEventArgs args)
+    {
+        return args.Uri != null && args.Uri.AbsolutePath.Equals(AppSearchUri, StringComparison.OrdinalIgnoreCase);
+    }
+
+    protected async override Task HandleInternalAsync(ProtocolActivatedEventArgs args)
+    {
+        await AppActivationFlowAsync(args.Uri.Query);
+    }
+
+    private async Task AppActivationFlowAsync(string query)
+    {
+        try
+        {
+            // Don't interrupt the user if the machine configuration is in progress
+            if (_setupFlowOrchestrator.IsMachineConfigurationInProgress)
+            {
+                _log.Warning("Cannot activate the add-apps-to-cart flow because the machine configuration is in progress");
+                return;
+            }
+            else
+            {
+                _log.Information("Starting add-apps-to-cart activation");
+                _navigationService.NavigateTo(typeof(SetupFlowViewModel).FullName!);
+                _setupFlowViewModel.StartAppManagementFlow(query);
+                await SearchAndSelectAsync(query);
+            }
+        }
+        catch (Exception ex)
+        {
+            _log.Error(ex, "Error executing the add-apps-to-cart activation flow");
+        }
+    }
+
+    private async Task SearchAndSelectAsync(string query)
+    {
+        var parameters = HttpUtility.ParseQueryString(query);
+        var searchParameter = parameters["search"];
+
+        if (string.IsNullOrEmpty(searchParameter))
+        {
+            _log.Warning("Search parameter is missing or empty in the query.");
+            return;
+        }
+
+        // Currently using the first search term only
+        var firstSearchTerm = searchParameter.Split(Separator, StringSplitOptions.RemoveEmptyEntries)
+                                             .Select(term => term.Trim(' ', '"'))
+                                             .FirstOrDefault();
+
+        if (string.IsNullOrEmpty(firstSearchTerm))
+        {
+            _log.Warning("No valid search term was extracted from the query.");
+            return;
+        }
+
+        var searchResults = await _windowsPackageManager.SearchAsync(firstSearchTerm, 1);
+        if (searchResults.Count == 0)
+        {
+            _log.Warning("No results found for the search term: {SearchTerm}", firstSearchTerm);
+            return;
+        }
+
+        var firstResult = _packageProvider.CreateOrGet(searchResults[0]);
+        firstResult.IsSelected = true;
+    }
+}
diff --git a/src/Services/ProtocolActivationHandler.cs b/src/Services/ProtocolActivationHandler.cs
@@ -19,14 +19,14 @@ public ProtocolActivationHandler(INavigationService navigationService)
         this._navigationService = navigationService;
     }
 
-    protected override Task HandleInternalAsync(ProtocolActivatedEventArgs args)
+    protected override bool CanHandleInternal(ProtocolActivatedEventArgs args)
     {
-        if (args.Uri.AbsolutePath == SettingsAccountsUri)
-        {
-            _navigationService.DefaultPage = typeof(AccountsViewModel).FullName!;
-            _navigationService.NavigateTo(_navigationService.DefaultPage);
-        }
+        return args.Uri != null && args.Uri.AbsolutePath.Equals(SettingsAccountsUri, StringComparison.OrdinalIgnoreCase);
+    }
 
+    protected override Task HandleInternalAsync(ProtocolActivatedEventArgs args)
+    {
+        _navigationService.NavigateTo(typeof(AccountsViewModel).FullName!);
         return Task.CompletedTask;
     }
 }
diff --git a/src/ViewModels/ShellViewModel.cs b/src/ViewModels/ShellViewModel.cs
@@ -57,6 +57,8 @@ public async Task OnLoaded()
             case ExtendedActivationKind.File:
                 // Allow the file activation handler to navigate to the appropriate page.
                 break;
+            case ExtendedActivationKind.Protocol:
+                break;
             case ExtendedActivationKind.Launch:
             default:
                 var isNotFirstRun = await _localSettingsService.ReadSettingAsync<bool>(WellKnownSettingsKeys.IsNotFirstRun);

diff --git a/tools/SetupFlow/DevHome.SetupFlow/TaskGroups/AppManagementTaskGroup.cs b/tools/SetupFlow/DevHome.SetupFlow/TaskGroups/AppManagementTaskGroup.cs
@@ -3,6 +3,7 @@
 
 using System.Collections.Generic;
 using System.Linq;
+using System.Web;
 using DevHome.SetupFlow.Models;
 using DevHome.SetupFlow.Services;
 using DevHome.SetupFlow.ViewModels;
@@ -35,4 +36,14 @@ public AppManagementTaskGroup(
     public SetupPageViewModelBase GetSetupPageViewModel() => _appManagementViewModel;
 
     public ReviewTabViewModelBase GetReviewTabViewModel() => _appManagementReviewViewModel;
+
+    public void HandleSearchQuery(string query)
+    {
+        var searchParameter = HttpUtility.ParseQueryString(query)["search"];
+        if (!string.IsNullOrEmpty(searchParameter))
+        {
+            var trimmedSearchParameter = searchParameter.Trim('\"');
+            _appManagementViewModel.PerformSearch(trimmedSearchParameter);
+        }
+    }
 }
diff --git a/tools/SetupFlow/DevHome.SetupFlow/ViewModels/AppManagementViewModel.cs b/tools/SetupFlow/DevHome.SetupFlow/ViewModels/AppManagementViewModel.cs
@@ -23,6 +23,9 @@ public partial class AppManagementViewModel : SetupPageViewModelBase
     private readonly PackageCatalogListViewModel _packageCatalogListViewModel;
     private readonly PackageProvider _packageProvider;
 
+    [ObservableProperty]
+    private string _searchText;
+
     /// <summary>
     /// Current view to display in the main content control
     /// </summary>
@@ -127,4 +130,9 @@ private void OnPackageSelectionChanged(object sender, EventArgs args)
         // Show warning if any selected package is installed
         ShowInstalledPackageWarning = SelectedPackages.Any(p => !p.CanInstall);
     }
+
+    internal void PerformSearch(string searchParameter)
+    {
+        SearchText = searchParameter;
+    }
 }
diff --git a/tools/SetupFlow/DevHome.SetupFlow/ViewModels/MainPageViewModel.cs b/tools/SetupFlow/DevHome.SetupFlow/ViewModels/MainPageViewModel.cs
@@ -103,6 +103,14 @@ public async Task StartConfigurationFileAsync(StorageFile file)
         }
     }
 
+    internal void StartAppManagementFlow(string query)
+    {
+        _log.Information($"Launching app management flow for query:{query}");
+        var appManagementSetupFlow = _host.GetService<AppManagementTaskGroup>();
+        StartSetupFlowForTaskGroups(null, "App Search URI", appManagementSetupFlow);
+        appManagementSetupFlow.HandleSearchQuery(query);
+    }
+
     protected async override Task OnFirstNavigateToAsync()
     {
         if (await ValidateAppInstallerAsync())

diff --git a/tools/SetupFlow/DevHome.SetupFlow/ViewModels/SetupFlowViewModel.cs b/tools/SetupFlow/DevHome.SetupFlow/ViewModels/SetupFlowViewModel.cs
@@ -143,4 +143,10 @@ public void OnNavigatedTo(NavigationEventArgs args)
             StartCreationFlowAsync();
         }
     }
+
+    public void StartAppManagementFlow(string query)
+    {
+        Orchestrator.FlowPages = [_mainPageViewModel];
+        _mainPageViewModel.StartAppManagementFlow(query);
+    }
 }
diff --git a/tools/SetupFlow/DevHome.SetupFlow/Views/AppManagementView.xaml b/tools/SetupFlow/DevHome.SetupFlow/Views/AppManagementView.xaml
@@ -57,6 +57,7 @@
                 <!-- Search bar -->
                 <AutoSuggestBox
                     x:Name="SearchBox"
+                    Text="{x:Bind ViewModel.SearchText, Mode=TwoWay}"
                     QueryIcon="Find"
                     HorizontalAlignment="Left"
                     Width="400"