add syntax text component

pull/84/head
Takayuki Maeda 3 years ago
parent a4d881f3f6
commit 50028e9c9b

@ -1,206 +1 @@
use super::{CommandInfo, Component, DrawableComponent, EventState};
use crate::config::KeyConfig;
use crate::event::Key;
use crate::ui::scrollbar::draw_scrollbar;
use crate::ui::{
self, common_nav, style::SharedTheme, AsyncSyntaxJob, ParagraphState, ScrollPos,
StatefulParagraph,
};
use anyhow::Result;
use asyncgit::{
asyncjob::AsyncSingleJob,
sync::{self, TreeFile},
AsyncGitNotification, CWD,
};
use crossterm::event::Event;
use database_tree::MoveSelection;
use itertools::Either;
use std::{cell::Cell, convert::From, path::Path};
use tui::{
backend::Backend,
layout::Rect,
text::Text,
widgets::{Block, Borders, Wrap},
Frame,
};
pub struct SyntaxTextComponent {
current_file: Option<(String, Either<ui::SyntaxText, String>)>,
async_highlighting: AsyncSingleJob<AsyncSyntaxJob, AsyncGitNotification>,
key_config: SharedKeyConfig,
paragraph_state: Cell<ParagraphState>,
focused: bool,
theme: SharedTheme,
}
impl SyntaxTextComponent {
///
pub fn new(key_config: KeyConfig, theme: SharedTheme) -> Self {
Self {
async_highlighting: AsyncSingleJob::new(
sender.clone(),
AsyncGitNotification::SyntaxHighlighting,
),
current_file: None,
paragraph_state: Cell::new(ParagraphState::default()),
focused: false,
key_config,
theme,
}
}
///
pub fn update(&mut self, ev: AsyncGitNotification) {
if ev == AsyncGitNotification::SyntaxHighlighting {
if let Some(job) = self.async_highlighting.take_last() {
if let Some((path, content)) = self.current_file.as_mut() {
if let Some(syntax) = job.result() {
if syntax.path() == Path::new(path) {
*content = Either::Left(syntax);
}
}
}
}
}
}
///
pub fn any_work_pending(&self) -> bool {
self.async_highlighting.is_pending()
}
pub fn clear(&mut self) {
self.current_file = None;
}
///
pub fn load_file(&mut self, path: String, item: &TreeFile) {
let already_loaded = self
.current_file
.as_ref()
.map(|(current_file, _)| current_file == &path)
.unwrap_or_default();
if !already_loaded {
//TODO: fetch file content async aswell
match sync::tree_file_content(CWD, item) {
Ok(content) => {
self.async_highlighting
.spawn(AsyncSyntaxJob::new(content.clone(), path.clone()));
self.current_file = Some((path, Either::Right(content)));
}
Err(e) => {
self.current_file =
Some((path, Either::Right(format!("error loading file: {}", e))));
}
}
}
}
fn scroll(&self, nav: MoveSelection) -> bool {
let state = self.paragraph_state.get();
let new_scroll_pos = match nav {
MoveSelection::Down => state.scroll().y.saturating_add(1),
MoveSelection::Up => state.scroll().y.saturating_sub(1),
MoveSelection::Top => 0,
MoveSelection::End => state
.lines()
.saturating_sub(state.height().saturating_sub(2)),
MoveSelection::PageUp => state
.scroll()
.y
.saturating_sub(state.height().saturating_sub(2)),
MoveSelection::PageDown => state
.scroll()
.y
.saturating_add(state.height().saturating_sub(2)),
_ => state.scroll().y,
};
self.set_scroll(new_scroll_pos)
}
fn set_scroll(&self, pos: u16) -> bool {
let mut state = self.paragraph_state.get();
let new_scroll_pos = pos.min(
state
.lines()
.saturating_sub(state.height().saturating_sub(2)),
);
if new_scroll_pos == state.scroll().y {
return false;
}
state.set_scroll(ScrollPos {
x: 0,
y: new_scroll_pos,
});
self.paragraph_state.set(state);
true
}
}
impl DrawableComponent for SyntaxTextComponent {
fn draw<B: Backend>(&self, f: &mut Frame<B>, area: Rect, focused: bool) -> Result<()> {
let text = self.current_file.as_ref().map_or_else(
|| Text::from(""),
|(_, content)| match content {
Either::Left(syn) => syn.into(),
Either::Right(s) => Text::from(s.as_str()),
},
);
let content = StatefulParagraph::new(text)
.wrap(Wrap { trim: false })
.block(
Block::default()
.title(
self.current_file
.as_ref()
.map(|(name, _)| name.clone())
.unwrap_or_default(),
)
.borders(Borders::ALL)
.border_style(self.theme.title(self.focused())),
);
let mut state = self.paragraph_state.get();
f.render_stateful_widget(content, area, &mut state);
self.paragraph_state.set(state);
self.set_scroll(state.scroll().y);
if self.focused() {
draw_scrollbar(
f,
area,
usize::from(
state
.lines()
.saturating_sub(state.height().saturating_sub(2)),
),
usize::from(state.scroll().y),
false,
false,
);
}
Ok(())
}
}
impl Component for SyntaxTextComponent {
fn commands(&self, out: &mut Vec<CommandInfo>) {}
fn event(&mut self, key: Key) -> Result<EventState> {
Ok(EventState::NotConsumed)
}
}

@ -1,4 +1,4 @@
use std::{ops::Range, path::Path};
use std::ops::Range;
use syntect::{
highlighting::{
FontStyle, HighlightState, Highlighter, RangedHighlightIterator, Style, ThemeSet,
@ -43,6 +43,25 @@ impl SyntaxText {
lines: syntax_lines,
}
}
pub fn convert(&self) -> tui::text::Text<'_> {
let mut result_lines: Vec<Spans> = Vec::with_capacity(self.lines.len());
for (syntax_line, line_content) in self.lines.iter().zip(self.text.lines()) {
let mut line_span = Spans(Vec::with_capacity(syntax_line.items.len()));
for (style, _, range) in &syntax_line.items {
let item_content = &line_content[range.clone()];
let item_style = syntact_style_to_tui(style);
line_span.0.push(Span::styled(item_content, item_style));
}
result_lines.push(line_span);
}
result_lines.into()
}
}
impl<'a> From<&'a SyntaxText> for tui::text::Text<'a> {

Loading…
Cancel
Save