pub mod parser; use std::collections::{HashMap, HashSet};use std::iter;use std::mem;use std::ops::Range;use std::rc::Rc;use std::sync::Arc;use std::time::Duration; use gpui::{ actions, point, quad, AnyElement, App, Bounds, ClipboardItem, CursorStyle, DispatchPhase, Edges, Entity, FocusHandle, Focusable, FontStyle, FontWeight, GlobalElementId, Hitbox, Hsla, KeyContext, Length, MouseDownEvent, MouseEvent, MouseMoveEvent, MouseUpEvent, Point, Render, Stateful, StrikethroughStyle, StyleRefinement, StyledText, Task, TextLayout, TextRun, TextStyle, TextStyleRefinement,};use language::{Language, LanguageRegistry, Rope};use parser::{parse_links_only, parse_markdown, MarkdownEvent, MarkdownTag, MarkdownTagEnd};use pulldown_cmark::Alignment;use theme::SyntaxTheme;use ui::{prelude::*, Tooltip};use util::{ResultExt, TryFutureExt}; use crate::parser::CodeBlockKind; #[derive(Clone)]pub struct MarkdownStyle { pub base_text_style: TextStyle, pub code_block: StyleRefinement, pub code_block_overflow_x_scroll: bool, pub inline_code: TextStyleRefinement, pub block_quote: TextStyleRefinement, pub link: TextStyleRefinement, pub rule_color: Hsla, pub block_quote_border_color: Hsla, pub syntax: Arc<SyntaxTheme>, pub selection_background_color: Hsla, pub heading: StyleRefinement, pub table_overflow_x_scroll: bool,} impl Default for MarkdownStyle { fn default() -> Self { Self { base_text_style: Default::default(), code_block: Default::default(), code_block_overflow_x_scroll: false, inline_code: Default::default(), block_quote: Default::default(), link: Default::default(), rule_color: Default::default(), block_quote_border_color: Default::default(), syntax: Arc::new(SyntaxTheme::default()), selection_background_color: Default::default(), heading: Default::default(), table_overflow_x_scroll: false, } }} pub struct Markdown { source: SharedString, selection: Selection, pressed_link: Option<RenderedLink>, autoscroll_request: Option<usize>, style: MarkdownStyle, parsed_markdown: ParsedMarkdown, should_reparse: bool, pending_parse: Option<Task<Option<()>>>, focus_handle: FocusHandle, language_registry: Option<Arc<LanguageRegistry>>, fallback_code_block_language: Option<String>, open_url: Option<Box<dyn Fn(SharedString, &mut Window, &mut App)>>, options: Options, copied_code_blocks: HashSet<ElementId>,} #[derive(Debug)]struct Options { parse_links_only: bool, copy_code_block_buttons: bool,} actions!(markdown, [Copy]); impl Markdown { pub fn new( source: SharedString, style: MarkdownStyle, language_registry: Option<Arc<LanguageRegistry>>, fallback_code_block_language: Option<String>, cx: &mut Context<Self>, ) -> Self { let focus_handle = cx.focus_handle(); let mut this = Self { source, selection: Selection::default(), pressed_link: None, autoscroll_request: None, style, should_reparse: false, parsed_markdown: ParsedMarkdown::default(), pending_parse: None, focus_handle, language_registry, fallback_code_block_language, options: Options { parse_links_only: false, copy_code_block_buttons: true, }, open_url: None, copied_code_blocks: HashSet::new(), }; this.parse(cx); this } pub fn open_url( self, open_url: impl Fn(SharedString, &mut Window, &mut App) + 'static, ) -> Self { Self { open_url: Some(Box::new(open_url)), ..self } } pub fn new_text(source: SharedString, style: MarkdownStyle, cx: &mut Context<Self>) -> Self { let focus_handle = cx.focus_handle(); let mut this = Self { source, selection: Selection::default(), pressed_link: None, autoscroll_request: None, style, should_reparse: false, parsed_markdown: ParsedMarkdown::default(), pending_parse: None, focus_handle, language_registry: None, fallback_code_block_language: None, options: Options { parse_links_only: true, copy_code_block_buttons: true, }, open_url: None, copied_code_blocks: HashSet::new(), }; this.parse(cx); this } pub fn source(&self) -> &str { &self.source } pub fn append(&mut self, text: &str, cx: &mut Context<Self>) { self.source = SharedString::new(self.source.to_string() + text); self.parse(cx); } pub fn reset(&mut self, source: SharedString, cx: &mut Context<Self>) { if source == self.source() { return; } self.source = source; self.selection = Selection::default(); self.autoscroll_request = None; self.pending_parse = None; self.should_reparse = false; self.parsed_markdown = ParsedMarkdown::default(); self.parse(cx); } pub fn parsed_markdown(&self) -> &ParsedMarkdown { &self.parsed_markdown } fn copy(&self, text: &RenderedText, _: &mut Window, cx: &mut Context<Self>) { if self.selection.end <= self.selection.start { return; } let text = text.text_for_range(self.selection.start..self.selection.end); cx.write_to_clipboard(ClipboardItem::new_string(text)); } fn parse(&mut self, cx: &mut Context<Self>) { if self.source.is_empty() { return; } if self.pending_parse.is_some() { self.should_reparse = true; return; } let source = self.source.clone(); let parse_text_only = self.options.parse_links_only; let language_registry = self.language_registry.clone(); let fallback = self.fallback_code_block_language.clone(); let parsed = cx.background_spawn(async move { if parse_text_only { return anyhow::Ok(ParsedMarkdown { events: Arc::from(parse_links_only(source.as_ref())), source, languages: HashMap::default(), }); } let (events, language_names) = parse_markdown(&source); let mut languages = HashMap::with_capacity(language_names.len()); for name in language_names { if let Some(registry) = language_registry.as_ref() { let language = if !name.is_empty() { registry.language_for_name(&name) } else if let Some(fallback) = &fallback { registry.language_for_name(fallback) } else { continue; }; if let Ok(language) = language.await { languages.insert(name, language); } } } anyhow::Ok(ParsedMarkdown { source, events: Arc::from(events), languages, }) }); self.should_reparse = false; self.pending_parse = Some(cx.spawn(|this, mut cx| { async move { let parsed = parsed.await?; this.update(&mut cx, |this, cx| { this.parsed_markdown = parsed; this.pending_parse.take(); if this.should_reparse { this.parse(cx); } cx.notify(); }) .ok(); anyhow::Ok(()) } .log_err() })); } pub fn copy_code_block_buttons(mut self, should_copy: bool) -> Self { self.options.copy_code_block_buttons = should_copy; self }} impl Render for Markdown { fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement { MarkdownElement::new(cx.entity().clone(), self.style.clone()) }} impl Focusable for Markdown { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.focus_handle.clone() }} #[derive(Copy, Clone, Default, Debug)]struct Selection { start: usize, end: usize, reversed: bool, pending: bool,} impl Selection { fn set_head(&mut self, head: usize) { if head < self.tail() { if !self.reversed { self.end = self.start; self.reversed = true; } self.start = head; } else { if self.reversed { self.start = self.end; self.reversed = false; } self.end = head; } } fn tail(&self) -> usize { if self.reversed { self.end } else { self.start } }} #[derive(Default)]pub struct ParsedMarkdown { source: SharedString, events: Arc<[(Range<usize>, MarkdownEvent)]>, languages: HashMap<SharedString, Arc<Language>>,} impl ParsedMarkdown { pub fn source(&self) -> &SharedString { &self.source } pub fn events(&self) -> &Arc<[(Range<usize>, MarkdownEvent)]> { &self.events }} pub struct MarkdownElement { markdown: Entity<Markdown>, style: MarkdownStyle,} impl MarkdownElement { fn new(markdown: Entity<Markdown>, style: MarkdownStyle) -> Self { Self { markdown, style } } fn paint_selection( &self, bounds: Bounds<Pixels>, rendered_text: &RenderedText, window: &mut Window, cx: &mut App, ) { let selection = self.markdown.read(cx).selection; let selection_start = rendered_text.position_for_source_index(selection.start); let selection_end = rendered_text.position_for_source_index(selection.end); if let Some(((start_position, start_line_height), (end_position, end_line_height))) = selection_start.zip(selection_end) { if start_position.y == end_position.y { window.paint_quad(quad( Bounds::from_corners( start_position, point(end_position.x, end_position.y + end_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } else { window.paint_quad(quad( Bounds::from_corners( start_position, point(bounds.right(), start_position.y + start_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); if end_position.y > start_position.y + start_line_height { window.paint_quad(quad( Bounds::from_corners( point(bounds.left(), start_position.y + start_line_height), point(bounds.right(), end_position.y), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } window.paint_quad(quad( Bounds::from_corners( point(bounds.left(), end_position.y), point(end_position.x, end_position.y + end_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } } } fn paint_mouse_listeners( &self, hitbox: &Hitbox, rendered_text: &RenderedText, window: &mut Window, cx: &mut App, ) { let is_hovering_link = hitbox.is_hovered(window) && !self.markdown.read(cx).selection.pending && rendered_text .link_for_position(window.mouse_position()) .is_some(); if is_hovering_link { window.set_cursor_style(CursorStyle::PointingHand, hitbox); } else { window.set_cursor_style(CursorStyle::IBeam, hitbox); } self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); let hitbox = hitbox.clone(); move |markdown, event: &MouseDownEvent, phase, window, cx| { if hitbox.is_hovered(window) { if phase.bubble() { if let Some(link) = rendered_text.link_for_position(event.position) { markdown.pressed_link = Some(link.clone()); } else { let source_index = match rendered_text.source_index_for_position(event.position) { Ok(ix) | Err(ix) => ix, }; let range = if event.click_count == 2 { rendered_text.surrounding_word_range(source_index) } else if event.click_count == 3 { rendered_text.surrounding_line_range(source_index) } else { source_index..source_index }; markdown.selection = Selection { start: range.start, end: range.end, reversed: false, pending: true, }; window.focus(&markdown.focus_handle); window.prevent_default(); } cx.notify(); } } else if phase.capture() { markdown.selection = Selection::default(); markdown.pressed_link = None; cx.notify(); } } }); self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); let hitbox = hitbox.clone(); let was_hovering_link = is_hovering_link; move |markdown, event: &MouseMoveEvent, phase, window, cx| { if phase.capture() { return; } if markdown.selection.pending { let source_index = match rendered_text.source_index_for_position(event.position) { Ok(ix) | Err(ix) => ix, }; markdown.selection.set_head(source_index); markdown.autoscroll_request = Some(source_index); cx.notify(); } else { let is_hovering_link = hitbox.is_hovered(window) && rendered_text.link_for_position(event.position).is_some(); if is_hovering_link != was_hovering_link { cx.notify(); } } } }); self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); move |markdown, event: &MouseUpEvent, phase, window, cx| { if phase.bubble() { if let Some(pressed_link) = markdown.pressed_link.take() { if Some(&pressed_link) == rendered_text.link_for_position(event.position) { if let Some(open_url) = markdown.open_url.as_mut() { open_url(pressed_link.destination_url, window, cx); } else { cx.open_url(&pressed_link.destination_url); } } } } else if markdown.selection.pending { markdown.selection.pending = false; #[cfg(any(target_os = "linux", target_os = "freebsd"))] { let text = rendered_text .text_for_range(markdown.selection.start..markdown.selection.end); cx.write_to_primary(ClipboardItem::new_string(text)) } cx.notify();Lines 1–500 of 1431. Load more
pub mod parser; use std::collections::{HashMap, HashSet}; use std::iter; use std::mem; use std::ops::Range; use std::rc::Rc; use std::sync::Arc; use std::time::Duration; use gpui::{ actions, point, quad, AnyElement, App, Bounds, ClipboardItem, CursorStyle, DispatchPhase, Edges, Entity, FocusHandle, Focusable, FontStyle, FontWeight, GlobalElementId, Hitbox, Hsla, KeyContext, Length, MouseDownEvent, MouseEvent, MouseMoveEvent, MouseUpEvent, Point, Render, Stateful, StrikethroughStyle, StyleRefinement, StyledText, Task, TextLayout, TextRun, TextStyle, TextStyleRefinement, }; use language::{Language, LanguageRegistry, Rope}; use parser::{parse_links_only, parse_markdown, MarkdownEvent, MarkdownTag, MarkdownTagEnd}; use pulldown_cmark::Alignment; use theme::SyntaxTheme; use ui::{prelude::*, Tooltip}; use util::{ResultExt, TryFutureExt}; use crate::parser::CodeBlockKind; #[derive(Clone)] pub struct MarkdownStyle { pub base_text_style: TextStyle, pub code_block: StyleRefinement, pub code_block_overflow_x_scroll: bool, pub inline_code: TextStyleRefinement, pub block_quote: TextStyleRefinement, pub link: TextStyleRefinement, pub rule_color: Hsla, pub block_quote_border_color: Hsla, pub syntax: Arc<SyntaxTheme>, pub selection_background_color: Hsla, pub heading: StyleRefinement, pub table_overflow_x_scroll: bool, } impl Default for MarkdownStyle { fn default() -> Self { Self { base_text_style: Default::default(), code_block: Default::default(), code_block_overflow_x_scroll: false, inline_code: Default::default(), block_quote: Default::default(), link: Default::default(), rule_color: Default::default(), block_quote_border_color: Default::default(), syntax: Arc::new(SyntaxTheme::default()), selection_background_color: Default::default(), heading: Default::default(), table_overflow_x_scroll: false, } } } pub struct Markdown { source: SharedString, selection: Selection, pressed_link: Option<RenderedLink>, autoscroll_request: Option<usize>, style: MarkdownStyle, parsed_markdown: ParsedMarkdown, should_reparse: bool, pending_parse: Option<Task<Option<()>>>, focus_handle: FocusHandle, language_registry: Option<Arc<LanguageRegistry>>, fallback_code_block_language: Option<String>, open_url: Option<Box<dyn Fn(SharedString, &mut Window, &mut App)>>, options: Options, copied_code_blocks: HashSet<ElementId>, } #[derive(Debug)] struct Options { parse_links_only: bool, copy_code_block_buttons: bool, } actions!(markdown, [Copy]); impl Markdown { pub fn new( source: SharedString, style: MarkdownStyle, language_registry: Option<Arc<LanguageRegistry>>, fallback_code_block_language: Option<String>, cx: &mut Context<Self>, ) -> Self { let focus_handle = cx.focus_handle(); let mut this = Self { source, selection: Selection::default(), pressed_link: None, autoscroll_request: None, style, should_reparse: false, parsed_markdown: ParsedMarkdown::default(), pending_parse: None, focus_handle, language_registry, fallback_code_block_language, options: Options { parse_links_only: false, copy_code_block_buttons: true, }, open_url: None, copied_code_blocks: HashSet::new(), }; this.parse(cx); this } pub fn open_url( self, open_url: impl Fn(SharedString, &mut Window, &mut App) + 'static, ) -> Self { Self { open_url: Some(Box::new(open_url)), ..self } } pub fn new_text(source: SharedString, style: MarkdownStyle, cx: &mut Context<Self>) -> Self { let focus_handle = cx.focus_handle(); let mut this = Self { source, selection: Selection::default(), pressed_link: None, autoscroll_request: None, style, should_reparse: false, parsed_markdown: ParsedMarkdown::default(), pending_parse: None, focus_handle, language_registry: None, fallback_code_block_language: None, options: Options { parse_links_only: true, copy_code_block_buttons: true, }, open_url: None, copied_code_blocks: HashSet::new(), }; this.parse(cx); this } pub fn source(&self) -> &str { &self.source } pub fn append(&mut self, text: &str, cx: &mut Context<Self>) { self.source = SharedString::new(self.source.to_string() + text); self.parse(cx); } pub fn reset(&mut self, source: SharedString, cx: &mut Context<Self>) { if source == self.source() { return; } self.source = source; self.selection = Selection::default(); self.autoscroll_request = None; self.pending_parse = None; self.should_reparse = false; self.parsed_markdown = ParsedMarkdown::default(); self.parse(cx); } pub fn parsed_markdown(&self) -> &ParsedMarkdown { &self.parsed_markdown } fn copy(&self, text: &RenderedText, _: &mut Window, cx: &mut Context<Self>) { if self.selection.end <= self.selection.start { return; } let text = text.text_for_range(self.selection.start..self.selection.end); cx.write_to_clipboard(ClipboardItem::new_string(text)); } fn parse(&mut self, cx: &mut Context<Self>) { if self.source.is_empty() { return; } if self.pending_parse.is_some() { self.should_reparse = true; return; } let source = self.source.clone(); let parse_text_only = self.options.parse_links_only; let language_registry = self.language_registry.clone(); let fallback = self.fallback_code_block_language.clone(); let parsed = cx.background_spawn(async move { if parse_text_only { return anyhow::Ok(ParsedMarkdown { events: Arc::from(parse_links_only(source.as_ref())), source, languages: HashMap::default(), }); } let (events, language_names) = parse_markdown(&source); let mut languages = HashMap::with_capacity(language_names.len()); for name in language_names { if let Some(registry) = language_registry.as_ref() { let language = if !name.is_empty() { registry.language_for_name(&name) } else if let Some(fallback) = &fallback { registry.language_for_name(fallback) } else { continue; }; if let Ok(language) = language.await { languages.insert(name, language); } } } anyhow::Ok(ParsedMarkdown { source, events: Arc::from(events), languages, }) }); self.should_reparse = false; self.pending_parse = Some(cx.spawn(|this, mut cx| { async move { let parsed = parsed.await?; this.update(&mut cx, |this, cx| { this.parsed_markdown = parsed; this.pending_parse.take(); if this.should_reparse { this.parse(cx); } cx.notify(); }) .ok(); anyhow::Ok(()) } .log_err() })); } pub fn copy_code_block_buttons(mut self, should_copy: bool) -> Self { self.options.copy_code_block_buttons = should_copy; self } } impl Render for Markdown { fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement { MarkdownElement::new(cx.entity().clone(), self.style.clone()) } } impl Focusable for Markdown { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.focus_handle.clone() } } #[derive(Copy, Clone, Default, Debug)] struct Selection { start: usize, end: usize, reversed: bool, pending: bool, } impl Selection { fn set_head(&mut self, head: usize) { if head < self.tail() { if !self.reversed { self.end = self.start; self.reversed = true; } self.start = head; } else { if self.reversed { self.start = self.end; self.reversed = false; } self.end = head; } } fn tail(&self) -> usize { if self.reversed { self.end } else { self.start } } } #[derive(Default)] pub struct ParsedMarkdown { source: SharedString, events: Arc<[(Range<usize>, MarkdownEvent)]>, languages: HashMap<SharedString, Arc<Language>>, } impl ParsedMarkdown { pub fn source(&self) -> &SharedString { &self.source } pub fn events(&self) -> &Arc<[(Range<usize>, MarkdownEvent)]> { &self.events } } pub struct MarkdownElement { markdown: Entity<Markdown>, style: MarkdownStyle, } impl MarkdownElement { fn new(markdown: Entity<Markdown>, style: MarkdownStyle) -> Self { Self { markdown, style } } fn paint_selection( &self, bounds: Bounds<Pixels>, rendered_text: &RenderedText, window: &mut Window, cx: &mut App, ) { let selection = self.markdown.read(cx).selection; let selection_start = rendered_text.position_for_source_index(selection.start); let selection_end = rendered_text.position_for_source_index(selection.end); if let Some(((start_position, start_line_height), (end_position, end_line_height))) = selection_start.zip(selection_end) { if start_position.y == end_position.y { window.paint_quad(quad( Bounds::from_corners( start_position, point(end_position.x, end_position.y + end_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } else { window.paint_quad(quad( Bounds::from_corners( start_position, point(bounds.right(), start_position.y + start_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); if end_position.y > start_position.y + start_line_height { window.paint_quad(quad( Bounds::from_corners( point(bounds.left(), start_position.y + start_line_height), point(bounds.right(), end_position.y), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } window.paint_quad(quad( Bounds::from_corners( point(bounds.left(), end_position.y), point(end_position.x, end_position.y + end_line_height), ), Pixels::ZERO, self.style.selection_background_color, Edges::default(), Hsla::transparent_black(), )); } } } fn paint_mouse_listeners( &self, hitbox: &Hitbox, rendered_text: &RenderedText, window: &mut Window, cx: &mut App, ) { let is_hovering_link = hitbox.is_hovered(window) && !self.markdown.read(cx).selection.pending && rendered_text .link_for_position(window.mouse_position()) .is_some(); if is_hovering_link { window.set_cursor_style(CursorStyle::PointingHand, hitbox); } else { window.set_cursor_style(CursorStyle::IBeam, hitbox); } self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); let hitbox = hitbox.clone(); move |markdown, event: &MouseDownEvent, phase, window, cx| { if hitbox.is_hovered(window) { if phase.bubble() { if let Some(link) = rendered_text.link_for_position(event.position) { markdown.pressed_link = Some(link.clone()); } else { let source_index = match rendered_text.source_index_for_position(event.position) { Ok(ix) | Err(ix) => ix, }; let range = if event.click_count == 2 { rendered_text.surrounding_word_range(source_index) } else if event.click_count == 3 { rendered_text.surrounding_line_range(source_index) } else { source_index..source_index }; markdown.selection = Selection { start: range.start, end: range.end, reversed: false, pending: true, }; window.focus(&markdown.focus_handle); window.prevent_default(); } cx.notify(); } } else if phase.capture() { markdown.selection = Selection::default(); markdown.pressed_link = None; cx.notify(); } } }); self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); let hitbox = hitbox.clone(); let was_hovering_link = is_hovering_link; move |markdown, event: &MouseMoveEvent, phase, window, cx| { if phase.capture() { return; } if markdown.selection.pending { let source_index = match rendered_text.source_index_for_position(event.position) { Ok(ix) | Err(ix) => ix, }; markdown.selection.set_head(source_index); markdown.autoscroll_request = Some(source_index); cx.notify(); } else { let is_hovering_link = hitbox.is_hovered(window) && rendered_text.link_for_position(event.position).is_some(); if is_hovering_link != was_hovering_link { cx.notify(); } } } }); self.on_mouse_event(window, cx, { let rendered_text = rendered_text.clone(); move |markdown, event: &MouseUpEvent, phase, window, cx| { if phase.bubble() { if let Some(pressed_link) = markdown.pressed_link.take() { if Some(&pressed_link) == rendered_text.link_for_position(event.position) { if let Some(open_url) = markdown.open_url.as_mut() { open_url(pressed_link.destination_url, window, cx); } else { cx.open_url(&pressed_link.destination_url); } } } } else if markdown.selection.pending { markdown.selection.pending = false; #[cfg(any(target_os = "linux", target_os = "freebsd"))] { let text = rendered_text .text_for_range(markdown.selection.start..markdown.selection.end); cx.write_to_primary(ClipboardItem::new_string(text)) } cx.notify();