mirror of https://github.com/TaKO8Ki/gobang
syntax text
parent
566f9ebb43
commit
c8ad6566c0
@ -0,0 +1,206 @@
|
|||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,87 @@
|
|||||||
|
use std::{ops::Range, path::Path};
|
||||||
|
use syntect::{
|
||||||
|
highlighting::{
|
||||||
|
FontStyle, HighlightState, Highlighter, RangedHighlightIterator, Style, ThemeSet,
|
||||||
|
},
|
||||||
|
parsing::{ParseState, ScopeStack, SyntaxSet},
|
||||||
|
};
|
||||||
|
use tui::text::{Span, Spans};
|
||||||
|
|
||||||
|
struct SyntaxLine {
|
||||||
|
items: Vec<(Style, usize, Range<usize>)>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct SyntaxText {
|
||||||
|
text: String,
|
||||||
|
lines: Vec<SyntaxLine>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SyntaxText {
|
||||||
|
pub fn new(text: String) -> Self {
|
||||||
|
const SYNTAX_SET: SyntaxSet = SyntaxSet::load_defaults_nonewlines();
|
||||||
|
const THEME_SET: ThemeSet = ThemeSet::load_defaults();
|
||||||
|
|
||||||
|
let mut state = ParseState::new(SYNTAX_SET.find_syntax_by_extension("sql").unwrap());
|
||||||
|
let highlighter = Highlighter::new(&THEME_SET.themes["base16-eighties.dark"]);
|
||||||
|
let mut syntax_lines: Vec<SyntaxLine> = Vec::new();
|
||||||
|
let mut highlight_state = HighlightState::new(&highlighter, ScopeStack::new());
|
||||||
|
|
||||||
|
for (number, line) in text.lines().enumerate() {
|
||||||
|
let ops = state.parse_line(line, &SYNTAX_SET);
|
||||||
|
let iter =
|
||||||
|
RangedHighlightIterator::new(&mut highlight_state, &ops[..], line, &highlighter);
|
||||||
|
|
||||||
|
syntax_lines.push(SyntaxLine {
|
||||||
|
items: iter
|
||||||
|
.map(|(style, _, range)| (style, number, range))
|
||||||
|
.collect(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
Self {
|
||||||
|
text,
|
||||||
|
lines: syntax_lines,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a> From<&'a SyntaxText> for tui::text::Text<'a> {
|
||||||
|
fn from(v: &'a SyntaxText) -> Self {
|
||||||
|
let mut result_lines: Vec<Spans> = Vec::with_capacity(v.lines.len());
|
||||||
|
|
||||||
|
for (syntax_line, line_content) in v.lines.iter().zip(v.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()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn syntact_style_to_tui(style: &Style) -> tui::style::Style {
|
||||||
|
let mut res = tui::style::Style::default().fg(tui::style::Color::Rgb(
|
||||||
|
style.foreground.r,
|
||||||
|
style.foreground.g,
|
||||||
|
style.foreground.b,
|
||||||
|
));
|
||||||
|
|
||||||
|
if style.font_style.contains(FontStyle::BOLD) {
|
||||||
|
res = res.add_modifier(tui::style::Modifier::BOLD);
|
||||||
|
}
|
||||||
|
if style.font_style.contains(FontStyle::ITALIC) {
|
||||||
|
res = res.add_modifier(tui::style::Modifier::ITALIC);
|
||||||
|
}
|
||||||
|
if style.font_style.contains(FontStyle::UNDERLINE) {
|
||||||
|
res = res.add_modifier(tui::style::Modifier::UNDERLINED);
|
||||||
|
}
|
||||||
|
|
||||||
|
res
|
||||||
|
}
|
Loading…
Reference in New Issue