When truncating the progress bar, take ANSI escape sequences into account

This commit is contained in:
Eelco Dolstra 2017-08-25 15:44:36 +02:00
parent a3015db6c3
commit ec9e0c03c3
No known key found for this signature in database
GPG key ID: 8170B4726D7198DE

View file

@ -24,6 +24,44 @@ static uint64_t getI(const std::vector<Logger::Field> & fields, size_t n)
return fields[n].i; return fields[n].i;
} }
/* Truncate a string to 'width' printable characters. ANSI escape
sequences are copied but not included in the character count. Also,
tabs are expanded to spaces. */
static std::string ansiTruncate(const std::string & s, int width)
{
if (width <= 0) return s;
std::string t;
size_t w = 0;
auto i = s.begin();
while (w < (size_t) width && i != s.end()) {
if (*i == '\e') {
t += *i++;
if (i != s.end() && *i == '[') {
t += *i++;
while (i != s.end() && (*i < 0x40 || *i > 0x7e)) {
t += *i++;
}
if (i != s.end()) t += *i++;
}
}
else if (*i == '\t') {
t += ' '; w++;
while (w < (size_t) width && w & 8) {
t += ' '; w++;
}
}
else {
t += *i++; w++;
}
}
return t;
}
class ProgressBar : public Logger class ProgressBar : public Logger
{ {
private: private:
@ -89,7 +127,7 @@ public:
void log(State & state, Verbosity lvl, const std::string & s) void log(State & state, Verbosity lvl, const std::string & s)
{ {
writeToStderr("\r\e[K" + s + "\n"); writeToStderr("\r\e[K" + s + ANSI_NORMAL "\n");
update(state); update(state);
} }
@ -207,7 +245,7 @@ public:
void update(State & state) void update(State & state)
{ {
std::string line = "\r"; std::string line;
std::string status = getStatus(state); std::string status = getStatus(state);
if (!status.empty()) { if (!status.empty()) {
@ -232,8 +270,7 @@ public:
} }
} }
line += "\e[K"; writeToStderr("\r" + ansiTruncate(line, width) + "\e[K");
writeToStderr(std::string(line, 0, width - 1));
} }
std::string getStatus(State & state) std::string getStatus(State & state)