/* 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;
  bool skipNotificationServer;
};

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 mozilla::Maybe<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
