#pragma once
#include <condition_variable>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <queue>
#include <vector>
#include <thread>

namespace utils {

class ThreadPool {
 public:
  ThreadPool();
  ThreadPool(std::uint32_t numThreads);
  ~ThreadPool();

  void init(std::uint32_t numThreads);
  void terminate();

  void respawnWorker(std::uint32_t numThreads);

  template <typename _Callable, typename... _Args>
    requires(
        !std::is_void_v<std::invoke_result_t<_Callable, ThreadPool*, _Args...>>)
  int enqueueJob(_Callable&& __job,
                 std::invoke_result_t<_Callable, _Args...>& retVal,
                 _Args&&... __args) {
    if (terminate_) {
      spdlog::error("Cannot run jobs on threads that terminating...");
      return -1;
    }

    std::lock_guard<std::mutex> lock(jobQueueMutex);
    auto boundFunc = [this, &retVal, __job,
                      ... __args = std::move(__args)]() mutable {
      retVal = __job(this, std::move(__args)...);
    };
    auto task = std::packaged_task<void()>(std::move(boundFunc));
    jobs_.push(std::move(task));
    jobQueueCV_.notify_one();

    return 0;
  }
  template <typename _Callable, typename... _Args>
    requires std::is_void_v<
        std::invoke_result_t<_Callable, ThreadPool*, _Args...>>
  int enqueueJob(_Callable&& __job, _Args&&... __args) {
    if (terminate_) {
      spdlog::error("Cannot run jobs on threads that terminating...");
      return -1;
    }

    std::lock_guard<std::mutex> lock(jobQueueMutex);
    auto boundFunc = [this, __job, ... __args = std::move(__args)]() mutable {
      __job(this, std::move(__args)...);
    };
    auto task = std::packaged_task<void()>(std::move(boundFunc));
    jobs_.push(std::move(task));
    jobQueueCV_.notify_one();

    return 0;
  }

  int threadCount = 0;

 private:
  void* Worker();

  std::condition_variable jobQueueCV_;
  std::mutex jobQueueMutex;
  std::queue<std::packaged_task<void()>> jobs_;
  std::vector<std::thread> workers_;
  bool terminate_ = false;
};

}  // namespace Chattr