intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 505 insertions, 0 deletions

diff --git a/library/cpp/colorizer/colors.cpp b/library/cpp/colorizer/colors.cpp
new file mode 100644
index 0000000000..decc5c9847
--- /dev/null
+++ b/library/cpp/colorizer/colors.cpp
@@ -0,0 +1,505 @@
+#include "colors.h"
+
+#include <util/stream/output.h>
+#include <util/generic/singleton.h>
+#include <util/system/env.h>
+
+#include <cstdlib>
+
+#if defined(_unix_)
+#include <unistd.h>
+#endif
+
+using namespace NColorizer;
+
+namespace {
+    constexpr TStringBuf ToStringBufC(NColorizer::EAnsiCode x) {
+        switch(x) {
+            case RESET:
+                return "\033[0m";
+
+            case ST_LIGHT:
+                return "\033[1m";
+            case ST_DARK:
+                return "\033[2m";
+            case ST_NORMAL:
+                return "\033[22m";
+
+            case ITALIC_ON:
+                return "\033[3m";
+            case ITALIC_OFF:
+                return "\033[23m";
+            case UNDERLINE_ON:
+                return "\033[4m";
+            case UNDERLINE_OFF:
+                return "\033[24m";
+
+            case FG_DEFAULT:
+                return "\033[39m";
+            case FG_BLACK:
+                return "\033[30m";
+            case FG_RED:
+                return "\033[31m";
+            case FG_GREEN:
+                return "\033[32m";
+            case FG_YELLOW:
+                return "\033[33m";
+            case FG_BLUE:
+                return "\033[34m";
+            case FG_MAGENTA:
+                return "\033[35m";
+            case FG_CYAN:
+                return "\033[36m";
+            case FG_WHITE:
+                return "\033[37m";
+
+            case BG_DEFAULT:
+                return "\033[49m";
+            case BG_BLACK:
+                return "\033[40m";
+            case BG_RED:
+                return "\033[41m";
+            case BG_GREEN:
+                return "\033[42m";
+            case BG_YELLOW:
+                return "\033[43m";
+            case BG_BLUE:
+                return "\033[44m";
+            case BG_MAGENTA:
+                return "\033[45m";
+            case BG_CYAN:
+                return "\033[46m";
+            case BG_WHITE:
+                return "\033[47m";
+
+            // Note: the following codes are split into two escabe sequences because of how ya.make handles them.
+
+            case DEFAULT:
+                return "\033[0m\033[0;39m";
+            case BLACK:
+                return "\033[0m\033[0;30m";
+            case RED:
+                return "\033[0m\033[0;31m";
+            case GREEN:
+                return "\033[0m\033[0;32m";
+            case YELLOW:
+                return "\033[0m\033[0;33m";
+            case BLUE:
+                return "\033[0m\033[0;34m";
+            case MAGENTA:
+                return "\033[0m\033[0;35m";
+            case CYAN:
+                return "\033[0m\033[0;36m";
+            case WHITE:
+                return "\033[0m\033[0;37m";
+
+            case LIGHT_DEFAULT:
+                return "\033[0m\033[1;39m";
+            case LIGHT_BLACK:
+                return "\033[0m\033[1;30m";
+            case LIGHT_RED:
+                return "\033[0m\033[1;31m";
+            case LIGHT_GREEN:
+                return "\033[0m\033[1;32m";
+            case LIGHT_YELLOW:
+                return "\033[0m\033[1;33m";
+            case LIGHT_BLUE:
+                return "\033[0m\033[1;34m";
+            case LIGHT_MAGENTA:
+                return "\033[0m\033[1;35m";
+            case LIGHT_CYAN:
+                return "\033[0m\033[1;36m";
+            case LIGHT_WHITE:
+                return "\033[0m\033[1;37m";
+
+            case DARK_DEFAULT:
+                return "\033[0m\033[2;39m";
+            case DARK_BLACK:
+                return "\033[0m\033[2;30m";
+            case DARK_RED:
+                return "\033[0m\033[2;31m";
+            case DARK_GREEN:
+                return "\033[0m\033[2;32m";
+            case DARK_YELLOW:
+                return "\033[0m\033[2;33m";
+            case DARK_BLUE:
+                return "\033[0m\033[2;34m";
+            case DARK_MAGENTA:
+                return "\033[0m\033[2;35m";
+            case DARK_CYAN:
+                return "\033[0m\033[2;36m";
+            case DARK_WHITE:
+                return "\033[0m\033[2;37m";
+
+            case INVALID:
+            default:
+                return "";
+        }
+    }
+}
+
+TStringBuf ToStringBuf(NColorizer::EAnsiCode x) {
+    return ToStringBufC(x);
+}
+
+TString ToString(NColorizer::EAnsiCode x) {
+    return TString(ToStringBufC(x));
+}
+
+template<>
+void Out<NColorizer::EAnsiCode>(IOutputStream& os, TTypeTraits<NColorizer::EAnsiCode>::TFuncParam x) {
+    if (AutoColors(os).IsTTY()) {
+        os << ToStringBufC(x);
+    }
+}
+
+bool TColors::CalcIsTTY(FILE* file) {
+    if (GetEnv("ENFORCE_TTY")) {
+        return true;
+    }
+
+#if defined(_unix_)
+    return isatty(fileno(file));
+#else
+    Y_UNUSED(file);
+    return false;
+#endif
+}
+
+TColors::TColors(FILE* f)
+    : IsTTY_(true)
+{
+    SetIsTTY(CalcIsTTY(f));
+}
+
+TColors::TColors(bool ontty)
+    : IsTTY_(true)
+{
+    SetIsTTY(ontty);
+}
+
+TStringBuf TColors::Reset() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::RESET) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::StyleLight() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::ST_LIGHT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::StyleDark() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::ST_DARK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::StyleNormal() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::ST_NORMAL) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::ItalicOn() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::ITALIC_ON) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ItalicOff() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::ITALIC_OFF) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::UnderlineOn() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::UNDERLINE_ON) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::UnderlineOff() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::UNDERLINE_OFF) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::ForeDefault() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_DEFAULT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeBlack() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_BLACK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeRed() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_RED) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeGreen() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_GREEN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeYellow() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_YELLOW) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeBlue() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_BLUE) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeMagenta() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_MAGENTA) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeCyan() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_CYAN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::ForeWhite() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::FG_WHITE) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::BackDefault() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_DEFAULT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackBlack() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_BLACK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackRed() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_RED) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackGreen() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_GREEN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackYellow() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_YELLOW) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackBlue() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_BLUE) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackMagenta() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_MAGENTA) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackCyan() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_CYAN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::BackWhite() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BG_WHITE) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::Default() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DEFAULT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Black() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BLACK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Red() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::RED) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Green() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::GREEN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Yellow() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::YELLOW) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Blue() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::BLUE) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Magenta() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::MAGENTA) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::Cyan() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::CYAN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::White() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::WHITE) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::LightDefault() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_DEFAULT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightBlack() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_BLACK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightRed() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_RED) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightGreen() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_GREEN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightYellow() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_YELLOW) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightBlue() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_BLUE) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightMagenta() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_MAGENTA) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightCyan() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_CYAN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::LightWhite() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::LIGHT_WHITE) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::DarkDefault() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_DEFAULT) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkBlack() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_BLACK) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkRed() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_RED) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkGreen() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_GREEN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkYellow() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_YELLOW) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkBlue() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_BLUE) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkMagenta() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_MAGENTA) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkCyan() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_CYAN) : ToStringBufC(EAnsiCode::INVALID);
+}
+TStringBuf TColors::DarkWhite() const noexcept {
+    return IsTTY() ? ToStringBufC(EAnsiCode::DARK_WHITE) : ToStringBufC(EAnsiCode::INVALID);
+}
+
+TStringBuf TColors::OldColor() const noexcept {
+    return IsTTY() ? "\033[22;39m" : "";
+}
+
+TStringBuf TColors::BoldColor() const noexcept {
+    return IsTTY() ? "\033[1m" : "";
+}
+
+TStringBuf TColors::BlackColor() const noexcept {
+    return IsTTY() ? "\033[22;30m" : "";
+}
+
+TStringBuf TColors::BlueColor() const noexcept {
+    return IsTTY() ? "\033[22;34m" : "";
+}
+
+TStringBuf TColors::GreenColor() const noexcept {
+    return IsTTY() ? "\033[22;32m" : "";
+}
+
+TStringBuf TColors::CyanColor() const noexcept {
+    return IsTTY() ? "\033[22;36m" : "";
+}
+
+TStringBuf TColors::RedColor() const noexcept {
+    return IsTTY() ? "\033[22;31m" : "";
+}
+
+TStringBuf TColors::PurpleColor() const noexcept {
+    return IsTTY() ? "\033[22;35m" : "";
+}
+
+TStringBuf TColors::BrownColor() const noexcept {
+    return IsTTY() ? "\033[22;33m" : "";
+}
+
+TStringBuf TColors::LightGrayColor() const noexcept {
+    return IsTTY() ? "\033[22;37m" : "";
+}
+
+TStringBuf TColors::DarkGrayColor() const noexcept {
+    return IsTTY() ? "\033[1;30m" : "";
+}
+
+TStringBuf TColors::LightBlueColor() const noexcept {
+    return IsTTY() ? "\033[1;34m" : "";
+}
+
+TStringBuf TColors::LightGreenColor() const noexcept {
+    return IsTTY() ? "\033[1;32m" : "";
+}
+
+TStringBuf TColors::LightCyanColor() const noexcept {
+    return IsTTY() ? "\033[1;36m" : "";
+}
+
+TStringBuf TColors::LightRedColor() const noexcept {
+    return IsTTY() ? "\033[1;31m" : "";
+}
+
+TStringBuf TColors::LightPurpleColor() const noexcept {
+    return IsTTY() ? "\033[1;35m" : "";
+}
+
+TStringBuf TColors::YellowColor() const noexcept {
+    return IsTTY() ? "\033[1;33m" : "";
+}
+
+TStringBuf TColors::WhiteColor() const noexcept {
+    return IsTTY() ? "\033[1;37m" : "";
+}
+
+
+namespace {
+    class TStdErrColors: public TColors {
+    public:
+        TStdErrColors()
+            : TColors(stderr)
+        {
+        }
+    };
+
+    class TStdOutColors: public TColors {
+    public:
+        TStdOutColors()
+            : TColors(stdout)
+        {
+        }
+    };
+
+    class TDisabledColors: public TColors {
+    public:
+        TDisabledColors()
+            : TColors(false)
+        {
+        }
+    };
+} // anonymous namespace
+
+TColors& NColorizer::StdErr() {
+    return *Singleton<TStdErrColors>();
+}
+
+TColors& NColorizer::StdOut() {
+    return *Singleton<TStdOutColors>();
+}
+
+TColors& NColorizer::AutoColors(IOutputStream& os) {
+    if (&os == &Cerr) {
+        return StdErr();
+    }
+    if (&os == &Cout) {
+        return StdOut();
+    }
+    return *Singleton<TDisabledColors>();
+}
+
+size_t NColorizer::TotalAnsiEscapeCodeLen(TStringBuf text) {
+    enum {
+        TEXT,
+        BEFORE_CODE,
+        IN_CODE,
+    } state = TEXT;
+
+    size_t totalLen = 0;
+    size_t curLen = 0;
+
+    for (auto it = text.begin(); it < text.end(); ++it) {
+        switch (state) {
+            case TEXT:
+                if (*it == '\033') {
+                    state = BEFORE_CODE;
+                    curLen = 1;
+                }
+                break;
+            case BEFORE_CODE:
+                if (*it == '[') {
+                    state = IN_CODE;
+                    curLen++;
+                } else {
+                    state = TEXT;
+                }
+                break;
+            case IN_CODE:
+                if (*it == ';' || isdigit(*it)) {
+                    curLen++;
+                } else {
+                    if (*it == 'm') {
+                        totalLen += curLen + 1;
+                    }
+                    state = TEXT;
+                }
+                break;
+        }
+    }
+
+    return totalLen;
+}