Logs with color

Author: Chaleshka

Core/LocalAdmin.cs

@@ -611,6 +611,7 @@ private void StartSession()
         Menu();
         SetTerminalTitle();
         Logger.Initialize();
+        LoggerWithColors.Initialize();
 
         ConsoleUtil.WriteLine($"Started new session on port {GamePort}.", ConsoleColor.DarkGreen);
         ConsoleUtil.WriteLine("Trying to start server...", ConsoleColor.Gray);
@@ -1186,6 +1187,7 @@ public void Exit(int code = -1, bool waitForKey = false, bool restart = false)
             _processClosing = true;
             LogCleaner.Abort();
             Logger.EndLogging();
+            LoggerWithColors.EndLogging();
             TerminateGame(); // Forcefully terminating the process
             _gameProcess?.Dispose();
 

IO/ConsoleUtil.cs

@@ -100,6 +100,9 @@ public static void Write(string content, ConsoleColor color = ConsoleColor.White
 
             if (log)
                 Logger.Log($"{GetLogsTimestamp()} {(multiline ? content.Replace("\n", GetLogsPadding(), StringComparison.Ordinal) : content)}");
+
+            if (display)
+                LoggerWithColors.Log($"{GetLogsTimestamp()} {(multiline ? content.Replace("\n", GetLogsPadding(), StringComparison.Ordinal) : content)}", color);
         }
     }
 
@@ -131,6 +134,9 @@ public static void WriteLine(string? content, ConsoleColor color = ConsoleColor.
 
             if (log)
                 Logger.Log($"{GetLogsTimestamp()} {(multiline ? content.Replace("\n", GetLogsPadding(), StringComparison.Ordinal) : content)}");
+
+            if (display)
+                LoggerWithColors.Log($"{GetLogsTimestamp()} {(multiline ? content.Replace("\n", GetLogsPadding(), StringComparison.Ordinal) : content)}", color);
         }
     }
 }

IO/Logging/LoggerWithColors.cs

@@ -0,0 +1,119 @@
+using System;
+using System.IO;
+using System.Text;
+
+namespace LocalAdmin.V2.IO.Logging;
+
+public static class LoggerWithColors
+{
+    private static StringBuilder? _sb;
+    private static bool _logging;
+    private static string? _logPath;
+    private static ulong _totalLength, _totalEntries;
+    internal const string LogFolderName = "LocalAdminLogsWithColors";
+
+    public static void Initialize()
+    {
+        if (!Core.LocalAdmin.EnableLogging)
+            return;
+        if (_logging)
+            EndLogging();
+
+        string dir = Core.LocalAdmin.LaLogsPath ?? PathManager.GameUserDataRoot + LogFolderName + Path.DirectorySeparatorChar + Core.LocalAdmin.GamePort + Path.DirectorySeparatorChar;
+        if (!Directory.Exists(dir))
+            Directory.CreateDirectory(dir);
+
+        _totalLength = 0;
+        _totalEntries = 0;
+        _logPath = dir + $"LocalAdmin Log {DateTime.Now:yyyy-MM-dd HH.mm.ss}.txt";
+        _logging = true;
+
+        Log($"{ConsoleUtil.GetLogsTimestamp()} Logging started.", ConsoleColor.White);
+        Log($"{ConsoleUtil.GetLogsTimestamp()} Timezone offset: {DateTimeOffset.Now:zzz}", ConsoleColor.White);
+    }
+
+    public static void EndLogging(bool bypass = false)
+    {
+        if (!_logging && !bypass) return;
+        AppendLog($"{ConsoleUtil.GetLogsTimestamp()} --- END OF LOG ---", ConsoleColor.White, true, true);
+
+        _logging = false;
+        _sb = null;
+    }
+
+    private static string GetAnsiCode(ConsoleColor color)
+    {
+        return color switch
+        {
+            ConsoleColor.Black => "\x1b[30m",
+            ConsoleColor.DarkBlue => "\x1b[34m",
+            ConsoleColor.DarkGreen => "\x1b[32m",
+            ConsoleColor.DarkCyan => "\x1b[36m",
+            ConsoleColor.DarkRed => "\x1b[31m",
+            ConsoleColor.DarkMagenta => "\x1b[35m",
+            ConsoleColor.DarkYellow => "\x1b[33m",
+            ConsoleColor.Gray => "\x1b[37m",
+            ConsoleColor.DarkGray => "\x1b[90m",
+            ConsoleColor.Blue => "\x1b[94m",
+            ConsoleColor.Green => "\x1b[92m",
+            ConsoleColor.Cyan => "\x1b[96m",
+            ConsoleColor.Red => "\x1b[91m",
+            ConsoleColor.Magenta => "\x1b[95m",
+            ConsoleColor.Yellow => "\x1b[93m",
+            ConsoleColor.White => "\x1b[97m",
+            _ => "\x1b[0m"
+        };
+    }
+
+    private static void AppendLog(string text, ConsoleColor color, bool flush = false, bool bypass = false)
+    {
+        if (!_logging && !bypass) return;
+
+        try
+        {
+            string ansiPrefix = GetAnsiCode(color);
+            string ansiSuffix = "\x1b[0m";
+            string coloredText = ansiPrefix + text + ansiSuffix;
+
+            if (Core.LocalAdmin.AutoFlush)
+                File.AppendAllText(_logPath!, coloredText + Environment.NewLine);
+            else
+            {
+                _sb ??= new StringBuilder();
+                _sb.AppendLine(coloredText);
+
+                if (_sb.Length > 1000 || flush)
+                {
+                    File.AppendAllText(_logPath!, _sb.ToString());
+                    _sb.Clear();
+                }
+            }
+
+            if (bypass)
+                return;
+
+            _totalEntries++;
+            _totalLength += (uint)coloredText.Length;
+
+            if (_totalEntries > Core.LocalAdmin.LogEntriesLimit && Core.LocalAdmin.LogEntriesLimit > 0)
+            {
+                _logging = false;
+                AppendLog($"{ConsoleUtil.GetLogsTimestamp()} Log entries limit exceeded. Logging stopped.", ConsoleColor.White, flush: true, bypass: true);
+                EndLogging(true);
+            }
+
+            if (_totalLength > Core.LocalAdmin.LogLengthLimit && Core.LocalAdmin.LogLengthLimit > 0)
+            {
+                _logging = false;
+                AppendLog($"{ConsoleUtil.GetLogsTimestamp()} Log length limit exceeded. Logging stopped.", ConsoleColor.White, flush: true, bypass: true);
+                EndLogging(true);
+            }
+        }
+        catch (Exception e)
+        {
+            Console.Write("Failed to write log: " + e.Message);
+        }
+    }
+
+    public static void Log(string text, ConsoleColor color, bool flush = false) => AppendLog(text, color, flush);
+}