PowerToys/src/modules/cmdpal/ext/Microsoft.CmdPal.Ext.PowerToys/Program.cs

// 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 System.Threading;
using ManagedCommon;
using Microsoft.CommandPalette.Extensions;

namespace PowerToysExtension;

public class Program
{
    [MTAThread]
    public static void Main(string[] args)
    {
        try
        {
            // Initialize per-extension log under CmdPal/PowerToysExtension.
            Logger.InitializeLogger("\\CmdPal\\PowerToysExtension\\Logs");
            Logger.LogInfo("PowerToysExtension starting (args: " + string.Join(' ', args) + ")");
        }
        catch
        {
            // Continue even if logging fails.
        }

        try
        {
            if (args.Length > 0 && args[0] == "-RegisterProcessAsComServer")
            {
                Logger.LogInfo("RegisterProcessAsComServer mode detected.");
                using ExtensionServer server = new();

                ManualResetEvent extensionDisposedEvent = new(false);

                // We are instantiating an extension instance once above, and returning it every time the callback in RegisterExtension below is called.
                // This makes sure that only one instance of SampleExtension is alive, which is returned every time the host asks for the IExtension object.
                // If you want to instantiate a new instance each time the host asks, create the new instance inside the delegate.
                PowerToysExtension extensionInstance = new(extensionDisposedEvent);
                server.RegisterExtension(() => extensionInstance);
                Logger.LogInfo("Extension instance registered; waiting for disposal signal.");

                // This will make the main thread wait until the event is signalled by the extension class.
                // Since we have single instance of the extension object, we exit as soon as it is disposed.
                extensionDisposedEvent.WaitOne();
                Logger.LogInfo("Extension disposed signal received; exiting server loop.");
            }
            else
            {
                Console.WriteLine("Not being launched as a Extension... exiting.");
                Logger.LogInfo("Exited: not launched with -RegisterProcessAsComServer.");
            }
        }
        finally
        {
        }
    }
}