mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-09 13:10:25 +00:00
85 lines
2.5 KiB
C++
85 lines
2.5 KiB
C++
#if defined(_WIN32)
|
|
#include <util/win32_logger.hpp>
|
|
#include <util/logger_internal.hpp>
|
|
|
|
static CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
|
|
static short old_attrs;
|
|
|
|
namespace llarp
|
|
{
|
|
Win32LogStream::Win32LogStream(std::ostream& out) : OStreamLogStream(out)
|
|
{
|
|
// Attempt to use ANSI escapes directly
|
|
// if the modern console is active.
|
|
DWORD mode_flags;
|
|
|
|
GetConsoleMode(fd1, &mode_flags);
|
|
// since release SDKs don't have ANSI escape support yet
|
|
// we get all or nothing: if we can't get it, then we wouldn't
|
|
// be able to get any of them individually
|
|
mode_flags |= 0x0004 | 0x0008;
|
|
BOOL t = SetConsoleMode(fd1, mode_flags);
|
|
if(!t)
|
|
this->isConsoleModern = false; // fall back to setting colours manually
|
|
}
|
|
|
|
void
|
|
Win32LogStream::PreLog(std::stringstream& ss, LogLevel lvl, const char* fname,
|
|
int lineno, const std::string& nodename) const
|
|
{
|
|
if(!isConsoleModern)
|
|
{
|
|
GetConsoleScreenBufferInfo(fd1, &consoleInfo);
|
|
old_attrs = consoleInfo.wAttributes;
|
|
switch(lvl)
|
|
{
|
|
case eLogNone:
|
|
break;
|
|
case eLogDebug:
|
|
SetConsoleTextAttribute(fd1,
|
|
FOREGROUND_RED | FOREGROUND_GREEN
|
|
| FOREGROUND_BLUE); // low white on black
|
|
ss << "[DBG] ";
|
|
break;
|
|
case eLogInfo:
|
|
SetConsoleTextAttribute(
|
|
fd1,
|
|
FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN
|
|
| FOREGROUND_BLUE); // high white on black
|
|
ss << "[NFO] ";
|
|
break;
|
|
case eLogWarn:
|
|
SetConsoleTextAttribute(fd1,
|
|
FOREGROUND_RED | FOREGROUND_GREEN
|
|
| FOREGROUND_INTENSITY); // bright yellow
|
|
ss << "[WRN] ";
|
|
break;
|
|
case eLogError:
|
|
SetConsoleTextAttribute(
|
|
fd1, FOREGROUND_RED | FOREGROUND_INTENSITY); // bright red
|
|
ss << "[ERR] ";
|
|
break;
|
|
}
|
|
ss << "[" << nodename << "]"
|
|
<< "(" << thread_id_string() << ") " << log_timestamp() << " " << fname
|
|
<< ":" << lineno << "\t";
|
|
}
|
|
else
|
|
OStreamLogStream::PreLog(ss, lvl, fname, lineno, nodename);
|
|
}
|
|
|
|
void
|
|
Win32LogStream::PostLog(std::stringstream& ss) const
|
|
{
|
|
if(!isConsoleModern)
|
|
{
|
|
SetConsoleTextAttribute(
|
|
fd1, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
|
|
ss << std::endl;
|
|
}
|
|
else
|
|
OStreamLogStream::PostLog(ss);
|
|
}
|
|
} // namespace llarp
|
|
#endif
|