forked from Mirrors/helix
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
649 lines
24 KiB
Rust
649 lines
24 KiB
Rust
use clap::ArgMatches as Args;
|
|
use helix_core::{indent::TAB_WIDTH, syntax::HighlightEvent, Position, Range, State};
|
|
use helix_view::{
|
|
commands,
|
|
document::Mode,
|
|
keymap::{self, Keymaps},
|
|
prompt::Prompt,
|
|
Document, Editor, Theme, View,
|
|
};
|
|
|
|
use std::{
|
|
borrow::Cow,
|
|
io::{self, stdout, Stdout, Write},
|
|
path::PathBuf,
|
|
time::Duration,
|
|
};
|
|
|
|
use smol::prelude::*;
|
|
|
|
use anyhow::Error;
|
|
|
|
use crossterm::{
|
|
cursor,
|
|
event::{read, Event, EventStream, KeyCode, KeyEvent},
|
|
execute, queue,
|
|
terminal::{self, disable_raw_mode, enable_raw_mode},
|
|
};
|
|
|
|
use tui::{
|
|
backend::CrosstermBackend,
|
|
buffer::Buffer as Surface,
|
|
layout::Rect,
|
|
style::{Color, Modifier, Style},
|
|
};
|
|
|
|
const OFFSET: u16 = 7; // 1 diagnostic + 5 linenr + 1 gutter
|
|
|
|
type Terminal = tui::Terminal<CrosstermBackend<std::io::Stdout>>;
|
|
|
|
const BASE_WIDTH: u16 = 30;
|
|
|
|
pub struct Application<'a> {
|
|
editor: Editor,
|
|
prompt: Option<Prompt>,
|
|
terminal: Renderer,
|
|
|
|
keymap: Keymaps,
|
|
executor: &'a smol::Executor<'a>,
|
|
lsp: helix_lsp::Client,
|
|
}
|
|
|
|
struct Renderer {
|
|
size: (u16, u16),
|
|
terminal: Terminal,
|
|
surface: Surface,
|
|
cache: Surface,
|
|
text_color: Style,
|
|
}
|
|
|
|
impl Renderer {
|
|
pub fn new() -> Result<Self, Error> {
|
|
let backend = CrosstermBackend::new(stdout());
|
|
let mut terminal = Terminal::new(backend)?;
|
|
let size = terminal::size().unwrap();
|
|
let text_color: Style = Style::default().fg(Color::Rgb(219, 191, 239)); // lilac
|
|
|
|
let area = Rect::new(0, 0, size.0, size.1);
|
|
|
|
Ok(Self {
|
|
size,
|
|
terminal,
|
|
surface: Surface::empty(area),
|
|
cache: Surface::empty(area),
|
|
text_color,
|
|
})
|
|
}
|
|
|
|
pub fn resize(&mut self, width: u16, height: u16) {
|
|
self.size = (width, height);
|
|
let area = Rect::new(0, 0, width, height);
|
|
self.surface = Surface::empty(area);
|
|
self.cache = Surface::empty(area);
|
|
}
|
|
|
|
pub fn render_view(&mut self, view: &mut View, viewport: Rect, theme: &Theme) {
|
|
self.render_buffer(view, viewport, theme);
|
|
self.render_statusline(view, theme);
|
|
}
|
|
|
|
// TODO: ideally not &mut View but highlights require it because of cursor cache
|
|
pub fn render_buffer(&mut self, view: &mut View, viewport: Rect, theme: &Theme) {
|
|
let area = Rect::new(0, 0, self.size.0, self.size.1);
|
|
self.surface.reset(); // reset is faster than allocating new empty surface
|
|
|
|
// clear with background color
|
|
self.surface.set_style(area, theme.get("ui.background"));
|
|
|
|
// TODO: inefficient, should feed chunks.iter() to tree_sitter.parse_with(|offset, pos|)
|
|
let source_code = view.doc.text().to_string();
|
|
|
|
let last_line = view.last_line();
|
|
|
|
let range = {
|
|
// calculate viewport byte ranges
|
|
let start = view.doc.text().line_to_byte(view.first_line);
|
|
let end = view.doc.text().line_to_byte(last_line)
|
|
+ view.doc.text().line(last_line).len_bytes();
|
|
|
|
start..end
|
|
};
|
|
|
|
// TODO: range doesn't actually restrict source, just highlight range
|
|
// TODO: cache highlight results
|
|
// TODO: only recalculate when state.doc is actually modified
|
|
let highlights: Vec<_> = match view.doc.syntax.as_mut() {
|
|
Some(syntax) => {
|
|
syntax
|
|
.highlight_iter(source_code.as_bytes(), Some(range), None, |_| None)
|
|
.unwrap()
|
|
.collect() // TODO: we collect here to avoid double borrow, fix later
|
|
}
|
|
None => vec![Ok(HighlightEvent::Source {
|
|
start: range.start,
|
|
end: range.end,
|
|
})],
|
|
};
|
|
let mut spans = Vec::new();
|
|
let mut visual_x = 0;
|
|
let mut line = 0u16;
|
|
let visible_selections: Vec<Range> = view
|
|
.doc
|
|
.state
|
|
.selection()
|
|
.ranges()
|
|
.iter()
|
|
// TODO: limit selection to one in viewport
|
|
// .filter(|range| !range.is_empty()) // && range.overlaps(&Range::new(start, end + 1))
|
|
.copied()
|
|
.collect();
|
|
|
|
'outer: for event in highlights {
|
|
match event.unwrap() {
|
|
HighlightEvent::HighlightStart(span) => {
|
|
spans.push(span);
|
|
}
|
|
HighlightEvent::HighlightEnd => {
|
|
spans.pop();
|
|
}
|
|
HighlightEvent::Source { start, end } => {
|
|
// TODO: filter out spans out of viewport for now..
|
|
|
|
let start = view.doc.text().byte_to_char(start);
|
|
let end = view.doc.text().byte_to_char(end); // <-- index 744, len 743
|
|
|
|
let text = view.doc.text().slice(start..end);
|
|
|
|
use helix_core::graphemes::{grapheme_width, RopeGraphemes};
|
|
|
|
let style = match spans.first() {
|
|
Some(span) => theme.get(theme.scopes()[span.0].as_str()),
|
|
None => Style::default().fg(Color::Rgb(164, 160, 232)), // lavender
|
|
};
|
|
|
|
// TODO: we could render the text to a surface, then cache that, that
|
|
// way if only the selection/cursor changes we can copy from cache
|
|
// and paint the new cursor.
|
|
|
|
let mut char_index = start;
|
|
|
|
// iterate over range char by char
|
|
for grapheme in RopeGraphemes::new(&text) {
|
|
// TODO: track current char_index
|
|
|
|
if grapheme == "\n" {
|
|
visual_x = 0;
|
|
line += 1;
|
|
|
|
// TODO: with proper iter this shouldn't be necessary
|
|
if line >= viewport.height {
|
|
break 'outer;
|
|
}
|
|
} else if grapheme == "\t" {
|
|
visual_x += (TAB_WIDTH as u16);
|
|
} else {
|
|
// Cow will prevent allocations if span contained in a single slice
|
|
// which should really be the majority case
|
|
let grapheme = Cow::from(grapheme);
|
|
let width = grapheme_width(&grapheme) as u16;
|
|
|
|
// TODO: this should really happen as an after pass
|
|
let style = if visible_selections
|
|
.iter()
|
|
.any(|range| range.contains(char_index))
|
|
{
|
|
// cedar
|
|
style.clone().bg(Color::Rgb(128, 47, 0))
|
|
} else {
|
|
style
|
|
};
|
|
|
|
let style = if visible_selections
|
|
.iter()
|
|
.any(|range| range.head == char_index)
|
|
{
|
|
style.clone().bg(Color::Rgb(255, 255, 255))
|
|
} else {
|
|
style
|
|
};
|
|
|
|
// ugh, improve with a traverse method
|
|
// or interleave highlight spans with selection and diagnostic spans
|
|
let style = if view.doc.diagnostics.iter().any(|diagnostic| {
|
|
diagnostic.range.0 <= char_index && diagnostic.range.1 > char_index
|
|
}) {
|
|
style.clone().add_modifier(Modifier::UNDERLINED)
|
|
} else {
|
|
style
|
|
};
|
|
|
|
// TODO: paint cursor heads except primary
|
|
|
|
self.surface
|
|
.set_string(OFFSET + visual_x, line, grapheme, style);
|
|
|
|
visual_x += width;
|
|
}
|
|
|
|
char_index += 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
let style: Style = theme.get("ui.linenr");
|
|
let warning: Style = theme.get("warning");
|
|
let last_line = view.last_line();
|
|
for (i, line) in (view.first_line..last_line).enumerate() {
|
|
if view.doc.diagnostics.iter().any(|d| d.line == line) {
|
|
self.surface.set_stringn(0, i as u16, "●", 1, warning);
|
|
}
|
|
|
|
self.surface
|
|
.set_stringn(1, i as u16, format!("{:>5}", line + 1), 5, style);
|
|
}
|
|
}
|
|
|
|
pub fn render_statusline(&mut self, view: &View, theme: &Theme) {
|
|
let mode = match view.doc.mode() {
|
|
Mode::Insert => "INS",
|
|
Mode::Normal => "NOR",
|
|
Mode::Goto => "GOTO",
|
|
};
|
|
// statusline
|
|
self.surface.set_style(
|
|
Rect::new(0, self.size.1 - 2, self.size.0, 1),
|
|
theme.get("ui.statusline"),
|
|
);
|
|
self.surface
|
|
.set_string(1, self.size.1 - 2, mode, self.text_color);
|
|
|
|
self.surface.set_string(
|
|
self.size.0 - 10,
|
|
self.size.1 - 2,
|
|
format!("{}", view.doc.diagnostics.len()),
|
|
self.text_color,
|
|
);
|
|
}
|
|
|
|
pub fn render_prompt(&mut self, view: &View, prompt: &Prompt, theme: &Theme) {
|
|
// completion
|
|
if !prompt.completion.is_empty() {
|
|
// TODO: find out better way of clearing individual lines of the screen
|
|
let mut row = 0;
|
|
let mut col = 0;
|
|
let max_col = self.size.0 / BASE_WIDTH;
|
|
let col_height = ((prompt.completion.len() as u16 + max_col - 1) / max_col);
|
|
|
|
for i in (3..col_height + 3) {
|
|
self.surface.set_string(
|
|
0,
|
|
self.size.1 - i as u16,
|
|
" ".repeat(self.size.0 as usize),
|
|
self.text_color,
|
|
);
|
|
}
|
|
self.surface.set_style(
|
|
Rect::new(0, self.size.1 - col_height - 2, self.size.0, col_height),
|
|
theme.get("ui.statusline"),
|
|
);
|
|
for (i, command) in prompt.completion.iter().enumerate() {
|
|
let color = if prompt.completion_selection_index.is_some()
|
|
&& i == prompt.completion_selection_index.unwrap()
|
|
{
|
|
Style::default().bg(Color::Rgb(104, 060, 232))
|
|
} else {
|
|
self.text_color
|
|
};
|
|
self.surface.set_stringn(
|
|
1 + col * BASE_WIDTH,
|
|
self.size.1 - col_height - 2 + row,
|
|
&command,
|
|
BASE_WIDTH as usize - 1,
|
|
color,
|
|
);
|
|
row += 1;
|
|
if row > col_height - 1 {
|
|
row = 0;
|
|
col += 1;
|
|
}
|
|
if col > max_col {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
// render buffer text
|
|
self.surface
|
|
.set_string(1, self.size.1 - 1, &prompt.prompt, self.text_color);
|
|
self.surface
|
|
.set_string(2, self.size.1 - 1, &prompt.line, self.text_color);
|
|
}
|
|
|
|
pub fn draw(&mut self) {
|
|
use tui::backend::Backend;
|
|
// TODO: theres probably a better place for this
|
|
self.terminal
|
|
.backend_mut()
|
|
.draw(self.cache.diff(&self.surface).into_iter());
|
|
// swap the buffer
|
|
std::mem::swap(&mut self.surface, &mut self.cache);
|
|
}
|
|
|
|
pub fn render_cursor(&mut self, view: &View, prompt: Option<&Prompt>, viewport: Rect) {
|
|
let mut stdout = stdout();
|
|
match view.doc.mode() {
|
|
Mode::Insert => write!(stdout, "\x1B[6 q"),
|
|
mode => write!(stdout, "\x1B[2 q"),
|
|
};
|
|
let pos = if let Some(prompt) = prompt {
|
|
Position::new(self.size.0 as usize, 2 + prompt.cursor)
|
|
} else {
|
|
if let Some(path) = view.doc.path() {
|
|
self.surface.set_string(
|
|
6,
|
|
self.size.1 - 1,
|
|
path.to_string_lossy(),
|
|
self.text_color,
|
|
);
|
|
}
|
|
|
|
let cursor = view.doc.state.selection().cursor();
|
|
|
|
let mut pos = view
|
|
.screen_coords_at_pos(&view.doc.text().slice(..), cursor)
|
|
.expect("Cursor is out of bounds.");
|
|
pos.col += viewport.x as usize;
|
|
pos.row += viewport.y as usize;
|
|
pos
|
|
};
|
|
|
|
execute!(stdout, cursor::MoveTo(pos.col as u16, pos.row as u16));
|
|
}
|
|
}
|
|
|
|
impl<'a> Application<'a> {
|
|
pub fn new(mut args: Args, executor: &'a smol::Executor<'a>) -> Result<Self, Error> {
|
|
let terminal = Renderer::new()?;
|
|
let mut editor = Editor::new();
|
|
|
|
if let Some(file) = args.values_of_t::<PathBuf>("files").unwrap().pop() {
|
|
editor.open(file, terminal.size)?;
|
|
}
|
|
|
|
let lsp = helix_lsp::Client::start(&executor, "rust-analyzer", &[]);
|
|
|
|
let mut app = Self {
|
|
editor,
|
|
terminal,
|
|
// TODO; move to state
|
|
prompt: None,
|
|
|
|
//
|
|
keymap: keymap::default(),
|
|
executor,
|
|
lsp,
|
|
};
|
|
|
|
Ok(app)
|
|
}
|
|
|
|
fn render(&mut self) {
|
|
let viewport = Rect::new(OFFSET, 0, self.terminal.size.0, self.terminal.size.1 - 2); // - 2 for statusline and prompt
|
|
|
|
// SAFETY: we cheat around the view_mut() borrow because it doesn't allow us to also borrow
|
|
// theme. Theme is immutable mutating view won't disrupt theme_ref.
|
|
let theme_ref = unsafe { &*(&self.editor.theme as *const Theme) };
|
|
if let Some(view) = self.editor.view_mut() {
|
|
self.terminal.render_view(view, viewport, theme_ref);
|
|
if let Some(prompt) = &self.prompt {
|
|
if prompt.should_close {
|
|
self.prompt = None;
|
|
} else {
|
|
self.terminal.render_prompt(view, prompt, theme_ref);
|
|
}
|
|
}
|
|
}
|
|
|
|
self.terminal.draw();
|
|
|
|
// TODO: drop unwrap
|
|
self.terminal
|
|
.render_cursor(self.editor.view().unwrap(), self.prompt.as_ref(), viewport);
|
|
}
|
|
|
|
pub async fn event_loop(&mut self) {
|
|
let mut reader = EventStream::new();
|
|
|
|
// initialize lsp
|
|
let res = self.lsp.initialize().await;
|
|
let res = self
|
|
.lsp
|
|
.text_document_did_open(&self.editor.view().unwrap().doc)
|
|
.await;
|
|
|
|
self.render();
|
|
|
|
loop {
|
|
if self.editor.should_close {
|
|
break;
|
|
}
|
|
|
|
use futures_util::{select, FutureExt};
|
|
select! {
|
|
event = reader.next().fuse() => {
|
|
self.handle_terminal_events(event).await
|
|
}
|
|
notification = self.lsp.incoming.next().fuse() => {
|
|
self.handle_lsp_notification(notification).await
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub async fn handle_terminal_events(
|
|
&mut self,
|
|
event: Option<Result<Event, crossterm::ErrorKind>>,
|
|
) {
|
|
// Handle key events
|
|
match event {
|
|
Some(Ok(Event::Resize(width, height))) => {
|
|
self.terminal.resize(width, height);
|
|
|
|
// TODO: simplistic ensure cursor in view for now
|
|
// TODO: loop over views
|
|
if let Some(view) = self.editor.view_mut() {
|
|
view.size = self.terminal.size;
|
|
view.ensure_cursor_in_view()
|
|
};
|
|
|
|
self.render();
|
|
}
|
|
Some(Ok(Event::Key(event))) => {
|
|
// if there's a prompt, it takes priority
|
|
if let Some(prompt) = &mut self.prompt {
|
|
self.prompt
|
|
.as_mut()
|
|
.unwrap()
|
|
.handle_input(event, &mut self.editor);
|
|
|
|
self.render();
|
|
} else if let Some(view) = self.editor.view_mut() {
|
|
let keys = vec![event];
|
|
// TODO: sequences (`gg`)
|
|
// TODO: handle count other than 1
|
|
match view.doc.mode() {
|
|
Mode::Insert => {
|
|
if let Some(command) = self.keymap[&Mode::Insert].get(&keys) {
|
|
command(view, 1);
|
|
} else if let KeyEvent {
|
|
code: KeyCode::Char(c),
|
|
..
|
|
} = event
|
|
{
|
|
commands::insert::insert_char(view, c);
|
|
}
|
|
view.ensure_cursor_in_view();
|
|
}
|
|
Mode::Normal => {
|
|
if let &[KeyEvent {
|
|
code: KeyCode::Char(':'),
|
|
..
|
|
}] = keys.as_slice()
|
|
{
|
|
let prompt = Prompt::new(
|
|
":".to_owned(),
|
|
|_input: &str| {
|
|
// TODO: i need this duplicate list right now to avoid borrow checker issues
|
|
let command_list = vec![
|
|
String::from("q"),
|
|
String::from("aaa"),
|
|
String::from("bbb"),
|
|
String::from("ccc"),
|
|
String::from("ddd"),
|
|
String::from("eee"),
|
|
String::from("averylongcommandaverylongcommandaverylongcommandaverylongcommandaverylongcommand"),
|
|
String::from("q"),
|
|
String::from("aaa"),
|
|
String::from("bbb"),
|
|
String::from("ccc"),
|
|
String::from("ddd"),
|
|
String::from("eee"),
|
|
String::from("q"),
|
|
String::from("aaa"),
|
|
String::from("bbb"),
|
|
String::from("ccc"),
|
|
String::from("ddd"),
|
|
String::from("eee"),
|
|
String::from("q"),
|
|
String::from("aaa"),
|
|
String::from("bbb"),
|
|
String::from("ccc"),
|
|
String::from("ddd"),
|
|
String::from("eee"),
|
|
String::from("q"),
|
|
String::from("aaa"),
|
|
String::from("bbb"),
|
|
String::from("ccc"),
|
|
String::from("ddd"),
|
|
String::from("eee"),
|
|
];
|
|
command_list
|
|
.into_iter()
|
|
.filter(|command| command.contains(_input))
|
|
.collect()
|
|
}, // completion
|
|
|editor: &mut Editor, input: &str| match input {
|
|
"q" => editor.should_close = true,
|
|
_ => (),
|
|
},
|
|
);
|
|
|
|
self.prompt = Some(prompt);
|
|
|
|
// HAXX: special casing for command mode
|
|
} else if let Some(command) = self.keymap[&Mode::Normal].get(&keys) {
|
|
command(view, 1);
|
|
|
|
// TODO: simplistic ensure cursor in view for now
|
|
view.ensure_cursor_in_view();
|
|
}
|
|
}
|
|
mode => {
|
|
if let Some(command) = self.keymap[&mode].get(&keys) {
|
|
command(view, 1);
|
|
|
|
// TODO: simplistic ensure cursor in view for now
|
|
view.ensure_cursor_in_view();
|
|
}
|
|
}
|
|
}
|
|
self.render();
|
|
}
|
|
}
|
|
Some(Ok(Event::Mouse(_))) => (), // unhandled
|
|
Some(Err(x)) => panic!(x),
|
|
None => panic!(),
|
|
};
|
|
}
|
|
|
|
pub async fn handle_lsp_notification(&mut self, notification: Option<helix_lsp::Notification>) {
|
|
use helix_lsp::Notification;
|
|
match notification {
|
|
Some(Notification::PublishDiagnostics(params)) => {
|
|
let path = Some(params.uri.to_file_path().unwrap());
|
|
let view = self
|
|
.editor
|
|
.views
|
|
.iter_mut()
|
|
.find(|view| view.doc.path == path);
|
|
|
|
if let Some(view) = view {
|
|
let doc = view.doc.text().slice(..);
|
|
let diagnostics = params
|
|
.diagnostics
|
|
.into_iter()
|
|
.map(|diagnostic| {
|
|
use helix_lsp::util::lsp_pos_to_pos;
|
|
let start = lsp_pos_to_pos(&doc, diagnostic.range.start);
|
|
let end = lsp_pos_to_pos(&doc, diagnostic.range.end);
|
|
|
|
helix_core::Diagnostic {
|
|
range: (start, end),
|
|
line: diagnostic.range.start.line as usize,
|
|
message: diagnostic.message,
|
|
// severity
|
|
// code
|
|
// source
|
|
}
|
|
})
|
|
.collect();
|
|
|
|
view.doc.diagnostics = diagnostics;
|
|
|
|
self.render();
|
|
}
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
pub async fn run(&mut self) -> Result<(), Error> {
|
|
enable_raw_mode()?;
|
|
|
|
let mut stdout = stdout();
|
|
|
|
execute!(stdout, terminal::EnterAlternateScreen)?;
|
|
|
|
// Exit the alternate screen and disable raw mode before panicking
|
|
let hook = std::panic::take_hook();
|
|
std::panic::set_hook(Box::new(move |info| {
|
|
execute!(std::io::stdout(), terminal::LeaveAlternateScreen);
|
|
disable_raw_mode();
|
|
hook(info);
|
|
}));
|
|
|
|
self.event_loop().await;
|
|
|
|
// reset cursor shape
|
|
write!(stdout, "\x1B[2 q");
|
|
|
|
execute!(stdout, terminal::LeaveAlternateScreen)?;
|
|
|
|
disable_raw_mode()?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
// TODO: language configs:
|
|
// tabSize, fileExtension etc, mapping to tree sitter parser
|
|
// themes:
|
|
// map tree sitter highlights to color values
|
|
//
|
|
// TODO: expand highlight thing so we're able to render only viewport range
|
|
// TODO: async: maybe pre-cache scopes as empty so we render all graphemes initially as regular
|
|
////text until calc finishes
|
|
// TODO: scope matching: biggest union match? [string] & [html, string], [string, html] & [ string, html]
|
|
// can do this by sorting our theme matches based on array len (longest first) then stopping at the
|
|
// first rule that matches (rule.all(|scope| scopes.contains(scope)))
|