1343181Sdim//===-- Highlighter.cpp -----------------------------------------*- C++ -*-===// 2343181Sdim// 3353358Sdim// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4353358Sdim// See https://llvm.org/LICENSE.txt for license information. 5353358Sdim// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6343181Sdim// 7343181Sdim//===----------------------------------------------------------------------===// 8343181Sdim 9343181Sdim#include "lldb/Core/Highlighter.h" 10343181Sdim 11343181Sdim#include "lldb/Target/Language.h" 12343181Sdim#include "lldb/Utility/AnsiTerminal.h" 13343181Sdim#include "lldb/Utility/StreamString.h" 14343181Sdim 15343181Sdimusing namespace lldb_private; 16360784Sdimusing namespace lldb_private::ansi; 17343181Sdim 18343181Sdimvoid HighlightStyle::ColorStyle::Apply(Stream &s, llvm::StringRef value) const { 19343181Sdim s << m_prefix << value << m_suffix; 20343181Sdim} 21343181Sdim 22343181Sdimvoid HighlightStyle::ColorStyle::Set(llvm::StringRef prefix, 23343181Sdim llvm::StringRef suffix) { 24360784Sdim m_prefix = FormatAnsiTerminalCodes(prefix); 25360784Sdim m_suffix = FormatAnsiTerminalCodes(suffix); 26343181Sdim} 27343181Sdim 28343181Sdimvoid DefaultHighlighter::Highlight(const HighlightStyle &options, 29343181Sdim llvm::StringRef line, 30343181Sdim llvm::Optional<size_t> cursor_pos, 31343181Sdim llvm::StringRef previous_lines, 32343181Sdim Stream &s) const { 33343181Sdim // If we don't have a valid cursor, then we just print the line as-is. 34343181Sdim if (!cursor_pos || *cursor_pos >= line.size()) { 35343181Sdim s << line; 36343181Sdim return; 37343181Sdim } 38343181Sdim 39343181Sdim // If we have a valid cursor, we have to apply the 'selected' style around 40343181Sdim // the character below the cursor. 41343181Sdim 42343181Sdim // Split the line around the character which is below the cursor. 43343181Sdim size_t column = *cursor_pos; 44343181Sdim // Print the characters before the cursor. 45343181Sdim s << line.substr(0, column); 46343181Sdim // Print the selected character with the defined color codes. 47343181Sdim options.selected.Apply(s, line.substr(column, 1)); 48343181Sdim // Print the rest of the line. 49343181Sdim s << line.substr(column + 1U); 50343181Sdim} 51343181Sdim 52343181Sdimstatic HighlightStyle::ColorStyle GetColor(const char *c) { 53343181Sdim return HighlightStyle::ColorStyle(c, "${ansi.normal}"); 54343181Sdim} 55343181Sdim 56343181SdimHighlightStyle HighlightStyle::MakeVimStyle() { 57343181Sdim HighlightStyle result; 58343181Sdim result.comment = GetColor("${ansi.fg.purple}"); 59343181Sdim result.scalar_literal = GetColor("${ansi.fg.red}"); 60343181Sdim result.keyword = GetColor("${ansi.fg.green}"); 61343181Sdim return result; 62343181Sdim} 63343181Sdim 64343181Sdimconst Highlighter & 65343181SdimHighlighterManager::getHighlighterFor(lldb::LanguageType language_type, 66343181Sdim llvm::StringRef path) const { 67343181Sdim Language *language = lldb_private::Language::FindPlugin(language_type, path); 68343181Sdim if (language && language->GetHighlighter()) 69343181Sdim return *language->GetHighlighter(); 70343181Sdim return m_default; 71343181Sdim} 72343181Sdim 73343181Sdimstd::string Highlighter::Highlight(const HighlightStyle &options, 74343181Sdim llvm::StringRef line, 75343181Sdim llvm::Optional<size_t> cursor_pos, 76343181Sdim llvm::StringRef previous_lines) const { 77343181Sdim StreamString s; 78343181Sdim Highlight(options, line, cursor_pos, previous_lines, s); 79343181Sdim s.Flush(); 80343181Sdim return s.GetString().str(); 81343181Sdim} 82