105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
#include <hex/helpers/logger.hpp>
|
|
#include <hex/helpers/fs.hpp>
|
|
#include <hex/helpers/fmt.hpp>
|
|
|
|
#include <wolv/io/file.hpp>
|
|
|
|
#include <chrono>
|
|
#include <fmt/chrono.h>
|
|
#include <hex/api/task_manager.hpp>
|
|
|
|
#if defined(OS_WINDOWS)
|
|
#include <Windows.h>
|
|
#endif
|
|
|
|
namespace hex::log::impl {
|
|
|
|
static wolv::io::File s_loggerFile;
|
|
std::mutex g_loggerMutex;
|
|
|
|
FILE *getDestination() {
|
|
if (s_loggerFile.isValid())
|
|
return s_loggerFile.getHandle();
|
|
else
|
|
return stdout;
|
|
}
|
|
|
|
wolv::io::File& getFile() {
|
|
return s_loggerFile;
|
|
}
|
|
|
|
bool isRedirected() {
|
|
return s_loggerFile.isValid();
|
|
}
|
|
|
|
void redirectToFile() {
|
|
if (s_loggerFile.isValid()) return;
|
|
|
|
for (const auto &path : fs::getDefaultPaths(fs::ImHexPath::Logs, true)) {
|
|
wolv::io::fs::createDirectories(path);
|
|
s_loggerFile = wolv::io::File(path / hex::format("{0:%Y%m%d_%H%M%S}.log", fmt::localtime(std::chrono::system_clock::to_time_t(std::chrono::system_clock::now()))), wolv::io::File::Mode::Create);
|
|
s_loggerFile.disableBuffering();
|
|
|
|
if (s_loggerFile.isValid()) break;
|
|
}
|
|
}
|
|
|
|
void enableColorPrinting() {
|
|
#if defined(OS_WINDOWS)
|
|
auto hConsole = ::GetStdHandle(STD_OUTPUT_HANDLE);
|
|
if (hConsole != INVALID_HANDLE_VALUE) {
|
|
DWORD mode = 0;
|
|
if (::GetConsoleMode(hConsole, &mode) == TRUE) {
|
|
mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING | ENABLE_PROCESSED_OUTPUT;
|
|
::SetConsoleMode(hConsole, mode);
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
std::vector<LogEntry>& getLogEntries() {
|
|
static std::vector<LogEntry> logEntries;
|
|
return logEntries;
|
|
}
|
|
|
|
void addLogEntry(std::string_view project, std::string_view level, std::string_view message) {
|
|
getLogEntries().emplace_back(project.data(), level.data(), message.data());
|
|
}
|
|
|
|
|
|
void printPrefix(FILE *dest, const fmt::text_style &ts, const std::string &level, const char *projectName) {
|
|
const auto now = fmt::localtime(std::chrono::system_clock::to_time_t(std::chrono::system_clock::now()));
|
|
|
|
fmt::print(dest, "[{0:%H:%M:%S}] ", now);
|
|
|
|
if (isRedirected())
|
|
fmt::print(dest, "{0} ", level);
|
|
else
|
|
fmt::print(dest, ts, "{0} ", level);
|
|
|
|
std::string projectThreadTag = projectName;
|
|
if (auto threadName = TaskManager::getCurrentThreadName(); !threadName.empty())
|
|
projectThreadTag += fmt::format(" | {0}", threadName);
|
|
|
|
constexpr static auto MaxTagLength = 25;
|
|
if (projectThreadTag.length() > MaxTagLength)
|
|
projectThreadTag.resize(MaxTagLength);
|
|
|
|
fmt::print(dest, "[{0}] ", projectThreadTag);
|
|
|
|
const auto projectNameLength = projectThreadTag.length();
|
|
fmt::print(dest, "{0}", std::string(projectNameLength > MaxTagLength ? 0 : MaxTagLength - projectNameLength, ' '));
|
|
}
|
|
|
|
void assertionHandler(bool expr, const char* exprString, const char* file, int line) {
|
|
if (!expr) {
|
|
log::error("Assertion failed: {} at {}:{}", exprString, file, line);
|
|
|
|
#if defined (DEBUG)
|
|
std::abort();
|
|
#endif
|
|
}
|
|
}
|
|
|
|
} |