NotificationCallback.h

Enable keyboard shortcuts

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim: set ts=2 et sw=2 tw=80: */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef NotificationCallback_h__

#define NotificationCallback_h__

#include <filesystem>

#include <tuple>

#include <unknwn.h>

#include <wrl.h>

#include "mozilla/Maybe.h"

#include "nsWindowsHelpers.h"

using namespace Microsoft::WRL;

using namespace std::filesystem;

// Windows 10+ declarations.

// TODO remove declarations and add `#include

// <notificationactivationcallback.h>` when Windows 10 is the minimum supported.

typedef struct NOTIFICATION_USER_INPUT_DATA {

  LPCWSTR Key;

  LPCWSTR Value;

} NOTIFICATION_USER_INPUT_DATA;

MIDL_INTERFACE("53E31837-6600-4A81-9395-75CFFE746F94")

INotificationActivationCallback : public IUnknown {

 public:

  virtual HRESULT STDMETHODCALLTYPE Activate(

      LPCWSTR appUserModelId, LPCWSTR invokedArgs,

      const NOTIFICATION_USER_INPUT_DATA* data, ULONG count) = 0;

};

struct ToastArgs {

  std::wstring profile;

  std::wstring windowsTag;

  std::wstring action;

};

class NotificationCallback final

    : public RuntimeClass<RuntimeClassFlags<ClassicCom>,

                          INotificationActivationCallback> {

 public:

  HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject) final;

  HRESULT STDMETHODCALLTYPE Activate(LPCWSTR appUserModelId,

                                     LPCWSTR invokedArgs,

                                     const NOTIFICATION_USER_INPUT_DATA* data,

                                     ULONG dataCount) final;

  explicit NotificationCallback(const GUID& runtimeGuid,

                                const path& dllInstallDir)

      : guid(runtimeGuid), installDir(dllInstallDir) {}

 private:

  const GUID guid = {};

  const path installDir = {};

  void HandleActivation(LPCWSTR invokedArgs);

  mozilla::Maybe<ToastArgs> ParseToastArguments(LPCWSTR invokedArgs);

  std::tuple<path, mozilla::UniquePtr<wchar_t[]>> BuildRunCommand(

      const ToastArgs& args);

  static mozilla::Maybe<nsAutoHandle> CreatePipe(const std::wstring& tag);

  static bool ConnectPipeWithTimeout(const nsAutoHandle& pipe);

  static void HandlePipeMessages(const nsAutoHandle& pipe);

  static DWORD TransferForegroundPermission(const DWORD pid);

};

#endif