/**
 * Copyright (c) 2019 Kylart <kylart.dev@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * Coming right from https://github.com/nodejs/node-addon-api/issues/269#issuecomment-455580129
 */

#ifdef _MSC_VER

#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif

#include <windows.h>
#include <Shlwapi.h>
#include <delayimp.h>
#include <string.h>

#pragma comment(lib, "Shlwapi.lib")

static FARPROC WINAPI load_exe_hook(unsigned int event, DelayLoadInfo* info) {
  if (event != dliNotePreLoadLibrary)
    return NULL;

  if (_stricmp(info->szDll, "iojs.exe") != 0 &&
    _stricmp(info->szDll, "node.exe") != 0 &&
    _stricmp(info->szDll, "node.dll") != 0)
    return NULL;

  // Get a handle to the current process executable.
  HMODULE processModule = GetModuleHandle(NULL);

  // Get the path to the executable.
  TCHAR processPath[_MAX_PATH];  // NOLINT
  GetModuleFileName(processModule, processPath, _MAX_PATH);

  // Get the name of the current executable.
  LPSTR processName = PathFindFileName(processPath);

  // If the current process is node or iojs, then just return the proccess module.
  if (_stricmp(processName, "node.exe") == 0 ||
    _stricmp(processName, "iojs.exe") == 0) {
    return (FARPROC)processModule;
  }

  // If it is another process, attempt to load 'node.dll' from the same directory.
  PathRemoveFileSpec(processPath);
  PathAppend(processPath, "node.dll");

  HMODULE nodeDllModule = GetModuleHandle(processPath);
  if (nodeDllModule != NULL) {
    // This application has a node.dll in the same directory as the executable, use that.
    return (FARPROC)nodeDllModule;
  }

  // Fallback to the current executable, which must statically link to node.lib.
  return (FARPROC)processModule;
}

// See https://docs.microsoft.com/en-us/cpp/build/reference/notification-hooks
decltype(__pfnDliNotifyHook2) __pfnDliNotifyHook2 = load_exe_hook;

#endif