Robot
/
lnav
mirror of https://github.com/tstack/lnav.git
2
0
Fork 0
Code Issues Releases Wiki Activity
lnav/src/base/ansi_scrubber.cc

240 lines
8.7 KiB
C++
Raw Blame History

/**
* Copyright (c) 2013, Timothy Stack
*
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* * Neither the name of Timothy Stack nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* @file ansi_scrubber.cc
*/
#include <algorithm>
#include "ansi_scrubber.hh"
#include "base/opt_util.hh"
#include "config.h"
#include "pcrepp/pcrepp.hh"
#include "scn/scn.h"
#include "view_curses.hh"
static pcrepp&
ansi_regex()
{
static pcrepp retval("\x1b\\[([\\d=;\\?]*)([a-zA-Z])|(.)\b\\3|.\b_|_\b.");
return retval;
}
void
scrub_ansi_string(std::string& str, string_attrs_t* sa)
{
pcre_context_static<60> context;
auto& regex = ansi_regex();
pcre_input pi(str);
int64_t origin_offset = 0;
int last_origin_offset_end = 0;
replace(str.begin(), str.end(), '\0', ' ');
while (regex.match(context, pi)) {
auto* caps = context.all();
const auto sf = pi.get_string_fragment(caps);
if (sf.length() == 3 && sf[1] == '\b') {
if (sa != nullptr) {
shift_string_attrs(*sa, caps->c_begin + 1, -2);
if (sf[0] == '_' || sf[2] == '_') {
sa->emplace_back(
line_range{caps->c_begin, caps->c_begin + 1},
VC_STYLE.value(text_attrs{A_UNDERLINE}));
} else {
sa->emplace_back(
line_range{caps->c_begin, caps->c_begin + 1},
VC_STYLE.value(text_attrs{A_BOLD}));
}
sa->emplace_back(
line_range{last_origin_offset_end, caps->c_begin + 1},
SA_ORIGIN_OFFSET.value(origin_offset));
}
if (sf[0] == '_') {
str.erase(str.begin() + caps->c_begin,
str.begin() + caps->c_end - 1);
last_origin_offset_end = caps->c_begin + 1;
} else {
str.erase(str.begin() + caps->c_begin + 1,
str.begin() + caps->c_end);
last_origin_offset_end = caps->c_begin + 1;
}
origin_offset += 2;
pi.reset(str);
continue;
}
struct line_range lr;
bool has_attrs = false;
text_attrs attrs;
auto role = nonstd::optional<role_t>();
size_t lpc;
switch (pi.get_substr_start(&caps[2])[0]) {
case 'm':
for (lpc = caps[1].c_begin;
lpc != std::string::npos && lpc < (size_t) caps[1].c_end;)
{
auto ansi_code_res = scn::scan_value<int>(
scn::string_view{&str[lpc], &str[caps[1].c_end]});
if (ansi_code_res) {
auto ansi_code = ansi_code_res.value();
if (90 <= ansi_code && ansi_code <= 97) {
ansi_code -= 60;
attrs.ta_attrs |= A_STANDOUT;
}
if (30 <= ansi_code && ansi_code <= 37) {
attrs.ta_fg_color = ansi_code - 30;
}
if (40 <= ansi_code && ansi_code <= 47) {
attrs.ta_bg_color = ansi_code - 40;
}
switch (ansi_code) {
case 1:
attrs.ta_attrs |= A_BOLD;
break;
case 2:
attrs.ta_attrs |= A_DIM;
break;
case 4:
attrs.ta_attrs |= A_UNDERLINE;
break;
case 7:
attrs.ta_attrs |= A_REVERSE;
break;
}
}
lpc = str.find(';', lpc);
if (lpc != std::string::npos) {
lpc += 1;
}
}
has_attrs = true;
break;
case 'C': {
auto spaces_res = scn::scan_value<unsigned int>(
pi.to_string_view(&caps[1]));
if (spaces_res && spaces_res.value() > 0) {
str.insert((std::string::size_type) caps[0].c_end,
spaces_res.value(),
' ');
}
break;
}
case 'H': {
unsigned int row = 0, spaces = 0;
if (scn::scan(pi.to_string_view(&caps[1]), "{};{}", row, spaces)
&& spaces > 1)
{
int ispaces = spaces - 1;
if (ispaces > caps[0].c_begin) {
str.insert((unsigned long) caps[0].c_end,
ispaces - caps[0].c_begin,
' ');
}
}
break;
}
case 'O': {
auto role_res
= scn::scan_value<int>(pi.to_string_view(&caps[1]));
if (role_res) {
role_t role_tmp = (role_t) role_res.value();
if (role_tmp > role_t::VCR_NONE
&& role_tmp < role_t::VCR__MAX)
{
role = role_tmp;
has_attrs = true;
}
}
break;
}
}
str.erase(str.begin() + caps[0].c_begin, str.begin() + caps[0].c_end);
if (sa != nullptr) {
shift_string_attrs(*sa, caps[0].c_begin, -caps[0].length());
if (has_attrs) {
for (auto rit = sa->rbegin(); rit != sa->rend(); rit++) {
if (rit->sa_range.lr_end != -1) {
continue;
}
rit->sa_range.lr_end = caps[0].c_begin;
}
lr.lr_start = caps[0].c_begin;
lr.lr_end = -1;
if (attrs.ta_attrs || attrs.ta_fg_color || attrs.ta_bg_color) {
sa->emplace_back(lr, VC_STYLE.value(attrs));
}
role | [&lr, &sa](role_t r) {
sa->emplace_back(lr, VC_ROLE.value(r));
};
}
sa->emplace_back(line_range{last_origin_offset_end, caps->c_begin},
SA_ORIGIN_OFFSET.value(origin_offset));
origin_offset += caps->length();
}
pi.reset(str);
}
if (sa != nullptr && last_origin_offset_end > 0) {
sa->emplace_back(line_range{last_origin_offset_end, (int) str.size()},
SA_ORIGIN_OFFSET.value(origin_offset));
}
}
void
add_ansi_vars(std::map<std::string, scoped_value_t>& vars)
{
vars["ansi_csi"] = ANSI_CSI;
vars["ansi_norm"] = ANSI_NORM;
vars["ansi_bold"] = ANSI_BOLD_START;
vars["ansi_underline"] = ANSI_UNDERLINE_START;
vars["ansi_black"] = ANSI_COLOR(COLOR_BLACK);
vars["ansi_red"] = ANSI_COLOR(COLOR_RED);
vars["ansi_green"] = ANSI_COLOR(COLOR_GREEN);
vars["ansi_yellow"] = ANSI_COLOR(COLOR_YELLOW);
vars["ansi_blue"] = ANSI_COLOR(COLOR_BLUE);
vars["ansi_magenta"] = ANSI_COLOR(COLOR_MAGENTA);
vars["ansi_cyan"] = ANSI_COLOR(COLOR_CYAN);
vars["ansi_white"] = ANSI_COLOR(COLOR_WHITE);
}
Reference in New Issue View Git Blame Copy Permalink