Refactoring: move language_servers into Editor, proper load for doc.

imgbot
Blaž Hrastnik 3 years ago
parent 15dd7ca6d8
commit 05c7fb98df

@ -114,7 +114,7 @@ impl Registry {
pub fn new() -> Self { pub fn new() -> Self {
let mut inner = HashMap::new(); let mut inner = HashMap::new();
inner.insert("rust".to_string(), OnceCell::new()); inner.insert("source.rust".to_string(), OnceCell::new());
Self { Self {
inner, inner,

@ -33,8 +33,6 @@ pub struct Application {
terminal: Terminal, terminal: Terminal,
executor: &'static smol::Executor<'static>, executor: &'static smol::Executor<'static>,
language_server: Arc<helix_lsp::Client>,
language_servers: helix_lsp::Registry,
} }
impl Application { impl Application {
@ -44,18 +42,8 @@ impl Application {
let mut editor = Editor::new(); let mut editor = Editor::new();
let size = terminal.size()?; let size = terminal.size()?;
let language_servers = helix_lsp::Registry::new();
let language_server = language_servers.get("rust", &executor).unwrap();
if let Some(file) = args.values_of_t::<PathBuf>("files").unwrap().pop() { if let Some(file) = args.values_of_t::<PathBuf>("files").unwrap().pop() {
editor.open(file, (size.width, size.height))?; editor.open(file, (size.width, size.height), executor)?;
// TODO: do this everywhere
editor
.view_mut()
.unwrap()
.doc
.set_language_server(Some(language_server.clone()));
} }
let mut compositor = Compositor::new(); let mut compositor = Compositor::new();
@ -67,8 +55,6 @@ impl Application {
compositor, compositor,
executor, executor,
language_server,
language_servers,
}; };
Ok(app) Ok(app)
@ -76,15 +62,10 @@ impl Application {
fn render(&mut self) { fn render(&mut self) {
let executor = &self.executor; let executor = &self.executor;
let language_servers = &self.language_servers;
let editor = &mut self.editor; let editor = &mut self.editor;
let compositor = &self.compositor; let compositor = &self.compositor;
let mut cx = crate::compositor::Context { let mut cx = crate::compositor::Context { editor, executor };
editor,
executor,
language_servers,
};
let area = self.terminal.size().unwrap(); let area = self.terminal.size().unwrap();
compositor.render(area, self.terminal.current_buffer_mut(), &mut cx); compositor.render(area, self.terminal.current_buffer_mut(), &mut cx);
@ -97,12 +78,6 @@ impl Application {
pub async fn event_loop(&mut self) { pub async fn event_loop(&mut self) {
let mut reader = EventStream::new(); let mut reader = EventStream::new();
let doc = &self.editor.view().unwrap().doc;
self.language_server
.text_document_did_open(doc.url().unwrap(), doc.version, doc.text())
.await
.unwrap();
self.render(); self.render();
loop { loop {
@ -115,7 +90,7 @@ impl Application {
event = reader.next().fuse() => { event = reader.next().fuse() => {
self.handle_terminal_events(event) self.handle_terminal_events(event)
} }
call = self.language_servers.incoming.next().fuse() => { call = self.editor.language_servers.incoming.next().fuse() => {
self.handle_language_server_message(call).await self.handle_language_server_message(call).await
} }
} }
@ -126,7 +101,6 @@ impl Application {
let mut cx = crate::compositor::Context { let mut cx = crate::compositor::Context {
editor: &mut self.editor, editor: &mut self.editor,
executor: &self.executor, executor: &self.executor,
language_servers: &self.language_servers,
}; };
// Handle key events // Handle key events
let should_redraw = match event { let should_redraw = match event {
@ -197,16 +171,17 @@ impl Application {
Some(Call::MethodCall(call)) => { Some(Call::MethodCall(call)) => {
debug!("Method not found {}", call.method); debug!("Method not found {}", call.method);
self.language_server.reply( // self.language_server.reply(
call.id, // call.id,
// TODO: make a Into trait that can cast to Err(jsonrpc::Error) // // TODO: make a Into trait that can cast to Err(jsonrpc::Error)
Err(helix_lsp::jsonrpc::Error { // Err(helix_lsp::jsonrpc::Error {
code: helix_lsp::jsonrpc::ErrorCode::MethodNotFound, // code: helix_lsp::jsonrpc::ErrorCode::MethodNotFound,
message: "Method not found".to_string(), // message: "Method not found".to_string(),
data: None, // data: None,
}), // }),
); // );
} }
None => (),
e => unreachable!("{:?}", e), e => unreachable!("{:?}", e),
} }
} }

@ -18,57 +18,66 @@ use helix_view::{
Editor, Editor,
}; };
pub struct Context<'a, 'b> { pub struct Context<'a> {
pub count: usize, pub count: usize,
pub view: &'a mut View, pub editor: &'a mut Editor,
pub executor: &'a smol::Executor<'b>, pub executor: &'static smol::Executor<'static>,
pub language_servers: &'a helix_lsp::Registry,
pub callback: Option<crate::compositor::Callback>, pub callback: Option<crate::compositor::Callback>,
} }
impl<'a> Context<'a> {
pub fn view(&mut self) -> &mut View {
self.editor.view_mut()
}
}
/// A command is a function that takes the current state and a count, and does a side-effect on the /// A command is a function that takes the current state and a count, and does a side-effect on the
/// state (usually by creating and applying a transaction). /// state (usually by creating and applying a transaction).
pub type Command = fn(cx: &mut Context); pub type Command = fn(cx: &mut Context);
pub fn move_char_left(cx: &mut Context) { pub fn move_char_left(cx: &mut Context) {
let count = cx.count;
let selection = let selection =
cx.view cx.view()
.doc .doc
.state .state
.move_selection(Direction::Backward, Granularity::Character, cx.count); .move_selection(Direction::Backward, Granularity::Character, count);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_char_right(cx: &mut Context) { pub fn move_char_right(cx: &mut Context) {
let count = cx.count;
let selection = let selection =
cx.view cx.view()
.doc .doc
.state .state
.move_selection(Direction::Forward, Granularity::Character, cx.count); .move_selection(Direction::Forward, Granularity::Character, count);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_line_up(cx: &mut Context) { pub fn move_line_up(cx: &mut Context) {
let count = cx.count;
let selection = let selection =
cx.view cx.view()
.doc .doc
.state .state
.move_selection(Direction::Backward, Granularity::Line, cx.count); .move_selection(Direction::Backward, Granularity::Line, count);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_line_down(cx: &mut Context) { pub fn move_line_down(cx: &mut Context) {
let count = cx.count;
let selection = let selection =
cx.view cx.view()
.doc .doc
.state .state
.move_selection(Direction::Forward, Granularity::Line, cx.count); .move_selection(Direction::Forward, Granularity::Line, count);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_line_end(cx: &mut Context) { pub fn move_line_end(cx: &mut Context) {
let lines = selection_lines(&cx.view.doc.state); let lines = selection_lines(&cx.view().doc.state);
let positions = lines let positions = lines
.into_iter() .into_iter()
@ -77,75 +86,85 @@ pub fn move_line_end(cx: &mut Context) {
// Line end is pos at the start of next line - 1 // Line end is pos at the start of next line - 1
// subtract another 1 because the line ends with \n // subtract another 1 because the line ends with \n
cx.view.doc.text().line_to_char(index + 1).saturating_sub(2) cx.view()
.doc
.text()
.line_to_char(index + 1)
.saturating_sub(2)
}) })
.map(|pos| Range::new(pos, pos)); .map(|pos| Range::new(pos, pos));
let selection = Selection::new(positions.collect(), 0); let selection = Selection::new(positions.collect(), 0);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_line_start(cx: &mut Context) { pub fn move_line_start(cx: &mut Context) {
let lines = selection_lines(&cx.view.doc.state); let lines = selection_lines(&cx.view().doc.state);
let positions = lines let positions = lines
.into_iter() .into_iter()
.map(|index| { .map(|index| {
// adjust all positions to the start of the line. // adjust all positions to the start of the line.
cx.view.doc.text().line_to_char(index) cx.view().doc.text().line_to_char(index)
}) })
.map(|pos| Range::new(pos, pos)); .map(|pos| Range::new(pos, pos));
let selection = Selection::new(positions.collect(), 0); let selection = Selection::new(positions.collect(), 0);
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn move_next_word_start(cx: &mut Context) { pub fn move_next_word_start(cx: &mut Context) {
let pos = cx.view.doc.state.move_pos( let count = cx.count;
cx.view.doc.selection().cursor(), let view = cx.view();
let pos = view.doc.state.move_pos(
view.doc.selection().cursor(),
Direction::Forward, Direction::Forward,
Granularity::Word, Granularity::Word,
cx.count, count,
); );
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
pub fn move_prev_word_start(cx: &mut Context) { pub fn move_prev_word_start(cx: &mut Context) {
let pos = cx.view.doc.state.move_pos( let count = cx.count;
cx.view.doc.selection().cursor(), let view = cx.view();
let pos = view.doc.state.move_pos(
view.doc.selection().cursor(),
Direction::Backward, Direction::Backward,
Granularity::Word, Granularity::Word,
cx.count, count,
); );
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
pub fn move_next_word_end(cx: &mut Context) { pub fn move_next_word_end(cx: &mut Context) {
let count = cx.count;
let view = cx.view();
let pos = State::move_next_word_end( let pos = State::move_next_word_end(
&cx.view.doc.text().slice(..), &view.doc.text().slice(..),
cx.view.doc.selection().cursor(), view.doc.selection().cursor(),
cx.count, count,
); );
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
pub fn move_file_start(cx: &mut Context) { pub fn move_file_start(cx: &mut Context) {
cx.view.doc.set_selection(Selection::point(0)); cx.view().doc.set_selection(Selection::point(0));
cx.view.doc.mode = Mode::Normal; cx.view().doc.mode = Mode::Normal;
} }
pub fn move_file_end(cx: &mut Context) { pub fn move_file_end(cx: &mut Context) {
let text = &cx.view.doc.text(); let text = &cx.view().doc.text();
let last_line = text.line_to_char(text.len_lines().saturating_sub(2)); let last_line = text.line_to_char(text.len_lines().saturating_sub(2));
cx.view.doc.set_selection(Selection::point(last_line)); cx.view().doc.set_selection(Selection::point(last_line));
cx.view.doc.mode = Mode::Normal; cx.view().doc.mode = Mode::Normal;
} }
pub fn check_cursor_in_view(view: &View) -> bool { pub fn check_cursor_in_view(view: &View) -> bool {
@ -160,93 +179,98 @@ pub fn check_cursor_in_view(view: &View) -> bool {
} }
pub fn page_up(cx: &mut Context) { pub fn page_up(cx: &mut Context) {
if cx.view.first_line < PADDING { let view = cx.view();
if view.first_line < PADDING {
return; return;
} }
cx.view.first_line = cx.view.first_line.saturating_sub(cx.view.size.1 as usize); view.first_line = view.first_line.saturating_sub(view.size.1 as usize);
if !check_cursor_in_view(cx.view) { if !check_cursor_in_view(view) {
let text = cx.view.doc.text(); let text = view.doc.text();
let pos = text.line_to_char(cx.view.last_line().saturating_sub(PADDING)); let pos = text.line_to_char(view.last_line().saturating_sub(PADDING));
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
} }
pub fn page_down(cx: &mut Context) { pub fn page_down(cx: &mut Context) {
cx.view.first_line += cx.view.size.1 as usize + PADDING; let view = cx.view();
view.first_line += view.size.1 as usize + PADDING;
if cx.view.first_line < cx.view.doc.text().len_lines() { if view.first_line < view.doc.text().len_lines() {
let text = cx.view.doc.text(); let text = view.doc.text();
let pos = text.line_to_char(cx.view.first_line as usize); let pos = text.line_to_char(view.first_line as usize);
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
} }
pub fn half_page_up(cx: &mut Context) { pub fn half_page_up(cx: &mut Context) {
if cx.view.first_line < PADDING { let view = cx.view();
if view.first_line < PADDING {
return; return;
} }
cx.view.first_line = cx view.first_line = view.first_line.saturating_sub(view.size.1 as usize / 2);
.view
.first_line
.saturating_sub(cx.view.size.1 as usize / 2);
if !check_cursor_in_view(cx.view) { if !check_cursor_in_view(view) {
let text = &cx.view.doc.text(); let text = &view.doc.text();
let pos = text.line_to_char(cx.view.last_line() - PADDING); let pos = text.line_to_char(view.last_line() - PADDING);
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
} }
pub fn half_page_down(cx: &mut Context) { pub fn half_page_down(cx: &mut Context) {
let lines = cx.view.doc.text().len_lines(); let view = cx.view();
if cx.view.first_line < lines.saturating_sub(cx.view.size.1 as usize) { let lines = view.doc.text().len_lines();
cx.view.first_line += cx.view.size.1 as usize / 2; if view.first_line < lines.saturating_sub(view.size.1 as usize) {
view.first_line += view.size.1 as usize / 2;
} }
if !check_cursor_in_view(cx.view) { if !check_cursor_in_view(view) {
let text = cx.view.doc.text(); let text = view.doc.text();
let pos = text.line_to_char(cx.view.first_line as usize); let pos = text.line_to_char(view.first_line as usize);
cx.view.doc.set_selection(Selection::point(pos)); view.doc.set_selection(Selection::point(pos));
} }
} }
// avoid select by default by having a visual mode switch that makes movements into selects // avoid select by default by having a visual mode switch that makes movements into selects
pub fn extend_char_left(cx: &mut Context) { pub fn extend_char_left(cx: &mut Context) {
let count = cx.count;
let view = cx.view();
let selection = let selection =
cx.view view.doc
.doc
.state .state
.extend_selection(Direction::Backward, Granularity::Character, cx.count); .extend_selection(Direction::Backward, Granularity::Character, count);
cx.view.doc.set_selection(selection); view.doc.set_selection(selection);
} }
pub fn extend_char_right(cx: &mut Context) { pub fn extend_char_right(cx: &mut Context) {
let count = cx.count;
let view = cx.view();
let selection = let selection =
cx.view view.doc
.doc
.state .state
.extend_selection(Direction::Forward, Granularity::Character, cx.count); .extend_selection(Direction::Forward, Granularity::Character, count);
cx.view.doc.set_selection(selection); view.doc.set_selection(selection);
} }
pub fn extend_line_up(cx: &mut Context) { pub fn extend_line_up(cx: &mut Context) {
let selection = let count = cx.count;
cx.view let view = cx.view();
.doc let selection = view
.state .doc
.extend_selection(Direction::Backward, Granularity::Line, cx.count); .state
cx.view.doc.set_selection(selection); .extend_selection(Direction::Backward, Granularity::Line, count);
view.doc.set_selection(selection);
} }
pub fn extend_line_down(cx: &mut Context) { pub fn extend_line_down(cx: &mut Context) {
let selection = let count = cx.count;
cx.view let view = cx.view();
.doc let selection = view
.state .doc
.extend_selection(Direction::Forward, Granularity::Line, cx.count); .state
cx.view.doc.set_selection(selection); .extend_selection(Direction::Forward, Granularity::Line, count);
view.doc.set_selection(selection);
} }
pub fn split_selection(cx: &mut Context) { pub fn split_selection(cx: &mut Context) {
@ -263,7 +287,7 @@ pub fn split_selection(cx: &mut Context) {
// # update state // # update state
// } // }
let snapshot = cx.view.doc.state.clone(); let snapshot = cx.view().doc.state.clone();
let prompt = Prompt::new( let prompt = Prompt::new(
"split:".to_string(), "split:".to_string(),
@ -272,7 +296,7 @@ pub fn split_selection(cx: &mut Context) {
match event { match event {
PromptEvent::Abort => { PromptEvent::Abort => {
// revert state // revert state
let view = editor.view_mut().unwrap(); let view = editor.view_mut();
view.doc.state = snapshot.clone(); view.doc.state = snapshot.clone();
} }
PromptEvent::Validate => { PromptEvent::Validate => {
@ -281,7 +305,7 @@ pub fn split_selection(cx: &mut Context) {
PromptEvent::Update => { PromptEvent::Update => {
match Regex::new(input) { match Regex::new(input) {
Ok(regex) => { Ok(regex) => {
let view = editor.view_mut().unwrap(); let view = editor.view_mut();
// revert state to what it was before the last update // revert state to what it was before the last update
view.doc.state = snapshot.clone(); view.doc.state = snapshot.clone();
@ -306,32 +330,33 @@ pub fn split_selection(cx: &mut Context) {
} }
pub fn split_selection_on_newline(cx: &mut Context) { pub fn split_selection_on_newline(cx: &mut Context) {
let text = &cx.view.doc.text().slice(..); let view = cx.view();
let text = &view.doc.text().slice(..);
// only compile the regex once // only compile the regex once
#[allow(clippy::trivial_regex)] #[allow(clippy::trivial_regex)]
static REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r"\n").unwrap()); static REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r"\n").unwrap());
let selection = selection::split_on_matches(text, cx.view.doc.selection(), &REGEX); let selection = selection::split_on_matches(text, view.doc.selection(), &REGEX);
cx.view.doc.set_selection(selection); view.doc.set_selection(selection);
} }
pub fn select_line(cx: &mut Context) { pub fn select_line(cx: &mut Context) {
// TODO: count // TODO: count
let pos = cx.view.doc.selection().primary(); let pos = cx.view().doc.selection().primary();
let text = cx.view.doc.text(); let text = cx.view().doc.text();
let line = text.char_to_line(pos.head); let line = text.char_to_line(pos.head);
let start = text.line_to_char(line); let start = text.line_to_char(line);
let end = text.line_to_char(line + 1).saturating_sub(1); let end = text.line_to_char(line + 1).saturating_sub(1);
cx.view.doc.set_selection(Selection::single(start, end)); cx.view().doc.set_selection(Selection::single(start, end));
} }
// heuristic: append changes to history after each command, unless we're in insert mode // heuristic: append changes to history after each command, unless we're in insert mode
fn _delete_selection(cx: &mut Context) { fn _delete_selection(cx: &mut Context) {
let transaction = Transaction::change_by_selection(&cx.view.doc.state, |range| { let transaction = Transaction::change_by_selection(&cx.view().doc.state, |range| {
(range.from(), range.to() + 1, None) (range.from(), range.to() + 1, None)
}); });
cx.view.doc.apply(&transaction); cx.view().doc.apply(&transaction);
} }
pub fn delete_selection(cx: &mut Context) { pub fn delete_selection(cx: &mut Context) {
@ -347,26 +372,26 @@ pub fn change_selection(cx: &mut Context) {
pub fn collapse_selection(cx: &mut Context) { pub fn collapse_selection(cx: &mut Context) {
let selection = cx let selection = cx
.view .view()
.doc .doc
.selection() .selection()
.transform(|range| Range::new(range.head, range.head)); .transform(|range| Range::new(range.head, range.head));
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
pub fn flip_selections(cx: &mut Context) { pub fn flip_selections(cx: &mut Context) {
let selection = cx let selection = cx
.view .view()
.doc .doc
.selection() .selection()
.transform(|range| Range::new(range.head, range.anchor)); .transform(|range| Range::new(range.head, range.anchor));
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
fn enter_insert_mode(cx: &mut Context) { fn enter_insert_mode(cx: &mut Context) {
cx.view.doc.mode = Mode::Insert; cx.view().doc.mode = Mode::Insert;
// TODO: store selection for undo // TODO: store selection for undo
} }
@ -376,34 +401,36 @@ pub fn insert_mode(cx: &mut Context) {
enter_insert_mode(cx); enter_insert_mode(cx);
let selection = cx let selection = cx
.view .view()
.doc .doc
.selection() .selection()
.transform(|range| Range::new(range.to(), range.from())); .transform(|range| Range::new(range.to(), range.from()));
cx.view.doc.set_selection(selection); cx.view().doc.set_selection(selection);
} }
// inserts at the end of each selection // inserts at the end of each selection
pub fn append_mode(cx: &mut Context) { pub fn append_mode(cx: &mut Context) {
enter_insert_mode(cx); enter_insert_mode(cx);
cx.view.doc.restore_cursor = true; let view = cx.view();
view.doc.restore_cursor = true;
// TODO: as transaction // TODO: as transaction
let text = &cx.view.doc.text().slice(..); let text = &view.doc.text().slice(..);
let selection = cx.view.doc.selection().transform(|range| { let selection = view.doc.selection().transform(|range| {
// TODO: to() + next char // TODO: to() + next char
Range::new( Range::new(
range.from(), range.from(),
graphemes::next_grapheme_boundary(text, range.to()), graphemes::next_grapheme_boundary(text, range.to()),
) )
}); });
cx.view.doc.set_selection(selection); view.doc.set_selection(selection);
} }
// TODO: I, A, o and O can share a lot of the primitives. // TODO: I, A, o and O can share a lot of the primitives.
pub fn command_mode(cx: &mut Context) { pub fn command_mode(cx: &mut Context) {
let executor = cx.executor;
cx.callback = Some(Box::new( cx.callback = Some(Box::new(
|compositor: &mut Compositor, editor: &mut Editor| { move |compositor: &mut Compositor, editor: &mut Editor| {
let prompt = Prompt::new( let prompt = Prompt::new(
":".to_owned(), ":".to_owned(),
|_input: &str| { |_input: &str| {
@ -446,7 +473,7 @@ pub fn command_mode(cx: &mut Context) {
.filter(|command| command.contains(_input)) .filter(|command| command.contains(_input))
.collect() .collect()
}, // completion }, // completion
|editor: &mut Editor, input: &str, event: PromptEvent| { move |editor: &mut Editor, input: &str, event: PromptEvent| {
if event != PromptEvent::Validate { if event != PromptEvent::Validate {
return; return;
} }
@ -456,9 +483,8 @@ pub fn command_mode(cx: &mut Context) {
match *parts.as_slice() { match *parts.as_slice() {
["q"] => editor.should_close = true, ["q"] => editor.should_close = true,
["o", path] => { ["o", path] => {
// TODO: make view()/view_mut() always contain a view. let size = editor.view().size;
let size = editor.view().unwrap().size; editor.open(path.into(), size, executor);
editor.open(path.into(), size);
} }
_ => (), _ => (),
} }
@ -469,9 +495,9 @@ pub fn command_mode(cx: &mut Context) {
)); ));
} }
pub fn file_picker(cx: &mut Context) { pub fn file_picker(cx: &mut Context) {
let picker = ui::file_picker("./", cx.executor);
cx.callback = Some(Box::new( cx.callback = Some(Box::new(
|compositor: &mut Compositor, editor: &mut Editor| { |compositor: &mut Compositor, editor: &mut Editor| {
let picker = ui::file_picker("./");
compositor.push(Box::new(picker)); compositor.push(Box::new(picker));
}, },
)); ));
@ -519,21 +545,23 @@ pub fn append_to_line(cx: &mut Context) {
pub fn open_below(cx: &mut Context) { pub fn open_below(cx: &mut Context) {
enter_insert_mode(cx); enter_insert_mode(cx);
let lines = selection_lines(&cx.view.doc.state); let view = cx.view();
let lines = selection_lines(&view.doc.state);
let positions: Vec<_> = lines let positions: Vec<_> = lines
.into_iter() .into_iter()
.map(|index| { .map(|index| {
// adjust all positions to the end of the line/start of the next one. // adjust all positions to the end of the line/start of the next one.
cx.view.doc.text().line_to_char(index + 1) view.doc.text().line_to_char(index + 1)
}) })
.collect(); .collect();
let changes = positions.iter().copied().map(|index| { let changes = positions.iter().copied().map(|index| {
// TODO: share logic with insert_newline for indentation // TODO: share logic with insert_newline for indentation
let indent_level = helix_core::indent::suggested_indent_for_pos( let indent_level = helix_core::indent::suggested_indent_for_pos(
cx.view.doc.syntax.as_ref(), view.doc.syntax.as_ref(),
&cx.view.doc.state, &view.doc.state,
index, index,
true, true,
); );
@ -567,42 +595,42 @@ pub fn open_below(cx: &mut Context) {
0, 0,
); );
let transaction = Transaction::change(&cx.view.doc.state, changes).with_selection(selection); let transaction = Transaction::change(&view.doc.state, changes).with_selection(selection);
cx.view.doc.apply(&transaction); view.doc.apply(&transaction);
} }
// O inserts a new line before each line with a selection // O inserts a new line before each line with a selection
fn append_changes_to_history(cx: &mut Context) { fn append_changes_to_history(cx: &mut Context) {
if cx.view.doc.changes.is_empty() { if cx.view().doc.changes.is_empty() {
return; return;
} }
// TODO: change -> change -> undo -> change -> change fails, probably old_state needs reset // TODO: change -> change -> undo -> change -> change fails, probably old_state needs reset
let new_changeset = ChangeSet::new(cx.view.doc.text()); let new_changeset = ChangeSet::new(cx.view().doc.text());
let changes = std::mem::replace(&mut cx.view.doc.changes, new_changeset); let changes = std::mem::replace(&mut cx.view().doc.changes, new_changeset);
// Instead of doing this messy merge we could always commit, and based on transaction // Instead of doing this messy merge we could always commit, and based on transaction
// annotations either add a new layer or compose into the previous one. // annotations either add a new layer or compose into the previous one.
let transaction = Transaction::from(changes).with_selection(cx.view.doc.selection().clone()); let transaction = Transaction::from(changes).with_selection(cx.view().doc.selection().clone());
// increment document version // increment document version
// TODO: needs to happen on undo/redo too // TODO: needs to happen on undo/redo too
cx.view.doc.version += 1; cx.view().doc.version += 1;
// TODO: trigger lsp/documentDidChange with changes // TODO: trigger lsp/documentDidChange with changes
// HAXX: we need to reconstruct the state as it was before the changes.. // HAXX: we need to reconstruct the state as it was before the changes..
let old_state = cx let old_state = cx
.view .view()
.doc .doc
.old_state .old_state
.take() .take()
.expect("no old_state available"); .expect("no old_state available");
// TODO: take transaction by value? // TODO: take transaction by value?
cx.view cx.view()
.doc .doc
.history .history
.commit_revision(&transaction, &old_state); .commit_revision(&transaction, &old_state);
@ -611,27 +639,29 @@ fn append_changes_to_history(cx: &mut Context) {
} }
pub fn normal_mode(cx: &mut Context) { pub fn normal_mode(cx: &mut Context) {
cx.view.doc.mode = Mode::Normal; cx.view().doc.mode = Mode::Normal;
append_changes_to_history(cx); append_changes_to_history(cx);
let view = cx.view();
// if leaving append mode, move cursor back by 1 // if leaving append mode, move cursor back by 1
if cx.view.doc.restore_cursor { if view.doc.restore_cursor {
let text = &cx.view.doc.text().slice(..); let text = &view.doc.text().slice(..);
let selection = cx.view.doc.selection().transform(|range| { let selection = view.doc.selection().transform(|range| {
Range::new( Range::new(
range.from(), range.from(),
graphemes::prev_grapheme_boundary(text, range.to()), graphemes::prev_grapheme_boundary(text, range.to()),
) )
}); });
cx.view.doc.set_selection(selection); view.doc.set_selection(selection);
cx.view.doc.restore_cursor = false; view.doc.restore_cursor = false;
} }
} }
pub fn goto_mode(cx: &mut Context) { pub fn goto_mode(cx: &mut Context) {
cx.view.doc.mode = Mode::Goto; cx.view().doc.mode = Mode::Goto;
} }
// NOTE: Transactions in this module get appended to history when we switch back to normal mode. // NOTE: Transactions in this module get appended to history when we switch back to normal mode.
@ -640,9 +670,9 @@ pub mod insert {
// TODO: insert means add text just before cursor, on exit we should be on the last letter. // TODO: insert means add text just before cursor, on exit we should be on the last letter.
pub fn insert_char(cx: &mut Context, c: char) { pub fn insert_char(cx: &mut Context, c: char) {
let c = Tendril::from_char(c); let c = Tendril::from_char(c);
let transaction = Transaction::insert(&cx.view.doc.state, c); let transaction = Transaction::insert(&cx.view().doc.state, c);
cx.view.doc.apply(&transaction); cx.view().doc.apply(&transaction);
} }
pub fn insert_tab(cx: &mut Context) { pub fn insert_tab(cx: &mut Context) {
@ -650,10 +680,11 @@ pub mod insert {
} }
pub fn insert_newline(cx: &mut Context) { pub fn insert_newline(cx: &mut Context) {
let transaction = Transaction::change_by_selection(&cx.view.doc.state, |range| { let view = cx.view();
let transaction = Transaction::change_by_selection(&view.doc.state, |range| {
let indent_level = helix_core::indent::suggested_indent_for_pos( let indent_level = helix_core::indent::suggested_indent_for_pos(
cx.view.doc.syntax.as_ref(), view.doc.syntax.as_ref(),
&cx.view.doc.state, &view.doc.state,
range.head, range.head,
true, true,
); );
@ -663,32 +694,36 @@ pub mod insert {
text.push_str(&indent); text.push_str(&indent);
(range.head, range.head, Some(text.into())) (range.head, range.head, Some(text.into()))
}); });
cx.view.doc.apply(&transaction); view.doc.apply(&transaction);
} }
// TODO: handle indent-aware delete // TODO: handle indent-aware delete
pub fn delete_char_backward(cx: &mut Context) { pub fn delete_char_backward(cx: &mut Context) {
let text = &cx.view.doc.text().slice(..); let count = cx.count;
let transaction = Transaction::change_by_selection(&cx.view.doc.state, |range| { let view = cx.view();
let text = &view.doc.text().slice(..);
let transaction = Transaction::change_by_selection(&view.doc.state, |range| {
( (
graphemes::nth_prev_grapheme_boundary(text, range.head, cx.count), graphemes::nth_prev_grapheme_boundary(text, range.head, count),
range.head, range.head,
None, None,
) )
}); });
cx.view.doc.apply(&transaction); view.doc.apply(&transaction);
} }
pub fn delete_char_forward(cx: &mut Context) { pub fn delete_char_forward(cx: &mut Context) {
let text = &cx.view.doc.text().slice(..); let count = cx.count;
let transaction = Transaction::change_by_selection(&cx.view.doc.state, |range| { let view = cx.view();
let text = &view.doc.text().slice(..);
let transaction = Transaction::change_by_selection(&view.doc.state, |range| {
( (
range.head, range.head,
graphemes::nth_next_grapheme_boundary(text, range.head, cx.count), graphemes::nth_next_grapheme_boundary(text, range.head, count),
None, None,
) )
}); });
cx.view.doc.apply(&transaction); view.doc.apply(&transaction);
} }
} }
@ -702,23 +737,23 @@ pub fn insert_char_prompt(prompt: &mut Prompt, c: char) {
// storing it? // storing it?
pub fn undo(cx: &mut Context) { pub fn undo(cx: &mut Context) {
cx.view.doc.undo(); cx.view().doc.undo();
} }
pub fn redo(cx: &mut Context) { pub fn redo(cx: &mut Context) {
cx.view.doc.redo(); cx.view().doc.redo();
} }
// Yank / Paste // Yank / Paste
pub fn yank(cx: &mut Context) { pub fn yank(cx: &mut Context) {
// TODO: should selections be made end inclusive? // TODO: should selections be made end inclusive?
let values = cx let view = cx.view();
.view let values = view
.doc .doc
.state .state
.selection() .selection()
.fragments(&cx.view.doc.text().slice(..)) .fragments(&view.doc.text().slice(..))
.map(|cow| cow.into_owned()) .map(|cow| cow.into_owned())
.collect(); .collect();
@ -756,21 +791,23 @@ pub fn paste(cx: &mut Context) {
let mut values = values.into_iter().map(Tendril::from).chain(repeat); let mut values = values.into_iter().map(Tendril::from).chain(repeat);
let view = cx.view();
let transaction = if linewise { let transaction = if linewise {
// paste on the next line // paste on the next line
// TODO: can simply take a range + modifier and compute the right pos without ifs // TODO: can simply take a range + modifier and compute the right pos without ifs
let text = cx.view.doc.text(); let text = view.doc.text();
Transaction::change_by_selection(&cx.view.doc.state, |range| { Transaction::change_by_selection(&view.doc.state, |range| {
let line_end = text.line_to_char(text.char_to_line(range.head) + 1); let line_end = text.line_to_char(text.char_to_line(range.head) + 1);
(line_end, line_end, Some(values.next().unwrap())) (line_end, line_end, Some(values.next().unwrap()))
}) })
} else { } else {
Transaction::change_by_selection(&cx.view.doc.state, |range| { Transaction::change_by_selection(&view.doc.state, |range| {
(range.head + 1, range.head + 1, Some(values.next().unwrap())) (range.head + 1, range.head + 1, Some(values.next().unwrap()))
}) })
}; };
cx.view.doc.apply(&transaction); view.doc.apply(&transaction);
append_changes_to_history(cx); append_changes_to_history(cx);
} }
} }
@ -793,28 +830,30 @@ fn get_lines(view: &View) -> Vec<usize> {
} }
pub fn indent(cx: &mut Context) { pub fn indent(cx: &mut Context) {
let lines = get_lines(cx.view); let lines = get_lines(cx.view());
// Indent by one level // Indent by one level
let indent = Tendril::from(" ".repeat(TAB_WIDTH)); let indent = Tendril::from(" ".repeat(TAB_WIDTH));
let view = cx.view();
let transaction = Transaction::change( let transaction = Transaction::change(
&cx.view.doc.state, &view.doc.state,
lines.into_iter().map(|line| { lines.into_iter().map(|line| {
let pos = cx.view.doc.text().line_to_char(line); let pos = view.doc.text().line_to_char(line);
(pos, pos, Some(indent.clone())) (pos, pos, Some(indent.clone()))
}), }),
); );
cx.view.doc.apply(&transaction); cx.view().doc.apply(&transaction);
append_changes_to_history(cx); append_changes_to_history(cx);
} }
pub fn unindent(cx: &mut Context) { pub fn unindent(cx: &mut Context) {
let lines = get_lines(cx.view); let lines = get_lines(cx.view());
let mut changes = Vec::with_capacity(lines.len()); let mut changes = Vec::with_capacity(lines.len());
for line_idx in lines { for line_idx in lines {
let line = cx.view.doc.text().line(line_idx); let line = cx.view().doc.text().line(line_idx);
let mut width = 0; let mut width = 0;
for ch in line.chars() { for ch in line.chars() {
@ -830,14 +869,14 @@ pub fn unindent(cx: &mut Context) {
} }
if width > 0 { if width > 0 {
let start = cx.view.doc.text().line_to_char(line_idx); let start = cx.view().doc.text().line_to_char(line_idx);
changes.push((start, start + width, None)) changes.push((start, start + width, None))
} }
} }
let transaction = Transaction::change(&cx.view.doc.state, changes.into_iter()); let transaction = Transaction::change(&cx.view().doc.state, changes.into_iter());
cx.view.doc.apply(&transaction); cx.view().doc.apply(&transaction);
append_changes_to_history(cx); append_changes_to_history(cx);
} }
@ -852,21 +891,25 @@ pub fn save(cx: &mut Context) {
// Spawns an async task to actually do the saving. This way we prevent blocking. // Spawns an async task to actually do the saving. This way we prevent blocking.
// TODO: handle save errors somehow? // TODO: handle save errors somehow?
cx.executor.spawn(cx.view.doc.save()).detach(); cx.executor.spawn(cx.view().doc.save()).detach();
} }
pub fn completion(cx: &mut Context) { pub fn completion(cx: &mut Context) {
let language_server = cx.language_servers.get("rust", &cx.executor).unwrap(); let language_server = cx
.editor
.language_servers
.get("source.rust", &cx.executor)
.unwrap();
use log::info; use log::info;
let view = cx.view();
// TODO: blocking here is not ideal // TODO: blocking here is not ideal
let pos = helix_lsp::util::pos_to_lsp_pos( let pos =
&cx.view.doc.text().slice(..), helix_lsp::util::pos_to_lsp_pos(&view.doc.text().slice(..), view.doc.selection().cursor());
cx.view.doc.selection().cursor(),
);
// TODO: handle fails // TODO: handle fails
let res = smol::block_on(language_server.completion(cx.view.doc.identifier(), pos)) let res = smol::block_on(language_server.completion(cx.view().doc.identifier(), pos))
.unwrap_or_default(); .unwrap_or_default();
// TODO: if no completion, show some message or something // TODO: if no completion, show some message or something
@ -903,7 +946,7 @@ pub fn completion(cx: &mut Context) {
} }
} }
let view = editor.view_mut().unwrap(); let view = editor.view_mut();
let transaction = let transaction =
util::generate_transaction_from_edits(&view.doc.state, vec![edit]); util::generate_transaction_from_edits(&view.doc.state, vec![edit]);
view.doc.apply(&transaction); view.doc.apply(&transaction);

@ -39,7 +39,6 @@ use helix_view::{Editor, View};
// shared with commands.rs // shared with commands.rs
pub struct Context<'a> { pub struct Context<'a> {
pub editor: &'a mut Editor, pub editor: &'a mut Editor,
pub language_servers: &'a helix_lsp::Registry,
pub executor: &'static smol::Executor<'static>, pub executor: &'static smol::Executor<'static>,
} }

@ -251,55 +251,52 @@ impl Component for EditorView {
Event::Resize(width, height) => { Event::Resize(width, height) => {
// TODO: simplistic ensure cursor in view for now // TODO: simplistic ensure cursor in view for now
// TODO: loop over views // TODO: loop over views
if let Some(view) = cx.editor.view_mut() { let view = cx.editor.view_mut();
view.size = (width, height); view.size = (width, height);
view.ensure_cursor_in_view() view.ensure_cursor_in_view();
};
EventResult::Consumed(None) EventResult::Consumed(None)
} }
Event::Key(event) => { Event::Key(event) => {
if let Some(view) = cx.editor.view_mut() { let view = cx.editor.view_mut();
let keys = vec![event];
// TODO: sequences (`gg`) let keys = vec![event];
let mode = view.doc.mode(); // TODO: sequences (`gg`)
// TODO: handle count other than 1 let mode = view.doc.mode();
let mut cx = commands::Context { // TODO: handle count other than 1
view, let mut cxt = commands::Context {
executor: cx.executor, executor: cx.executor,
language_servers: cx.language_servers, editor: &mut cx.editor,
count: 1, count: 1,
callback: None, callback: None,
}; };
match mode { match mode {
Mode::Insert => { Mode::Insert => {
if let Some(command) = self.keymap[&Mode::Insert].get(&keys) { if let Some(command) = self.keymap[&Mode::Insert].get(&keys) {
command(&mut cx); command(&mut cxt);
} else if let KeyEvent { } else if let KeyEvent {
code: KeyCode::Char(c), code: KeyCode::Char(c),
.. ..
} = event } = event
{ {
commands::insert::insert_char(&mut cx, c); commands::insert::insert_char(&mut cxt, c);
}
} }
mode => { }
if let Some(command) = self.keymap[&mode].get(&keys) { mode => {
command(&mut cx); if let Some(command) = self.keymap[&mode].get(&keys) {
command(&mut cxt);
// TODO: simplistic ensure cursor in view for now // TODO: simplistic ensure cursor in view for now
}
} }
} }
// appease borrowck }
let callback = cx.callback.take();
view.ensure_cursor_in_view(); // appease borrowck
let callback = cxt.callback.take();
drop(cxt);
cx.editor.view_mut().ensure_cursor_in_view();
EventResult::Consumed(callback) EventResult::Consumed(callback)
} else {
EventResult::Ignored
}
} }
Event::Mouse(_) => EventResult::Ignored, Event::Mouse(_) => EventResult::Ignored,
} }
@ -309,9 +306,8 @@ impl Component for EditorView {
// SAFETY: we cheat around the view_mut() borrow because it doesn't allow us to also borrow // 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. // theme. Theme is immutable mutating view won't disrupt theme_ref.
let theme_ref = unsafe { &*(&cx.editor.theme as *const Theme) }; let theme_ref = unsafe { &*(&cx.editor.theme as *const Theme) };
if let Some(view) = cx.editor.view_mut() { let view = cx.editor.view_mut();
self.render_view(view, area, surface, theme_ref); self.render_view(view, area, surface, theme_ref);
}
// TODO: drop unwrap // TODO: drop unwrap
} }
@ -321,7 +317,7 @@ impl Component for EditorView {
// Mode::Insert => write!(stdout, "\x1B[6 q"), // Mode::Insert => write!(stdout, "\x1B[6 q"),
// mode => write!(stdout, "\x1B[2 q"), // mode => write!(stdout, "\x1B[2 q"),
// }; // };
let view = ctx.editor.view().unwrap(); let view = ctx.editor.view();
let cursor = view.doc.state.selection().cursor(); let cursor = view.doc.state.selection().cursor();
let mut pos = view let mut pos = view

@ -16,7 +16,7 @@ pub fn text_color() -> Style {
} }
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
pub fn file_picker(root: &str) -> Picker<PathBuf> { pub fn file_picker(root: &str, ex: &'static smol::Executor) -> Picker<PathBuf> {
use ignore::Walk; use ignore::Walk;
// TODO: determine root based on git root // TODO: determine root based on git root
let files = Walk::new(root).filter_map(|entry| match entry { let files = Walk::new(root).filter_map(|entry| match entry {
@ -40,9 +40,9 @@ pub fn file_picker(root: &str) -> Picker<PathBuf> {
// format_fn // format_fn
path.strip_prefix("./").unwrap().to_str().unwrap().into() path.strip_prefix("./").unwrap().to_str().unwrap().into()
}, },
|editor: &mut Editor, path: &PathBuf| { move |editor: &mut Editor, path: &PathBuf| {
let size = editor.view().unwrap().size; let size = editor.view().size;
editor.open(path.into(), size); editor.open(path.into(), size, ex);
}, },
) )
} }

@ -10,6 +10,7 @@ pub struct Editor {
pub focus: usize, pub focus: usize,
pub should_close: bool, pub should_close: bool,
pub theme: Theme, // TODO: share one instance pub theme: Theme, // TODO: share one instance
pub language_servers: helix_lsp::Registry,
} }
impl Default for Editor { impl Default for Editor {
@ -21,28 +22,54 @@ impl Default for Editor {
impl Editor { impl Editor {
pub fn new() -> Self { pub fn new() -> Self {
let theme = Theme::default(); let theme = Theme::default();
let language_servers = helix_lsp::Registry::new();
Self { Self {
views: Vec::new(), views: Vec::new(),
focus: 0, focus: 0,
should_close: false, should_close: false,
theme, theme,
language_servers,
} }
} }
pub fn open(&mut self, path: PathBuf, size: (u16, u16)) -> Result<(), Error> { pub fn open(
&mut self,
path: PathBuf,
size: (u16, u16),
executor: &smol::Executor,
) -> Result<(), Error> {
let pos = self.views.len(); let pos = self.views.len();
let doc = Document::load(path, self.theme.scopes())?; let mut doc = Document::load(path, self.theme.scopes())?;
// try to find a language server based on the language name
let language_server = doc
.language
.as_ref()
.and_then(|language| self.language_servers.get(&language, &executor));
if let Some(language_server) = language_server {
// TODO: do this everywhere
doc.set_language_server(Some(language_server.clone()));
smol::block_on(language_server.text_document_did_open(
doc.url().unwrap(),
doc.version,
doc.text(),
))
.unwrap();
}
self.views.push(View::new(doc, size)?); self.views.push(View::new(doc, size)?);
self.focus = pos; self.focus = pos;
Ok(()) Ok(())
} }
pub fn view(&self) -> Option<&View> { pub fn view(&self) -> &View {
self.views.get(self.focus) self.views.get(self.focus).unwrap()
} }
pub fn view_mut(&mut self) -> Option<&mut View> { pub fn view_mut(&mut self) -> &mut View {
self.views.get_mut(self.focus) self.views.get_mut(self.focus).unwrap()
} }
} }

Loading…
Cancel
Save