use crate::compositor::{Component, Context}; use tui::{ buffer::Buffer as Surface, text::{Span, Spans, Text}, }; use std::sync::Arc; use pulldown_cmark::{CodeBlockKind, Event, HeadingLevel, Options, Parser, Tag}; use helix_core::{ syntax::{self, HighlightEvent, Syntax}, Rope, }; use helix_view::{ graphics::{Margin, Rect, Style}, Theme, }; fn styled_multiline_text<'a>(text: String, style: Style) -> Text<'a> { let spans: Vec<_> = text .lines() .map(|line| Span::styled(line.to_string(), style)) .map(Spans::from) .collect(); Text::from(spans) } pub fn highlighted_code_block<'a>( text: String, language: &str, theme: Option<&Theme>, config_loader: Arc, additional_highlight_spans: Option)>>, ) -> Text<'a> { let mut spans = Vec::new(); let mut lines = Vec::new(); let get_theme = |key: &str| -> Style { theme.map(|t| t.get(key)).unwrap_or_default() }; let text_style = get_theme(Markdown::TEXT_STYLE); let code_style = get_theme(Markdown::BLOCK_STYLE); let theme = match theme { Some(t) => t, None => return styled_multiline_text(text, code_style), }; let rope = Rope::from(text.as_ref()); let syntax = config_loader .language_configuration_for_injection_string(language) .and_then(|config| config.highlight_config(theme.scopes())) .map(|config| Syntax::new(&rope, config, Arc::clone(&config_loader))); let syntax = match syntax { Some(s) => s, None => return styled_multiline_text(text, code_style), }; let highlight_iter = syntax .highlight_iter(rope.slice(..), None, None) .map(|e| e.unwrap()); let highlight_iter: Box> = if let Some(spans) = additional_highlight_spans { Box::new(helix_core::syntax::merge(highlight_iter, spans)) } else { Box::new(highlight_iter) }; let mut highlights = Vec::new(); for event in highlight_iter { match event { HighlightEvent::HighlightStart(span) => { highlights.push(span); } HighlightEvent::HighlightEnd => { highlights.pop(); } HighlightEvent::Source { start, end } => { let style = highlights .iter() .fold(text_style, |acc, span| acc.patch(theme.highlight(span.0))); let mut slice = &text[start..end]; // TODO: do we need to handle all unicode line endings // here, or is just '\n' okay? while let Some(end) = slice.find('\n') { // emit span up to newline let text = &slice[..end]; let text = text.replace('\t', " "); // replace tabs let span = Span::styled(text, style); spans.push(span); // truncate slice to after newline slice = &slice[end + 1..]; // make a new line let spans = std::mem::take(&mut spans); lines.push(Spans::from(spans)); } // if there's anything left, emit it too if !slice.is_empty() { let span = Span::styled(slice.replace('\t', " "), style); spans.push(span); } } } } if !spans.is_empty() { let spans = std::mem::take(&mut spans); lines.push(Spans::from(spans)); } Text::from(lines) } pub struct Markdown { contents: String, config_loader: Arc, } // TODO: pre-render and self reference via Pin // better yet, just use Tendril + subtendril for references impl Markdown { const TEXT_STYLE: &'static str = "ui.text"; const BLOCK_STYLE: &'static str = "markup.raw.inline"; const HEADING_STYLES: [&'static str; 6] = [ "markup.heading.1", "markup.heading.2", "markup.heading.3", "markup.heading.4", "markup.heading.5", "markup.heading.6", ]; const INDENT: &'static str = " "; pub fn new(contents: String, config_loader: Arc) -> Self { Self { contents, config_loader, } } pub fn parse(&self, theme: Option<&Theme>) -> tui::text::Text<'_> { fn push_line<'a>(spans: &mut Vec>, lines: &mut Vec>) { let spans = std::mem::take(spans); if !spans.is_empty() { lines.push(Spans::from(spans)); } } let mut options = Options::empty(); options.insert(Options::ENABLE_STRIKETHROUGH); let parser = Parser::new_ext(&self.contents, options); // TODO: if possible, render links as terminal hyperlinks: https://gist.github.com/egmontkob/eb114294efbcd5adb1944c9f3cb5feda let mut tags = Vec::new(); let mut spans = Vec::new(); let mut lines = Vec::new(); let mut list_stack = Vec::new(); let get_indent = |level: usize| { if level < 1 { String::new() } else { Self::INDENT.repeat(level - 1) } }; let get_theme = |key: &str| -> Style { theme.map(|t| t.get(key)).unwrap_or_default() }; let text_style = get_theme(Self::TEXT_STYLE); let code_style = get_theme(Self::BLOCK_STYLE); let heading_styles: Vec