Merge pull request #129 from RLBot/win-color

Force-enable colors for cmd.exe

Author: VirxEC

RLBotCS/Main.cs

@@ -10,13 +10,17 @@
 if (args.Length > 0 && args[0] == "--version")
 {
     Console.WriteLine(
-        $"RLBotServer v5.beta.7.4\n"
+        "RLBotServer v5.beta.7.5\n"
             + $"Bridge {BridgeVersion.Version}\n"
-            + $"@ https://www.rlbot.org & https://github.com/RLBot/core"
+            + "@ https://www.rlbot.org & https://github.com/RLBot/core"
     );
     Environment.Exit(0);
 }
 
+#if WINDOWS
+WinTermColor.EnableVirtualTerminal();
+#endif
+
 var logger = Logging.GetLogger("Main");
 
 int rlbotSocketsPort;

RLBotCS/ManagerTools/Logging.cs

@@ -10,7 +10,7 @@ public class Logging : ILogger
     private const string Yellow = "\x1b[33;20m";
     private const string Green = "\x1b[32;20m";
     private const string Red = "\x1b[31;20m";
-    private const string BoldRen = "\x1b[31;1m";
+    private const string BoldRed = "\x1b[31;1m";
     private const string Reset = "\x1b[0m";
 
     private static readonly LogLevel LoggingLevel = Environment.GetEnvironmentVariable(
@@ -80,7 +80,7 @@ private string[] GetLogLevelColors(LogLevel logLevel) =>
             LogLevel.Information => new[] { Grey, LightBlue, Grey, LightBlue },
             LogLevel.Warning => new[] { Yellow, Yellow, Yellow, Yellow },
             LogLevel.Error => new[] { Red, Red, Red, Red },
-            LogLevel.Critical => new[] { Red, BoldRen, Red, BoldRen },
+            LogLevel.Critical => new[] { Red, BoldRed, Red, BoldRed },
             _ => new[] { Grey, Grey, Grey, Grey },
         };
 

RLBotCS/ManagerTools/WinTermColor.cs

@@ -0,0 +1,37 @@
+#if WINDOWS
+using System.Runtime.InteropServices;
+
+public class WinTermColor
+{
+    const int STD_OUTPUT_HANDLE = -11;
+    const uint ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004;
+
+    [DllImport("kernel32.dll", SetLastError = true)]
+    static extern IntPtr GetStdHandle(int nStdHandle);
+
+    [DllImport("kernel32.dll")]
+    static extern bool GetConsoleMode(IntPtr hConsoleHandle, out uint lpMode);
+
+    [DllImport("kernel32.dll")]
+    static extern bool SetConsoleMode(IntPtr hConsoleHandle, uint dwMode);
+
+    public static void EnableVirtualTerminal()
+    {
+        var handle = GetStdHandle(STD_OUTPUT_HANDLE);
+        if (!GetConsoleMode(handle, out uint mode))
+        {
+            // Don't use the logger here because we couldn't enable colors
+            Console.WriteLine("Failed to get console mode.");
+            return;
+        }
+
+        mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
+
+        if (!SetConsoleMode(handle, mode))
+        {
+            // Don't use logger for the same reason
+            Console.WriteLine("Failed to set console mode.");
+        }
+    }
+}
+#endif