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 System.Threading.Tasks;
using ManagedCommon;
using Microsoft.CommandPalette.Extensions;
using Shmuelie.WinRTServer;
using Shmuelie.WinRTServer.CsWinRT;

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");
        }
        catch
        {
            // Continue even if logging fails.
        }

        if (args.Length > 0 && args[0] == "-RegisterProcessAsComServer")
        {
            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);

            // 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();
        }
        else
        {
            Console.WriteLine("Not being launched as a Extension... exiting.");
        }
    }
}