Merge pull request #1451 from zed-industries/tab-auto-indent

Auto-indent on tab for cursors left of the suggested indent level
This commit is contained in:
Julia 2022-08-02 15:52:12 -04:00 committed by GitHub
commit 8115c1ab24
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 403 additions and 211 deletions

View file

@ -2921,133 +2921,174 @@ impl Editor {
} }
let mut selections = self.selections.all_adjusted(cx); let mut selections = self.selections.all_adjusted(cx);
if selections.iter().all(|s| s.is_empty()) { let buffer = self.buffer.read(cx);
self.transact(cx, |this, cx| { let snapshot = buffer.snapshot(cx);
this.buffer.update(cx, |buffer, cx| { let suggested_indents =
let mut prev_cursor_row = 0; snapshot.suggested_indents(selections.iter().map(|s| s.head().row), cx);
let mut row_delta = 0;
for selection in &mut selections { let mut edits = Vec::new();
let mut cursor = selection.start; let mut prev_edited_row = 0;
if cursor.row != prev_cursor_row { let mut row_delta = 0;
row_delta = 0; for selection in &mut selections {
prev_cursor_row = cursor.row; if selection.start.row != prev_edited_row {
row_delta = 0;
}
prev_edited_row = selection.end.row;
if selection.is_empty() {
let cursor = selection.head();
let mut did_auto_indent = false;
if let Some(suggested_indent) = suggested_indents.get(&cursor.row).copied() {
let current_indent = snapshot.indent_size_for_line(cursor.row);
// Don't account for `row_delta` as we only care if the cursor *started* in the leading whitespace
if cursor.column < suggested_indent.len
&& cursor.column <= current_indent.len
&& current_indent.len <= suggested_indent.len
{
selection.start = Point::new(cursor.row, suggested_indent.len);
selection.end = selection.start;
if row_delta == 0 {
edits.extend(Buffer::edit_for_indent_size_adjustment(
cursor.row,
current_indent,
suggested_indent,
));
row_delta = suggested_indent.len - current_indent.len;
} }
cursor.column += row_delta; did_auto_indent = true;
let language_name = buffer.language_at(cursor, cx).map(|l| l.name());
let settings = cx.global::<Settings>();
let tab_size = if settings.hard_tabs(language_name.as_deref()) {
IndentSize::tab()
} else {
let tab_size = settings.tab_size(language_name.as_deref()).get();
let char_column = buffer
.read(cx)
.text_for_range(Point::new(cursor.row, 0)..cursor)
.flat_map(str::chars)
.count();
let chars_to_next_tab_stop = tab_size - (char_column as u32 % tab_size);
IndentSize::spaces(chars_to_next_tab_stop)
};
buffer.edit(
[(cursor..cursor, tab_size.chars().collect::<String>())],
None,
cx,
);
cursor.column += tab_size.len;
selection.start = cursor;
selection.end = cursor;
row_delta += tab_size.len;
} }
}); }
this.change_selections(Some(Autoscroll::Fit), cx, |s| {
s.select(selections); if !did_auto_indent {
}); let language_name = buffer.language_at(cursor, cx).map(|l| l.name());
}); let settings = cx.global::<Settings>();
} else { let tab_size = if settings.hard_tabs(language_name.as_deref()) {
self.indent(&Indent, cx); IndentSize::tab()
} else {
let tab_size = settings.tab_size(language_name.as_deref()).get();
let char_column = snapshot
.text_for_range(Point::new(cursor.row, 0)..cursor)
.flat_map(str::chars)
.count()
+ row_delta as usize;
let chars_to_next_tab_stop = tab_size - (char_column as u32 % tab_size);
IndentSize::spaces(chars_to_next_tab_stop)
};
selection.start =
Point::new(cursor.row, cursor.column + row_delta + tab_size.len);
selection.end = selection.start;
edits.push((cursor..cursor, tab_size.chars().collect::<String>()));
row_delta += tab_size.len;
}
} else {
row_delta =
Self::indent_selection(buffer, &snapshot, selection, &mut edits, row_delta, cx);
}
} }
self.transact(cx, |this, cx| {
this.buffer
.update(cx, |buffer, cx| buffer.edit(edits, None, cx));
this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections))
});
} }
pub fn indent(&mut self, _: &Indent, cx: &mut ViewContext<Self>) { pub fn indent(&mut self, _: &Indent, cx: &mut ViewContext<Self>) {
let mut selections = self.selections.all::<Point>(cx); let mut selections = self.selections.all::<Point>(cx);
let mut prev_edited_row = 0;
let mut row_delta = 0;
let mut edits = Vec::new();
let buffer = self.buffer.read(cx);
let snapshot = buffer.snapshot(cx);
for selection in &mut selections {
if selection.start.row != prev_edited_row {
row_delta = 0;
}
prev_edited_row = selection.end.row;
row_delta =
Self::indent_selection(buffer, &snapshot, selection, &mut edits, row_delta, cx);
}
self.transact(cx, |this, cx| { self.transact(cx, |this, cx| {
let mut last_indent = None; this.buffer
this.buffer.update(cx, |buffer, cx| { .update(cx, |buffer, cx| buffer.edit(edits, None, cx));
let snapshot = buffer.snapshot(cx); this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections));
for selection in &mut selections {
let language_name = buffer.language_at(selection.start, cx).map(|l| l.name());
let settings = &cx.global::<Settings>();
let tab_size = settings.tab_size(language_name.as_deref()).get();
let indent_kind = if settings.hard_tabs(language_name.as_deref()) {
IndentKind::Tab
} else {
IndentKind::Space
};
let mut start_row = selection.start.row;
let mut end_row = selection.end.row + 1;
// If a selection ends at the beginning of a line, don't indent
// that last line.
if selection.end.column == 0 {
end_row -= 1;
}
// Avoid re-indenting a row that has already been indented by a
// previous selection, but still update this selection's column
// to reflect that indentation.
if let Some((last_indent_row, last_indent_len)) = last_indent {
if last_indent_row == selection.start.row {
selection.start.column += last_indent_len;
start_row += 1;
}
if last_indent_row == selection.end.row {
selection.end.column += last_indent_len;
}
}
for row in start_row..end_row {
let current_indent = snapshot.indent_size_for_line(row);
let indent_delta = match (current_indent.kind, indent_kind) {
(IndentKind::Space, IndentKind::Space) => {
let columns_to_next_tab_stop =
tab_size - (current_indent.len % tab_size);
IndentSize::spaces(columns_to_next_tab_stop)
}
(IndentKind::Tab, IndentKind::Space) => IndentSize::spaces(tab_size),
(_, IndentKind::Tab) => IndentSize::tab(),
};
let row_start = Point::new(row, 0);
buffer.edit(
[(
row_start..row_start,
indent_delta.chars().collect::<String>(),
)],
None,
cx,
);
// Update this selection's endpoints to reflect the indentation.
if row == selection.start.row {
selection.start.column += indent_delta.len;
}
if row == selection.end.row {
selection.end.column += indent_delta.len as u32;
}
last_indent = Some((row, indent_delta.len));
}
}
});
this.change_selections(Some(Autoscroll::Fit), cx, |s| {
s.select(selections);
});
}); });
} }
fn indent_selection(
buffer: &MultiBuffer,
snapshot: &MultiBufferSnapshot,
selection: &mut Selection<Point>,
edits: &mut Vec<(Range<Point>, String)>,
delta_for_start_row: u32,
cx: &AppContext,
) -> u32 {
let language_name = buffer.language_at(selection.start, cx).map(|l| l.name());
let settings = cx.global::<Settings>();
let tab_size = settings.tab_size(language_name.as_deref()).get();
let indent_kind = if settings.hard_tabs(language_name.as_deref()) {
IndentKind::Tab
} else {
IndentKind::Space
};
let mut start_row = selection.start.row;
let mut end_row = selection.end.row + 1;
// If a selection ends at the beginning of a line, don't indent
// that last line.
if selection.end.column == 0 {
end_row -= 1;
}
// Avoid re-indenting a row that has already been indented by a
// previous selection, but still update this selection's column
// to reflect that indentation.
if delta_for_start_row > 0 {
start_row += 1;
selection.start.column += delta_for_start_row;
if selection.end.row == selection.start.row {
selection.end.column += delta_for_start_row;
}
}
let mut delta_for_end_row = 0;
for row in start_row..end_row {
let current_indent = snapshot.indent_size_for_line(row);
let indent_delta = match (current_indent.kind, indent_kind) {
(IndentKind::Space, IndentKind::Space) => {
let columns_to_next_tab_stop = tab_size - (current_indent.len % tab_size);
IndentSize::spaces(columns_to_next_tab_stop)
}
(IndentKind::Tab, IndentKind::Space) => IndentSize::spaces(tab_size),
(_, IndentKind::Tab) => IndentSize::tab(),
};
let row_start = Point::new(row, 0);
edits.push((
row_start..row_start,
indent_delta.chars().collect::<String>(),
));
// Update this selection's endpoints to reflect the indentation.
if row == selection.start.row {
selection.start.column += indent_delta.len;
}
if row == selection.end.row {
selection.end.column += indent_delta.len;
delta_for_end_row = indent_delta.len;
}
}
if selection.start.row == selection.end.row {
delta_for_start_row + delta_for_end_row
} else {
delta_for_end_row
}
}
pub fn outdent(&mut self, _: &Outdent, cx: &mut ViewContext<Self>) { pub fn outdent(&mut self, _: &Outdent, cx: &mut ViewContext<Self>) {
let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx)); let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
let selections = self.selections.all::<Point>(cx); let selections = self.selections.all::<Point>(cx);
@ -3574,6 +3615,8 @@ impl Editor {
let old_selections = this.selections.all::<usize>(cx); let old_selections = this.selections.all::<usize>(cx);
let all_selections_were_entire_line = let all_selections_were_entire_line =
clipboard_selections.iter().all(|s| s.is_entire_line); clipboard_selections.iter().all(|s| s.is_entire_line);
let first_selection_indent_column =
clipboard_selections.first().map(|s| s.first_line_indent);
if clipboard_selections.len() != old_selections.len() { if clipboard_selections.len() != old_selections.len() {
let mut newline_separated_text = String::new(); let mut newline_separated_text = String::new();
let mut clipboard_selections = clipboard_selections.drain(..).peekable(); let mut clipboard_selections = clipboard_selections.drain(..).peekable();
@ -3593,22 +3636,23 @@ impl Editor {
let snapshot = buffer.read(cx); let snapshot = buffer.read(cx);
let mut start_offset = 0; let mut start_offset = 0;
let mut edits = Vec::new(); let mut edits = Vec::new();
let mut start_columns = Vec::new(); let mut original_indent_columns = Vec::new();
let line_mode = this.selections.line_mode; let line_mode = this.selections.line_mode;
for (ix, selection) in old_selections.iter().enumerate() { for (ix, selection) in old_selections.iter().enumerate() {
let to_insert; let to_insert;
let entire_line; let entire_line;
let start_column; let original_indent_column;
if let Some(clipboard_selection) = clipboard_selections.get(ix) { if let Some(clipboard_selection) = clipboard_selections.get(ix) {
let end_offset = start_offset + clipboard_selection.len; let end_offset = start_offset + clipboard_selection.len;
to_insert = &clipboard_text[start_offset..end_offset]; to_insert = &clipboard_text[start_offset..end_offset];
entire_line = clipboard_selection.is_entire_line; entire_line = clipboard_selection.is_entire_line;
start_offset = end_offset; start_offset = end_offset;
start_column = clipboard_selection.first_line_indent; original_indent_column =
Some(clipboard_selection.first_line_indent);
} else { } else {
to_insert = clipboard_text.as_str(); to_insert = clipboard_text.as_str();
entire_line = all_selections_were_entire_line; entire_line = all_selections_were_entire_line;
start_column = 0; original_indent_column = first_selection_indent_column
} }
// If the corresponding selection was empty when this slice of the // If the corresponding selection was empty when this slice of the
@ -3624,13 +3668,14 @@ impl Editor {
}; };
edits.push((range, to_insert)); edits.push((range, to_insert));
start_columns.push(start_column); original_indent_columns.extend(original_indent_column);
} }
drop(snapshot); drop(snapshot);
buffer.edit( buffer.edit(
edits, edits,
Some(AutoindentMode::Block { Some(AutoindentMode::Block {
original_indent_columns: start_columns, original_indent_columns,
}), }),
cx, cx,
); );
@ -7987,6 +8032,74 @@ mod tests {
|🏀 |🏀 |efg |🏀 |🏀 |efg
d | d |
"}); "});
cx.set_state(indoc! {"
a
[🏀}🏀[🏀}🏀[🏀}
"});
cx.update_editor(|e, cx| e.tab(&Tab, cx));
cx.assert_editor_state(indoc! {"
a
[🏀}🏀[🏀}🏀[🏀}
"});
}
#[gpui::test]
async fn test_tab_on_blank_line_auto_indents(cx: &mut gpui::TestAppContext) {
let mut cx = EditorTestContext::new(cx).await;
let language = Arc::new(
Language::new(
LanguageConfig::default(),
Some(tree_sitter_rust::language()),
)
.with_indents_query(r#"(_ "(" ")" @end) @indent"#)
.unwrap(),
);
cx.update_buffer(|buffer, cx| buffer.set_language(Some(language), cx));
// cursors that are already at the suggested indent level insert
// a soft tab. cursors that are to the left of the suggested indent
// auto-indent their line.
cx.set_state(indoc! {"
|
const a: B = (
c(
d(
|
)
|
| )
);
"});
cx.update_editor(|e, cx| e.tab(&Tab, cx));
cx.assert_editor_state(indoc! {"
|
const a: B = (
c(
d(
|
)
|
|)
);
"});
// handle auto-indent when there are multiple cursors on the same line
cx.set_state(indoc! {"
const a: B = (
c(
| |
| )
);
"});
cx.update_editor(|e, cx| e.tab(&Tab, cx));
cx.assert_editor_state(indoc! {"
const a: B = (
c(
|
|)
);
"});
} }
#[gpui::test] #[gpui::test]
@ -8701,18 +8814,18 @@ mod tests {
// Cut an indented block, without the leading whitespace. // Cut an indented block, without the leading whitespace.
cx.set_state(indoc! {" cx.set_state(indoc! {"
const a = ( const a: B = (
b(), c(),
[c( [d(
d, e,
e f
)} )}
); );
"}); "});
cx.update_editor(|e, cx| e.cut(&Cut, cx)); cx.update_editor(|e, cx| e.cut(&Cut, cx));
cx.assert_editor_state(indoc! {" cx.assert_editor_state(indoc! {"
const a = ( const a: B = (
b(), c(),
| |
); );
"}); "});
@ -8720,83 +8833,83 @@ mod tests {
// Paste it at the same position. // Paste it at the same position.
cx.update_editor(|e, cx| e.paste(&Paste, cx)); cx.update_editor(|e, cx| e.paste(&Paste, cx));
cx.assert_editor_state(indoc! {" cx.assert_editor_state(indoc! {"
const a = ( const a: B = (
b(), c(),
c( d(
d, e,
e f
)| )|
); );
"}); "});
// Paste it at a line with a lower indent level. // Paste it at a line with a lower indent level.
cx.update_editor(|e, cx| e.paste(&Paste, cx));
cx.set_state(indoc! {" cx.set_state(indoc! {"
| |
const a = ( const a: B = (
b(), c(),
); );
"}); "});
cx.update_editor(|e, cx| e.paste(&Paste, cx)); cx.update_editor(|e, cx| e.paste(&Paste, cx));
cx.assert_editor_state(indoc! {" cx.assert_editor_state(indoc! {"
c( d(
d, e,
e f
)| )|
const a = ( const a: B = (
b(), c(),
); );
"}); "});
// Cut an indented block, with the leading whitespace. // Cut an indented block, with the leading whitespace.
cx.set_state(indoc! {" cx.set_state(indoc! {"
const a = ( const a: B = (
b(), c(),
[ c( [ d(
d, e,
e f
) )
}); });
"}); "});
cx.update_editor(|e, cx| e.cut(&Cut, cx)); cx.update_editor(|e, cx| e.cut(&Cut, cx));
cx.assert_editor_state(indoc! {" cx.assert_editor_state(indoc! {"
const a = ( const a: B = (
b(), c(),
|); |);
"}); "});
// Paste it at the same position. // Paste it at the same position.
cx.update_editor(|e, cx| e.paste(&Paste, cx)); cx.update_editor(|e, cx| e.paste(&Paste, cx));
cx.assert_editor_state(indoc! {" cx.assert_editor_state(indoc! {"
const a = ( const a: B = (
b(), c(),
c( d(
d, e,
e f
) )
|); |);
"}); "});
// Paste it at a line with a higher indent level. // Paste it at a line with a higher indent level.
cx.set_state(indoc! {" cx.set_state(indoc! {"
const a = ( const a: B = (
b(), c(),
c( d(
d, e,
e| f|
) )
); );
"}); "});
cx.update_editor(|e, cx| e.paste(&Paste, cx)); cx.update_editor(|e, cx| e.paste(&Paste, cx));
cx.set_state(indoc! {" cx.assert_editor_state(indoc! {"
const a = ( const a: B = (
b(), c(),
c( d(
d, e,
ec( f d(
d, e,
e f
)| )
|
) )
); );
"}); "});

View file

@ -3,7 +3,7 @@ mod anchor;
pub use anchor::{Anchor, AnchorRangeExt}; pub use anchor::{Anchor, AnchorRangeExt};
use anyhow::Result; use anyhow::Result;
use clock::ReplicaId; use clock::ReplicaId;
use collections::{Bound, HashMap, HashSet}; use collections::{BTreeMap, Bound, HashMap, HashSet};
use gpui::{AppContext, Entity, ModelContext, ModelHandle, Task}; use gpui::{AppContext, Entity, ModelContext, ModelHandle, Task};
pub use language::Completion; pub use language::Completion;
use language::{ use language::{
@ -1939,6 +1939,53 @@ impl MultiBufferSnapshot {
} }
} }
pub fn suggested_indents(
&self,
rows: impl IntoIterator<Item = u32>,
cx: &AppContext,
) -> BTreeMap<u32, IndentSize> {
let mut result = BTreeMap::new();
let mut rows_for_excerpt = Vec::new();
let mut cursor = self.excerpts.cursor::<Point>();
let mut rows = rows.into_iter().peekable();
while let Some(row) = rows.next() {
cursor.seek(&Point::new(row, 0), Bias::Right, &());
let excerpt = match cursor.item() {
Some(excerpt) => excerpt,
_ => continue,
};
let single_indent_size = excerpt.buffer.single_indent_size(cx);
let start_buffer_row = excerpt.range.context.start.to_point(&excerpt.buffer).row;
let start_multibuffer_row = cursor.start().row;
rows_for_excerpt.push(row);
while let Some(next_row) = rows.peek().copied() {
if cursor.end(&()).row > next_row {
rows_for_excerpt.push(next_row);
rows.next();
} else {
break;
}
}
let buffer_rows = rows_for_excerpt
.drain(..)
.map(|row| start_buffer_row + row - start_multibuffer_row);
let buffer_indents = excerpt
.buffer
.suggested_indents(buffer_rows, single_indent_size);
let multibuffer_indents = buffer_indents
.into_iter()
.map(|(row, indent)| (start_multibuffer_row + row - start_buffer_row, indent));
result.extend(multibuffer_indents);
}
result
}
pub fn indent_size_for_line(&self, row: u32) -> IndentSize { pub fn indent_size_for_line(&self, row: u32) -> IndentSize {
if let Some((buffer, range)) = self.buffer_line_for_row(row) { if let Some((buffer, range)) = self.buffer_line_for_row(row) {
let mut size = buffer.indent_size_for_line(range.start.row); let mut size = buffer.indent_size_for_line(range.start.row);

View file

@ -957,45 +957,42 @@ impl Buffer {
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) { ) {
self.autoindent_requests.clear(); self.autoindent_requests.clear();
self.start_transaction();
for (row, indent_size) in &indent_sizes { let edits: Vec<_> = indent_sizes
self.set_indent_size_for_line(*row, *indent_size, cx); .into_iter()
} .filter_map(|(row, indent_size)| {
self.end_transaction(cx); let current_size = indent_size_for_line(&self, row);
Self::edit_for_indent_size_adjustment(row, current_size, indent_size)
})
.collect();
self.edit(edits, None, cx);
} }
fn set_indent_size_for_line( pub fn edit_for_indent_size_adjustment(
&mut self,
row: u32, row: u32,
size: IndentSize, current_size: IndentSize,
cx: &mut ModelContext<Self>, new_size: IndentSize,
) { ) -> Option<(Range<Point>, String)> {
let current_size = indent_size_for_line(&self, row); if new_size.kind != current_size.kind && current_size.len > 0 {
if size.kind != current_size.kind && current_size.len > 0 { return None;
return;
} }
if size.len > current_size.len { if new_size.len > current_size.len {
let offset = Point::new(row, 0).to_offset(&*self); let point = Point::new(row, 0);
self.edit( Some((
[( point..point,
offset..offset, iter::repeat(new_size.char())
iter::repeat(size.char()) .take((new_size.len - current_size.len) as usize)
.take((size.len - current_size.len) as usize) .collect::<String>(),
.collect::<String>(), ))
)], } else if new_size.len < current_size.len {
None, Some((
cx, Point::new(row, 0)..Point::new(row, current_size.len - new_size.len),
); String::new(),
} else if size.len < current_size.len { ))
self.edit( } else {
[( None
Point::new(row, 0)..Point::new(row, current_size.len - size.len),
"",
)],
None,
cx,
);
} }
} }
@ -1225,13 +1222,7 @@ impl Buffer {
let edit_id = edit_operation.local_timestamp(); let edit_id = edit_operation.local_timestamp();
if let Some((before_edit, mode)) = autoindent_request { if let Some((before_edit, mode)) = autoindent_request {
let language_name = self.language().map(|language| language.name()); let indent_size = before_edit.single_indent_size(cx);
let settings = cx.global::<Settings>();
let indent_size = if settings.hard_tabs(language_name.as_deref()) {
IndentSize::tab()
} else {
IndentSize::spaces(settings.tab_size(language_name.as_deref()).get())
};
let (start_columns, is_block_mode) = match mode { let (start_columns, is_block_mode) = match mode {
AutoindentMode::Block { AutoindentMode::Block {
original_indent_columns: start_columns, original_indent_columns: start_columns,
@ -1269,7 +1260,7 @@ impl Buffer {
first_line_is_new = true; first_line_is_new = true;
} }
// Avoid auto-indenting before the insertion. // Avoid auto-indenting after the insertion.
if is_block_mode { if is_block_mode {
start_column = start_columns.get(ix).copied(); start_column = start_columns.get(ix).copied();
if new_text[range_of_insertion_to_indent.clone()].ends_with('\n') { if new_text[range_of_insertion_to_indent.clone()].ends_with('\n') {
@ -1611,6 +1602,47 @@ impl BufferSnapshot {
indent_size_for_line(&self, row) indent_size_for_line(&self, row)
} }
pub fn single_indent_size(&self, cx: &AppContext) -> IndentSize {
let language_name = self.language().map(|language| language.name());
let settings = cx.global::<Settings>();
if settings.hard_tabs(language_name.as_deref()) {
IndentSize::tab()
} else {
IndentSize::spaces(settings.tab_size(language_name.as_deref()).get())
}
}
pub fn suggested_indents(
&self,
rows: impl Iterator<Item = u32>,
single_indent_size: IndentSize,
) -> BTreeMap<u32, IndentSize> {
let mut result = BTreeMap::new();
for row_range in contiguous_ranges(rows, 10) {
let suggestions = match self.suggest_autoindents(row_range.clone()) {
Some(suggestions) => suggestions,
_ => break,
};
for (row, suggestion) in row_range.zip(suggestions) {
let indent_size = if let Some(suggestion) = suggestion {
result
.get(&suggestion.basis_row)
.copied()
.unwrap_or_else(|| self.indent_size_for_line(suggestion.basis_row))
.with_delta(suggestion.delta, single_indent_size)
} else {
self.indent_size_for_line(row)
};
result.insert(row, indent_size);
}
}
result
}
fn suggest_autoindents<'a>( fn suggest_autoindents<'a>(
&'a self, &'a self,
row_range: Range<u32>, row_range: Range<u32>,