///
/// Created by Anonymous275 on 1/21/22
/// Copyright (c) 2021-present Anonymous275 read the LICENSE file for more info.
///
#define WIN32_LEAN_AND_MEAN
#include <MinHook.h>
#include "Memory/Memory.h"

#pragma once
template<class FuncType>
class Hook {
   FuncType targetPtr;
   FuncType detourFunc;
   bool Attached = false;

   public:
   Hook(FuncType src, FuncType dest) : targetPtr(src), detourFunc(dest) {
      auto status =
          MH_CreateHook((void*)targetPtr, (void*)detourFunc, (void**)&Original);
      if (status != MH_OK) {
         Memory::Print(std::string("MH Error -> ") + MH_StatusToString(status));
         return;
      }
   }

   void Enable() {
      if (!Attached) {
         auto status = MH_EnableHook((void*)targetPtr);
         if (status != MH_OK) {
            Memory::Print(std::string("MH Error -> ") +
                          MH_StatusToString(status));
            return;
         }
         Attached = true;
      }
   }

   void Disable() {
      if (Attached) {
         auto status = MH_DisableHook((void*)targetPtr);
         if (status != MH_OK) {
            Memory::Print(std::string("MH Error -> ") +
                          MH_StatusToString(status));
            return;
         }
         Attached = false;
      }
   }

   FuncType Original{};
};