#pragma once
#include "pre.hpp"

#include <functional>

#include "base.hpp"
#include "util/format.hpp"

namespace cru::debug {
#ifdef CRU_DEBUG
void DebugMessage(const StringView& message);
#else
inline void DebugMessage(const StringView& message) {}
#endif

#ifdef CRU_DEBUG
inline void DebugTime(const std::function<void()>& action,
                      const StringView& hint_message) {
  const auto before = std::chrono::steady_clock::now();
  action();
  const auto after = std::chrono::steady_clock::now();
  const auto duration =
      std::chrono::duration_cast<std::chrono::milliseconds>(after - before);
  DebugMessage(util::Format(L"{}: {}ms.\n", hint_message, duration.count()));
}

template <typename TReturn>
TReturn DebugTime(const std::function<TReturn()>& action,
                  const StringView& hint_message) {
  const auto before = std::chrono::steady_clock::now();
  auto&& result = action();
  const auto after = std::chrono::steady_clock::now();
  const auto duration =
      std::chrono::duration_cast<std::chrono::milliseconds>(after - before);
  DebugMessage(util::Format(L"{}: {}ms.\n", hint_message, duration.count()));
  return std::move(result);
}
#else
inline void DebugTime(const std::function<void()>& action,
                      const StringView& hint_message) {
  action();
}

template <typename TReturn>
TReturn DebugTime(const std::function<TReturn()>& action,
                  const StringView& hint_message) {
  return action();
}
#endif
}  // namespace cru::debug