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 };