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