537 lines
18 KiB
Rust
537 lines
18 KiB
Rust
use gpui::{
|
||
fill, point, px, relative, size, Bounds, Corners, Element, ElementId, ElementInputHandler,
|
||
GlobalElementId, IntoElement, LayoutId, MouseButton, MouseMoveEvent, PaintQuad, Path, Pixels,
|
||
Point, Style, TextRun, UnderlineStyle, View, WindowContext, WrappedLine,
|
||
};
|
||
use smallvec::SmallVec;
|
||
|
||
use crate::theme::ActiveTheme as _;
|
||
|
||
use super::TextInput;
|
||
|
||
const RIGHT_MARGIN: Pixels = px(5.);
|
||
const CURSOR_INSET: Pixels = px(0.5);
|
||
|
||
pub(super) struct TextElement {
|
||
input: View<TextInput>,
|
||
}
|
||
|
||
impl TextElement {
|
||
pub(super) fn new(input: View<TextInput>) -> Self {
|
||
Self { input }
|
||
}
|
||
|
||
fn paint_mouse_listeners(&mut self, cx: &mut WindowContext) {
|
||
cx.on_mouse_event({
|
||
let input = self.input.clone();
|
||
|
||
move |event: &MouseMoveEvent, _, cx| {
|
||
if event.pressed_button == Some(MouseButton::Left) {
|
||
input.update(cx, |input, cx| {
|
||
input.on_drag_move(event, cx);
|
||
});
|
||
}
|
||
}
|
||
});
|
||
}
|
||
|
||
fn layout_cursor(
|
||
&self,
|
||
lines: &[WrappedLine],
|
||
line_height: Pixels,
|
||
bounds: &mut Bounds<Pixels>,
|
||
cx: &mut WindowContext,
|
||
) -> (Option<PaintQuad>, Point<Pixels>) {
|
||
let input = self.input.read(cx);
|
||
let selected_range = &input.selected_range;
|
||
let cursor_offset = input.cursor_offset();
|
||
let mut scroll_offset = input.scroll_handle.offset();
|
||
let mut cursor = None;
|
||
|
||
// The cursor corresponds to the current cursor position in the text no only the line.
|
||
let mut cursor_pos = None;
|
||
let mut cursor_start = None;
|
||
let mut cursor_end = None;
|
||
|
||
let mut prev_lines_offset = 0;
|
||
let mut offset_y = px(0.);
|
||
for line in lines.iter() {
|
||
// break loop if all cursor positions are found
|
||
if cursor_pos.is_some() && cursor_start.is_some() && cursor_end.is_some() {
|
||
break;
|
||
}
|
||
|
||
let line_origin = point(px(0.), offset_y);
|
||
if cursor_pos.is_none() {
|
||
let offset = cursor_offset.saturating_sub(prev_lines_offset);
|
||
if let Some(pos) = line.position_for_index(offset, line_height) {
|
||
cursor_pos = Some(line_origin + pos);
|
||
}
|
||
}
|
||
if cursor_start.is_none() {
|
||
let offset = selected_range.start.saturating_sub(prev_lines_offset);
|
||
if let Some(pos) = line.position_for_index(offset, line_height) {
|
||
cursor_start = Some(line_origin + pos);
|
||
}
|
||
}
|
||
if cursor_end.is_none() {
|
||
let offset = selected_range.end.saturating_sub(prev_lines_offset);
|
||
if let Some(pos) = line.position_for_index(offset, line_height) {
|
||
cursor_end = Some(line_origin + pos);
|
||
}
|
||
}
|
||
|
||
offset_y += line.size(line_height).height;
|
||
// +1 for skip the last `\n`
|
||
prev_lines_offset += line.len() + 1;
|
||
}
|
||
|
||
if let (Some(cursor_pos), Some(cursor_start), Some(cursor_end)) =
|
||
(cursor_pos, cursor_start, cursor_end)
|
||
{
|
||
let cursor_moved = input.last_cursor_offset != Some(cursor_offset);
|
||
let selection_changed = input.last_selected_range != Some(selected_range.clone());
|
||
|
||
if cursor_moved || selection_changed {
|
||
scroll_offset.x =
|
||
if scroll_offset.x + cursor_pos.x > (bounds.size.width - RIGHT_MARGIN) {
|
||
// cursor is out of right
|
||
bounds.size.width - RIGHT_MARGIN - cursor_pos.x
|
||
} else if scroll_offset.x + cursor_pos.x < px(0.) {
|
||
// cursor is out of left
|
||
scroll_offset.x - cursor_pos.x
|
||
} else {
|
||
scroll_offset.x
|
||
};
|
||
scroll_offset.y = if scroll_offset.y + cursor_pos.y > (bounds.size.height) {
|
||
// cursor is out of bottom
|
||
bounds.size.height - cursor_pos.y
|
||
} else if scroll_offset.y + cursor_pos.y < px(0.) {
|
||
// cursor is out of top
|
||
scroll_offset.y - cursor_pos.y
|
||
} else {
|
||
scroll_offset.y
|
||
};
|
||
|
||
if input.selection_reversed {
|
||
if scroll_offset.x + cursor_start.x < px(0.) {
|
||
// selection start is out of left
|
||
scroll_offset.x = -cursor_start.x;
|
||
}
|
||
if scroll_offset.y + cursor_start.y < px(0.) {
|
||
// selection start is out of top
|
||
scroll_offset.y = -cursor_start.y;
|
||
}
|
||
} else {
|
||
if scroll_offset.x + cursor_end.x <= px(0.) {
|
||
// selection end is out of left
|
||
scroll_offset.x = -cursor_end.x;
|
||
}
|
||
if scroll_offset.y + cursor_end.y <= px(0.) {
|
||
// selection end is out of top
|
||
scroll_offset.y = -cursor_end.y;
|
||
}
|
||
}
|
||
}
|
||
|
||
bounds.origin = bounds.origin + scroll_offset;
|
||
|
||
if input.show_cursor(cx) {
|
||
// cursor blink
|
||
cursor = Some(fill(
|
||
Bounds::new(
|
||
point(
|
||
bounds.left() + cursor_pos.x,
|
||
bounds.top() + cursor_pos.y + CURSOR_INSET,
|
||
),
|
||
size(px(1.5), line_height),
|
||
),
|
||
crate::blue_500(),
|
||
))
|
||
};
|
||
}
|
||
|
||
(cursor, scroll_offset)
|
||
}
|
||
|
||
fn layout_selections(
|
||
&self,
|
||
lines: &[WrappedLine],
|
||
line_height: Pixels,
|
||
bounds: &mut Bounds<Pixels>,
|
||
cx: &mut WindowContext,
|
||
) -> Option<Path<Pixels>> {
|
||
let input = self.input.read(cx);
|
||
let selected_range = &input.selected_range;
|
||
if selected_range.is_empty() {
|
||
return None;
|
||
}
|
||
|
||
let (start_ix, end_ix) = if selected_range.start < selected_range.end {
|
||
(selected_range.start, selected_range.end)
|
||
} else {
|
||
(selected_range.end, selected_range.start)
|
||
};
|
||
|
||
let mut prev_lines_offset = 0;
|
||
let mut line_corners = vec![];
|
||
|
||
let mut offset_y = px(0.);
|
||
for line in lines.iter() {
|
||
let line_size = line.size(line_height);
|
||
let line_wrap_width = line_size.width;
|
||
|
||
let line_origin = point(px(0.), offset_y);
|
||
|
||
let line_cursor_start =
|
||
line.position_for_index(start_ix.saturating_sub(prev_lines_offset), line_height);
|
||
let line_cursor_end =
|
||
line.position_for_index(end_ix.saturating_sub(prev_lines_offset), line_height);
|
||
|
||
if line_cursor_start.is_some() || line_cursor_end.is_some() {
|
||
let start = line_cursor_start
|
||
.unwrap_or_else(|| line.position_for_index(0, line_height).unwrap());
|
||
|
||
let end = line_cursor_end
|
||
.unwrap_or_else(|| line.position_for_index(line.len(), line_height).unwrap());
|
||
|
||
// Split the selection into multiple items
|
||
let wrapped_lines =
|
||
(end.y / line_height).ceil() as usize - (start.y / line_height).ceil() as usize;
|
||
|
||
let mut end_x = end.x;
|
||
if wrapped_lines > 0 {
|
||
end_x = line_wrap_width;
|
||
}
|
||
|
||
line_corners.push(Corners {
|
||
top_left: line_origin + point(start.x, start.y),
|
||
top_right: line_origin + point(end_x, start.y),
|
||
bottom_left: line_origin + point(start.x, start.y + line_height),
|
||
bottom_right: line_origin + point(end_x, start.y + line_height),
|
||
});
|
||
|
||
// wrapped lines
|
||
for i in 1..=wrapped_lines {
|
||
let start = point(px(0.), start.y + i as f32 * line_height);
|
||
let mut end = point(end.x, end.y + i as f32 * line_height);
|
||
if i < wrapped_lines {
|
||
end.x = line_size.width;
|
||
}
|
||
|
||
line_corners.push(Corners {
|
||
top_left: line_origin + point(start.x, start.y),
|
||
top_right: line_origin + point(end.x, start.y),
|
||
bottom_left: line_origin + point(start.x, start.y + line_height),
|
||
bottom_right: line_origin + point(end.x, start.y + line_height),
|
||
});
|
||
}
|
||
}
|
||
|
||
if line_cursor_start.is_some() && line_cursor_end.is_some() {
|
||
break;
|
||
}
|
||
|
||
offset_y += line_size.height;
|
||
// +1 for skip the last `\n`
|
||
prev_lines_offset += line.len() + 1;
|
||
}
|
||
|
||
let mut points = vec![];
|
||
if line_corners.is_empty() {
|
||
return None;
|
||
}
|
||
|
||
// Fix corners to make sure the left to right direction
|
||
for corners in &mut line_corners {
|
||
if corners.top_left.x > corners.top_right.x {
|
||
std::mem::swap(&mut corners.top_left, &mut corners.top_right);
|
||
std::mem::swap(&mut corners.bottom_left, &mut corners.bottom_right);
|
||
}
|
||
}
|
||
|
||
for corners in &line_corners {
|
||
points.push(corners.top_right);
|
||
points.push(corners.bottom_right);
|
||
points.push(corners.bottom_left);
|
||
}
|
||
|
||
let mut rev_line_corners = line_corners.iter().rev().peekable();
|
||
while let Some(corners) = rev_line_corners.next() {
|
||
points.push(corners.top_left);
|
||
if let Some(next) = rev_line_corners.peek() {
|
||
if next.top_left.x > corners.top_left.x {
|
||
points.push(point(next.top_left.x, corners.top_left.y));
|
||
}
|
||
}
|
||
}
|
||
|
||
// print_points_as_svg_path(&line_corners, &points);
|
||
|
||
let first_p = *points.get(0).unwrap();
|
||
let mut path = gpui::Path::new(bounds.origin + first_p);
|
||
for p in points.iter().skip(1) {
|
||
path.line_to(bounds.origin + *p);
|
||
}
|
||
Some(path)
|
||
}
|
||
}
|
||
|
||
pub(super) struct PrepaintState {
|
||
lines: SmallVec<[WrappedLine; 1]>,
|
||
cursor: Option<PaintQuad>,
|
||
cursor_scroll_offset: Point<Pixels>,
|
||
selection_path: Option<Path<Pixels>>,
|
||
bounds: Bounds<Pixels>,
|
||
}
|
||
|
||
impl IntoElement for TextElement {
|
||
type Element = Self;
|
||
|
||
fn into_element(self) -> Self::Element {
|
||
self
|
||
}
|
||
}
|
||
|
||
/// A debug function to print points as SVG path.
|
||
#[allow(unused)]
|
||
fn print_points_as_svg_path(
|
||
line_corners: &Vec<Corners<Point<Pixels>>>,
|
||
points: &Vec<Point<Pixels>>,
|
||
) {
|
||
for corners in line_corners {
|
||
println!(
|
||
"tl: ({}, {}), tr: ({}, {}), bl: ({}, {}), br: ({}, {})",
|
||
corners.top_left.x.0 as i32,
|
||
corners.top_left.y.0 as i32,
|
||
corners.top_right.x.0 as i32,
|
||
corners.top_right.y.0 as i32,
|
||
corners.bottom_left.x.0 as i32,
|
||
corners.bottom_left.y.0 as i32,
|
||
corners.bottom_right.x.0 as i32,
|
||
corners.bottom_right.y.0 as i32,
|
||
);
|
||
}
|
||
|
||
if points.len() > 0 {
|
||
println!("M{},{}", points[0].x.0 as i32, points[0].y.0 as i32);
|
||
for p in points.iter().skip(1) {
|
||
println!("L{},{}", p.x.0 as i32, p.y.0 as i32);
|
||
}
|
||
}
|
||
}
|
||
|
||
impl Element for TextElement {
|
||
type RequestLayoutState = ();
|
||
type PrepaintState = PrepaintState;
|
||
|
||
fn id(&self) -> Option<ElementId> {
|
||
None
|
||
}
|
||
|
||
fn request_layout(
|
||
&mut self,
|
||
_id: Option<&GlobalElementId>,
|
||
cx: &mut WindowContext,
|
||
) -> (LayoutId, Self::RequestLayoutState) {
|
||
let input = self.input.read(cx);
|
||
let mut style = Style::default();
|
||
style.size.width = relative(1.).into();
|
||
if self.input.read(cx).is_multi_line() {
|
||
style.size.height = relative(1.).into();
|
||
style.min_size.height = (input.rows.max(1) as f32 * cx.line_height()).into();
|
||
} else {
|
||
style.size.height = cx.line_height().into();
|
||
};
|
||
(cx.request_layout(style, []), ())
|
||
}
|
||
|
||
fn prepaint(
|
||
&mut self,
|
||
_id: Option<&GlobalElementId>,
|
||
bounds: Bounds<Pixels>,
|
||
_request_layout: &mut Self::RequestLayoutState,
|
||
cx: &mut WindowContext,
|
||
) -> Self::PrepaintState {
|
||
let multi_line = self.input.read(cx).is_multi_line();
|
||
let line_height = cx.line_height();
|
||
let input = self.input.read(cx);
|
||
let text = input.text.clone();
|
||
let placeholder = input.placeholder.clone();
|
||
let style = cx.text_style();
|
||
let mut bounds = bounds;
|
||
|
||
let (display_text, text_color) = if text.is_empty() {
|
||
(placeholder, cx.theme().muted_foreground)
|
||
} else if input.masked {
|
||
(
|
||
"*".repeat(text.chars().count()).into(),
|
||
cx.theme().foreground,
|
||
)
|
||
} else {
|
||
(text, cx.theme().foreground)
|
||
};
|
||
|
||
let run = TextRun {
|
||
len: display_text.len(),
|
||
font: style.font(),
|
||
color: text_color,
|
||
background_color: None,
|
||
underline: None,
|
||
strikethrough: None,
|
||
};
|
||
|
||
let runs = if let Some(marked_range) = input.marked_range.as_ref() {
|
||
vec![
|
||
TextRun {
|
||
len: marked_range.start,
|
||
..run.clone()
|
||
},
|
||
TextRun {
|
||
len: marked_range.end - marked_range.start,
|
||
underline: Some(UnderlineStyle {
|
||
color: Some(run.color),
|
||
thickness: px(1.0),
|
||
wavy: false,
|
||
}),
|
||
..run.clone()
|
||
},
|
||
TextRun {
|
||
len: display_text.len() - marked_range.end,
|
||
..run.clone()
|
||
},
|
||
]
|
||
.into_iter()
|
||
.filter(|run| run.len > 0)
|
||
.collect()
|
||
} else {
|
||
vec![run]
|
||
};
|
||
|
||
let font_size = style.font_size.to_pixels(cx.rem_size());
|
||
let wrap_width = if multi_line {
|
||
Some(bounds.size.width - RIGHT_MARGIN)
|
||
} else {
|
||
None
|
||
};
|
||
|
||
let lines = cx
|
||
.text_system()
|
||
.shape_text(display_text, font_size, &runs, wrap_width)
|
||
.unwrap();
|
||
|
||
// `position_for_index` for example
|
||
//
|
||
// #### text
|
||
//
|
||
// Hello 世界，this is GPUI component.
|
||
// The GPUI Component is a collection of UI components for
|
||
// GPUI framework, including Button, Input, Checkbox, Radio,
|
||
// Dropdown, Tab, and more...
|
||
//
|
||
// wrap_width: 444px, line_height: 20px
|
||
//
|
||
// #### lines[0]
|
||
//
|
||
// | index | pos | line |
|
||
// |-------|------------------|------|
|
||
// | 5 | (37 px, 0.0) | 0 |
|
||
// | 38 | (261.7 px, 20.0) | 0 |
|
||
// | 40 | None | - |
|
||
//
|
||
// #### lines[1]
|
||
//
|
||
// | index | position | line |
|
||
// |-------|-----------------------|------|
|
||
// | 5 | (43.578125 px, 0.0) | 0 |
|
||
// | 56 | (422.21094 px, 0.0) | 0 |
|
||
// | 57 | (11.6328125 px, 20.0) | 1 |
|
||
// | 114 | (429.85938 px, 20.0) | 1 |
|
||
// | 115 | (11.3125 px, 40.0) | 2 |
|
||
|
||
// Calculate the scroll offset to keep the cursor in view
|
||
|
||
let (cursor, cursor_scroll_offset) =
|
||
self.layout_cursor(&lines, line_height, &mut bounds, cx);
|
||
|
||
let selection_path = self.layout_selections(&lines, line_height, &mut bounds, cx);
|
||
|
||
PrepaintState {
|
||
bounds,
|
||
lines,
|
||
cursor,
|
||
cursor_scroll_offset,
|
||
selection_path,
|
||
}
|
||
}
|
||
|
||
fn paint(
|
||
&mut self,
|
||
_id: Option<&GlobalElementId>,
|
||
input_bounds: Bounds<Pixels>,
|
||
_request_layout: &mut Self::RequestLayoutState,
|
||
prepaint: &mut Self::PrepaintState,
|
||
cx: &mut WindowContext,
|
||
) {
|
||
let focus_handle = self.input.read(cx).focus_handle.clone();
|
||
let focused = focus_handle.is_focused(cx);
|
||
let bounds = prepaint.bounds;
|
||
let selected_range = self.input.read(cx).selected_range.clone();
|
||
|
||
cx.handle_input(
|
||
&focus_handle,
|
||
ElementInputHandler::new(bounds, self.input.clone()),
|
||
);
|
||
|
||
// Paint selections
|
||
if let Some(path) = prepaint.selection_path.take() {
|
||
cx.paint_path(path, cx.theme().selection);
|
||
}
|
||
|
||
// Paint multi line text
|
||
let line_height = cx.line_height();
|
||
let origin = bounds.origin;
|
||
|
||
let mut offset_y = px(0.);
|
||
for line in prepaint.lines.iter() {
|
||
let p = point(origin.x, origin.y + offset_y);
|
||
_ = line.paint(p, line_height, cx);
|
||
offset_y += line.size(line_height).height;
|
||
}
|
||
|
||
if focused {
|
||
if let Some(cursor) = prepaint.cursor.take() {
|
||
cx.paint_quad(cursor);
|
||
}
|
||
}
|
||
|
||
let width = prepaint
|
||
.lines
|
||
.iter()
|
||
.map(|l| l.width())
|
||
.max()
|
||
.unwrap_or_default();
|
||
let height = prepaint
|
||
.lines
|
||
.iter()
|
||
.map(|l| l.size(line_height).height.0)
|
||
.sum::<f32>();
|
||
|
||
let scroll_size = size(width, px(height));
|
||
|
||
self.input.update(cx, |input, _cx| {
|
||
input.last_layout = Some(prepaint.lines.clone());
|
||
input.last_bounds = Some(bounds);
|
||
input.last_cursor_offset = Some(input.cursor_offset());
|
||
input.last_line_height = line_height;
|
||
input.input_bounds = input_bounds;
|
||
input.last_selected_range = Some(selected_range);
|
||
input
|
||
.scroll_handle
|
||
.set_offset(prepaint.cursor_scroll_offset);
|
||
input.scroll_size = scroll_size;
|
||
});
|
||
|
||
self.paint_mouse_listeners(cx);
|
||
}
|
||
}
|