LinuxConsoleDriver.cs 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. using System;
  2. using System.IO;
  3. using System.Text;
  4. using BepInEx.Logging;
  5. using HarmonyLib;
  6. using UnityInjector.ConsoleUtil;
  7. namespace BepInEx.Unix
  8. {
  9. internal class LinuxConsoleDriver : IConsoleDriver
  10. {
  11. public TextWriter StandardOut { get; private set; }
  12. public TextWriter ConsoleOut { get; private set; }
  13. public bool ConsoleActive { get; private set; }
  14. public bool ConsoleIsExternal => false;
  15. public bool UseMonoTtyDriver { get; private set; }
  16. public bool StdoutRedirected { get; private set; }
  17. public TtyInfo TtyInfo { get; private set; }
  18. public void Initialize(bool alreadyActive)
  19. {
  20. // Console is always considered active on Unix
  21. ConsoleActive = true;
  22. UseMonoTtyDriver = typeof(Console).Assembly.GetType("System.ConsoleDriver") != null;
  23. StdoutRedirected = UnixStreamHelper.isatty(1) != 1;
  24. var duplicateStream = UnixStreamHelper.CreateDuplicateStream(1);
  25. if (UseMonoTtyDriver && !StdoutRedirected)
  26. {
  27. // Mono implementation handles xterm for us
  28. var writer = ConsoleWriter.CreateConsoleStreamWriter(duplicateStream, Console.Out.Encoding, true);
  29. StandardOut = TextWriter.Synchronized(writer);
  30. var driver = AccessTools.Field(AccessTools.TypeByName("System.ConsoleDriver"), "driver").GetValue(null);
  31. AccessTools.Field(AccessTools.TypeByName("System.TermInfoDriver"), "stdout").SetValue(driver, writer);
  32. }
  33. else
  34. {
  35. // Handle TTY ourselves
  36. var writer = new StreamWriter(duplicateStream, Console.Out.Encoding);
  37. writer.AutoFlush = true;
  38. StandardOut = TextWriter.Synchronized(writer);
  39. TtyInfo = TtyHandler.GetTtyInfo();
  40. }
  41. ConsoleOut = StandardOut;
  42. }
  43. public void CreateConsole(uint codepage)
  44. {
  45. Logger.LogWarning("An external console currently cannot be spawned on a Unix platform.");
  46. }
  47. public void DetachConsole()
  48. {
  49. throw new PlatformNotSupportedException("Cannot detach console on a Unix platform");
  50. }
  51. public void SetConsoleColor(ConsoleColor color)
  52. {
  53. if (StdoutRedirected)
  54. return;
  55. if (UseMonoTtyDriver)
  56. {
  57. // Use mono's inbuilt terminfo driver to set the foreground color for us
  58. SafeConsole.ForegroundColor = color;
  59. }
  60. else
  61. {
  62. ConsoleOut.Write(TtyInfo.GetAnsiCode(color));
  63. }
  64. }
  65. public void SetConsoleTitle(string title)
  66. {
  67. if (StdoutRedirected)
  68. return;
  69. if (UseMonoTtyDriver && SafeConsole.TitleExists)
  70. {
  71. SafeConsole.Title = title;
  72. }
  73. else
  74. {
  75. ConsoleOut.Write($"\u001B]2;{title.Replace("\\", "\\\\")}\u0007");
  76. }
  77. }
  78. }
  79. }