lnav/src/textview_curses.cc

#include "config.h"

#include <algorithm>

#include "lnav_util.hh"
#include "textview_curses.hh"

using namespace std;

bookmark_type_t textview_curses::BM_USER;
bookmark_type_t textview_curses::BM_SEARCH;

class ansi_scrubber {
    /* XXX move to view_curses.cc ; actually, move to it's own file and call
     * from there.
     */
public:
    static ansi_scrubber &singleton()
    {
	static ansi_scrubber s_as;

	return s_as;
    }

    void scrub_value(string &str, string_attrs_t &sa)
    {
	int rc, matches[60];

	do {
	    rc = pcre_exec(this->as_pcre,
			   NULL,
			   str.c_str(),
			   str.size(),
			   0,
			   0,
			   matches,
			   60);
	    if (rc > 0) {
		int c_start = matches[0];
		int c_end   = matches[1];
		struct line_range lr;
		bool has_attrs = false;
		int attrs = 0;
		int lpc;

		switch (str[matches[4]]) {
		case 'm':
		    for (lpc = matches[2]; lpc != string::npos && lpc < matches[3];) {
			int ansi_code = 0;
			
			if (sscanf(&(str[lpc]), "%d", &ansi_code) == 1) {
			    switch (ansi_code) {
			    case 1:
				attrs |= A_BOLD;
				break;
			    case 2:
				attrs |= A_DIM;
				break;
			    case 4:
				attrs |= A_UNDERLINE;
				break;
			    case 7:
				attrs |= A_REVERSE;
				break;
			    case 31:
				attrs |= COLOR_PAIR(view_colors::VC_RED);
				break;
			    case 32:
				attrs |= COLOR_PAIR(view_colors::VC_GREEN);
				break;
			    case 33:
				attrs |= COLOR_PAIR(view_colors::VC_YELLOW);
				break;
			    case 34:
				attrs |= COLOR_PAIR(view_colors::VC_BLUE);
				break;
			    case 35:
				attrs |= COLOR_PAIR(view_colors::VC_MAGENTA);
				break;
			    case 36:
				attrs |= COLOR_PAIR(view_colors::VC_CYAN);
				break;
			    case 37:
				attrs |= COLOR_PAIR(view_colors::VC_WHITE);
				break;
			    }
			}
			lpc = str.find(";", lpc);
			if (lpc != string::npos) {
			    lpc += 1;
			}
		    }
		    has_attrs = true;
		    break;
		case 'C':
		    {
			int spaces = 0;

			if (sscanf(&(str[matches[2]]), "%d", &spaces) == 1) {
			    str.insert(c_end, spaces, ' ');
			}
		    }
		    break;
		}
		str.erase(str.begin() + c_start, str.begin() + c_end);
		
		if (has_attrs) {
		    lr.lr_start = c_start;
		    lr.lr_end = -1;
		    sa[lr].insert(make_string_attr("style", attrs));
		}
	    }
	} while (rc > 0);
    };

private:
    pcre *build_pcre(const char *pattern) /* XXX refactor me */
    {
	const char *errptr;
	pcre       *retval;
	int        eoff;

	retval = pcre_compile(pattern, 0, &errptr, &eoff, NULL);
	if (retval == NULL) {
	    throw errptr;
	}

	return retval;
    };

    ansi_scrubber()
    {
	this->as_pcre = this->build_pcre("\x1b\\[([\\d=;]*)([a-zA-Z])");
    };

    pcre *as_pcre;
};

textview_curses::textview_curses()
    : tc_searching(false),
      tc_follow_search(false)
{
    this->set_data_source(this);
}

textview_curses::~textview_curses()
{ }

void textview_curses::reload_data(void)
{
    if (this->tc_sub_source != NULL) {
	this->tc_sub_source->text_update_marks(this->tc_bookmarks);
    }
    this->listview_curses::reload_data();
}

void textview_curses::grep_begin(grep_proc &gp)
{
    this->tc_searching   = true;
    this->tc_match_count = 0;
    this->tc_bookmarks[&BM_SEARCH].clear();

    this->tc_search_action.invoke(this);

    listview_curses::reload_data();
}

void textview_curses::grep_end(grep_proc &gp)
{
    this->tc_searching = false;
    this->tc_search_action.invoke(this);
}

void textview_curses::grep_match(grep_proc &gp,
				 grep_line_t line,
				 int start,
				 int end)
{
    this->tc_match_count += 1;
    this->tc_bookmarks[&BM_SEARCH].insert_once(vis_line_t(line));

    listview_curses::reload_data();
}


void textview_curses::listview_value_for_row(const listview_curses &lv,
					     vis_line_t row,
					     attr_line_t &value_out)
{
    bookmark_vector &bv = this->tc_bookmarks[&BM_USER];
    string_attrs_t &sa = value_out.get_attrs();
    string &str = value_out.get_string();
    highlight_map_t::iterator iter;
    string::iterator str_iter;
    
    this->tc_sub_source->text_value_for_line(*this, row, str);
    this->tc_sub_source->text_attrs_for_line(*this, row, sa);

    ansi_scrubber::singleton().scrub_value(str, sa);

    for (iter = this->tc_highlights.begin();
	 iter != this->tc_highlights.end();
	 iter++) {
	int off, hcount = 0;

	for (off = 0; off < str.size(); ) {
	    int rc, matches[30];
	    
	    rc = pcre_exec(iter->second.h_code,
			   NULL,
			   str.c_str(),
			   str.size(),
			   off,
			   0,
			   matches,
			   30);
	    if (rc > 0) {
		struct line_range lr;
	    
		if (rc == 2) {
		    lr.lr_start = matches[2];
		    lr.lr_end   = matches[3];
		}
		else {
		    lr.lr_start = matches[0];
		    lr.lr_end   = matches[1];
		}

		if (lr.lr_end > lr.lr_start) {
		  sa[lr].insert(make_string_attr("style", iter->second.
						 get_attrs(hcount)));
		  hcount++;

		  off = matches[1];
		}
		else {
		  off += 1;
		}
	    }
	    else {
		off = str.size();
	    }
	}
    }
    
    if (binary_search(bv.begin(), bv.end(), row)) {
	string_attrs_t::iterator iter;

	for (iter = sa.begin(); iter != sa.end(); iter++) {
	    attrs_map_t &am = iter->second;
	    attrs_map_t::iterator am_iter;

	    for (am_iter = am.begin(); am_iter != am.end(); am_iter++) {
		if (am_iter->first == "style") {
		    am_iter->second.sa_int ^= A_REVERSE;
		}
	    }
	}
    }
}