-
Notifications
You must be signed in to change notification settings - Fork 8
/
loggr.hpp
130 lines (100 loc) · 2.4 KB
/
loggr.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#pragma once
#include <windows.h>
#include <iostream>
#include <chrono>
#include "fmt/core.h"
class loggr
{
public:
loggr()
{
auto console_window = GetConsoleWindow();
if (console_window == 0x00)
{
if (!AllocConsole())
{
return;
}
std::freopen("CONOUT$", "w", stdout);
m_did_allocate_console = true;
}
}
~loggr()
{
if (m_did_allocate_console)
{
const auto freed = FreeConsole();
if (!freed)
{
// ??
}
}
}
template <class...T>
inline void log_raw(T... arguments) const
{
fmt::print(arguments...);
}
inline void log(std::string_view message) const
{
fmt::print("[+] {}\n", message);
}
inline void log_error(std::string_view message) const
{
fmt::print("[!] {}\n", message);
}
template <bool hex = false, class T>
inline void log(std::string_view variable_name, const T& value) const
{
constexpr auto format_string = hex ?
"[=] {:<15} {:X}\n" :
"[=] {:<15} {}\n";
fmt::print(format_string, variable_name, value);
}
template <std::size_t indentation>
inline void log_error_indented(std::string_view message) const
{
fmt::print("[!] {:<{}} {}\n", ' ', indentation, message);
}
template <std::size_t indentation>
inline void log_indented(std::string_view message) const
{
fmt::print("[+] {:<{}} {}\n", ' ', indentation, message);
}
template <std::size_t indentation, bool hex = false, class T>
inline void log_indented(std::string_view variable_name, const T& value) const
{
constexpr auto format_string = hex ?
"[=] {:<{}} {:.<15} {:X}\n" :
"[=] {:<{}} {:.<15} {}\n";
fmt::print(format_string, ' ', indentation, variable_name, value);
}
inline COORD get_position() const
{
CONSOLE_SCREEN_BUFFER_INFO info;
if (!GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &info))
{
this->log_error("Failed to get cursor position");
return { 0, 0 };
}
return info.dwCursorPosition;
}
inline void set_position(const COORD cursor) const
{
if (!SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), cursor))
{
this->log_error("Failed to set cursor position");
}
}
inline void clear_line() const
{
auto position = this->get_position();
position.X = 0;
DWORD count = 0;
const auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
FillConsoleOutputCharacter(handle, ' ', 150, position, &count);
set_position(position);
}
private:
bool m_did_allocate_console = false;
};