diff --git a/src/modules/espresso/Espresso.Shell/Espresso.Shell.csproj b/src/modules/espresso/Espresso.Shell/Espresso.Shell.csproj index 4e2bd2fb4c..ea61957bbd 100644 --- a/src/modules/espresso/Espresso.Shell/Espresso.Shell.csproj +++ b/src/modules/espresso/Espresso.Shell/Espresso.Shell.csproj @@ -43,6 +43,10 @@ + + + + Never diff --git a/src/modules/espresso/Espresso.Shell/Program.cs b/src/modules/espresso/Espresso.Shell/Program.cs index 91393ea80e..4e16cd1135 100644 --- a/src/modules/espresso/Espresso.Shell/Program.cs +++ b/src/modules/espresso/Espresso.Shell/Program.cs @@ -4,6 +4,7 @@ using Espresso.Shell.Core; using Espresso.Shell.Models; +using ManagedCommon; using Newtonsoft.Json; using NLog; using System; @@ -89,16 +90,30 @@ namespace Espresso.Shell timeOption.Required = false; + var powerToysPidOption = new Option( + aliases: new[] { "--ptpid", "-p" }, + getDefaultValue: () => 0, + description: "Reference to PowerToys PID, when executed under the application context.") + { + Argument = new Argument(() => 0) + { + Arity = ArgumentArity.ZeroOrOne, + }, + }; + + powerToysPidOption.Required = false; + var rootCommand = new RootCommand { configOption, displayOption, - timeOption + timeOption, + powerToysPidOption }; rootCommand.Description = appName; - rootCommand.Handler = CommandHandler.Create(HandleCommandLineArguments); + rootCommand.Handler = CommandHandler.Create(HandleCommandLineArguments); return rootCommand.InvokeAsync(args).Result; } @@ -111,11 +126,12 @@ namespace Espresso.Shell Environment.Exit(exitCode); } - private static void HandleCommandLineArguments(string config, bool displayOn, long timeLimit) + private static void HandleCommandLineArguments(string config, bool displayOn, long timeLimit, int ptpid) { log.Info($"The value for --config is: {config}"); log.Info($"The value for --display-on is: {displayOn}"); log.Info($"The value for --time-limit is: {timeLimit}"); + log.Info($"The value for --ptpid is: {ptpid}"); if (!string.IsNullOrWhiteSpace(config)) { @@ -150,6 +166,14 @@ namespace Espresso.Shell log.Info(errorString); log.Debug(errorString); } + + if (ptpid != 0) + { + RunnerHelper.WaitForPowerToysRunner(ptpid, () => + { + Environment.Exit(0); + }); + } } else { diff --git a/src/modules/espresso/Espresso/dllmain.cpp b/src/modules/espresso/Espresso/dllmain.cpp index 6b7b897ceb..1b877910a3 100644 --- a/src/modules/espresso/Espresso/dllmain.cpp +++ b/src/modules/espresso/Espresso/dllmain.cpp @@ -81,8 +81,8 @@ private: // Get the configuration file that will be passed to the process. std::wstring espresso_settings_location = PTSettingsHelper::get_module_save_file_location(MODULE_NAME); - std::wstring executable_args = L"--config " + espresso_settings_location; - //executable_args.append(std::to_wstring(powertoys_pid)); + std::wstring executable_args = L"--config " + espresso_settings_location + L" --ptpid " + std::to_wstring(powertoys_pid); + Logger::trace(L"Espresso launching with parameters: " + executable_args); SHELLEXECUTEINFOW sei{ sizeof(sei) }; sei.fMask = { SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_NO_UI };