You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
210 lines
7.8 KiB
210 lines
7.8 KiB
using HarmonyLib;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Reflection.Emit;
|
|
using System.Text;
|
|
|
|
namespace IPA.Logging
|
|
{
|
|
internal class StdoutInterceptor : TextWriter
|
|
{
|
|
public override Encoding Encoding => Encoding.Default;
|
|
|
|
private bool isStdErr;
|
|
|
|
public override void Write(char value)
|
|
{
|
|
Write(value.ToString());
|
|
}
|
|
|
|
private string lineBuffer = "";
|
|
private readonly object bufferLock = new object();
|
|
|
|
public override void Write(string value)
|
|
{
|
|
lock (bufferLock)
|
|
{ // avoid threading issues
|
|
lineBuffer += value;
|
|
|
|
var parts = lineBuffer.Split(new[] { Environment.NewLine, "\n", "\r" }, StringSplitOptions.None);
|
|
for (int i = 0; i < parts.Length; i++)
|
|
{
|
|
if (i + 1 == parts.Length) // last element
|
|
lineBuffer = parts[i];
|
|
else
|
|
{
|
|
var str = parts[i];
|
|
if (string.IsNullOrEmpty(str)) continue;
|
|
if (!isStdErr && WinConsole.IsInitialized)
|
|
str = ConsoleColorToForegroundSet(currentColor) + str;
|
|
|
|
if (isStdErr)
|
|
Logger.stdout.Error(str);
|
|
else
|
|
Logger.stdout.Info(str);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private const ConsoleColor defaultColor = ConsoleColor.Gray;
|
|
private ConsoleColor currentColor = defaultColor;
|
|
|
|
private static string ConsoleColorToForegroundSet(ConsoleColor col)
|
|
{
|
|
if (!WinConsole.UseVTEscapes) return "";
|
|
string code = "0"; // reset
|
|
|
|
switch (col)
|
|
{
|
|
case ConsoleColor.Black:
|
|
code = "30";
|
|
break;
|
|
case ConsoleColor.DarkBlue:
|
|
code = "34";
|
|
break;
|
|
case ConsoleColor.DarkGreen:
|
|
code = "32";
|
|
break;
|
|
case ConsoleColor.DarkCyan:
|
|
code = "36";
|
|
break;
|
|
case ConsoleColor.DarkRed:
|
|
code = "31";
|
|
break;
|
|
case ConsoleColor.DarkMagenta:
|
|
code = "35";
|
|
break;
|
|
case ConsoleColor.DarkYellow:
|
|
code = "33";
|
|
break;
|
|
case ConsoleColor.Gray:
|
|
code = "37";
|
|
break;
|
|
case ConsoleColor.DarkGray:
|
|
code = "90"; // literally bright black
|
|
break;
|
|
case ConsoleColor.Blue:
|
|
code = "94";
|
|
break;
|
|
case ConsoleColor.Green:
|
|
code = "92";
|
|
break;
|
|
case ConsoleColor.Cyan:
|
|
code = "96";
|
|
break;
|
|
case ConsoleColor.Red:
|
|
code = "91";
|
|
break;
|
|
case ConsoleColor.Magenta:
|
|
code = "95";
|
|
break;
|
|
case ConsoleColor.Yellow:
|
|
code = "93";
|
|
break;
|
|
case ConsoleColor.White:
|
|
code = "97";
|
|
break;
|
|
}
|
|
|
|
return "\x1b[" + code + "m";
|
|
}
|
|
|
|
private static StdoutInterceptor stdoutInterceptor;
|
|
private static StdoutInterceptor stderrInterceptor;
|
|
|
|
private static class ConsoleHarmonyPatches
|
|
{
|
|
public static void Patch(Harmony harmony)
|
|
{
|
|
var console = typeof(Console);
|
|
var resetColor = console.GetMethod("ResetColor");
|
|
var foregroundProperty = console.GetProperty("ForegroundColor");
|
|
var setFg = foregroundProperty?.GetSetMethod();
|
|
var getFg = foregroundProperty?.GetGetMethod();
|
|
|
|
try
|
|
{
|
|
if (resetColor != null)
|
|
harmony.Patch(resetColor, transpiler: new HarmonyMethod(typeof(ConsoleHarmonyPatches), nameof(PatchResetColor)));
|
|
if (foregroundProperty != null)
|
|
{
|
|
harmony.Patch(setFg, transpiler: new HarmonyMethod(typeof(ConsoleHarmonyPatches), nameof(PatchSetForegroundColor)));
|
|
harmony.Patch(getFg, transpiler: new HarmonyMethod(typeof(ConsoleHarmonyPatches), nameof(PatchGetForegroundColor)));
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
// Harmony might be fucked because of wierdness in Guid.NewGuid, don't let that kill us
|
|
Logger.log.Error("Error installing harmony patches to intercept Console color properties:");
|
|
Logger.log.Error(e);
|
|
}
|
|
}
|
|
|
|
public static ConsoleColor GetColor() => stdoutInterceptor.currentColor;
|
|
public static void SetColor(ConsoleColor col) => stdoutInterceptor.currentColor = col;
|
|
public static void ResetColor() => stdoutInterceptor.currentColor = defaultColor;
|
|
|
|
public static IEnumerable<CodeInstruction> PatchGetForegroundColor(IEnumerable<CodeInstruction> _)
|
|
{
|
|
var getColorM = typeof(ConsoleHarmonyPatches).GetMethod("GetColor");
|
|
return new[] {
|
|
new CodeInstruction(OpCodes.Tailcall),
|
|
new CodeInstruction(OpCodes.Call, getColorM),
|
|
new CodeInstruction(OpCodes.Ret)
|
|
};
|
|
}
|
|
|
|
public static IEnumerable<CodeInstruction> PatchSetForegroundColor(IEnumerable<CodeInstruction> _)
|
|
{
|
|
var setColorM = typeof(ConsoleHarmonyPatches).GetMethod("SetColor");
|
|
return new[] {
|
|
new CodeInstruction(OpCodes.Ldarg_0),
|
|
new CodeInstruction(OpCodes.Tailcall),
|
|
new CodeInstruction(OpCodes.Call, setColorM),
|
|
new CodeInstruction(OpCodes.Ret)
|
|
};
|
|
}
|
|
|
|
public static IEnumerable<CodeInstruction> PatchResetColor(IEnumerable<CodeInstruction> _)
|
|
{
|
|
var resetColor = typeof(ConsoleHarmonyPatches).GetMethod("ResetColor");
|
|
return new[] {
|
|
new CodeInstruction(OpCodes.Tailcall),
|
|
new CodeInstruction(OpCodes.Call, resetColor),
|
|
new CodeInstruction(OpCodes.Ret)
|
|
};
|
|
}
|
|
}
|
|
|
|
private static Harmony harmony;
|
|
private static bool usingInterceptor = false;
|
|
|
|
public static void Intercept()
|
|
{
|
|
if (!usingInterceptor)
|
|
{
|
|
usingInterceptor = true;
|
|
if (harmony == null)
|
|
harmony = new Harmony("BSIPA Console Redirector Patcher");
|
|
if (stdoutInterceptor == null)
|
|
stdoutInterceptor = new StdoutInterceptor();
|
|
if (stderrInterceptor == null)
|
|
stderrInterceptor = new StdoutInterceptor() { isStdErr = true };
|
|
|
|
RedirectConsole();
|
|
ConsoleHarmonyPatches.Patch(harmony);
|
|
}
|
|
}
|
|
|
|
public static void RedirectConsole()
|
|
{
|
|
if (usingInterceptor)
|
|
{
|
|
Console.SetOut(stdoutInterceptor);
|
|
Console.SetError(stderrInterceptor);
|
|
}
|
|
}
|
|
}
|
|
}
|