feat: spdlog also log to file

Author: std-microblock

src/inject/inject.cc

@@ -5,6 +5,9 @@
 #include <vector>
 
 #include <spdlog/spdlog.h>
+#include <spdlog/sinks/basic_file_sink.h>
+#include <spdlog/sinks/stdout_color_sinks.h>
+#include <spdlog/sinks/msvc_sink.h>
 
 #include "breeze_ui/animator.h"
 #include "breeze_ui/ui.h"
@@ -23,11 +26,28 @@ static unsigned char g_icon_png[] = {
 #include <shellapi.h>
 #include <taskschd.h>
 
-
 #include "Shlobj.h"
 
 namespace fs = std::filesystem;
 
+void init_inject_logger() {
+    try {
+        auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
+        auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(
+            (data_directory() / "inject.log").string(), true);
+        auto msvc_sink = std::make_shared<spdlog::sinks::msvc_sink_mt>();
+
+        std::vector<spdlog::sink_ptr> sinks{console_sink, file_sink, msvc_sink};
+        auto logger = std::make_shared<spdlog::logger>("inject", sinks.begin(), sinks.end());
+        spdlog::set_default_logger(logger);
+        spdlog::set_pattern("[%Y-%m-%d %H:%M:%S.%e] [%^%l%$] %v");
+        spdlog::set_level(spdlog::level::debug);
+        spdlog::flush_on(spdlog::level::info);
+    } catch (const spdlog::spdlog_ex &ex) {
+        printf("Log initialization failed: %s\n", ex.what());
+    }
+}
+
 std::wstring GetModuleDirectory() {
     wchar_t path[MAX_PATH];
     GetModuleFileNameW(NULL, path, MAX_PATH);
@@ -1197,6 +1217,8 @@ void UpdateDllPath() {
 
 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
                    LPSTR lpCmdLine, int nShowCmd) {
+    init_inject_logger();
+
     int argc = 0;
     auto argv = CommandLineToArgvW(GetCommandLineW(), &argc);
 

src/shell/entry.cc

@@ -50,6 +50,8 @@
 
 #include "qjs_sanitizer.h"
 
+#include "logger.h"
+
 namespace mb_shell {
 window_proc_hook entry::main_window_loop_hook{};
 void main() {
@@ -60,6 +62,7 @@ void main() {
     freopen("CONIN$", "r", stdin);
     ShowWindow(GetConsoleWindow(), SW_HIDE);
 
+    init_logger();
     install_error_handlers();
     config::run_config_loader();
 

src/shell/logger.cc

@@ -2,24 +2,29 @@
 #include "config.h"
 #include "utils.h"
 
-#include <fstream>
-#include <iostream>
-#include <string>
-
-#include <windows.h>
+#include <spdlog/spdlog.h>
+#include <spdlog/sinks/basic_file_sink.h>
+#include <spdlog/sinks/stdout_color_sinks.h>
+#include <spdlog/sinks/msvc_sink.h>
 
 namespace mb_shell {
-void append_debug_string(const std::string &str) {
-    static std::mutex mutex;
-    static std::ofstream file(config::data_directory() / "debug.log",
-                              std::ios::app);
+void init_logger() {
+    try {
+        auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
+        auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(
+            (config::data_directory() / "debug.log").string(), true);
+        auto msvc_sink = std::make_shared<spdlog::sinks::msvc_sink_mt>();
 
-    std::lock_guard<std::mutex> lock(mutex);
+        std::vector<spdlog::sink_ptr> sinks{console_sink, file_sink, msvc_sink};
 
-    file << str;
-    file.flush();
+        auto logger = std::make_shared<spdlog::logger>("multi_sink", sinks.begin(), sinks.end());
+        spdlog::set_default_logger(logger);
+        spdlog::set_pattern("[%Y-%m-%d %H:%M:%S.%e] [%^%l%$] %v");
+        spdlog::set_level(spdlog::level::debug);
+        spdlog::flush_on(spdlog::level::info);
 
-    printf("%s", str.c_str());
-    OutputDebugStringA(str.c_str());
+    } catch (const spdlog::spdlog_ex &ex) {
+        printf("Log initialization failed: %s\n", ex.what());
+    }
 }
 } // namespace mb_shell

src/shell/logger.h

@@ -1,3 +1,7 @@
 #pragma once
 #include <spdlog/spdlog.h>
-#include <fmt/format.h>
+#include <fmt/format.h>
+
+namespace mb_shell {
+void init_logger();
+}