1 /**
2 * Copyright (c) 2020-2021 Paul-Louis Ageneau
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include "plog/Appenders/ColorConsoleAppender.h"
20 #include "plog/Formatters/FuncMessageFormatter.h"
21 #include "plog/Formatters/TxtFormatter.h"
22 #include "plog/Init.h"
23 #include "plog/Log.h"
24 #include "plog/Logger.h"
25 //
26 #include "global.hpp"
27
28 #include "impl/init.hpp"
29
30 #include <mutex>
31
32 #ifdef _WIN32
33 #include <codecvt>
34 #include <locale>
35 #endif
36
37 namespace {
38
plogInit(plog::Severity severity,plog::IAppender * appender)39 void plogInit(plog::Severity severity, plog::IAppender *appender) {
40 using Logger = plog::Logger<PLOG_DEFAULT_INSTANCE_ID>;
41 static Logger *logger = nullptr;
42 if (!logger) {
43 PLOG_DEBUG << "Initializing logger";
44 logger = new Logger(severity);
45 if (appender) {
46 logger->addAppender(appender);
47 } else {
48 using ConsoleAppender = plog::ColorConsoleAppender<plog::TxtFormatter>;
49 static ConsoleAppender *consoleAppender = new ConsoleAppender();
50 logger->addAppender(consoleAppender);
51 }
52 } else {
53 logger->setMaxSeverity(severity);
54 if (appender)
55 logger->addAppender(appender);
56 }
57 }
58
59 } // namespace
60
61 namespace rtc {
62
63 struct LogAppender : public plog::IAppender {
64 synchronized_callback<LogLevel, string> callback;
65
writertc::LogAppender66 void write(const plog::Record &record) override {
67 const auto severity = record.getSeverity();
68 auto formatted = plog::FuncMessageFormatter::format(record);
69 formatted.pop_back(); // remove newline
70
71 #ifdef _WIN32
72 using convert_type = std::codecvt_utf8<wchar_t>;
73 std::wstring_convert<convert_type, wchar_t> converter;
74 std::string str = converter.to_bytes(formatted);
75 #else
76 std::string str = formatted;
77 #endif
78
79 if (!callback(static_cast<LogLevel>(severity), str))
80 std::cout << plog::severityToString(severity) << " " << str << std::endl;
81 }
82 };
83
InitLogger(LogLevel level,LogCallback callback)84 void InitLogger(LogLevel level, LogCallback callback) {
85 const auto severity = static_cast<plog::Severity>(level);
86 static LogAppender *appender = nullptr;
87 static std::mutex mutex;
88 std::lock_guard lock(mutex);
89 if (appender) {
90 appender->callback = std::move(callback);
91 plogInit(severity, nullptr); // change the severity
92 } else if (callback) {
93 appender = new LogAppender();
94 appender->callback = std::move(callback);
95 plogInit(severity, appender);
96 } else {
97 plogInit(severity, nullptr); // log to cout
98 }
99 }
100
InitLogger(plog::Severity severity,plog::IAppender * appender)101 void InitLogger(plog::Severity severity, plog::IAppender *appender) {
102 plogInit(severity, appender);
103 }
104
Preload()105 void Preload() { Init::Instance().preload(); }
Cleanup()106 std::shared_future<void> Cleanup() { return Init::Instance().cleanup(); }
107
SetSctpSettings(SctpSettings s)108 void SetSctpSettings(SctpSettings s) { Init::Instance().setSctpSettings(std::move(s)); }
109
110 } // namespace rtc
111
operator <<(std::ostream & out,rtc::LogLevel level)112 RTC_CPP_EXPORT std::ostream &operator<<(std::ostream &out, rtc::LogLevel level) {
113 switch (level) {
114 case rtc::LogLevel::Fatal:
115 out << "fatal";
116 break;
117 case rtc::LogLevel::Error:
118 out << "error";
119 break;
120 case rtc::LogLevel::Warning:
121 out << "warning";
122 break;
123 case rtc::LogLevel::Info:
124 out << "info";
125 break;
126 case rtc::LogLevel::Debug:
127 out << "debug";
128 break;
129 case rtc::LogLevel::Verbose:
130 out << "verbose";
131 break;
132 default:
133 out << "none";
134 break;
135 }
136 return out;
137 }
138