BepInEx/BepInEx.Core/Console/SafeConsole.cs

// --------------------------------------------------
// UnityInjector - SafeConsole.cs
// Copyright (c) Usagirei 2015 - 2015
// --------------------------------------------------

using System;
using System.Reflection;

namespace UnityInjector.ConsoleUtil
{
    /// <summary>
    ///     Console class with safe handlers for Unity 4.x, which does not have a proper Console implementation
    /// </summary>
    internal static class SafeConsole
    {
        private static GetColorDelegate _getBackgroundColor;
        private static SetColorDelegate _setBackgroundColor;

        private static GetColorDelegate _getForegroundColor;
        private static SetColorDelegate _setForegroundColor;

        private static GetStringDelegate _getTitle;
        private static SetStringDelegate _setTitle;

        static SafeConsole()
        {
            var tConsole = typeof(Console);
            InitColors(tConsole);
        }

        public static bool BackgroundColorExists { get; private set; }

        public static ConsoleColor BackgroundColor
        {
            get => _getBackgroundColor();
            set => _setBackgroundColor(value);
        }

        public static bool ForegroundColorExists { get; private set; }

        public static ConsoleColor ForegroundColor
        {
            get => _getForegroundColor();
            set => _setForegroundColor(value);
        }

        public static bool TitleExists { get; private set; }

        public static string Title
        {
            get => _getTitle();
            set => _setTitle(value);
        }

        private static void InitColors(Type tConsole)
        {
            const BindingFlags BINDING_FLAGS = BindingFlags.Public | BindingFlags.Static;

            var gfc = tConsole.GetMethod("get_ForegroundColor", BINDING_FLAGS);
            var sfc = tConsole.GetMethod("set_ForegroundColor", BINDING_FLAGS);

            var gbc = tConsole.GetMethod("get_BackgroundColor", BINDING_FLAGS);
            var sbc = tConsole.GetMethod("set_BackgroundColor", BINDING_FLAGS);

            var gtt = tConsole.GetMethod("get_Title", BINDING_FLAGS);
            var stt = tConsole.GetMethod("set_Title", BINDING_FLAGS);

            _setForegroundColor = sfc != null
                                      ? (SetColorDelegate) Delegate.CreateDelegate(typeof(SetColorDelegate), sfc)
                                      : value => { };

            _setBackgroundColor = sbc != null
                                      ? (SetColorDelegate) Delegate.CreateDelegate(typeof(SetColorDelegate), sbc)
                                      : value => { };

            _getForegroundColor = gfc != null
                                      ? (GetColorDelegate) Delegate.CreateDelegate(typeof(GetColorDelegate), gfc)
                                      : () => ConsoleColor.Gray;

            _getBackgroundColor = gbc != null
                                      ? (GetColorDelegate) Delegate.CreateDelegate(typeof(GetColorDelegate), gbc)
                                      : () => ConsoleColor.Black;

            _getTitle = gtt != null
                            ? (GetStringDelegate) Delegate.CreateDelegate(typeof(GetStringDelegate), gtt)
                            : () => string.Empty;

            _setTitle = stt != null
                            ? (SetStringDelegate) Delegate.CreateDelegate(typeof(SetStringDelegate), stt)
                            : value => { };

            BackgroundColorExists = _setBackgroundColor != null && _getBackgroundColor != null;
            ForegroundColorExists = _setForegroundColor != null && _getForegroundColor != null;
            TitleExists = _setTitle != null && _getTitle != null;
        }

        private delegate ConsoleColor GetColorDelegate();

        private delegate void SetColorDelegate(ConsoleColor value);

        private delegate string GetStringDelegate();

        private delegate void SetStringDelegate(string value);
    }
}