diff options
Diffstat (limited to 'src/term_extra.hh')
| -rw-r--r-- | src/term_extra.hh | 119 |
1 files changed, 119 insertions, 0 deletions
diff --git a/src/term_extra.hh b/src/term_extra.hh new file mode 100644 index 0000000..9740315 --- /dev/null +++ b/src/term_extra.hh @@ -0,0 +1,119 @@ +/** + * Copyright (c) 2014, 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 term_extra.hh + */ + +#ifndef term_extra_hh +#define term_extra_hh + +#include <string> + +#include <pwd.h> +#include <string.h> +#include <sys/param.h> +#include <sys/types.h> +#include <unistd.h> + +#include "listview_curses.hh" +#include "log_format.hh" +#include "logfile.hh" + +class term_extra { +public: + term_extra() + { + const char* term_name = getenv("TERM"); + + this->te_enabled + = (term_name != nullptr && strstr(term_name, "xterm") != nullptr); + + if (getenv("SSH_CONNECTION") != nullptr) { + char hostname[MAXHOSTNAMELEN] = "UNKNOWN"; + struct passwd* userent; + + gethostname(hostname, sizeof(hostname)); + this->te_prefix = hostname; + if ((userent = getpwuid(getuid())) != nullptr) { + this->te_prefix + = std::string(userent->pw_name) + "@" + this->te_prefix; + } + this->te_prefix += ":"; + } + } + + void update_title(listview_curses* lc) + { + static const char* xterm_title_fmt = "\033]0;%s\007"; + + if (!this->te_enabled) { + return; + } + + if (lc->get_inner_height() > 0) { + std::vector<attr_line_t> rows(1); + + lc->get_data_source()->listview_value_for_rows( + *lc, lc->get_top(), rows); + string_attrs_t& sa = rows[0].get_attrs(); + auto line_attr_opt = get_string_attr(sa, logline::L_FILE); + if (line_attr_opt) { + auto lf = line_attr_opt.value().get(); + const std::string& filename = lf->get_unique_path(); + + if (filename != this->te_last_title) { + std::string title = this->te_prefix + filename; + + printf(xterm_title_fmt, title.c_str()); + fflush(stdout); + + this->te_last_title = filename; + } + return; + } + } + + const std::string& view_title = lc->get_title(); + + if (view_title != this->te_last_title) { + std::string title = this->te_prefix + view_title; + + printf(xterm_title_fmt, title.c_str()); + fflush(stdout); + + this->te_last_title = view_title; + } + } + +private: + bool te_enabled; + std::string te_prefix; + std::string te_last_title; +}; + +#endif |